feat: generate project title for wezterm

pbogut · Jan 18, 2024 · f8c7cbc · f8c7cbc
1 parent c9d019a
commit f8c7cbc
Showing 1 changed file with 26 additions and 4 deletions.
diff --git a/wezterm-status/src/main.rs b/wezterm-status/src/main.rs
@@ -1,3 +1,8 @@
+enum Status {
+    Right,
+    Title,
+}
+
 fn main() {
     let args: Vec<String> = std::env::args().collect();
     let cwd = std::env::current_dir()
@@ -10,6 +15,14 @@ fn main() {
         Some(path) => path,
         None => &cwd,
     };
+    let status_type = match args.get(2) {
+        Some(arg) => match arg.as_str() {
+            "--right" => Status::Right,
+            "--title" => Status::Title,
+            _ => Status::Right,
+        },
+        None => Status::Right,
+    };
 
     let abs_path = cmd_out("git", &["-C", &path, "rev-parse", "--absolute-git-dir"]);
     let top_path = cmd_out("git", &["-C", &path, "rev-parse", "--show-toplevel"]);
@@ -30,10 +43,19 @@ fn main() {
     };
 
     let base_name = path.split("/").last().unwrap();
-    if branch_name != "" {
-        println!("{}", format!(" {}   {}", base_name, branch_name));
-    } else {
-        println!("{}", format!(" {}", base_name));
+    match status_type {
+        Status::Right => {
+            if branch_name != "" {
+                println!("{}", format!(" {}   {}", base_name, branch_name));
+            } else {
+                println!("{}", format!(" {}", base_name));
+            }
+        }
+        Status::Title => {
+            if branch_name != "" {
+                println!("{}", format!("{}  {}", base_name, branch_name));
+            }
+        }
     }
 }