stackql-deploy/shell_completions/stackql-deploy-completion.bash at 9cb488de4fc82bfcbada919f319d0ccd9f4b09a0 · stackql/stackql-deploy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
_stackql_deploy_completion() {
    local cur prev opts base
    COMPREPLY=()
    cur="${COMP_WORDS[COMP_CWORD]}"
    prev="${COMP_WORDS[COMP_CWORD-1]}"

    # Main commands
    local commands="build test teardown info init upgrade shell completion"

    # Common options for build/test/teardown
    local common_opts="--log-level --env-file -e --env --dry-run --show-queries --on-failure --custom-registry --download-dir --help"

    # Get the command (first non-option argument)
    local cmd=""
    for ((i=1; i<${#COMP_WORDS[@]}-1; i++)); do
        if [[ ${COMP_WORDS[i]} != -* ]]; then
            cmd=${COMP_WORDS[i]}
            break
        fi
    done

    # Completion logic
    case "${cmd}" in
        build|test|teardown)
            # After command, need stack_dir then stack_env
            local args=()
            for ((i=2; i<${#COMP_WORDS[@]}-1; i++)); do
                if [[ ${COMP_WORDS[i]} != -* ]]; then
                    args+=("${COMP_WORDS[i]}")
                fi
            done

            if [ ${#args[@]} -eq 0 ]; then
                # Complete directory names for stack_dir
                compopt -o dirnames
                COMPREPLY=( $(compgen -d -- "${cur}") )
            elif [ ${#args[@]} -eq 1 ]; then
                # Complete common environment names
                COMPREPLY=( $(compgen -W "dev staging prod test prd sit uat" -- "${cur}") )
            else
                # Complete options
                COMPREPLY=( $(compgen -W "${common_opts}" -- "${cur}") )
            fi
            ;;

        init)
            # init <stack_name> [--provider]
            case "${prev}" in
                --provider)
                    COMPREPLY=( $(compgen -W "aws google azure" -- "${cur}") )
                    ;;
                init)
                    # Just type the stack name, no completion
                    ;;
                *)
                    COMPREPLY=( $(compgen -W "--provider --help" -- "${cur}") )
                    ;;
            esac
            ;;

        completion)
            COMPREPLY=( $(compgen -W "bash zsh fish powershell" -- "${cur}") )
            ;;

        info|upgrade|shell)
            COMPREPLY=( $(compgen -W "--help --custom-registry --download-dir" -- "${cur}") )
            ;;

        *)
            # No command yet, show main commands and global options
            if [[ ${cur} == -* ]]; then
                COMPREPLY=( $(compgen -W "--help --version" -- "${cur}") )
            else
                COMPREPLY=( $(compgen -W "${commands}" -- "${cur}") )
            fi
            ;;
    esac

    # Handle option arguments
    case "${prev}" in
        --log-level)
            COMPREPLY=( $(compgen -W "DEBUG INFO WARNING ERROR CRITICAL" -- "${cur}") )
            return 0
            ;;
        --env-file)
            compopt -o default
            COMPREPLY=( $(compgen -f -X '!*.env' -- "${cur}") $(compgen -d -- "${cur}") )
            return 0
            ;;
        --on-failure)
            COMPREPLY=( $(compgen -W "rollback ignore error" -- "${cur}") )
            return 0
            ;;
        --custom-registry)
            # URL completion - just let user type
            return 0
            ;;
        --download-dir)
            compopt -o dirnames
            COMPREPLY=( $(compgen -d -- "${cur}") )
            return 0
            ;;
    esac

    return 0
}

complete -F _stackql_deploy_completion stackql-deploy