install xrepo complete in scripts

1 files changed, 22 insertions, 0 deletions

diff --git a/scripts/register-completions.sh b/scripts/register-completions.sh
index e274605b2..40a8edac3 100755
--- a/scripts/register-completions.sh
+++ b/scripts/register-completions.sh
@@ -9,6 +9,15 @@ if   [[ "$SHELL" = */zsh ]]; then
     reply=( "${(ps:\n:)completions}" )
   }
   compctl -f -S "" -K _xmake_zsh_complete xmake
+
+  # zsh parameter completion for xrepo
+  _xrepo_zsh_complete() 
+  {
+    local completions=("$(XMAKE_SKIP_HISTORY=1 XMAKE_ROOT=y xmake lua private.xrepo.complete 0 nospace "$words")")
+    reply=( "${(ps:\n:)completions}" )
+  }
+  compctl -f -S "" -K _xrepo_zsh_complete xmake
+
 elif [[ "$SHELL" = */bash ]]; then
   # bash parameter completion for xmake
   _xmake_bash_complete()
@@ -22,5 +31,18 @@ elif [[ "$SHELL" = */bash ]]; then
     COMPREPLY=( $(compgen -W "$completions") )
   }
   complete -o default -o nospace -F _xmake_bash_complete xmake
+
+  # bash parameter completion for xrepo
+  _xrepo_bash_complete()
+  {
+    local word=${COMP_WORDS[COMP_CWORD]}
+    local completions
+    completions="$(XMAKE_SKIP_HISTORY=1 XMAKE_ROOT=y xmake lua private.xrepo.complete "${COMP_POINT}" "nospace-nokey" "${COMP_LINE}")"
+    if [ $? -ne 0 ]; then
+      completions=""
+    fi
+    COMPREPLY=( $(compgen -W "$completions") )
+  }
+  complete -o default -o nospace -F _xrepo_bash_complete xrepo
 fi