From e43fd9d7a1ab7d818c65b6400dfabccb6dc2c4bb Mon Sep 17 00:00:00 2001
From: candifloss <candifloss@candifloss.cc>
Date: Sat, 17 Aug 2024 23:57:14 +0530
Subject: [PATCH] Git repo name & path abrev

---
 src/main.rs | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 5b7a17c..22abac3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,7 +1,7 @@
 use std::env::{current_dir,var_os,args};
-use std::path::PathBuf;
 use colored::{Colorize,ColoredString};
 use std::process::Command;
+//use std::path::PathBuf;
 //use std::ffi::OsString;
 
 fn get_shell_char (shell: String) -> String {
@@ -71,15 +71,14 @@ fn get_git_repo_name (git_repo_root: String) -> String {
 
 fn get_git_char (git_branch: String) -> ColoredString {
     match git_branch.as_str() {
+        ""   => "".clear(),
         "main"   => " 󰊢 ".truecolor(178,98,44),
         "master" => " 󰊢 ".truecolor(196,132,29),
         _        => " 󰊢 ".truecolor(82,82,82),
     }
 }
 
-fn abrev_path (pwd: PathBuf, homedir: String) -> String {
-    let mut path = pwd.display().to_string();   
-    path = path.replace(&homedir, "~");    // Abreviate homedir with "~"
+fn abrev_path (path: String) -> String {
     let mut short_dir = path.clone();
 
     let slashes = path.matches("/").count();
@@ -106,13 +105,9 @@ fn main() -> std::io::Result<()> {
     let _user = var_os("USER").expect("UnknownUser").to_str().expect("UnknownUser").to_string();
     /*
     if user == "root" {
-        println!("roo_user");
+        println!("root_user");
     }*/
 
-    //pwd
-    let pwd = current_dir()?;
-    let homedir = var_os("HOME").expect("UnknownDir").to_str().expect("UnknownDir").to_string();
-    
     //Shell symbol
     let args: Vec<String> = args().collect();
     let shell: String;
@@ -137,15 +132,24 @@ fn main() -> std::io::Result<()> {
     //Git status
     let git_branch      = get_git_branch();//.clear();
     let git_repo_root = get_git_root();
-    let git_repo_name = get_git_repo_name(git_repo_root);
+    let git_repo_name = get_git_repo_name(git_repo_root.clone()).truecolor(200, 150, 80);
     let git_char = get_git_char(git_branch); //.to_string());
+
+    //pwd
+    let homedir = var_os("HOME").expect("UnknownDir").to_str().expect("UnknownDir").to_string();
+    let pwd = current_dir()?;
+    let mut cur_dir = pwd.display().to_string();
+    cur_dir = cur_dir.replace(&git_repo_root, "");    // Remove git repo root
+    cur_dir = cur_dir.replace(&homedir, "~");    // Abreviate homedir with "~"
+    cur_dir = abrev_path(cur_dir);
     
-    print!("{}{}{}{} {}{} ",
+    print!("{}{}{}{}{}{} ",
                 ssh_char,
                 get_shell_char(shell).truecolor(75,75,75),
                 git_repo_name,
                 git_char,
-                abrev_path(pwd,homedir).italic().truecolor(82,82,82),
+                //abrev_path(pwd,homedir).italic().truecolor(82,82,82),
+                cur_dir.italic().truecolor(82,82,82),
                 angle.truecolor(0, 255, 180),
     );