fixed: panics on empty local bare repos

1 files changed, 4 insertions, 1 deletions

diff --git a/src/vcs.rs b/src/vcs.rs
index 5bc194d..26edf81 100644
--- a/src/vcs.rs
+++ b/src/vcs.rs
@@ -21,7 +21,10 @@ pub fn vcs_status() -> Option<(colored::ColoredString, colored::ColoredString)>
         return None
     }
     let repo = repo.unwrap();
-    let reference = repo.head().unwrap();
+    let reference = match repo.head() {
+        Ok(r) => r,
+        Err(_) => return None
+    };
     let mut branch;
 
     if reference.is_branch() {