From 0708bd6cc06c4802e4dacb976fde8ee9193f779c Mon Sep 17 00:00:00 2001
From: drymer <drymer@autistici.org>
Date: Sun, 10 Jun 2018 21:31:42 +0200
Subject: [PATCH] Fix ku and kops completion

---
 roles/shell-dotfiles/files/functions | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/roles/shell-dotfiles/files/functions b/roles/shell-dotfiles/files/functions
index 161833d..6a015b0 100644
--- a/roles/shell-dotfiles/files/functions
+++ b/roles/shell-dotfiles/files/functions
@@ -125,12 +125,24 @@ function kubectl (){
     if [[ -z $KUBECTL_SHOW_CONTEXT ]]
     then
 	source <($KUBECTL_BIN completion zsh)
-	source <(kops completion zsh)
+    # Add support to completion when using the kubectl alias ku
+	source <($KUBECTL_BIN completion zsh | sed "s/kubectl/ku/g")
     fi
     KUBECTL_SHOW_CONTEXT="k8s-$($KUBECTL_BIN config current-context)"
     $KUBECTL_BIN "$@"
 }
 
+## kops
+KOPS_BIN=`which kops`
+function kops(){
+    if [[ -z $KOPS_SHOW_CONTEXT ]]
+    then
+	source <($KOPS_BIN completion zsh)
+    fi
+    KOPS_SHOW_CONTEXT="k8s-$($KOPS_BIN config current-context)"
+    $KOPS_BIN "$@"
+}
+
 
 ## molecule
 MOLECULE_BIN=`which molecule`