Add setting to disable kubie's prompt

README.md

@@ -92,6 +92,11 @@ configs:
 
 # Prompt settings.
 prompt:
+    # Disable kubie's custom prompt inside of a kubie shell. This is useful
+    # when you already have a prompt displaying kubernetes information.
+    # Default: false
+    disable: true
+
     # When using recursive contexts, show depth when larger than 1.
     # Default: true
     show_depth: true

src/settings.rs

@@ -123,6 +123,8 @@ fn default_exclude_path() -> Vec<String> {
 
 #[derive(Debug, Deserialize)]
 pub struct Prompt {
+    #[serde(default = "def_bool_false")]
+    pub disable: bool,
     #[serde(default = "def_bool_true")]
     pub show_depth: bool,
     #[serde(default = "def_bool_false")]
@@ -132,6 +134,7 @@ pub struct Prompt {
 impl Default for Prompt {
     fn default() -> Self {
         Prompt {
+            disable: false,
             show_depth: true,
             zsh_use_rps1: false,
         }

src/shell/bash.rs

@@ -25,13 +25,21 @@ function __kubie_cmd_pre_exec__() {{
 }}
 
 trap '__kubie_cmd_pre_exec__' DEBUG
+"#
+    )?;
 
+    if !info.settings.prompt.disable {
+        write!(
+            temp_rc_file,
+            r#"
 KUBIE_PROMPT='{}'
 PS1="$KUBIE_PROMPT $PS1"
 unset KUBIE_PROMPT
 "#,
-        info.prompt,
-    )?;
+            info.prompt,
+        )?;
+    }
+
     temp_rc_file.flush()?;
 
     let mut cmd = Command::new("bash");

src/shell/mod.rs

@@ -35,7 +35,8 @@ impl<'n> EnvVars<'n> {
     }
 }
 
-pub struct ShellSpawnInfo<'n> {
+pub struct ShellSpawnInfo<'s, 'n> {
+    settings: &'s Settings,
     env_vars: EnvVars<'n>,
     prompt: String,
 }
@@ -62,13 +63,17 @@ pub fn spawn_shell(settings: &Settings, config: KubeConfig, session: &Session) -
     let next_depth = depth + 1;
 
     let mut env_vars = EnvVars::new();
-    // pre-insert the KUBECONFIG variable into the shell.
-    // This will make sure any shell plugins/add-ons which require this env variable will have it available at the beginninng of the .rc file
+
+    // Pre-insert the KUBECONFIG variable into the shell.
+    // This will make sure any shell plugins/add-ons which require this env variable
+    // will have it available at the beginninng of the .rc file
     env_vars.insert("KUBECONFIG", temp_config_file.path());
     env_vars.insert("KUBIE_ACTIVE", "1");
     env_vars.insert("KUBIE_DEPTH", next_depth.to_string());
     env_vars.insert("KUBIE_KUBECONFIG", temp_config_file.path());
     env_vars.insert("KUBIE_SESSION", temp_session_file.path());
+
+    env_vars.insert("KUBIE_PROMPT_DISABLE", if settings.prompt.disable { "1" } else { "0" });
     env_vars.insert(
         "KUBIE_ZSH_USE_RPS1",
         if settings.prompt.zsh_use_rps1 { "1" } else { "0" },
@@ -87,6 +92,7 @@ pub fn spawn_shell(settings: &Settings, config: KubeConfig, session: &Session) -
     }
 
     let info = ShellSpawnInfo {
+        settings,
         env_vars,
         prompt: prompt::generate_ps1(settings, next_depth, kind),
     };

src/shell/zsh.rs

@@ -28,7 +28,13 @@ function __kubie_cmd_pre_exec__() {{
 }}
 
 add-zsh-hook preexec __kubie_cmd_pre_exec__
+"#,
+        )?;
 
+        if !info.settings.prompt.disable {
+            write!(
+                zshrc,
+                r#"
 # Activate prompt substitution.
 setopt PROMPT_SUBST
 
@@ -62,8 +68,9 @@ function __kubie_cmd_pre_cmd__() {{
 # also add our own precmd hook which modifies PS1 after promptinit themes.
 add-zsh-hook precmd __kubie_cmd_pre_cmd__
 "#,
-            info.prompt
-        )?;
+                info.prompt
+            )?;
+        }
     }
 
     let mut cmd = Command::new("zsh");