#!/bin/bash

set -e

test $USER = www-data

thumbnailsconf=${1:-${0##*/}.conf}
test -r "$thumbnailsconf" || exit 1
shift

declare -A imgsize width height

imgsize[xxlarge]="1656 1242"
imgsize[xlarge]=" 1224  918"
imgsize[large]="  1008  756"
imgsize[medium]=" 792   594"
imgsize[small]="  576   432"
imgsize[xsmall]=" 432   324"
imgsize[2small]=" 240   180"
imgsize[square]=" 120   120"

if test $# -gt 0; then
    filter=$(echo $@ | sed 's/ /|/g')
    for size in ${!imgsize[@]}; do
        if ! [[ $size =~ ^(${filter})$ ]]; then
            unset imgsize[$size]
        else
            geometry=(${imgsize[$size]})
            width[$size]=${geometry[0]}
            height[$size]=${geometry[1]}
        fi
    done
fi

for size in ${!imgsize[@]}; do
    geometry=(${imgsize[$size]})
    width[$size]=${geometry[0]}
    height[$size]=${geometry[1]}
done

if which readconf > /dev/null; then
    eval $(readconf --conf "$thumbnailsconf" --section=mysql --section=path --case-sensitive)
else
    echo readconf needed !
    echo PATH = $PATH
    echo get readconf at https://seb.lautre.net/git/seb/scripts/raw/master/readconf
    exit 1
fi

if ! which convert > /dev/null; then
    echo please install ImageMagick
    exit 1
fi

sqlcmd="select distinct path from ${dbprefix}images
           where id in (
              select distinct image_id from ${dbprefix}image_category
                  where category_id in (
                      select id from ${dbprefix}categories where dir is ${IMPORT:+not} NULL
                  )
           )"

function get_ext_fn () {
    fnNoExt="${1%.*}"
    fnExt="${1##*.}"
    fnPath="${1%/*}"
}

function get_formats () {
    local IFS=$'\n'
    for file in $(mysql -N -u $dbuser -p${dbpasswd} $dbname <<< $sqlcmd); do
        get_ext_fn "$file"
        test $fnExt = 'mp4' && continue
        for size in ${!imgsize[@]}; do
            if ! test -e "${destDir}/${fnNoExt}-${size:0:2}.$fnExt" -a -z "$FORCE"; then
                formats[$file]+="$size "
                let total_ext++
            fi
        done
    done
}

function calc_left () {
    local format=(${formats[$file]})
    unset formats[$file]
    total_ext=$(( $total_ext - ${#format[@]} ))
}

declare -A formats
total_ext=1
get_formats
let total_ext--

if test -n "$LIST"; then
    for file in "${!formats[@]}"; do
        echo "[${#formats[@]}/$total_ext] $file : ${formats[$file]}"
        calc_left
    done
else
    for file in "${!formats[@]}"; do
        unset cmd
        sleep_duration=1
        get_ext_fn "$file"
        for size in ${formats[$file]}; do
            cmd+=(mpr:image -resize ${width[$size]}x${height[$size]} -write "${destDir}/${fnNoExt}-${size:0:2}.$fnExt" +delete)
            let sleep_duration++
        done
        printf "\r\033[K[${#formats[@]}/$total_ext] $file : ${formats[$file]}"
        cmd[-1]="null:"
        nice -20 convert \
            -filter Lanczos -strip -auto-orient -quality 95 -interlace line -sampling-factor 4:2:2 \
            "$site/$file" +write "${cmd[@]}"
        for size in ${!imgsize[@]}; do
            test -s "${destDir}/${fnNoExt}-${size:0:2}.$fnExt" || echo -e "\nmissing extension ${size:0:2}.$fnExt"
        done
        sleep $sleep_duration
        calc_left
    done
fi