From 8dac47564f5efe8843bead982033feb8839b7e9d Mon Sep 17 00:00:00 2001
From: hophacker <j@paiyou.co>
Date: Thu, 27 Feb 2020 21:56:17 +0800
Subject: [PATCH] use traditional way to parse kexec arguments, making it
 support namespace and left arguments

---
 zsh/functions.zsh | 38 +++++++++++++++++++++++++-------------
 1 file changed, 25 insertions(+), 13 deletions(-)

diff --git a/zsh/functions.zsh b/zsh/functions.zsh
index ec9872e..67058ff 100644
--- a/zsh/functions.zsh
+++ b/zsh/functions.zsh
@@ -318,28 +318,40 @@ function helm() {
 function kexec {
   RAN=false
   KCONTEXT=${KCONTEXT:-gcloud}
-  while getopts ":c:rp:" opt; do
-    case "${opt}" in
-      c)
-        KCONTEXT=$OPTARG
+  NAMESPACE=default
+  finalopts=()
+  while [[ $@ != "" ]] do
+    case $1 in
+      -c)
+        KCONTEXT="$2"
+        shift; shift
         ;;
-      r)
+      --context=*)
+        KCONTEXT="${i#*=}"
+        shift
+        ;;
+      -r)
         RAN=true
+        shift
         ;;
-      p)
-        PROJECT=$OPTARG
+      -p)
+        PROJECT="$2"
+        shift; shift
+        ;;
+      -n)
+        NAMESPACE="$2"
+        shift; shift
         ;;
       *)
-        echo "Usage: cmd [-h]"
-        return
+        finalopts+=($1)
+        shift
         ;;
     esac
   done
-  shift $(($OPTIND - 1))
 
   RUNNING_POD_INDEX=-1
   while true; do
-    ALL_PODS=$(kubectl -c $KCONTEXT get pods | grep "$PROJECT")
+    ALL_PODS=$(kubectl -c $KCONTEXT -n $NAMESPACE get pods | grep "$PROJECT")
     echo $fg[green]"All Pods:"$reset_color
     echo $ALL_PODS
     if  [[ ${#ALL_PODS[@]} == 0 ]]; then
@@ -383,8 +395,8 @@ function kexec {
     fi
   done
   if [[ $RUNNING_POD_INDEX != -1 ]]; then
-    echo "executing pod $fg[green]$RUNNING_PODS[$RUNNING_POD_INDEX]$reset_color"
-    kubectl -c $KCONTEXT exec -it $RUNNING_PODS[$RUNNING_POD_INDEX] $@
+    echo "kubectl -c $KCONTEXT exec -it $RUNNING_PODS[$RUNNING_POD_INDEX] $finalopts"
+    kubectl -n $NAMESPACE -c $KCONTEXT exec -it $RUNNING_PODS[$RUNNING_POD_INDEX] $finalopts
   fi
 }