From 0a6003a9a02846959fb15bc7f2f7bdf066e8f438 Mon Sep 17 00:00:00 2001
From: Charalampos Kardaris <ckardaris@outlook.com>
Date: Sun, 19 Sep 2021 09:07:41 +0200
Subject: [PATCH] Issue #16: Add better expand directory options

---
 src/filelist.sh | 42 +++++++++++++++++++++++++++++++++++++-----
 1 file changed, 37 insertions(+), 5 deletions(-)

diff --git a/src/filelist.sh b/src/filelist.sh
index d52b325..52895d0 100644
--- a/src/filelist.sh
+++ b/src/filelist.sh
@@ -47,13 +47,45 @@ read_filenames() {
         done
     elif [[ "$expand_dirs" == "ask" ]]
     then
+        local yesall=0
+        local first_item=1
         for item in "${filelist[@]}"
         do
-            [[ -d "$item" ]] \
-                && read -p "Expand $item? (n, Esc: no, N: no to all)" -rsN1 char \
-                && echo
-            [[ "$char" == "N" ]] && break
-            [[ ! "$char" =~ ^(n|$'\e')$ ]] && filelist+=("$item"/*)
+            if [[ -d "$item" ]]
+            then
+                if [[ "$yesall" -eq 1 ]]
+                then
+                    filelist+=("$item"/*)
+                else
+                    [[ "$first_item" -eq 1 ]] \
+                        && printf "Expand directory options:\n" \
+                        && printf "  y/Enter\tyes\n" \
+                        && printf "  Y\t\tyes to all\n" \
+                        && printf "  n\t\tno\n" \
+                        && printf "  N\t\tno to all\n" \
+                        && printf "%s\n" "------------------------" \
+                        && first_item=0
+                    read -p "Expand $item? " -rsN1 char
+                    case "$char" in
+                        "N")
+                            echo N
+                            break
+                            ;;
+                        "Y")
+                            echo Y
+                            filelist+=("$item"/*)
+                            yesall=1
+                            ;;
+                        "y"|$'\n')
+                            echo y
+                            filelist+=("$item"/*)
+                            ;;
+                        *)
+                            echo n
+                            ;;
+                    esac
+                fi
+            fi
         done
     fi
     argc="${#filelist[@]}"