From e16bd69b1d7f347902110e29467681531af52694 Mon Sep 17 00:00:00 2001
From: taleksovska <taleksovska@ibasis.net>
Date: Tue, 19 Mar 2024 09:40:47 +0100
Subject: [PATCH] Improve autocompletion functionality for 'enabler' command

---
 enabler/commands/enabler_completion.sh | 55 ++++++++++++++++++++
 enabler/enabler_autocomplete.py        | 70 --------------------------
 2 files changed, 55 insertions(+), 70 deletions(-)
 create mode 100755 enabler/commands/enabler_completion.sh
 delete mode 100755 enabler/enabler_autocomplete.py

diff --git a/enabler/commands/enabler_completion.sh b/enabler/commands/enabler_completion.sh
new file mode 100755
index 0000000..b761d59
--- /dev/null
+++ b/enabler/commands/enabler_completion.sh
@@ -0,0 +1,55 @@
+_enabler_complete() {
+    local cur_word prev_word commands
+
+    # Get the current and previous words
+    cur_word="${COMP_WORDS[COMP_CWORD]}"
+    prev_word="${COMP_WORDS[COMP_CWORD-1]}"
+    local categories="apps kind preflight platform setup version"
+
+    case "$prev_word" in
+        "enabler") # noqa
+            commands="$categories"
+            ;;
+    esac
+
+    # Initialize the variable to store previous words
+    prev_words=""
+    local apps="namespace"
+    local kind="create delete status start stop"
+    local platform="init info keys release version"
+    local setup="init metallb istio"
+
+    # Loop through previous words and concatenate them
+    for ((i=1; i<COMP_CWORD; i++)); 
+    do
+        prev_words="${prev_words}${COMP_WORDS[i]} "
+    done
+
+    # Trim any trailing whitespace
+    prev_words="${prev_words% }"
+
+    case "$prev_words" in
+        "enabler apps")
+            commands="$commands $apps"
+            ;;
+        "enabler kind")
+            commands="$commands $kind"
+            ;;
+        "enabler platform")
+            commands="$commands $platform"
+            ;;
+        "enabler setup")
+            commands="$commands $setup"
+            ;;
+    esac
+    
+    echo ""
+    echo "$commands"
+
+    if [[ "$cur_word" == "$prev_word"* ]]; then
+        COMPREPLY=( $(compgen -W "$commands" -- "$cur_word") )
+    fi
+}
+
+# Register _enabler_complete to provide completion for the enabler command
+complete -F _enabler_complete enabler
\ No newline at end of file
diff --git a/enabler/enabler_autocomplete.py b/enabler/enabler_autocomplete.py
deleted file mode 100755
index 2139513..0000000
--- a/enabler/enabler_autocomplete.py
+++ /dev/null
@@ -1,70 +0,0 @@
-#!/usr/bin/env python
-import argcomplete
-import argparse
-
-
-def apps_namespace_subparser(subparsers):
-    parser = subparsers.add_parser('apps')
-    parser.add_argument('namespace')
-
-
-def platform_subparser(subparsers):
-    parser = subparsers.add_parser('platform')
-    subsubparsers = parser.add_subparsers(dest='platform_command')
-    subsubparsers.required = True
-
-    subparser_init = subsubparsers.add_parser('init') # noqa
-    subparser_info = subsubparsers.add_parser('info') # noqa
-    subparser_keys = subsubparsers.add_parser('keys') # noqa
-    subparser_release = subsubparsers.add_parser('release') # noqa
-    subparser_version = subsubparsers.add_parser('version') # noqa
-
-
-def kind_subparser(subparsers):
-    parser = subparsers.add_parser('kind')
-    subsubparsers = parser.add_subparsers(dest='kind_command')
-    subsubparsers.required = True
-
-    subparser_create = subsubparsers.add_parser('create') # noqa 
-    subparser_delete = subsubparsers.add_parser('delete') # noqa
-    subparser_status = subsubparsers.add_parser('status') # noqa
-    subparser_start = subsubparsers.add_parser('start')   # noqa
-    subparser_stop = subsubparsers.add_parser('stop')     # noqa
-
-
-def setup_subparser(subparsers):
-    parser = subparsers.add_parser('setup')
-    subsubparsers = parser.add_subparsers(dest='setup_command')
-    subsubparsers.required = True
-
-    subparser_init = subsubparsers.add_parser('init') # noqa 
-    subparser_metallb = subsubparsers.add_parser('metallb') # noqa
-    subparser_istio = subsubparsers.add_parser('istio')  # noqa
-
-
-def version_subparser(subparsers):
-    subparsers.add_parser('version')
-
-
-def preflight_subparser(subparsers):
-    subparsers.add_parser('preflight')
-
-
-def main():
-    parser = argparse.ArgumentParser()
-    subparsers = parser.add_subparsers(dest='command')
-    subparsers.required = True
-
-    apps_namespace_subparser(subparsers)
-    platform_subparser(subparsers)
-    kind_subparser(subparsers)
-    setup_subparser(subparsers)
-    version_subparser(subparsers)
-    preflight_subparser(subparsers)
-
-    argcomplete.autocomplete(parser, exclude=[None])  # Modify this line
-    args = parser.parse_args()  # noqa
-
-
-if __name__ == "__main__":
-    main()