get worktree status in parallel

This can take a long time on massive repos.

Author: Byron

src/info/mod.rs

@@ -1,7 +1,7 @@
 use crate::cli::{self, Config};
 use crate::ui::get_ascii_colors;
 use crate::ui::text_colors::TextColors;
-use anyhow::Result;
+use anyhow::{Context, Result};
 use author::Author;
 use deps::DependencyDetector;
 use git2::Repository;
@@ -166,8 +166,15 @@ impl Info {
             config.number_of_authors,
         )?;
         let (repo_name, repo_url) = internal_repo.get_name_and_url()?;
+        let pending_changes = std::thread::spawn({
+            let git_dir = repo.path().to_owned();
+            move || {
+                let repo = git2::Repository::open(git_dir)?;
+                repo::get_pending_changes(&repo)
+            }
+        });
+
         let head_refs = internal_repo.get_head_refs()?;
-        let pending_changes = internal_repo.get_pending_changes()?;
         let version = internal_repo.get_version()?;
         let git_username = internal_repo.get_git_username()?;
         let number_of_tags = internal_repo.get_number_of_tags()?;
@@ -202,7 +209,10 @@ impl Info {
             number_of_tags,
             number_of_branches,
             head_refs,
-            pending_changes,
+            pending_changes: pending_changes
+                .join()
+                .ok()
+                .context("BUG: panic in pending-changes thread")??,
             version,
             creation_date,
             languages,

src/info/repo.rs

@@ -205,46 +205,6 @@ impl<'a> Repo<'a> {
         Ok(version_name)
     }
 
-    pub fn get_pending_changes(&self) -> Result<String> {
-        let statuses = self.git2_repo.statuses(Some(
-            StatusOptions::default()
-                .show(StatusShow::Workdir)
-                .update_index(true)
-                .include_untracked(true)
-                .renames_head_to_index(true)
-                .recurse_untracked_dirs(true),
-        ))?;
-
-        let (added, deleted, modified) =
-            statuses
-                .iter()
-                .fold((0, 0, 0), |(added, deleted, modified), e| {
-                    let s: Status = e.status();
-                    if s.is_index_new() || s.is_wt_new() {
-                        (added + 1, deleted, modified)
-                    } else if s.is_index_deleted() || s.is_wt_deleted() {
-                        (added, deleted + 1, modified)
-                    } else {
-                        (added, deleted, modified + 1)
-                    }
-                });
-
-        let mut result = String::new();
-        if modified > 0 {
-            result = format!("{}+-", modified)
-        }
-
-        if added > 0 {
-            result = format!("{} {}+", result, added);
-        }
-
-        if deleted > 0 {
-            result = format!("{} {}-", result, deleted);
-        }
-
-        Ok(result.trim().into())
-    }
-
     pub fn get_name_and_url(&self) -> Result<(String, String)> {
         let config = self.git2_repo.config()?;
         let mut remote_origin_url: Option<String> = None;
@@ -397,3 +357,43 @@ mod tests {
         assert_eq!(result, "1970-01-01T00:00:00Z");
     }
 }
+
+pub fn get_pending_changes(repo: &git2::Repository) -> Result<String> {
+    let statuses = repo.statuses(Some(
+        StatusOptions::default()
+            .show(StatusShow::Workdir)
+            .update_index(true)
+            .include_untracked(true)
+            .renames_head_to_index(true)
+            .recurse_untracked_dirs(true),
+    ))?;
+
+    let (added, deleted, modified) =
+        statuses
+            .iter()
+            .fold((0, 0, 0), |(added, deleted, modified), e| {
+                let s: Status = e.status();
+                if s.is_index_new() || s.is_wt_new() {
+                    (added + 1, deleted, modified)
+                } else if s.is_index_deleted() || s.is_wt_deleted() {
+                    (added, deleted + 1, modified)
+                } else {
+                    (added, deleted, modified + 1)
+                }
+            });
+
+    let mut result = String::new();
+    if modified > 0 {
+        result = format!("{}+-", modified)
+    }
+
+    if added > 0 {
+        result = format!("{} {}+", result, added);
+    }
+
+    if deleted > 0 {
+        result = format!("{} {}-", result, deleted);
+    }
+
+    Ok(result.trim().into())
+}