feat(git_status): dont render module output if not in git repository (#2897)

* refactor get_repo method to not swallow error

* module git status don't render if not in git repository

src/context.rs

@@ -221,29 +221,18 @@ impl<'a> Context<'a> {
     }
 
     /// Will lazily get repo root and branch when a module requests it.
-    pub fn get_repo(&self) -> Result<&Repo, std::io::Error> {
+    pub fn get_repo(&self) -> Result<&Repo, git2::Error> {
-        self.repo
+        self.repo.get_or_try_init(|| -> Result<Repo, git2::Error> {
-            .get_or_try_init(|| -> Result<Repo, std::io::Error> {
             let repository = if env::var("GIT_DIR").is_ok() {
-                    Repository::open_from_env().ok()
+                Repository::open_from_env()
             } else {
-                    Repository::discover(&self.current_dir).ok()
+                Repository::discover(&self.current_dir)
-                };
+            }?;
-                let branch = repository
-                    .as_ref()
-                    .and_then(|repo| get_current_branch(repo));
-                let root = repository
-                    .as_ref()
-                    .and_then(|repo| repo.workdir().map(Path::to_path_buf));
-                let state = repository.as_ref().map(git2::Repository::state);
-                let remote = repository
-                    .as_ref()
-                    .and_then(|repo| get_remote_repository_info(repo));
             Ok(Repo {
-                    branch,
+                branch: get_current_branch(&repository),
-                    root,
+                root: repository.workdir().map(Path::to_path_buf),
-                    state,
+                state: repository.state(),
-                    remote,
+                remote: get_remote_repository_info(&repository),
             })
         })
     }
@@ -397,7 +386,7 @@ pub struct Repo {
     pub root: Option<PathBuf>,
 
     /// State
-    pub state: Option<RepositoryState>,
+    pub state: RepositoryState,
 
     /// Remote repository
     pub remote: Option<Remote>,

src/modules/git_state.rs

@@ -15,7 +15,7 @@ pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
 
     let repo = context.get_repo().ok()?;
     let repo_root = repo.root.as_ref()?;
-    let repo_state = repo.state?;
+    let repo_state = repo.state;
 
     let state_description = get_state_description(repo_state, repo_root, &config)?;
 

src/modules/git_status.rs

@@ -31,6 +31,9 @@ pub fn module<'a>(context: &'a Context) -> Option<Module<'a>> {
     let mut module = context.new_module("git_status");
     let config: GitStatusConfig = GitStatusConfig::try_load(module.config);
 
+    //Return None if not in git repository
+    context.get_repo().ok()?;
+
     let parsed = StringFormatter::new(config.format).and_then(|formatter| {
         formatter
             .map_meta(|variable, _| match variable {