Add Repository::in_progress_operation()

git-repository/src/lib.rs

@@ -184,7 +184,9 @@ pub enum Path {
 
 ///
 mod types;
-pub use types::{Commit, DetachedObject, Head, Id, Object, Reference, Repository, Tag, ThreadSafeRepository, Tree};
+pub use types::{
+    Commit, DetachedObject, Head, Id, Object, Reference, Repository, RepositoryState, Tag, ThreadSafeRepository, Tree,
+};
 
 pub mod commit;
 pub mod head;

git-repository/src/repository/mod.rs

@@ -78,6 +78,8 @@ mod location;
 
 mod snapshots;
 
+mod state;
+
 mod impls;
 
 mod cache;

git-repository/src/repository/state.rs

@@ -0,0 +1,42 @@
+use crate::RepositoryState;
+
+impl crate::Repository {
+    /// Returns the status of an in progress operation on a repository or [`RepositoryState::None`]
+    /// if nothing is happening.
+    pub fn in_progress_operation(&self) -> RepositoryState {
+        let repo_path = self.path();
+
+        // This is modeled on the logic from wt_status_get_state in git's wt-status.c and
+        // ps1 from git-prompt.sh.
+
+        if repo_path.join("rebase-apply/applying").is_file() {
+            return RepositoryState::ApplyMailbox;
+        } else if repo_path.join("rebase-apply/rebasing").is_file() {
+            return RepositoryState::Rebase;
+        } else if repo_path.join("rebase-apply").is_dir() {
+            return RepositoryState::ApplyMailboxRebase;
+        } else if repo_path.join("rebase-merge/interactive").is_file() {
+            return RepositoryState::RebaseInteractive;
+        } else if repo_path.join("rebase-merge").is_dir() {
+            return RepositoryState::Rebase;
+        } else if repo_path.join("CHERRY_PICK_HEAD").is_file() {
+            if repo_path.join("todo").is_file() {
+                return RepositoryState::CherryPickSequence;
+            } else {
+                return RepositoryState::CherryPick;
+            }
+        } else if repo_path.join("MERGE_HEAD").is_file() {
+            return RepositoryState::Merge;
+        } else if repo_path.join("BISECT_LOG").is_file() {
+            return RepositoryState::Bisect;
+        } else if repo_path.join("REVERT_HEAD").is_file() {
+            if repo_path.join("todo").is_file() {
+                return RepositoryState::RevertSequence;
+            } else {
+                return RepositoryState::Revert;
+            }
+        } else {
+            return RepositoryState::None;
+        }
+    }
+}

git-repository/src/types.rs

@@ -124,6 +124,33 @@ pub struct Repository {
     pub(crate) config: crate::config::Cache,
 }
 
+/// The state of a git repository
+#[derive(Debug, PartialEq)]
+pub enum RepositoryState {
+    /// No operations are in progress
+    None,
+    /// Apply mailbox in progress
+    ApplyMailbox,
+    /// Rebase while an apply mailbox operation is in progress
+    ApplyMailboxRebase,
+    /// Bisect in progress
+    Bisect,
+    /// Cherry pick operation in progress
+    CherryPick,
+    /// Cherry pick with multiple commits pending in the sequencer in progress
+    CherryPickSequence,
+    /// Merge operation in progress
+    Merge,
+    /// Rebase in progress
+    Rebase,
+    /// Interactive rebase in progress
+    RebaseInteractive,
+    /// Revert operation in progress
+    Revert,
+    /// Revert operation with multiple commits pending in the sequencer in progress
+    RevertSequence,
+}
+
 /// An instance with access to everything a git repository entails, best imagined as container implementing `Sync + Send` for _most_
 /// for system resources required to interact with a `git` repository which are loaded in once the instance is created.
 ///