fix: resolve git branch names in reftable repos (#423)

* fix: resolve git branch names in reftable repos

refs #384

* fix: limit git plumbing to reftable refs

refs #384

---------

Co-authored-by: Can Celik <ogulcancelik@gmail.com>

1 files changed, 68 insertions, 4 deletions

diff --git a/src/workspace/git/status.rs b/src/workspace/git/status.rs
index 24e59e57..024ffdde 100644
--- a/src/workspace/git/status.rs
+++ b/src/workspace/git/status.rs
@@ -5,7 +5,8 @@ use crate::workspace::WorkspaceGitStatusSnapshot;
 use super::{
     config::{read_branch_config, upstream_full_ref},
     discovery::{
-        canonicalize_best_effort_path, git_branch, git_space_metadata, git_worktree_info,
+        canonicalize_best_effort_path, git_branch, git_ref_storage_is_reftable,
+        git_rev_parse_verify, git_space_metadata, git_symbolic_head_full, git_worktree_info,
         read_ref_oid, GitWorktreeInfo,
     },
 };
@@ -52,18 +53,18 @@ pub fn git_status_snapshot_for_cwd(
     cwd: &Path,
     cached: Option<&GitStatusCacheEntry>,
 ) -> (WorkspaceGitStatusSnapshot, Option<GitStatusCacheEntry>) {
-    let branch = git_branch(cwd);
     let space = git_space_metadata(cwd);
     let Some(fingerprint) = git_status_fingerprint(cwd) else {
         return (
             WorkspaceGitStatusSnapshot {
-                branch,
+                branch: git_branch(cwd),
                 ahead_behind: None,
                 space,
             },
             None,
         );
     };
+    let branch = fingerprint.branch_name().map(str::to_string);
 
     if let Some(cached) = cached.filter(|entry| entry.fingerprint == fingerprint) {
         let snapshot = WorkspaceGitStatusSnapshot {
@@ -115,6 +116,13 @@ pub(super) fn git_status_fingerprint(cwd: &Path) -> Option<GitStatusFingerprint>
 }
 
 impl GitStatusFingerprint {
+    fn branch_name(&self) -> Option<&str> {
+        match &self.head {
+            GitHeadIdentity::Branch { short_name, .. } => Some(short_name.as_str()),
+            GitHeadIdentity::Detached { .. } => None,
+        }
+    }
+
     fn head_oid(&self) -> Option<&str> {
         match &self.head {
             GitHeadIdentity::Branch { oid, .. } => oid.as_deref(),
@@ -130,6 +138,28 @@ impl GitStatusFingerprint {
 }
 
 fn read_head_identity(info: &GitWorktreeInfo) -> Option<GitHeadIdentity> {
+    if git_ref_storage_is_reftable(&info.git_common_dir) {
+        return read_head_identity_from_git(info);
+    }
+
+    read_head_identity_from_files(info)
+}
+
+fn read_head_identity_from_git(info: &GitWorktreeInfo) -> Option<GitHeadIdentity> {
+    if let Some(full_ref) = git_symbolic_head_full(&info.repo_root) {
+        let short_name = full_ref.strip_prefix("refs/heads/")?.to_string();
+        let oid = git_rev_parse_verify(&info.repo_root, &full_ref);
+        return Some(GitHeadIdentity::Branch {
+            full_ref,
+            short_name,
+            oid,
+        });
+    }
+
+    git_rev_parse_verify(&info.repo_root, "HEAD").map(|oid| GitHeadIdentity::Detached { oid })
+}
+
+fn read_head_identity_from_files(info: &GitWorktreeInfo) -> Option<GitHeadIdentity> {
     let head = std::fs::read_to_string(info.git_dir.join("HEAD")).ok()?;
     let head = head.trim();
     if let Some(full_ref) = head.strip_prefix("ref: ") {
@@ -150,7 +180,11 @@ fn read_head_identity(info: &GitWorktreeInfo) -> Option<GitHeadIdentity> {
 fn read_upstream_identity(info: &GitWorktreeInfo, branch: &str) -> Option<GitUpstreamIdentity> {
     let config = read_branch_config(info, branch)?;
     let full_ref = upstream_full_ref(&config)?;
-    let oid = read_ref_oid(&info.git_common_dir, &full_ref);
+    let oid = if git_ref_storage_is_reftable(&info.git_common_dir) {
+        git_rev_parse_verify(&info.repo_root, &full_ref)
+    } else {
+        read_ref_oid(&info.git_common_dir, &full_ref)
+    };
     Some(GitUpstreamIdentity {
         remote: config.remote,
         merge_ref: config.merge_ref,
@@ -357,6 +391,36 @@ mod tests {
     }
 
     #[test]
+    fn git_status_fingerprint_reads_reftable_branch_identity() {
+        let root = temp_test_dir("reftable-fingerprint");
+        let root_arg = root.to_string_lossy().to_string();
+        let output = std::process::Command::new("git")
+            .args(["init", "--ref-format=reftable", "-b", "main", &root_arg])
+            .output()
+            .unwrap();
+        if !output.status.success() {
+            std::fs::remove_dir_all(root).unwrap();
+            return;
+        }
+        run_git(&root, &["config", "user.email", "herdr@example.invalid"]);
+        run_git(&root, &["config", "user.name", "Herdr Test"]);
+        run_git(&root, &["commit", "--allow-empty", "-m", "initial"]);
+
+        let fingerprint = git_status_fingerprint(&root).unwrap();
+
+        assert_eq!(
+            fingerprint.head,
+            GitHeadIdentity::Branch {
+                full_ref: "refs/heads/main".into(),
+                short_name: "main".into(),
+                oid: git_rev_parse_verify(&root, "HEAD"),
+            }
+        );
+
+        std::fs::remove_dir_all(root).unwrap();
+    }
+
+    #[test]
     fn git_status_recomputes_ahead_behind_when_head_moves() {
         let base = temp_test_dir("head-moves");
         let remote = base.join("remote.git");