#!/usr/bin/env python3 """ Working Tree Review Script Reviews all git branches, stashes, and working trees for the repository """ import subprocess import sys from pathlib import Path from datetime import datetime from collections import defaultdict REPO_ROOT = Path(__file__).resolve().parents[1] # ⚡ Bolt: Global cache for branch metadata to avoid redundant git subprocess calls. BRANCH_METADATA_CACHE = {} def get_git_version(): """⚡ Bolt: Get the git version as a tuple of integers.""" res = run_git_command(["version"]) if not res or res.returncode != 0: return (0, 0, 0) # Extract version numbers from 'git version 2.41.0' try: ver_str = res.stdout.strip().split()[-1] return tuple(int(x) for x in ver_str.split('.')[:3] if x.isdigit()) except (IndexError, ValueError): return (0, 0, 0) def run_git_command(cmd, capture_output=True): """Run a git command and return the result.""" try: result = subprocess.run( ["git"] + cmd, cwd=REPO_ROOT, capture_output=capture_output, text=True, timeout=30, encoding='utf-8', errors='replace' ) return result except subprocess.TimeoutExpired: return None except Exception as e: print(f"Error running git command: {e}", file=sys.stderr) return None def get_branch_info(): """⚡ Bolt: Get information about all branches using optimized bulk calls.""" print("=" * 80) print("BRANCH REVIEW") print("=" * 80) print() # Detect current branch correctly current_branch = "" res = run_git_command(["symbolic-ref", "--short", "HEAD"]) if res and res.returncode == 0: current_branch = res.stdout.strip() # ⚡ Bolt: Check Git version and main branch availability for optimal querying. git_ver = get_git_version() use_ahead_behind = git_ver >= (2, 41, 0) main_exists = run_git_command(["rev-parse", "--verify", "main"]).returncode == 0 # ⚡ Bolt: Consolidate branch calls into one 'for-each-ref' call. # Use multiple refs to include all remotes, not just 'origin'. fmt = "%(refname:short)" if use_ahead_behind and main_exists: fmt += "|%(ahead-behind:main)" result = run_git_command(["for-each-ref", f"--format={fmt}", "refs/heads", "refs/remotes"]) if not result or result.returncode != 0: return local_branches = [] remote_branches = [] merged_branches = set() unmerged_branches = [] for line in result.stdout.strip().split("\n"): if not line: continue parts = line.split("|") name = parts[0] # Ignore HEAD and the main branch itself if "HEAD" in name or name == "main" or "/main" in name: continue is_remote = name.startswith("refs/remotes/") or "/" in name if is_remote: remote_branches.append(name) else: local_branches.append(name) # If we have ahead-behind info, track merge status if len(parts) > 1: counts = parts[1].split() if len(counts) == 2 and int(counts[0]) == 0: merged_branches.add(name) else: if is_remote: unmerged_branches.append(name) # ⚡ Bolt: Fallback for older Git versions without ahead-behind support. if (not use_ahead_behind or not main_exists) and main_exists: merged_res = run_git_command(["branch", "-r", "--merged", "main"]) if merged_res and merged_res.returncode == 0: for b in merged_res.stdout.strip().split("\n"): merged_branches.add(b.strip()) # Populate unmerged_branches based on merged set for remotes for b in remote_branches: if b not in merged_branches: unmerged_branches.append(b) # Print Local Branches if local_branches: print(f"📌 Local Branches: {len(local_branches)}") for branch in local_branches: current = "*" if branch == current_branch else " " print(f" {current} {branch}") print() # Print Remote Branches if remote_branches: print(f"🌐 Remote Branches: {len(remote_branches)}") branch_groups = defaultdict(list) for branch in remote_branches: parts = branch.replace("origin/", "").split("/") prefix = parts[0] if len(parts) > 1 else "other" branch_groups[prefix].append(branch) for prefix, branches in sorted(branch_groups.items()): print(f"\n {prefix.upper()}: {len(branches)} branches") for branch in branches[:5]: print(f" - {branch}") if len(branches) > 5: print(f" ... and {len(branches) - 5} more") print() # Print Merged Branches if merged_branches and main_exists: # Remote merged branches remote_merged = [b for b in merged_branches if b.startswith("origin/") or "/" in b] if remote_merged: print(f"✅ Merged into main: {len(remote_merged)} branches") print(" (These can potentially be deleted)") print() # Print Unmerged Branches if unmerged_branches and main_exists: print(f"⚠️ Not merged into main: {len(unmerged_branches)} branches") print(" (These may contain unmerged changes)") print() def get_stash_info(): """Get information about stashes.""" print("=" * 80) print("STASH REVIEW") print("=" * 80) print() result = run_git_command(["stash", "list"]) if result and result.returncode == 0 and result.stdout.strip(): stashes = result.stdout.strip().split("\n") print(f"📦 Stashes: {len(stashes)}") for stash in stashes: print(f" - {stash}") else: print("📦 No stashes found") print() def get_worktree_info(): """Get information about git worktrees.""" print("=" * 80) print("WORKTREE REVIEW") print("=" * 80) print() result = run_git_command(["worktree", "list"]) if result and result.returncode == 0: worktrees = [w.strip() for w in result.stdout.strip().split("\n") if w.strip()] print(f"🌳 Worktrees: {len(worktrees)}") for worktree in worktrees: print(f" - {worktree}") else: print("🌳 No additional worktrees found") print() def get_status_info(): """⚡ Bolt: Get current working tree status in a single consolidated call.""" print("=" * 80) print("WORKING TREE STATUS") print("=" * 80) print() # ⚡ Bolt: Use a single 'status -sb' call to retrieve both tracking info and changes. result = run_git_command(["status", "-sb"]) if result and result.returncode == 0: lines = result.stdout.strip().split("\n") if not lines: return # First line is always the branch/tracking info branch_line = lines[0] # Remaining lines are uncommitted changes uncommitted_changes = lines[1:] if uncommitted_changes: print("⚠️ Uncommitted changes:") print("\n".join(uncommitted_changes)) else: print("✅ Working tree is clean") print() if "ahead" in branch_line or "behind" in branch_line: print(f"📊 {branch_line}") print() def get_unpushed_commits(): """Get commits that haven't been pushed.""" print("=" * 80) print("UNPUSHED COMMITS") print("=" * 80) print() result = run_git_command(["log", "--oneline", "origin/main..HEAD"]) if result and result.returncode == 0 and result.stdout.strip(): commits = result.stdout.strip().split("\n") print(f"📤 Unpushed commits: {len(commits)}") for commit in commits: print(f" - {commit}") else: print("✅ All commits are pushed") print() def get_recent_activity(): """Get recent commit activity.""" print("=" * 80) print("RECENT ACTIVITY") print("=" * 80) print() result = run_git_command(["log", "--all", "--oneline", "--graph", "--decorate", "-15"]) if result and result.returncode == 0: print(result.stdout) print() def main(): """Main review function.""" print("\n" + "=" * 80) print(f"WORKING TREE REVIEW REPORT") print(f"Generated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") print("=" * 80) print() get_status_info() get_unpushed_commits() get_branch_info() get_stash_info() get_worktree_info() get_recent_activity() print("=" * 80) print("REVIEW COMPLETE") print("=" * 80) print() print("Recommendations:") print("1. Push any unpushed commits") print("2. Review and potentially delete merged branches") print("3. Consider merging or closing unmerged branches") print("4. Clean up old stashes if no longer needed") print() if __name__ == "__main__": main()