Updated mcollective.init according to OSCI-658

[packages/precise/mcollective.git] / ext / bash / mco_completion.sh

diff --git a/ext/bash/mco_completion.sh b/ext/bash/mco_completion.sh
new file mode 100644 (file)
index 0000000..f37b1d4
--- /dev/null
+++ b/ext/bash/mco_completion.sh
@@ -0,0 +1,57 @@
+_mco() {
+  local agents options
+
+  COMPREPLY=()
+  local cur=${COMP_WORDS[COMP_CWORD]}
+  local prev=${COMP_WORDS[COMP_CWORD-1]}
+
+  # Where are the plugins?
+  local libdir=$(sed -n 's@libdir = @@p' /etc/mcollective/client.cfg)
+
+  # All arguments by options
+  noopt=($(tr ' ' '\n' <<<${COMP_WORDS[@]} | \
+    grep -v "^$cur$" | grep -v -- '^-'))
+
+  local count_noopt=${#noopt[@]}
+  local cmd=${noopt[0]}
+  local app=${noopt[1]}
+
+  # A bug in the output of --help prevents
+  # from parsing all options, so we list the common ones here
+  local common_options="-T --target -c --config --dt --discovery-timeout \
+    -t --timeout -q --quiet -v --verbose -h --help -W --with -F \
+    --wf --with-fact -C --wc --with-class -A --wa --with-agent -I \
+    --wi --with-identity"
+
+  if [ $COMP_CWORD -eq 1 ]; then
+    apps=$($cmd completion --list-applications)
+    COMPREPLY=($(compgen -W "$apps" -- "$cur"))
+  elif [ $COMP_CWORD -gt 1 ]; then
+    options="${common_options} $($cmd $app --help | grep -o -- '-[^, ]\+')"
+
+    if [ "x${app}" = "xrpc" ]; then
+      if [[ $count_noopt -eq 2 || "x${prev}" = "x--agent" ]]; then
+        # Complete with agents
+        agents=$($cmd completion --list-agents)
+        options="$options $agents"
+      elif [[ $count_noopt -eq 3 || "x${prev}" = "x--action" ]]; then
+        # Complete with agent actions
+        rpcagent=${noopt[2]}
+        actions=$($cmd completion --list-actions \
+                       --agent "$rpcagent")
+        options="$options $actions"
+      elif [ $count_noopt -gt 3 ]; then
+        # Complete with key=value
+        rpcagent=${noopt[2]}
+        rpcaction=${noopt[3]}
+        inputs=$($cmd completion --list-inputs \
+                      --agent "$rpcagent" --action "$rpcaction")
+        options="$options $inputs"
+      fi
+    fi
+
+    COMPREPLY=($(compgen -W "$options" -S ' ' -- "$cur"))
+  fi
+}
+[ -n "${have:-}" ] && complete -o nospace -F _mco mco
+