@@ -1,6 +1,7 @@

 #!/bin/bash

 DB=/var/lib/navidrome/navidrome.db

+medias=/music

 no_cover_flag=.no_cover

 cover_img=cover

 mbz_agent='getCover/0.1 (https://seb.lautre.net/git/seb/scripts)'

@@ -11,9 +12,14 @@ coverartarchive_api="https://coverartarchive.org/release"

 coverart=$(mktemp --dry-run /dev/shm/XXXXXXXX)

 sizes=(small 250 500 large 1200)

-ERROR   () { echo -e "\e[3;31m ${1:-erreur} \e[0;m"; }

-OK      () { echo -e "\e[3;32m ${1:-OK} \e[0;m";     }

-WARNING () { echo -e "\e[3;33m ${1:-alerte} \e[0;m"; }

+OK      () { echo -e "\e[3;32m ${1:-OK} \e[0;m";     return 0; }

+WARNING () { echo -e "\e[3;33m ${1:-alerte} \e[0;m"; return 1; }

+ERROR   () { echo -e "\e[3;31m ${1:-erreur} \e[0;m"; return 1; }

+

+if ! test -d $medias; then

+    ERROR "$medias n'est pas un répertoire"

+    exit 1

+fi

 sql_request () {

     sqlite3 $DB <<< "$1"

@@ -22,7 +28,7 @@ sql_request () {

 covered () {

     case "${forcing_level:-none}" in

         retry)

-            if compgen -G "${album#*:}/$no_cover_flag" > /dev/null; then

+            if test -w "${album#*:}/$no_cover_flag"; then

                 rm -f "${album#*:}/$no_cover_flag" 2>/dev/null

                 return 1

             else

@@ -30,11 +36,17 @@ covered () {

             fi

             ;;

         force)

+            rm -f "${album#*:}/$no_cover_flag" 2>/dev/null

             return 1

             ;;

         none)

-            compgen -G "${album#*:}/$cover_img.*" > /dev/null \

-            || compgen -G "${album#*:}/$no_cover_flag" > /dev/null

+            if compgen -G "${album#*:}/$cover_img.*" > /dev/null; then

+                return 0

+            elif test -e "${album#*:}/$no_cover_flag"; then

+                return 0

+            else

+                return 1

+            fi

             ;;

     esac

 }

