From 19a52e06212b4fcd1b6a680240aed467680e86f6 Mon Sep 17 00:00:00 2001
From: Hunter Haugen <hunter@haugen.email>
Date: Wed, 7 Jan 2026 12:47:50 -0800
Subject: [PATCH] Add kubectl shell context to prompt

---
 .zshrc | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/.zshrc b/.zshrc
index 912727d..3f5eca1 100644
--- a/.zshrc
+++ b/.zshrc
@@ -172,7 +172,11 @@ terraform_wrapper() {
 }
 kubectl_wrapper() {
   if type -p kubectl > /dev/null ; then
-    local k8s_context=$(kubectl config current-context 2> /dev/null)
+    if [ -z "$KUBE_CONTEXT" ]; then
+      local k8s_context=$(kubectl config current-context 2> /dev/null)
+    else
+      local k8s_context="$KUBE_CONTEXT"
+    fi
     if [ -z "$k8s_context" ]; then
       return
     fi