Replace home dir with ~

apauley · Jul 26, 2016 · 65e18af · 65e18af
1 parent d5e0573
commit 65e18af
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/PromptLib.hs b/src/PromptLib.hs
@@ -101,10 +101,18 @@ rebaseNeeded currentBranch trackBranch = do
 hostPwd :: IO Text
 hostPwd = do
   h <- hostname
-  w <- pwd
+  w <- promptPwd
   let host = colourUnlessNull lightPurpleFG h
-  let cwd  = colourUnlessNull lightBlueFG   $ format fp w
+  let cwd  = colourUnlessNull lightBlueFG   w
   return $ host <> ":" <> cwd
 
+promptPwd :: IO Text
+promptPwd = do
+  h <- home
+  w <- pwd
+
+  let cwd = sed (text (format fp h) *> return "~") (return (format fp w)) :: Shell Text
+  fmap T.strip $ strict cwd
+
 recentNHashes :: Text -> Int -> Shell Text
 recentNHashes branch limit = gitDiscardErr "log" ["-n", repr limit, "--format=%H", branch]