1 files changed, 16 insertions, 0 deletions

diff --git a/src/lib/git/mod.rs b/src/lib/git/mod.rs
index b9711ae..516d9e2 100644
--- a/src/lib/git/mod.rs
+++ b/src/lib/git/mod.rs
@@ -91,6 +91,7 @@ pub trait RepoActions {
     ) -> Result<Oid>;
     fn parse_starting_commits(&self, starting_commits: &str) -> Result<Vec<Sha1Hash>>;
     fn ancestor_of(&self, decendant: &Sha1Hash, ancestor: &Sha1Hash) -> Result<bool>;
+    fn get_upstream_for_branch(&self, branch_name: &str) -> Result<Option<String>>;
     fn get_git_config_item(&self, item: &str, global: Option<bool>) -> Result<Option<String>>;
     fn save_git_config_item(&self, item: &str, value: &str, global: bool) -> Result<()>;
     fn remove_git_config_item(&self, item: &str, global: bool) -> Result<bool>;
@@ -734,6 +735,21 @@ impl RepoActions for Repo {
         }
     }
 
+    fn get_upstream_for_branch(&self, branch_name: &str) -> Result<Option<String>> {
+        let branch = self
+            .git_repo
+            .find_branch(branch_name, git2::BranchType::Local)
+            .context(format!("failed to find local branch {branch_name}"))?;
+        let upstream = branch.upstream();
+        match upstream {
+            Ok(upstream_branch) => {
+                let name = upstream_branch.name()?.map(|s| s.to_string());
+                Ok(name)
+            }
+            Err(_) => Ok(None),
+        }
+    }
+
     /// setting global to None will suppliment local config with global items
     /// not in local
     fn get_git_config_item(&self, item: &str, global: Option<bool>) -> Result<Option<String>> {