metadata extraction; slow

fuzique

@@ -4,6 +4,85 @@ set -eu
 
 APP_NAME="fuzique"
 
+if [ "${1:-}" = "--get-cache-line" ]; then
+  # Format: filename albumartist album year track artist tracknumber
+  shift
+  #ffprobe -v quiet -show_entries format -of json "$1" | jq -C '.format.tags'
+  tags="$(ffprobe -v quiet -show_entries format -of json "$1" |
+    jq -r '.format.tags |
+  [
+    [.album_artist, ."ALBUM ARTIST", ."album artist"],
+    [.album, .ALBUM],
+    [.originalyear, .ORIGINALYEAR, ."original year", ."ORIGINAL YEAR", .year, .YEAR, .date, .DATE],
+    [.title, .TITLE],
+    [.artist_credit, .ARTIST_CREDIT, .artist, .ARTIST],
+    [.track, .TRACK]
+  ] | map(map(select(. != null))[0]) | join("\t")')"
+  printf "%s\t%s\n" "$1" "$tags"
+  echo "$1" >>/tmp/cacheupd
+  exit 0
+  #sh -c 'printf "%s\t%s" "{}" "$(ffprobe -v quiet -show_entries format -of json "{}" | jq -r ".format.tags" | [ [.album_artist, .\"ALBUM ARTIST\", .\"album artist\"])"'
+  #jq -r '.format.tags | [ [.artist, .ARTIST], [.album, .ALBUM], [.title, .TITLE] ] | map(map(select(. != null))) | map(.[0]) | join("|")'
+fi
+
+if [ "${1:-}" = "--mpv-play" ]; then
+  if [ "$3" = "$MODE_PLAYLIST" ]; then
+    echo "{ \"command\": [\"set\", \"playlist-pos-1\", \"$FZF_POS\"] }" | socat - "$MPV_SOCKET"
+  else
+    echo "{ \"command\": [\"loadfile\", \"$2\"] }" | socat - "$MPV_SOCKET"
+  fi
+  exit 0
+fi
+if [ "${1:-}" = "--mpv-queue" ]; then
+  echo "{ \"command\": [\"loadfile\", \"$2\", \"append-play\"] }" | socat - "$MPV_SOCKET"
+  exit 0
+fi
+if [ "${1:-}" = "--mpv-next" ]; then
+  echo '{ "command": ["playlist-next"] }' | socat - "$MPV_SOCKET"
+  exit 0
+fi
+if [ "${1:-}" = "--mpv-prev" ]; then
+  echo '{ "command": ["playlist-prev"] }' | socat - "$MPV_SOCKET"
+  exit 0
+fi
+if [ "${1:-}" = "--mpv-seekf" ]; then
+  echo '{ "command": ["seek", "10"] }' | socat - "$MPV_SOCKET"
+  exit 0
+fi
+if [ "${1:-}" = "--mpv-seekb" ]; then
+  echo '{ "command": ["seek", "-10"] }' | socat - "$MPV_SOCKET"
+  exit 0
+fi
+if [ "${1:-}" = "--mpv-toggle-pause" ]; then
+  echo '{ "command": ["cycle", "pause"] }' | socat - "$MPV_SOCKET"
+  exit 0
+fi
+
+if [ "${1:-}" = "--browse" ]; then
+  rel=${3#"$ROOT/"}
+  d=$(echo "$rel" | awk -F'/' '{ print NF }')
+  if [ "$2" = "l" ]; then
+    if [ "$d" -eq 1 ]; then
+      grep -F "$3/" "$release_file_browse"
+    elif [ "$d" -eq 2 ]; then
+      grep -F "$3/" "$tracks_file_browse"
+    else
+      p=$(echo "$rel" | cut -d '/' -f 1,2)
+      grep -F "$ROOT/$p/" "$tracks_file_browse"
+    fi
+  elif [ "$2" = "h" ]; then
+    if [ "$d" -eq 3 ]; then
+      p=$(echo "$rel" | cut -d '/' -f 1)
+      grep -F "$ROOT/$p/" "$release_file_browse"
+    elif [ "$d" -eq 2 ]; then
+      cat "$artists_file_browse"
+    else
+      cat "$artists_file_browse"
+    fi
+  fi
+  exit 0
+fi
+
 if [ "${1:-}" = "--prompt" ]; then
   shift
   file="$1"
@@ -11,15 +90,16 @@ if [ "${1:-}" = "--prompt" ]; then
   mode="$1"
   case "$mode" in
   "$MODE_SEARCH")
