add trim_prompt option

kkpattern · Apr 8, 2024 · 5800c6b · 5800c6b
1 parent f2057f9
commit 5800c6b
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/.github/config-schema.json b/.github/config-schema.json
@@ -1908,6 +1908,10 @@
       "default": true,
       "type": "boolean"
     },
+    "trim_prompt": {
+      "default": false,
+      "type": "boolean"
+    },
     "follow_symlinks": {
       "default": true,
       "type": "boolean"

diff --git a/src/configs/starship_root.rs b/src/configs/starship_root.rs
@@ -18,6 +18,7 @@ pub struct StarshipRootConfig {
     pub scan_timeout: u64,
     pub command_timeout: u64,
     pub add_newline: bool,
+    pub trim_prompt: bool,
     pub follow_symlinks: bool,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub palette: Option<String>,
@@ -140,6 +141,7 @@ impl Default for StarshipRootConfig {
             scan_timeout: 30,
             command_timeout: 500,
             add_newline: true,
+            trim_prompt: false,
             follow_symlinks: true,
             palette: None,
             palettes: HashMap::default(),

diff --git a/src/print.rs b/src/print.rs
@@ -125,7 +125,12 @@ pub fn get_prompt(context: Context) -> String {
     // color sequences for this specific shell
     let shell_wrapped_output =
         wrap_colorseq_for_shell(AnsiStrings(&module_strings).to_string(), context.shell);
-    write!(buf, "{}", shell_wrapped_output).unwrap();
+
+    if config.trim_prompt {
+        write!(buf, "{}", shell_wrapped_output.trim()).unwrap();
+    } else {
+        write!(buf, "{}", shell_wrapped_output).unwrap();
+    }
 
     if context.target == Target::Right {
         // right prompts generally do not allow newlines