From e4fe8e8f37e9c9a3da90094cdd49fdd160862987 Mon Sep 17 00:00:00 2001
From: Darko Poljak <darko.poljak@gmail.com>
Date: Tue, 7 Jun 2016 13:44:35 +0200
Subject: [PATCH] Implement bash and zsh completions.

---
 completions/bash/cdist-completion.bash | 59 ++++++++++++++++++++++++++
 completions/zsh/_cdist                 | 48 +++++++++++++++++++++
 2 files changed, 107 insertions(+)
 create mode 100644 completions/bash/cdist-completion.bash
 create mode 100644 completions/zsh/_cdist

diff --git a/completions/bash/cdist-completion.bash b/completions/bash/cdist-completion.bash
new file mode 100644
index 00000000..3d396bb4
--- /dev/null
+++ b/completions/bash/cdist-completion.bash
@@ -0,0 +1,59 @@
+_cdist()
+{
+    local cur prev prevprev opts cmds projects
+    COMPREPLY=()
+    cur="${COMP_WORDS[COMP_CWORD]}"
+    prev="${COMP_WORDS[COMP_CWORD-1]}"
+    prevprev="${COMP_WORDS[COMP_CWORD-2]}"
+    opts="-h --help -d --debug -v --verbose -V --version"
+    cmds="banner shell config"
+
+    case "${prevprev}" in
+        shell)
+            case "${prev}" in
+                -s|--shell)
+                    shells=$(grep -v '^#' /etc/shells)
+                    COMPREPLY=( $(compgen -W "${shells}" -- ${cur}) )
+                    return 0
+                    ;;
+            esac
+            ;;
+    esac
+
+    case "${prev}" in
+        -*)
+            COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
+            return 0
+            ;;
+        banner)
+            opts="-h --help -d --debug -v --verbose"
+            COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
+            return 0
+            ;;
+        shell)
+            opts="-h --help -d --debug -v --verbose -s --shell"
+            COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
+            return 0
+            ;;
+        config)
+            opts="-h --help -d --debug -v --verbose \
+                -c --conf-dir -f --file -i --initial-manifest -n --dry-run \
+                -o --out-dir -p --parallel -s --sequential --remote-copy \
+                --remote-exec"
+            COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
+            return 0
+            ;;
+        *)
+            ;;
+    esac
+
+    if [[ ${cur} == -* ]]; then 
+    	COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
+    	return 0
+    fi
+
+    COMPREPLY=( $(compgen -W "${cmds}" -- ${cur}) )
+    return 0
+}
+
+complete -F _cdist cdist
diff --git a/completions/zsh/_cdist b/completions/zsh/_cdist
new file mode 100644
index 00000000..2bf324a6
--- /dev/null
+++ b/completions/zsh/_cdist
@@ -0,0 +1,48 @@
+#compdef cdist
+
+_cdist()
+{
+    local curcontext="$curcontext" state line
+    typeset -A opt_args
+
+    _arguments \
+        '1: :->opts_cmds'\
+        '*: :->opts'
+
+    case $state in
+        opts_cmds)
+            _arguments '1:Options and commands:(banner config shell -h --help -d --debug -v --verbose -V --version)'
+            ;;
+        *)
+            case $words[2] in
+                -*)
+                    opts=(-h --help -d --debug -v --verbose -V --version)
+                    compadd "$@" -- $opts
+                    ;;
+                banner)
+                    opts=(-h --help -d --debug -v --verbose)
+                    compadd "$@" -- $opts
+                    ;;
+                shell)
+                    case $words[3] in
+                        -s|--shell)
+                            shells=($(grep -v '^#' /etc/shells))
+                            compadd "$@" -- $shells
+                            ;;
+                        *)
+                            opts=(-h --help -d --debug -v --verbose -s --shell)
+                            compadd "$@" -- $opts
+                            ;;
+                    esac
+                    ;;
+                config)
+                    opts=(-h --help -d --debug -v --verbose -c --conf-dir -f --file -i --initial-manifest -n --dry-run -o --out-dir -p --parallel -s --sequential --remote-copy --remote-exec)
+                    compadd "$@" -- $opts
+                    ;;
+               *)
+                    ;;
+            esac
+    esac
+}
+
+_cdist "$@"