#!/usr/bin/env python3 """ Pull Request Review Script - Optimized Version ⚡ Reviews all pull requests and creates a comprehensive summary. Uses git for-each-ref for O(1) subprocess calls instead of O(N). """ import subprocess import sys import json from pathlib import Path from datetime import datetime from collections import defaultdict REPO_ROOT = Path(__file__).resolve().parents[1] def run_command(cmd, capture_output=True): """Run a command and return the result.""" try: result = subprocess.run( cmd, cwd=REPO_ROOT, capture_output=capture_output, text=True, timeout=30, encoding='utf-8', errors='replace' ) return result except Exception as e: print(f"Error running command: {e}", file=sys.stderr) return None def get_prs_via_gh_cli(): """Get PRs using GitHub CLI.""" result = run_command(["gh", "pr", "list", "--state", "all", "--json", "number,title,state,author,createdAt,updatedAt,headRefName,baseRefName,isDraft,labels"]) if result and result.returncode == 0: try: return json.loads(result.stdout) except json.JSONDecodeError: return [] return None def get_git_branch_metadata(): """Get metadata for all remote branches in a single git call.""" # ⚡ Bolt: Using git for-each-ref with ahead-behind atom (Git 2.41+) # to fetch all metadata in one subprocess call. cmd = [ "git", "for-each-ref", "--format=%(refname:short)|%(ahead-behind:main)|%(committerdate:iso8601)|%(subject)", "refs/remotes/origin" ] result = run_command(cmd) if not result or result.returncode != 0: return None metadata = [] for line in result.stdout.strip().split("\n"): if not line: continue try: # Use split('|', 3) to handle pipe characters that might be in the commit subject parts = line.split("|", 3) if len(parts) < 4: continue full_name = parts[0] # Filter out HEAD, main, and the shorthand 'origin' if full_name in ("origin", "origin/main", "origin/HEAD") or "/HEAD" in full_name: continue ahead_behind = parts[1].split() if len(ahead_behind) < 2: continue ahead = int(ahead_behind[0]) behind = int(ahead_behind[1]) last_commit_date = parts[2] subject = parts[3] metadata.append({ "full_name": full_name, "branch": full_name.replace("origin/", ""), "commit_count": ahead, # Equivalent to commits since main "ahead": ahead, "behind": behind, "last_commit_date": last_commit_date, "subject": subject }) except (ValueError, IndexError): continue return metadata def analyze_branch_name(branch_name): """Analyze branch name to extract PR information.""" branch = branch_name.replace("origin/", "") info = { "type": "unknown", "category": "other", "description": branch } # Categorize branches if branch.startswith("Cursor/"): info["type"] = "cursor" info["category"] = "ai-generated" info["description"] = branch.replace("Cursor/A6-9V/", "") elif branch.startswith("copilot/"): info["type"] = "copilot" info["category"] = "ai-generated" info["description"] = branch.replace("copilot/", "") elif branch.startswith("bolt-"): info["type"] = "bolt" info["category"] = "optimization" info["description"] = branch.replace("bolt-", "") elif branch.startswith("feat/"): info["type"] = "feature" info["category"] = "feature" info["description"] = branch.replace("feat/", "") elif branch.startswith("feature/"): info["type"] = "feature" info["category"] = "feature" info["description"] = branch.replace("feature/", "") return info def main(): """Main review function.""" print("=" * 80) print("PULL REQUEST REVIEW") print(f"Generated: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}") print("=" * 80) print() # Try GitHub CLI first prs = get_prs_via_gh_cli() if prs is not None: print(f"Found {len(prs)} pull requests via GitHub CLI") print() # Group by state by_state = defaultdict(list) for pr in prs: by_state[pr.get("state", "unknown")].append(pr) print("Pull Requests by State:") for state, pr_list in sorted(by_state.items()): print(f" {state.upper()}: {len(pr_list)}") print() # Show open PRs open_prs = by_state.get("OPEN", []) if open_prs: print("=" * 80) print("OPEN PULL REQUESTS") print("=" * 80) for pr in open_prs: print(f"\nPR #{pr.get('number', 'N/A')}: {pr.get('title', 'No title')}") print(f" Author: {pr.get('author', {}).get('login', 'Unknown')}") print(f" Branch: {pr.get('headRefName', 'N/A')} -> {pr.get('baseRefName', 'main')}") print(f" Created: {pr.get('createdAt', 'N/A')}") print(f" Updated: {pr.get('updatedAt', 'N/A')}") print(f" Draft: {'Yes' if pr.get('isDraft') else 'No'}") labels = [l.get('name') for l in pr.get('labels', [])] if labels: print(f" Labels: {', '.join(labels)}") # Show merged PRs merged_prs = by_state.get("MERGED", []) if merged_prs: print("\n" + "=" * 80) print(f"MERGED PULL REQUESTS ({len(merged_prs)} total)") print("=" * 80) print(f"\nShowing last 10 merged PRs:") for pr in merged_prs[-10:]: print(f" PR #{pr.get('number', 'N/A')}: {pr.get('title', 'No title')}") else: # Fallback to git branch analysis print("GitHub CLI not available, analyzing branches...") print() branch_metadata = get_git_branch_metadata() if branch_metadata is None: print("Error: Could not retrieve git branch metadata.") return open_branches = [m for m in branch_metadata if m["ahead"] > 0] merged_branches = [m for m in branch_metadata if m["ahead"] == 0] print(f"Open branches (potential PRs): {len(open_branches)}") print(f"Merged branches (completed PRs): {len(merged_branches)}") print() # Categorize open branches categories = defaultdict(list) for metadata in open_branches: info = analyze_branch_name(metadata["full_name"]) categories[info["category"]].append((metadata, info)) print("=" * 80) print("OPEN BRANCHES (Potential Pull Requests)") print("=" * 80) print() for category, branches in sorted(categories.items()): print(f"{category.upper()}: {len(branches)} branches") for metadata, info in branches[:10]: # Show first 10 print(f" - {info['description']}") print(f" Branch: {metadata['branch']}") print(f" Commits: {metadata['commit_count']}") if metadata['last_commit_date']: print(f" Last commit: {metadata['last_commit_date']}") if len(branches) > 10: print(f" ... and {len(branches) - 10} more") print() print("=" * 80) print("MERGED BRANCHES (Completed Pull Requests)") print("=" * 80) print(f"\nTotal merged: {len(merged_branches)}") print("\nRecent merged branches:") # Show last 20 merged branches sorted_merged = sorted(merged_branches, key=lambda x: x["last_commit_date"], reverse=True) for metadata in sorted_merged[:20]: info = analyze_branch_name(metadata["full_name"]) print(f" - {info['description']}") print("\n" + "=" * 80) print("REVIEW COMPLETE") print("=" * 80) print("\nNote: GitHub doesn't support 'pinning' pull requests directly.") print("Consider:") print("1. Creating a tracking issue for important PRs") print("2. Using labels to categorize PRs") print("3. Adding PRs to project boards") print("4. Creating a PR summary document") if __name__ == "__main__": main()