45c77a40cf
Same rule as 6e294f543d for the reads that arrived with the origin/main merge;
the Windows footguns lane is blocking.
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
"""Current git branch for the CLI status bar.
|
|
|
|
Reads ``.git/HEAD`` directly (no subprocess) so status-bar repaints stay cheap, with a
|
|
short per-directory TTL cache. Worktrees and submodules (``.git`` as a ``gitdir:``
|
|
pointer file) resolve to their private git dir, whose HEAD is per-worktree.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import time
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
_TTL_SECONDS = 5.0
|
|
_cache: dict = {}
|
|
|
|
|
|
def _resolve_git_dir(start: Path) -> Optional[Path]:
|
|
"""Nearest enclosing git dir for ``start``, following worktree pointer files."""
|
|
for parent in (start, *start.parents):
|
|
dotgit = parent / ".git"
|
|
if dotgit.is_dir():
|
|
return dotgit
|
|
if dotgit.is_file():
|
|
try:
|
|
line = dotgit.read_text(encoding="utf-8-sig", errors="replace").strip()
|
|
except OSError:
|
|
return None
|
|
if line.startswith("gitdir:"):
|
|
target = (parent / line.split(":", 1)[1].strip()).resolve()
|
|
return target if target.is_dir() else None
|
|
return None
|
|
return None
|
|
|
|
|
|
def current_git_branch(cwd: Optional[str] = None) -> str:
|
|
"""Branch name for ``cwd`` (defaults to the process cwd); ``""`` outside a repo.
|
|
|
|
A detached HEAD renders as the abbreviated commit (``a1b2c3d…``). Results are
|
|
cached ~5s per directory so per-repaint calls never re-walk the tree.
|
|
"""
|
|
try:
|
|
base = Path(cwd or os.getcwd()).resolve()
|
|
except OSError:
|
|
return ""
|
|
key = str(base)
|
|
now = time.monotonic()
|
|
hit = _cache.get(key)
|
|
if hit and now - hit[0] < _TTL_SECONDS:
|
|
return hit[1]
|
|
label = ""
|
|
git_dir = _resolve_git_dir(base)
|
|
if git_dir is not None:
|
|
try:
|
|
head = (git_dir / "HEAD").read_text(encoding="utf-8-sig", errors="replace").strip()
|
|
except OSError:
|
|
head = ""
|
|
if head.startswith("ref:"):
|
|
ref = head.split(":", 1)[1].strip()
|
|
label = ref[len("refs/heads/"):] if ref.startswith("refs/heads/") else ref.rsplit("/", 1)[-1]
|
|
elif head:
|
|
label = f"{head[:8]}…"
|
|
_cache[key] = (now, label)
|
|
return label
|