improved ZSH autocompletion script

2 files changed, 13 insertions, 5 deletions

diff --git a/.zed/tasks.json b/.zed/tasks.json
index 70f3bfe..3acc441 100644
--- a/.zed/tasks.json
+++ b/.zed/tasks.json
@@ -15,14 +15,14 @@
   },
   {
     "label": "install",
-    "command": "./install.sh && ls -lha $ZED_WORKTREE_ROOT/target/release/rsp | awk '{print $5}'",
+    "command": "./install.sh && ls -lha $ZED_WORKTREE_ROOT/target/release/rsp | awk '{print $2}'",
     "show_command": false,
     "show_summary": false,
     "save": "all",
   },
   {
     "label": "build release",
-    "command": "cargo b -r && ls -lha $ZED_WORKTREE_ROOT/target/release/rsp | awk '{print $5}'",
+    "command": "cargo b -r && ls -l $ZED_WORKTREE_ROOT/target/release/rsp | awk '{print $2}'",
     "show_command": false,
     "show_summary": false,
     "save": "all",
diff --git a/completions/_rsp b/completions/_rsp
index 9bf0579..89d122b 100644
--- a/completions/_rsp
+++ b/completions/_rsp
@@ -2,9 +2,17 @@
 functions[_rsp_clap]=$functions[_rsp]
 
 _rsp() {
+    setopt nobanghist
     case $words[2] in
         play|queue)
             if (( CURRENT > 2 )); then
+                local cur="${words[$CURRENT]}"
+                if [[ "$words[2]" == "play" && "$cur" == -* ]]; then
+                    local -a flags
+                    flags=('-i' '--now' '-n' '--next' '-a' '--album')
+                    _describe 'flags' flags
+                    return 0
+                fi
                 _rsp_complete_args
                 return 0
             fi
@@ -14,14 +22,14 @@ _rsp() {
 }
 
 _rsp_complete_args() {
-    setopt localoptions
     setopt nobanghist
     local -a args=("${(@)words[3,$CURRENT]}")
     local cur="${(j: :)args}"
+    # No completion if this is a flag (handled above)
+    [[ "$words[2]" == "play" && "$cur" == -* ]] && return 1
     local cur_clean="${cur//\\/}"
     local -a suggestions
     suggestions=("${(@f)"$(rsp db-complete "$cur" 2>/dev/null)"}")
-    # Filter out empty entries (e.g. from empty command output)
     local -a filtered=()
     for s in "${suggestions[@]}"; do
         if [[ -n "$s" ]]; then
@@ -49,7 +57,7 @@ _rsp_complete_args() {
             fi
         done
         if (( ${#suggestions} > 1 )); then
-            compstate[insert]=menu
+            compstate[insert]=
         fi
         if [[ ${vals[1]} == */ ]]; then
             compadd -U -S '' -M 'm:{a-z}={A-Z} l:|=*' -d display -- "${vals[@]}"