axzshctl set-theme completion: Look for all types in custom_themes/ and rpositories

[ax-zsh.git] / core / 50_axzsh / functions / _axzsh

diff --git a/core/50_axzsh/functions/_axzsh b/core/50_axzsh/functions/_axzsh
index e3ae1755dd2702ade47c96a63b1e7745296a195c..ea32b58ab9cf3c8ddb4f411d7797336f11235d28 100644 (file)
--- a/core/50_axzsh/functions/_axzsh
+++ b/core/50_axzsh/functions/_axzsh
@@ -21,10 +21,18 @@ _axzshctl() {
          'args')
                case "$words[2]" in
                  "enable-plugin"|"disable-plugin")
-                       compadd "$@" -- $(ls -1 "$AXZSH"/plugins)
+                       compadd "$@" -- \
+                               "$AXZSH"/plugins/*(/:t) \
+                               "$AXZSH"/custom_plugins/*(N/:t) \
+                               "$AXZSH"/repos/*(/:t:s/#/\\/)
                        ;;
                  "set-theme")
-                       compadd "$@" -- "-" $(ls -1 "$AXZSH"/themes | cut -d'.' -f1)
+                       compadd "$@" -- "-" \
+                               "$AXZSH"/themes/*.axzshtheme(:r:t) \
+                               "$AXZSH"/custom_themes/*.axzshtheme(N:r:t) \
+                               "$AXZSH"/custom_themes/*.zsh-theme(N:r:t) \
+                               "$AXZSH"/repos/*/*.axzshtheme(N:r:t) \
+                               "$AXZSH"/repos/*/*.zsh-theme(N:r:t)
                        ;;
                esac
                ;;