From 60bb4c89fb93aaea3d3e1a2441c5aedb26ad5758 Mon Sep 17 00:00:00 2001
From: Jarrod Johnson <jjohnson2@lenovo.com>
Date: Fri, 21 Sep 2018 14:09:49 -0400
Subject: [PATCH] Add specific completions for nodeboot and nodesetboot

---
 confluent_client/confluent_env.sh | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/confluent_client/confluent_env.sh b/confluent_client/confluent_env.sh
index 01556972..eb9f8c20 100644
--- a/confluent_client/confluent_env.sh
+++ b/confluent_client/confluent_env.sh
@@ -60,6 +60,19 @@ _confluent_nodeidentify_completion()
     fi
 }
 
+
+_confluent_nodesetboot_completion()
+{
+    _confluent_get_args
+    if [ $NUMARGS == 3 ]; then
+        COMPREPLY=($(compgen -W "default cd network setup hd" -- ${COMP_WORDS[-1]}))
+    fi
+    if [ $NUMARGS -lt 3 ]; then
+        _confluent_nr_completion
+        return;
+    fi
+}
+
 _confluent_nodepower_completion()
 {
     _confluent_get_args
@@ -175,7 +188,7 @@ _confluent_ng_completion()
 }
 complete -F _confluent_nr_completion nodeattrib
 complete -F _confluent_nr_completion nodebmcreset
-complete -F _confluent_nr_completion nodeboot
+complete -F _confluent_nodesetboot_completion nodeboot
 complete -F _confluent_nr_completion nodeconfig
 complete -F _confluent_nn_completion nodeconsole
 complete -F _confluent_nr_completion nodeeventlog
@@ -192,7 +205,7 @@ complete -F _confluent_nr_completion noderemove
 complete -F _confluent_nr_completion nodereseat
 complete -F _confluent_nr_completion noderun
 complete -F _confluent_nr_completion nodesensors
-complete -F _confluent_nr_completion nodesetboot
+complete -F _confluent_nodesetboot_completion nodesetboot
 complete -F _confluent_nr_completion nodeshell
 complete -F _confluent_nodesupport_completion nodesupport