Add variables for user, context and namespace (#15)

* Add variables for user, context and namespace * Fetching namespace only once
superbrothers · May 31, 2019 · c97dac5 · c97dac5
1 parent 414acd6
commit c97dac5
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -14,6 +14,21 @@ source /path/to/zsh-kubectl-prompt/kubectl.zsh
 RPROMPT='%{$fg[blue]%}($ZSH_KUBECTL_PROMPT)%{$reset_color%}'
 ```
 
+Or create different style depending on user, context, namespace.
+The plugin creates 4 variables:
+* ZSH_KUBECTL_CONTEXT
+* ZSH_KUBECTL_NAMESPACE
+* ZSH_KUBECTL_PROMPT
+* ZSH_KUBECTL_USER
+
+For example, make the prompt red when the username matches admin.
+```sh
+autoload -U colors; colors
+source /path/to/zsh-kubectl-prompt/kubectl.zsh
+[[ "$ZSH_KUBECTL_USER" =~ "admin" ]] && color=red || color=blue
+RPROMPT='%{$fg[$color]%}($ZSH_KUBECTL_PROMPT)%{$reset_color%}'
+```
+
 Also you can install with homebrew.
 
 ```console

diff --git a/kubectl.zsh b/kubectl.zsh
@@ -56,15 +56,18 @@ function _zsh_kubectl_prompt_precmd() {
         return 1
     fi
 
+    ZSH_KUBECTL_USER="$(kubectl config view -o "jsonpath={.contexts[?(@.name==\"$context\")].context.user}")"
+    ZSH_KUBECTL_CONTEXT="${context}"
+    ns="$(kubectl config view -o "jsonpath={.contexts[?(@.name==\"$context\")].context.namespace}")"
+    [[ -z "$ns" ]] && ns="default"
+    ZSH_KUBECTL_NAMESPACE="${ns}"
+
     zstyle -s ':zsh-kubectl-prompt:' namespace namespace
     if [[ "$namespace" != true ]]; then
         ZSH_KUBECTL_PROMPT="${context}"
         return 0
     fi
 
-    ns="$(kubectl config view -o "jsonpath={.contexts[?(@.name==\"$context\")].context.namespace}")"
-    [[ -z "$ns" ]] && ns="default"
-
     zstyle -s ':zsh-kubectl-prompt:' separator separator
     ZSH_KUBECTL_PROMPT="${context}${separator}${ns}"