doy · Game4Move78 · Apr 3, 2025 · Apr 3, 2025 · Apr 3, 2025 · Apr 3, 2025
diff --git a/src/bin/rbw/main-bash.sh b/src/bin/rbw/main-bash.sh
@@ -0,0 +1,55 @@
+_rbw_wrapper() {
+  local cur prev folder opts res
+  COMPREPLY=()
+
+  if [[ "${COMP_WORDS[1]}" == "get" ]]; then
+
+    # rbw get
+
+    cur="${COMP_WORDS[COMP_CWORD]}"
+    prev="${COMP_WORDS[COMP_CWORD-1]}"
+
+    for (( i=1; i < ${#COMP_WORDS[@]}; i++ )); do
+      if [[ "${COMP_WORDS[i-2]}" == "--folder" ]] && [[ "${COMP_WORDS[i-1]}" == "=" ]]; then
+        folder="${COMP_WORDS[i]}"
+        if [[ $i -eq $(( COMP_CWORD - 1 )) ]]; then
+          prev="--folder"
+        fi
+      elif [[ "${COMP_WORDS[i-1]}" == "--folder" ]] && [[ "${COMP_WORDS[i]}" == "=" ]]; then
+        folder=""
+        if [[ $i -eq $(( COMP_CWORD - 1 )) ]]; then
+          prev="--folder"
+          cur=""
+        fi
+      elif [[ "${COMP_WORDS[i-1]}" == "--folder" ]]; then
+        folder="${COMP_WORDS[i]}"
+      fi
+    done
+
+    if [[ "$prev" == --folder ]]; then
+      # rbw get --folder $folder
+      res=$(
+        rbw list --fields folder 2>/dev/null \
+          | awk -v folder="$folder" 'NF && $1 ~ folder {print $1}'
+      )
+
+      # Split output into an array (one entry per line)
+    elif [[ "$prev" != ^-- ]]; then
+      # rbw get ... $cur
+      res=$(
+        rbw list --fields name,folder 2>/dev/null \
+          | awk -F'\t' -v name="$cur" -v folder="$folder" '$1 ~ name && $1 && $2 ~ folder {print $1}'
+      )
+    else
+      return 0
+    fi
+
+    mapfile -t opts <<< "$res"
+    COMPREPLY=( $(compgen -W "${opts[*]}" -- "$cur") )
+    return 0
+  else
+    return 0
+  fi
+}
+
+complete -F _rbw_wrapper rbw
diff --git a/src/bin/rbw/main-zsh.sh b/src/bin/rbw/main-zsh.sh
@@ -0,0 +1,63 @@
+_rbw_wrapper() {
+  local -a opts
+  local cur prev folder res
+
+  if [[ "${words[2]}" == "get" ]]; then
+
+    # rbw get
+
+    for (( i=1; i < ${#words}; i++ )); do
+      cur="${words[$i]}"
+      if [[ "$cur" == --folder=* ]]; then
+        # rbw get ... --folder $folder
+        folder="${cur#--folder=}"
+        folder="${(Q)folder}"
+      elif [[ "${words[$(( i - 1 ))]}" == --folder ]]; then
+        # rbw get ... --folder $folder
+        folder="${(Q)words[$i]}"
+        prefix=""
+      fi
+    done
+
+    local cur="${(Q)words[CURRENT]}"
+
+    if [[ "$cur" == "--folder"* ]]; then
+      # rbw get --folder=$folder
+      if [[ "$cur" != "--folder" ]]; then
+        folder="${cur#--folder=}"
+        folder="${(Q)folder}"
+      fi
+      res=$(
+        rbw list --fields folder \
+          | awk -v folder="$folder" 'NF && $1 ~ folder {print "--folder=" $1}'
+         )
+    elif [[ "$prev" == "--folder" ]]; then
+      # rbw get --folder $folder
+      res=$(
+        rbw list --fields folder \
+          | awk -v folder="$folder" 'NF && $1 ~ folder {print $1}'
+      )
+    elif [[ "$prev" != -* ]] && [[ "$cur" != -* ]]; then
+      # rbw get ... $cur
+      res=$(
+        rbw list --fields name,folder \
+        | awk -F'\t' -v name="$cur" -v folder="$folder" '$1 ~ name && $1 && $2 ~ folder {print $1}'
+      )
+    else
+      _rbw
+      return
+    fi
+
+    if [[ "$res" == "$cur" ]]; then
+      compadd -S '' -- "${opts[@]}"
+    else
+      opts=("${(@f)${res}}")
+      compadd -S '' -- "${opts[@]}"
+    fi
+  else
+    _rbw
+  fi
+
+}
+
+compdef _rbw_wrapper rbw
diff --git a/src/bin/rbw/main.rs b/src/bin/rbw/main.rs
@@ -465,6 +465,12 @@ fn main() {
                 "rbw",
                 &mut std::io::stdout(),
             );
+            if *shell == clap_complete::Shell::Zsh {
+                println!("{}", include_str!("main-zsh.sh"));
+            }
+            if *shell == clap_complete::Shell::Bash {
+                println!("{}", include_str!("main-bash.sh"));
+            }
             Ok(())
         }
     }