refactor(bash): improve bash integration script

justfile

@@ -86,6 +86,11 @@ br: (build 'release')
 @generate-fish-completion:
 	cargo run -- init fish | sed '2,$ s/tv /.\/target\/debug\/tv /'
 
+# test the bash shell completion script in debug mode
+# Run: `source <(just generate-bash-completion)` directly in bash
+@generate-bash-completion:
+	cargo run -- init bash | sed 's/tv /.\/target\/debug\/tv /'
+
 # Update the project's changelog
 @update-changelog:
 	echo "Updating changelog..."

television/utils/shell/completion.bash

@@ -1,31 +1,136 @@
-function tv_smart_autocomplete() {
-  # prefix (lhs of cursor)
-  local current_prompt="${READLINE_LINE:0:$READLINE_POINT}"
-
-  local output=$(tv --autocomplete-prompt "$current_prompt" --inline | tr '\n' ' ')
-
-  if [[ -n $output ]]; then
-    # suffix (rhs of cursor)
-    local rhs="${READLINE_LINE:$READLINE_POINT}"
-    # add a space if the prompt does not end with one
-    [[ "${current_prompt}" != *" " ]] && current_prompt="${current_prompt} "
-
-    local lhs=$current_prompt$output
-    READLINE_LINE=$lhs$rhs
-    READLINE_POINT=${#lhs}
-  fi
+_disable_bracketed_paste() {
+    # Disable bracketed paste mode to prevent unwanted escape sequences
+    printf '\e[?2004l' > /dev/tty
 }
 
-function tv_shell_history() {
-  local current_prompt="${READLINE_LINE:0:$READLINE_POINT}"
-
-  local output=$(tv bash-history --input "$current_prompt" --inline)
-
-  if [[ -n $output ]]; then
-    READLINE_LINE=$output
-    READLINE_POINT=${#READLINE_LINE}
-  fi
+_enable_bracketed_paste() {
+    # Re-enable bracketed paste mode
+    printf '\e[?2004h' > /dev/tty
 }
 
+__tv_path_completion() {
+    local base="$1"
+    local lbuf="$2"
+    local suffix=""
+    local tail=" "
+    local dir leftover matches
+
+    # Evaluate the base path (handle ~, variables, etc.)
+    eval "base=\"$base\"" 2>/dev/null || return
+
+    # Extract directory part if base contains a slash
+    [[ "$base" == *"/"* ]] && dir="$base"
+    
+    while true; do
+        if [[ -z "$dir" || -d "$dir" ]]; then
+            # Calculate leftover part (what comes after the directory)
+            leftover="${base#"$dir"}"
+            leftover="${leftover#/}"
+            
+            # Set default directory if empty
+            [[ -z "$dir" ]] && dir='.'
+            
+            # Remove trailing slash unless it's root
+            [[ "$dir" != "/" ]] && dir="${dir%/}"
+
+            # move to the next line so that the prompt is not overwritten
+            printf "\n"
+            
+            # Call tv with proper arguments and process output
+            matches=$(
+                tv "$dir" --autocomplete-prompt "$lbuf" --inline --input "$leftover" < /dev/tty | while IFS= read -r item; do
+                    item="${item%$suffix}$suffix"
+                    dirP="$dir/"
+                    [[ "$dirP" == "./" ]] && dirP=""
+                    # Quote the item to handle special characters
+                    printf '%s ' "$dirP$(printf '%q' "$item")"
+                done
+            )
+            
+            # Remove trailing space
+            matches="${matches% }"
+            
+            if [[ -n "$matches" ]]; then
+                # Update readline buffer
+                local new_line="$lbuf$matches$tail"
+                local rhs="${READLINE_LINE:$READLINE_POINT}"
+                READLINE_LINE="$new_line$rhs"
+                READLINE_POINT=${#new_line}
+            fi
+            # move the cursor back to the previous line
+            printf "\033[A"
+
+            break
+        fi
+        
+        # Move up one directory level
+        dir=$(dirname "$dir")
+        dir="${dir%/}/"
+    done
+}
+
+tv_smart_autocomplete() {
+    _disable_bracketed_paste
+
+    local tokens prefix lbuf
+    local current_prompt="${READLINE_LINE:0:$READLINE_POINT}"
+    
+    # Split the current prompt into tokens
+    # This is a simplified version of zsh's word splitting
+    read -ra tokens <<< "$current_prompt"
+    
+    if [[ ${#tokens[@]} -lt 1 ]]; then
+        # Fall back to default completion if no tokens
+        _enable_bracketed_paste
+        return
+    fi
+
+    # Handle trailing space
+    [[ "${READLINE_LINE:$((READLINE_POINT-1)):1}" == " " ]] && tokens+=("")
+
+    # Get the last token as prefix
+    prefix="${tokens[-1]}"
+    
+    # Calculate lbuf (everything except the last token)
+    if [[ -n "$prefix" ]]; then
+        lbuf="${current_prompt:0:$((${#current_prompt} - ${#prefix}))}"
+    else
+        lbuf="$current_prompt"
+    fi
+
+    __tv_path_completion "$prefix" "$lbuf"
+
+    _enable_bracketed_paste
+}
+
+tv_shell_history() {
+    _disable_bracketed_paste
+
+    local current_prompt="${READLINE_LINE:0:$READLINE_POINT}"
+    local output
+
+    # move to the next line so that the prompt is not overwritten
+    printf "\n"
+
+    # Get history using tv with the same arguments as zsh version
+    output=$(tv bash-history --input "$current_prompt" --inline)
+
+    if [[ -n "$output" ]]; then
+        # Clear the right side of cursor and set new line
+        READLINE_LINE="$output"
+        READLINE_POINT=${#READLINE_LINE}
+
+        # Uncomment this to automatically accept the line
+        # (i.e. run the command without having to press enter twice)
+        # accept-line() { echo; }; accept-line
+    fi
+
+    # move the cursor back to the previous line
+    printf "\033[A"
+
+    _enable_bracketed_paste
+}
+
+# Bind the functions to key combinations
 bind -x '"{tv_smart_autocomplete_keybinding}": tv_smart_autocomplete'
 bind -x '"{tv_shell_history_keybinding}": tv_shell_history'