From 6276c8206d85cc4939df72d71a4c15e685e21f2b Mon Sep 17 00:00:00 2001
From: Jesse Luehrs <doy@tozt.net>
Date: Sat, 24 Feb 2018 02:47:06 -0500
Subject: display uninitialized repositories a bit better

---
 src/prompt.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/src/prompt.rs b/src/prompt.rs
index 940029f..374d143 100644
--- a/src/prompt.rs
+++ b/src/prompt.rs
@@ -283,6 +283,7 @@ fn format_vcs(vcs_info: &Option<Box<vcs::VcsInfo>>) -> Option<String> {
         }
         if !vcs_info.has_commits() {
             write!(vcs, "!").unwrap();
+            return vcs;
         }
 
         let branch = vcs_info.branch().map(|branch| {
@@ -656,5 +657,22 @@ mod test {
                 Some(String::from("g*+?:dev:-"))
             )
         }
+        {
+            let test_vcs = TestVcs {
+                vcs: vcs::VcsType::Git,
+                has_modified_files: false,
+                has_staged_files: false,
+                has_new_files: false,
+                has_commits: false,
+                active_operation: vcs::ActiveOperation::None,
+                branch: None,
+                remote_branch_diff: None,
+            };
+
+            assert_eq!(
+                format_vcs(&Some(Box::new(test_vcs))),
+                Some(String::from("g!"))
+            )
+        }
     }
 }
-- 
cgit v1.2.3-54-g00ecf