From 98cf251989d482c536d9c59f913bfc6fb1810a1d Mon Sep 17 00:00:00 2001
From: Gigitsu <gigitsu.23@gmail.com>
Date: Fri, 21 Aug 2020 11:37:04 +0200
Subject: [PATCH] Added support for k8s context

---
 agnoster.zsh-theme | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/agnoster.zsh-theme b/agnoster.zsh-theme
index fbf4af19..705e3a7e 100644
--- a/agnoster.zsh-theme
+++ b/agnoster.zsh-theme
@@ -129,6 +129,20 @@ prompt_virtualenv() {
   fi
 }
 
+prompt_kubecontext() {
+  local env='';
+
+  if [[ -n $KUBE_PS1_CONTEXT ]]; then
+    env="$KUBE_PS1_SYMBOL_DEFAULT$KUBE_PS1_SEPARATOR$KUBE_PS1_CONTEXT$KUBE_PS1_DIVIDER$KUBE_PS1_NAMESPACE"
+  fi
+
+  if [[ -n $env ]]; then
+    color=magenta
+    prompt_segment $color $PRIMARY_FG
+    print -Pn " $env "
+  fi
+}
+
 ## Main prompt
 prompt_agnoster_main() {
   RETVAL=$?
@@ -138,6 +152,7 @@ prompt_agnoster_main() {
   prompt_virtualenv
   prompt_dir
   prompt_git
+  prompt_kubecontext
   prompt_end
 }