1 files changed, 55 insertions, 0 deletions

diff --git a/completions/_rsp b/completions/_rsp
new file mode 100644
index 0000000..1ddc1fe
--- /dev/null
+++ b/completions/_rsp
@@ -0,0 +1,55 @@
+# Dynamic completion overrides for rsp play/queue arguments.
+
+# Save clap-generated function so we can delegate non-dynamic cases.
+functions[_rsp_clap]=$functions[_rsp]
+
+_rsp() {
+    case $words[2] in
+        play|queue)
+            _rsp_complete_args
+            ;;
+        *)
+            _rsp_clap "$@"
+            ;;
+    esac
+}
+
+_rsp_complete_args() {
+    setopt localoptions
+    setopt nobanghist
+    # Reconstruct full query from all argument words after subcommand.
+    # This handles spaces in names (e.g. "Master of Puppets") that get
+    # split across multiple $words entries when backslash escaping breaks.
+    local -a args=("${(@)words[3,$CURRENT]}")
+    local cur="${(j: :)args}"
+    local -a suggestions
+    local n_slashes=${#cur//[^\/]/}
+    case $n_slashes in
+        0) suggestions=(${(f)"$(rsp db-complete artists "$cur" 2>/dev/null)"}) ;;
+        1) suggestions=(${(f)"$(rsp db-complete albums "$cur" 2>/dev/null)"}) ;;
+        *) suggestions=(${(f)"$(rsp db-complete tracks "$cur" 2>/dev/null)"}) ;;
+    esac
+    # display: what the user sees in the completion menu (raw values)
+    local -a display=("${suggestions[@]}")
+    # vals: what gets inserted (with ZSH backslash escaping + ! escape)
+    local -a vals=()
+    local slash_idx=$cur[(I)/]
+    local s
+    for s in "${suggestions[@]}"; do
+        local esc="${(q)s:gs/\!/\\\!/}"
+        if (( slash_idx )); then
+            local dir_part=$cur[1,$slash_idx]
+            local edir="${(q)dir_part:gs/\!/\\\!/}"
+            vals+=("${edir}${esc}")
+        else
+            vals+=("${esc}")
+        fi
+    done
+    if (( ${#suggestions} )); then
+        if [[ ${vals[1]} == */ ]]; then
+            compadd -U -S '' -d display -- "${vals[@]}"
+        else
+            compadd -U -d display -- "${vals[@]}"
+        fi
+    fi
+}