display kube context in prompt

Signed-off-by: Takashi Iiguni <[email protected]>
guni1192 · Nov 6, 2024 · d7e7532 · d7e7532
1 parent c01fd3e
commit d7e7532
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/zsh/prompt.zsh b/zsh/prompt.zsh
@@ -15,9 +15,16 @@ auth="%F{white}%K{black}%# %b%k"
 
 precmd () {
   vcs_info
+  export kube_current_context=""
+  context=$(kubectl config current-context)
+  if [[ "$context" =~ .*"prd".* ]] || [[ "$context" =~ .*"production".* ]]; then
+      export kube_current_context="%F{red}%K{black} $context %b%k"
+  else
+      export kube_current_context="%F{yellow}%K{black} $context %b%k"
+  fi
 }
 
 # RPROMPT='${vcs_info_msg_0_}'
 
-PROMPT='$username$directory${vcs_info_msg_0_}$return_code
+PROMPT='$username$directory${vcs_info_msg_0_}${kube_current_context}$return_code
 $auth'