Add wrap set-sources to shell completion scripts

mesonbuild · Dec 2, 2024 · 484655e · 484655e
1 parent ac530b1
commit 484655e
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/data/shell-completions/bash/meson b/data/shell-completions/bash/meson
@@ -715,6 +715,7 @@ _meson-wrap() {
     list
     promote
     search
+    set-source
     status
     update
     update-db
@@ -865,6 +866,28 @@ _meson-wrap-search() {
   COMPREPLY+=($(compgen -W '${wraps[*]}' -- "$cur"))
 }
 
+_meson-wrap-set-source() {
+  shortopts=(
+    h
+  )
+
+  longopts=(
+    help
+  )
+
+  local cur prev
+  if ! _get_comp_words_by_ref cur prev &>/dev/null; then
+    cur="${COMP_WORDS[COMP_CWORD]}"
+  fi
+
+  if ! _meson_compgen_options "$cur"; then
+    if [[ -z $cur ]]; then
+      COMPREPLY+=($(compgen -P '--' -W '${longopts[*]}'))
+      COMPREPLY+=($(compgen -P '-' -W '${shortopts[*]}'))
+    fi
+  fi
+}
+
 _meson-wrap-status() {
   shortopts=(
     h

diff --git a/data/shell-completions/zsh/_meson b/data/shell-completions/zsh/_meson
@@ -271,6 +271,7 @@ _arguments \
     'update:Update a project to its newest available version'
     'info:Show info about a wrap'
     'status:Show the status of your subprojects'
+    'set-source:Set WrapDB source URL'
   )
 
   if (( CURRENT == 2 )); then
@@ -297,6 +298,8 @@ _arguments \
         # TODO: how do you figure out what wraps are provided by subprojects if
         # they haven't been fetched yet?
         _arguments '*:'
+      elif [[ $cmd == "set-source" ]]; then
+        _arguments '*:'
       fi
     else
       _message "unknown meson wrap command: $words[2]"