-    printf "search>  "
+    printf "$SEARCH_PROMPT"
     ;;
   "$MODE_BROWSE")
     rel=${file#"$ROOT/"}
     artist=$(echo "$rel" | cut -d "/" -f 1)
     release=$(echo "$rel" | cut -d "/" -f 2)
     d=$(echo "$rel" | awk -F'/' '{ print NF }')
-    [ "$d" -eq 2 ] && printf " $AFMT 〉  " "$artist"
-    [ "$d" -eq 3 ] && printf " $AFMT 》 $RFMT 〉  " "$artist" "$release"
+    [ "$d" -eq 1 ] && printf "$BROWSE_PROMPT_ROOT"
+    [ "$d" -eq 2 ] && printf "$BROWSE_PROMPT_ARTIST" "$artist"
+    [ "$d" -eq 3 ] && printf "$BROWSE_PROMPT_RELEASE" "$artist" "$release"
     ;;
   "$MODE_PLAYLIST")
     printf ""
@@ -29,37 +109,65 @@ if [ "${1:-}" = "--prompt" ]; then
 fi
 
 if [ "${1:-}" = "--preview" ]; then
-  echo "call to --preview" >>/tmp/foo
   shift
   file="$1"
-  tags=$(ffprobe -v quiet -show_entries format -of json "$file" | jq)
-  {
-    echo "## Lyrics"
-    echo ""
-    printf "%s" "$tags" | jq -C -r '.format.tags.LYRICS'
-  } | $CAT
+  rel=${file#"$ROOT/"}
+  artist=$(echo "$rel" | cut -d "/" -f 1)
+  release=$(echo "$rel" | cut -d "/" -f 2)
+  track=$(echo "$rel" | cut -d "/" -f 3)
+  if [ "$track" ]; then
+    echo "(not implemented: track info for $track)"
+  elif [ "$release" ]; then
+    echo "(not implemented: release info for $release)"
+  elif [ "$artist" ]; then
+    echo "(not implemented: artist info for $artist)"
+  fi
   exit 0
 fi
 
+mpv_text() {
+  echo "{ \"command\": [\"expand-text\",\"$1\"] }" | socat - "$MPV_SOCKET" | jq -r '.data'
+}
+
+tags_from_file() {
+  ffprobe \
+    -v quiet \
+    -show_entries format \
+    -of json \
+    "$1" |
+    jq -r '.format.tags | [ [.artist, .ARTIST], [.album, .ALBUM], [.title, .TITLE] ] | map(map(select(. != null))) | map(.[0]) | join("|")'
+}
+
 if [ "${1:-}" = "--show-playlist" ]; then
-  [ ! -S "${MPV_SOCKET:-}" ] && exit 1
-  count=$(echo "{ \"command\": [\"expand-text\",\"\${playlist/count}\"] }" | socat - "$MPV_SOCKET" | jq -r '.data')
+  count=$(mpv_text '${playlist/count}')
   if [ "$count" -eq 0 ]; then
-    printf "(empty playlist)\t\t%s\n" "$MODE_PLAYLIST"
+    printf "$EMPTY_PLAYLIST_STRING\t\t%s\n" "$MODE_PLAYLIST"
     exit 0
   fi
+  pos=$(mpv_text '${playlist-pos}')
+  cmd=""
   for i in $(seq 0 $((count - 1))); do
-    file=$(echo "{ \"command\": [\"expand-text\",\"\${playlist/$i/filename}\"] }" | socat - "$MPV_SOCKET" | jq -r '.data')
-    curr=$(echo "{ \"command\": [\"expand-text\",\"\${playlist/$i/current}\"] }" | socat - "$MPV_SOCKET" | jq -r '.data')
-    pnt="  "
-    [ "$curr" = "yes" ] && pnt="$PLAYLIST_POINTER"
-    artist=$(echo "$file" | rev | cut -d "/" -f 3 | rev)
-    release=$(echo "$file" | rev | cut -d "/" -f 2 | rev)
-    title=$(echo "$file" | rev | cut -d "/" -f 1 | rev | sed 's/\..*$//')
+    cmd="$cmd\${playlist/$i/filename}|"
+  done
+  fn=$(mpv_text "$cmd")
+  for i in $(seq 0 $((count - 1))); do
+    f=$(echo "$fn" | cut -d "|" -f "$((i + 1))")
+    [ ! "$f" ] && continue
+    if [ -f "$f" ]; then
+      tags=$(tags_from_file "$f")
+      artist=$(echo "$tags" | cut -d "|" -f 1)
+      release=$(echo "$tags" | cut -d "|" -f 2)
+      track=$(echo "$tags" | cut -d "|" -f 3)
+    fi
+    rel=${f#"$ROOT/"}
+    [ "${artist:-}" ] || artist=$(echo "$rel" | cut -d "/" -f 1)
+    [ "${release:-}" ] || release=$(echo "$rel" | cut -d "/" -f 2)
+    [ "${track:-}" ] || track=$(echo "$rel" | cut -d "/" -f 3 | sed 's/\..*$//')
     artist=$(printf "$AFMT" "$artist")
     release=$(printf "$RFMT" "$release")
-    title=$(printf "$TFMT" "$title")
-    printf "%s|%s|%s|%s\t%s\t%s\n" "$pnt" "$title" "$release" "$artist" "$file" "$MODE_PLAYLIST"
+    track=$(printf "$TFMT" "$track")
+    [ "$i" -eq "$pos" ] && pnt="$PLAYLIST_POINTER" || pnt="  "
+    printf "%s|%s|%s|%s\t%s\t%s\n" "$pnt" "$track" "$release" "$artist" "$f" "$MODE_PLAYLIST"
   done |
     grep '.' |
     column -t -s '|'
@@ -79,7 +187,7 @@ CAT=${CAT:+$CAT --color=always --style=plain --language=md}
 CAT=${CAT:-cat}
 export CAT
 
-if [ "${1:-}" = "--help" ]; then
+if [ ! "${1:-}" ] || [ "${1:-}" = "--help" ]; then
   $CAT <<EOF
 Usage: \`$0 [ --help | [ MUSIC_DIRECTORY [ --reset-cache ] ] ]\`
 
@@ -117,7 +225,7 @@ EOF
 fi
 
 # Load configuration
-[ "${1:-}" ] && ROOT="$(realpath "$1")" && shift
+[ "${1:?"You did not specify the music directory. Run $0 --help."}" ] && ROOT="$(realpath "$1")" && shift
 ROOT=${ROOT%%/}
 [ ! -d "${ROOT:-}" ] && echo "Faild to recognize music directory. Run $0 --help." && exit 1
 DEPTH=$(echo "$ROOT" | awk -F'/' '{ print NF }')
@@ -125,18 +233,22 @@ DEPTH=$(echo "$ROOT" | awk -F'/' '{ print NF }')
 # Cache support
 ROOTHASH=$(echo "$ROOT" | sha1sum | cut -d ' ' -f 1)
 CACHE_DIR="$HOME/.cache/$APP_NAME"
-CACHE_FILE="$CACHE_DIR/$ROOTHASH"
+CACHE_FILE="$CACHE_DIR/f-$ROOTHASH"
 
 # Theme
 ARTIST_COLOR=${ARTIST_COLOR:-'\033[38;5;202m'}
 RELEASE_COLOR=${RELEASE_COLOR:-'\033[38;5;208m'}
 TRACK_COLOR=${TRACK_COLOR:-'\033[38;5;215m'}
 OFF="\033[m"
-AFMT="${AFTM:-"🎤 ${ARTIST_COLOR}%s${OFF}"}"
+AFMT="${AFMT:-"🎤 ${ARTIST_COLOR}%s${OFF}"}"
 RFMT="${RFMT:-"💽 ${RELEASE_COLOR}%s${OFF}"}"
 TFMT="${TFMT:-"🎵 ${TRACK_COLOR}%s${OFF}"}"
-PLAYLIST_POINTER="${PLAYLIST_POINTER:-☞}"
-#PLAYLIST_POINTER="${PLAYLIST_POINTER:-👉}"
+PLAYLIST_POINTER="${PLAYLIST_POINTER:-👉}"
+EMPTY_PLAYLIST_STRING="${EMPTY_PLAYLIST_STRING:-"\033[38;5;251m(empty playlist)${OFF}"}"
+SEARCH_PROMPT=${SEARCH_PROMPT:-"🔎 search〉"}
+BROWSE_PROMPT_ROOT="${BROWSE_PROMPT_ROOT:-"$SEARCH_PROMPT"}"
+BROWSE_PROMPT_ARTIST="${BROWSE_PROMPT_ARTIST:-"$AFMT 〉"}"
+BROWSE_PROMPT_RELEASE="${BROWSE_PROMPT_RELEASE:-"$AFMT 》 $RFMT 〉"}"
 
 # Modes
 MODE_BROWSE="B"
@@ -146,27 +258,29 @@ MODE_PLAYLIST="P"
 # Make socket
 MPV_SOCKET="$(mktemp --suffix=.sock)"
 
-# Export
-export ROOT MPV_SOCKET AFMT RFMT TFMT PLAYLIST_POINTER MODE_SEARCH MODE_BROWSE MODE_PLAYLIST
-
 # Load
-if [ "${1:-}" = "--reset-cache" ] || [ ! -f "$CACHE_FILE" ]; then
-  rm -f "$CACHE_FILE"
+if [ ! -f "$CACHE_FILE" ]; then
   [ ! -d "$CACHE_DIR" ] && mkdir -p "$CACHE_DIR"
-  {
-    $FD --max-depth 2 --type d . "$ROOT"
-    $FD --exact-depth 3 --type f -i -e "mp3" -e "mp4" -e "m4a" -e "ogg" -e "flac" -e "wav" . "$ROOT"
-  } |
-    sort >"$CACHE_FILE"
+  $FD --exact-depth 3 --type f -i -e "mp3" -e "mp4" -e "m4a" -e "ogg" -e "flac" -e "wav" -0 . "$ROOT" |
+    xargs -0 -P4 -I {} "$0" --get-cache-line "{}" | sort >"$CACHE_FILE"
 fi
 
 # Parse
 artists_file_browse=$(mktemp)
-artists_file_search=$(mktemp)
 release_file_browse=$(mktemp)
 release_file_search=$(mktemp)
 tracks_file_browse=$(mktemp)
 tracks_file_search=$(mktemp)
+
+# Export
+export ROOT MPV_SOCKET \
+  AFMT RFMT TFMT \
+  PLAYLIST_POINTER EMPTY_PLAYLIST_STRING \
+  SEARCH_PROMPT BROWSE_PROMPT_ROOT BROWSE_PROMPT_ARTIST BROWSE_PROMPT_RELEASE \
+  MODE_SEARCH MODE_BROWSE MODE_PLAYLIST \
+  artists_file_browse release_file_browse tracks_file_browse \
+  tracks_file_search release_file_search
+
 awk \
   -F'/' \
   -v afmt="$AFMT" \
@@ -176,7 +290,6 @@ awk \
   -v artists_file_browse="$artists_file_browse" \
   -v release_file_browse="$release_file_browse" \
   -v tracks_file_browse="$tracks_file_browse" \
-  -v artists_file_search="$artists_file_search" \
   -v release_file_search="$release_file_search" \
   -v tracks_file_search="$tracks_file_search" \
   -v mode_browse="$MODE_BROWSE" \
@@ -193,56 +306,75 @@ NF >= depth + 3 { tr = $itrack; gsub(/\..*$/, "", tr); tr = sprintf(tfmt, tr) }
 NF == depth + 1 { print ar, $0, mode_browse >> artists_file_browse }
 NF == depth + 2 { print rl, $0, mode_browse >> release_file_browse }
 NF == depth + 3 { print tr, $0, mode_browse >> tracks_file_browse }
-NF == depth + 1 { print               ar, $0, mode_search >> artists_file_search }
 NF == depth + 2 { print        rl "|" ar, $0, mode_search >> release_file_search }
 NF == depth + 3 { print tr "|" rl "|" ar, $0, mode_search >> tracks_file_search }
 ' <"$CACHE_FILE"
 
-$MPV --no-config --no-terminal --input-ipc-server="$MPV_SOCKET" --idle &
+$MPV --no-config --no-terminal --input-ipc-server="$MPV_SOCKET" --idle --no-osc --no-input-default-bindings &
 
 $FZF \
   --reverse \
   --ansi \
-  --no-sort \
   --delimiter="\t" \
   --with-nth="{1}" \
   --cycle \
   --bind="ctrl-d:half-page-down,ctrl-u:half-page-up" \
-  --bind="enter:execute:printf '{ \"command\": [\"loadfile\", \"%s\"] }\n' {2} | socat - \"$MPV_SOCKET\"" \
-  --bind="alt-enter:execute:printf '{ \"command\": [\"loadfile\", \"%s\", \"append-play\"] }\n' {2} | socat - \"$MPV_SOCKET\"" \
+  --bind="down:preview-down,up:preview-up" \
+  --bind="ctrl-w:toggle-preview-wrap" \
+  --preview-window="right,30%" \
   --preview="$0 --preview {2}" \
-  --bind="alt-1:reload:column -t -s '|' -E 0 \"$artists_file_search\"" \
+  --bind="alt-1:reload:cat \"$artists_file_browse\"" \
   --bind="alt-2:reload:column -t -s '|' -E 0 \"$release_file_search\"" \
   --bind="alt-3:reload:column -t -s '|' -E 0 \"$tracks_file_search\"" \
   --bind="ctrl-p:reload:$0 --show-playlist" \
+  --bind="enter:execute-silent:$0 --mpv-play {2} {3}" \
+  --bind="alt-enter:execute-silent:$0 --mpv-queue {2}" \
+  --bind="l:execute-silent:$0 --mpv-next" \
+  --bind="h:execute-silent:$0 --mpv-prev" \
+  --bind="right:execute-silent:$0 --mpv-seekf" \
+  --bind="left:execute-silent:$0 --mpv-seekb" \
+  --bind="space:execute-silent:$0 --mpv-toggle-pause" \
+  --bind="q:abort" \
   --bind="load:transform:
-    echo \"l2:{}\" >> /tmp/foo
-    [ {3} = \"$MODE_SEARCH\" ] && printf \"+hide-preview\" || printf \"+show-preview\"
-    [ {3} = \"$MODE_PLAYLIST\" ] && printf \"+hide-input\" || printf \"+show-input\"
-    echo \"l3:{}\" >> /tmp/foo
+    if [ {3} = \"$MODE_PLAYLIST\" ]; then
+      printf \"+hide-input\"
+      printf \"+show-preview\"
+      printf \"+rebind(l)\"
+      printf \"+rebind(h)\"
+      printf \"+rebind(right)\"
+      printf \"+rebind(left)\"
+      printf \"+rebind(space)\"
+      printf \"+rebind(q)\"
+    fi
+    if [ {3} = \"$MODE_SEARCH\" ]; then
+      printf \"+show-input\"
+      printf \"+hide-preview\"
+      printf \"+unbind(l)\"
+      printf \"+unbind(h)\"
+      printf \"+unbind(right)\"
+      printf \"+unbind(left)\"
+      printf \"+unbind(space)\"
+      printf \"+unbind(q)\"
+    fi
+    if [ {3} = \"$MODE_BROWSE\" ]; then
+      printf \"+show-input\"
+      printf \"+show-preview\"
+      printf \"+unbind(l)\"
+      printf \"+unbind(h)\"
+      printf \"+unbind(right)\"
+      printf \"+unbind(left)\"
+      printf \"+unbind(space)\"
+      printf \"+unbind(q)\"
+    fi
     printf \"+transform-prompt:$0 --prompt {2} {3}\"
-    echo \"l4:{}\" >> /tmp/foo
-    echo \"load ended..\" >> /tmp/foo
     " \
-  --bind="ctrl-l:transform:
-    printf \"clear-query+pos(1)\"
-    d=\$(echo {2} | awk -F'/' '{ print NF }')
-    [ \"\$d\" -eq \"$((DEPTH + 1))\" ] && echo \"+reload:grep -F {2}/ \\\"$release_file_browse\\\" | column -t -s '|' || true\"
-    [ \"\$d\" -eq \"$((DEPTH + 2))\" ] && echo \"+reload:grep -F {2}/ \\\"$tracks_file_browse\\\" | column -t -s '|' || true\"
-    " \
-  --bind="ctrl-h:transform:
-    printf \"clear-query+pos(1)\"
-    d=\$(echo {2} | awk -F'/' '{ print NF }')
-    p=\$(echo {2} | rev | cut -d '/' -f 3- | rev)
-    [ \"\$d\" -eq \"$((DEPTH + 3))\" ] && echo \"+reload:grep -F \\\"\$p/\\\" \\\"$release_file_browse\\\" | column -t -s '|' || true\"
-    [ \"\$d\" -eq \"$((DEPTH + 2))\" ] && echo \"+reload:grep -F \\\"\$p/\\\" \\\"$artists_file_browse\\\" | column -t -s '|' || true\"
-    " \
-  <"$artists_file_search" || true
+  --bind="ctrl-l:reload($0 --browse l {2})+clear-query+pos(1)" \
+  --bind="ctrl-h:reload($0 --browse h {2})+clear-query+pos(1)" \
+  <"$artists_file_browse" || true
 
 printf '{ "command": ["quit"] }\n' | socat - "$MPV_SOCKET"
 rm -f "$MPV_SOCKET" \
   "$artists_file_browse" \
-  "$artists_file_search" \
   "$release_file_browse" \
   "$release_file_search" \
   "$tracks_file_browse" \