From a199156cae279b62917f626415c61cfddffbc9e2 Mon Sep 17 00:00:00 2001
From: Luuk van Baal <luukvbaal@gmail.com>
Date: Thu, 1 Jul 2021 02:26:47 +0200
Subject: [PATCH] Imgview hovered image/thumbnail dir

---
 plugins/imgview | 103 +++++++++++++++++++++---------------------------
 1 file changed, 46 insertions(+), 57 deletions(-)

diff --git a/plugins/imgview b/plugins/imgview
index 1b7b9761..21e45015 100755
--- a/plugins/imgview
+++ b/plugins/imgview
@@ -1,8 +1,7 @@
 #!/usr/bin/env sh
 
 # Description: Open hovered or current directory in image viewer.
-#              Supported image viewers open in thumbnail mode when
-#              the hovered entry is a directory.
+#              Generates media thumbnails with optional dependencies.
 #
 # Dependencies:
 #   - imv (https://github.com/eXeC64/imv) or,
@@ -16,8 +15,10 @@
 #
 # Shell: POSIX compliant
 # Author: Arun Prakash Jana, Luuk van Baal
-
-target="$(readlink -f "$1")"
+#
+# Consider setting NNN_PREVIEWDIR to $XDG_CACHE_HOME/nnn/previews
+# if you want to keep media thumbnails on disk between reboots.
+NNN_PREVIEWDIR="${NNN_PREVIEWDIR:-${TMPDIR:-/tmp}/nnn/previews}"
 
 exit_prompt() {
     [ -n "$1" ] && printf "%s\n" "$1"
@@ -27,47 +28,47 @@ exit_prompt() {
     exit
 }
 
-nthumb_cleanup() {
-    [ -n "$!" ] && while [ -e /proc/"$!" ]; do sleep 1; done
-    if [ -f "$target" ]; then
-        rm -r "$(dirname "$target")"/.nthumbs
-    elif [ -d "$target" ]; then
-        rm -r "$target"/.nthumbs
-    fi &
-}
-
 make_thumbs() {
-    if [ -d "$target" ]; then
-        [ "$1" -eq 4 ] && exit_prompt "$3 can only display a single image"
-        cd "$target" || exit_prompt
-    fi
-    if mkdir .nthumbs >/dev/null; then
-        thumbs=1
-    else
-        exit_prompt
-    fi
-    if [ "$1" -eq 4 ]; then
-      mime="$(file -bL --mime-type -- "$2")"
+    mkdir -p "$NNN_PREVIEWDIR$dir" || return
+    if [ "$1" -eq 3 ]; then
+        [ -d "$target" ] && exit_prompt "$2 can only display a single image"
+        mime="$(file -bL --mime-type -- "$target")"
         case "$mime" in
-          audio/*) ffmpeg -i "$2" ".nthumbs/${2%%.*}.jpg" -y >/dev/null 2>&1
-              ret=".nthumbs/${2%%.*}.jpg" ;;
-          video/*) ffmpegthumbnailer -i "$2" -o .nthumbs/"${2%%.*}".jpg 2> /dev/null
-              ret=".nthumbs/${2%%.*}.jpg" ;;
-          *) ret="$2" ;;
+          audio/*) ffmpeg -i "$target" "$NNN_PREVIEWDIR$target.jpg" -y >/dev/null 2>&1
+              ret="$NNN_PREVIEWDIR/$target.jpg" ;;
+          video/*) ffmpegthumbnailer -i "$target" -o "$NNN_PREVIEWDIR$target.jpg" 2> /dev/null
+              ret="$NNN_PREVIEWDIR/$target.jpg" ;;
+          *) ret="$target" ;;
         esac
     fi
-    for file in *; do
-        case "$(file -bL --mime-type -- "$file")" in
-          audio/*) [ "$1" -ne 0 ] && ffmpeg -i "$file" ".nthumbs/${file%%.*}.jpg" -y >/dev/null 2>&1 ;;
-          video/*) [ "$1" -ne 1 ] && ffmpegthumbnailer -i "$file" -o .nthumbs/"${file%%.*}".jpg 2> /dev/null ;;
-        esac
+    for file in "$dir"/*; do
+        if [ ! -f "$NNN_PREVIEWDIR$file.jpg" ]; then
+            case "$(file -bL --mime-type -- "$file")" in
+                audio/*) [ "$1" -ne 0 ] && ffmpeg -i "$file" "$NNN_PREVIEWDIR$file.jpg" -y >/dev/null 2>&1 ;;
+                video/*) [ "$1" -ne 1 ] && ffmpegthumbnailer -i "$file" -o "$NNN_PREVIEWDIR$file.jpg" 2> /dev/null ;;
+            esac
+        fi
+    done
+    for file in "$NNN_PREVIEWDIR$dir"/*; do
+        filename="$(basename "$file" .jpg)"
+        [ ! -e "$dir/$filename" ] && rm "$file" 2>/dev/null
     done
 }
 
-view_files() {
-    find . .nthumbs -maxdepth 1 -type f -print0 | xargs -0 "$@" --
+listimages() {
+    find -L "$dir" "$NNN_PREVIEWDIR$dir" -maxdepth 1 -type f -print0 2>/dev/null | sort -z
 }
 
+view_files() {
+    [ -f "$target" ] && count="-n $(listimages | grep -a -m 1 -ZznF "$target" | cut -d: -f1)"
+    case "$1" in
+        sxiv) listimages | xargs -0 sxiv -a "${count:--t}" -- ;;
+        imv*) listimages | xargs -0 "$1" "${count:-}" -- ;;
+    esac
+}
+
+target="$(readlink -f "$1")"
+[ -d "$target" ] && dir="$target" || dir="${target%/*}"
 if uname | grep -q "Darwin"; then
     [ -f "$1" ] && open "$1" >/dev/null 2>&1 &
 elif type lsix >/dev/null 2>&1; then
@@ -77,41 +78,29 @@ elif type lsix >/dev/null 2>&1; then
     make_thumbs ""
     clear
     lsix
-    cd .nthumbs && lsix
-    nthumb_cleanup
+    cd "$NNN_PREVIEWDIR$dir" && lsix
     exit_prompt
 elif type ucollage >/dev/null 2>&1; then
-    make_thumbs 1
-    if ! UCOLLAGE_EXPAND_DIRS=1 ucollage . .nthumbs; then
-      nthumb_cleanup && exit_prompt
-    fi
-    nthumb_cleanup
-    exit
+    type ffmpeg >/dev/null 2>&1 && make_thumbs 1
+    UCOLLAGE_EXPAND_DIRS=1 ucollage . "$NNN_PREVIEWDIR$dir" || exit_prompt
 elif type sxiv >/dev/null 2>&1; then
-    make_thumbs 0
-    if [ -f "$target" ]; then
-        view_files sxiv -a >/dev/null 2>&1 &
-    elif [ -d "$target" ]; then
-        view_files sxiv -aqt >/dev/null 2>&1 &
-    fi
+    type ffmpegthumbnailer >/dev/null 2>&1 && make_thumbs 0
+    view_files sxiv >/dev/null 2>&1 &
 elif type imv >/dev/null 2>&1; then
-    make_thumbs 3
+    make_thumbs ""
     view_files imv >/dev/null 2>&1 &
 elif type imvr >/dev/null 2>&1; then
-    make_thumbs 3
+    make_thumbs ""
     view_files imvr >/dev/null 2>&1 &
 elif type viu >/dev/null 2>&1; then
     clear
-    make_thumbs 4 "$1" viu
+    make_thumbs 3 viu
     viu -n "$ret"
-    nthumb_cleanup
     exit_prompt
 elif type catimg >/dev/null 2>&1; then
-    make_thumbs 4 "$1" catimg
+    make_thumbs 3 catimg
     catimg "$ret"
-    nthumb_cleanup
     exit_prompt
 else
     exit_prompt "Please install sxiv/imv/viu/catimg/lsix."
 fi
-[ "$thumbs" -ne 0 ] && nthumb_cleanup &