@@ -44,7 +56,6 @@ get_image () {

         mime_type=$(file -bn --mime-type /dev/shm/$2)

         if [[ ${mime_type:-erreur} =~ ^image/ ]]; then

             mv -f /dev/shm/$2 ${album#*:}/$cover_img.${2##*.} \

-            && OK \

             || ERROR "échec en écriture"

         else

             WARNING "${album%%:*} type $mime_type"

@@ -71,86 +82,109 @@ fi

 if declare -f pre_get_cover > /dev/null; then

     pre_get_cover

 fi

+

 IFS=$'\n'

 for arg in $@; do

     if [[ $arg =~ ^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}:/ ]]; then

-        albums[${#albums[@]}]=$arg

+        from_CLI=1

+        albums[${#albums[@]}]="${arg%/}"

+    elif [[ $arg =~ ^/ ]]; then

+        from_CLI=1

+        albums[${#albums[@]}]=":${arg%/}"

     elif [[ $arg =~ ^(force|retry)$ ]]; then

         forcing_level=$arg

     else

         WARNING "incohérence sur $arg"

     fi

 done

+if test -n "$from_CLI"; then

+    forcing_level=force

+fi

+

 if test ${#albums[@]} -eq 0; then

-    albums=($(sql_request "select mbz_album_id || ':' || path

-                           from media_file

-                           where mbz_album_id is not ''

-                           and path not like '/media/musique/la mule/%'

-                           and has_cover_art = FALSE" | sed -r 's|/[^/]+$||' | sort -u))

+    albums=($(sql_request 'select printf("%s:'${medias%/}'/%s/%s", mbz_album_id, artist, name)

+             from album where cover_art_path = ""

+             and name != "[Unknown Album]"'))

 fi

+get_from_mbz () {

+    curl -Ls $coverartarchive_api/${album%%:*} > $coverart

+    if test $(file -bn --mime-type $coverart) = application/json; then

+        unset img

+        for size in ${sizes[@]}; do

+            img=$(jq -r '.images | .[] | select(.front == true) | .thumbnails | ."'$size'"?' $coverart 2>/dev/null)

+            test -n "$img" && break

+        done

+        if test -n "$img"; then

+            get_image "$img" "${album%%:*}.${img##*.}"

+        fi

+    else

+        WARNING "${album%%:*} réponse inattendue de musicbrainz"

+    fi

+}

+

+get_from_spotify () {

+    eval "$(sql_request 'select printf("artist=""%s"";album_name=""%s""", artist, name)

+                        from album

+                        where artist || "/" || name = "'${album#*:$medias/}'"')"

+    if test -n "$album_name" -a -n "$artist"; then

+        curl --request GET --silent \

+            --header "Authorization: Bearer $spotify_access_token" \

+            --header 'Content-Type: application/json' \

+            --url "https://api.spotify.com/v1/search?type=album&query=album:${album_name// /%20}${artist:+%20artist:${artist// /%20}}" > $coverart

+        found_albums=$(jq -r '.albums .total // empty' $coverart)

+        if test ${found_albums:-0} -gt 0; then

+            img=$(jq --raw-output --arg artist "${artist^^}" --arg album "${album_name^^}" "

+                    .albums .items

+                    | .[]

+                    | select(.name | ascii_upcase == \$album)

+                    | select(.artists | .[].name | ascii_upcase == \$artist)

+                    | .images

+                    | .[]

+                    | select(.height == 300)

+                    | .url // empty" $coverart)

+            if test -n "$img"; then

+                get_image "$img" "${img##*/}.jpg"

+            else

+                WARNING "pas d'image trouvée sur spotify (artist: $artist; album: $album_name)"

+            fi

+        else

+            WARNING "aucun album trouvé sur spotify"

+        fi

+    else

+        ERROR "album: $album_name; artist: $artist"

+    fi

+}

+

 for album in ${albums[@]}; do

     if ! test -d "${album#*:}"; then

-        WARNING "${album#*:}: chemin inconnu"

+        if test -n "$from_CLI"; then

+            WARNING "${album#*:}: chemin inconnu"

+        fi

         continue

     fi

     if ! covered; then

         echo -n "${album#*:} "

-        curl -Ls $coverartarchive_api/${album%%:*} > $coverart

-        if test $(file -bn --mime-type $coverart) = application/json; then

-            unset img

-            for size in ${sizes[@]}; do

-                img=$(jq -r '.images | .[] | select(.front == true) | .thumbnails | ."'$size'"?' $coverart 2>/dev/null)

-                test -n "$img" && break

-            done

-            if test -n "$img"; then

-                get_image "$img" "${album%%:*}.${img##*.}"

-            fi

-        elif test -n "$spotify_access_token"; then

-            curl --user-agent "$mbz_agent" --silent \

-                "https://musicbrainz.org/ws/2/release/${album%%:*}?fmt=json&inc=artists" > $coverart

-            artist=$(jq -r '."artist-credit"? | .[0].name // empty' $coverart)

-            if [[ ${artist:-__erreur__} = @(Various Artists|__erreur__) ]]; then

-                unset artist

-                WARNING "artiste inconnu ou multiples (compilation ?)"

-                continue

-            fi

-            album_name=$(jq -r '.title' $coverart)

-            if test -n "$album_name"; then

-                curl --request GET --silent --show-error --http1.1 \

-                    --header "Authorization: Bearer $spotify_access_token" \

-                    --header 'Content-Type: application/json' \

-                    --url "https://api.spotify.com/v1/search?type=album&query=album:${album_name// /%20}${artist:+%20artist:${artist// /%20}}" > $coverart

-                found_albums=$(jq -r '.albums .total // empty' $coverart)

-                if test ${found_albums:-0} -gt 0; then

-                    img=$(jq --raw-output --arg artist "$artist" --arg album "$album_name" "

-                            .albums .items

-                            | .[]

-                            | select(.name == \$album${artist:+ and (.artists | .[].name == \$artist)})

-                            | .images

-                            | .[]

-                            | select(.height == 300)

-                            | .url // empty" $coverart)

-                    if test -n "$img"; then

-                        get_image "$img" "${img##*/}.jpg"

-                    else

-                        touch "${album#*:}/$no_cover_flag" 

-                        WARNING "${album%%:*} aucune image trouvée"

-                    fi

-                elif test ${found_albums:-1} -eq 0; then

-                    WARNING "album indisponible sur spotify"

+        if test -n "${album%%:*}"; then

+            if get_from_mbz; then

+                OK

+            elif test -n "$spotify_access_token"; then

+                if get_from_spotify; then

+                    OK

                 else

-                    WARNING "${found_albums:-NULL} albums trouvés sur spotify"

-                    echo $artist

-                    echo $album_name

+                    touch "${album#*:}/$no_cover_flag"

                 fi

             fi

-        else

-            touch "${album#*:}/$no_cover_flag" 

-            WARNING "${album%%:*} aucune image trouvée"

+        elif test -n "$spotify_access_token"; then

+            if get_from_spotify; then

+                OK

+            else

+                touch "${album#*:}/$no_cover_flag"

+            fi

         fi

     fi

 done

+

 rm -f $coverart 

 if declare -f post_get_cover > /dev/null; then

     post_get_cover