From ff37975452d4cbc8a3f117baa9444f3fd023bd66 Mon Sep 17 00:00:00 2001
From: NerdyPepper <akshayoppiliappan@gmail.com>
Date: Tue, 9 Jul 2019 22:56:27 +0530
Subject: handle edge cases

---
 src/cwd.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/cwd.rs b/src/cwd.rs
index 7672981..95b6659 100644
--- a/src/cwd.rs
+++ b/src/cwd.rs
@@ -8,7 +8,11 @@ pub fn cwd() -> colored::ColoredString {
     let tilde_expand = env::var("EXPAND_TILDE").unwrap_or("0".into());
 
     match tilde_expand.as_ref() {
-        "0" => path = path.replace(&home[..], "~"),
+        "0" => {
+            if (&path[..]).starts_with(&home[..]) && home.len() > 0 {
+                path = path.replacen(&home[..], "~", 1);
+            }
+        }
         _ => {}
     };
 
-- 
cgit v1.2.3