From 2af32850667abfc294ced76d5dfd866112f16b64 Mon Sep 17 00:00:00 2001
From: Ustun Ozgur <ustun@braid.health>
Date: Thu, 16 Dec 2021 14:21:04 +0300
Subject: [PATCH] add ability to toggle prompt by "kube_ps" without flags

Without any arguments, it toggles the env var and handles the initial
invocation.
---
 README.md              | 12 ++++++++++++
 functions/kube_ps.fish |  8 ++++++++
 2 files changed, 20 insertions(+)

diff --git a/README.md b/README.md
index cc99277..2b14323 100644
--- a/README.md
+++ b/README.md
@@ -23,6 +23,13 @@ function fish_prompt
 end
 ```
 
+Finally, enable the prompt by executing this once (the setting will be preserved
+across restarts of the shell.)
+
+```fish
+kube_fs
+```
+
 ## Speed
 
 Running the `kubectl` commands to get the context and namespace every time
@@ -38,6 +45,11 @@ This was not my idea, I took it from `jonmosco/kube-ps1`.
 
 You can toggle the prompt on or off like this:
 
+```fish
+kube_ps
+```
+
+You can enable or disable the prompt like this:
 ```fish
 kube_ps on
 kube_ps off
diff --git a/functions/kube_ps.fish b/functions/kube_ps.fish
index 6da506b..7c04e95 100644
--- a/functions/kube_ps.fish
+++ b/functions/kube_ps.fish
@@ -8,4 +8,12 @@ function kube_ps -a toggle
     set -U __kube_ps_enabled 0
     return
   end
+
+  if test "$__kube_ps_enabled" = 1
+    set -U __kube_ps_enabled 0
+    return
+  end
+
+  set -U __kube_ps_enabled 1
+
 end