Initial commit

skills/pr-creator/scripts/get_git_status.py

@@ -0,0 +1,67 @@
+#!/usr/bin/env python3
+"""Parallel git status fetcher using asyncio subprocess."""
+
+import asyncio
+import json
+import sys
+from pathlib import Path
+from typing import Any
+
+
+async def run_git_command(command: str) -> tuple[str, str, int]:
+    process = await asyncio.create_subprocess_shell(
+        command,
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.PIPE,
+    )
+    stdout, stderr = await process.communicate()
+    return (
+        stdout.decode("utf-8", errors="replace").strip(),
+        stderr.decode("utf-8", errors="replace").strip(),
+        process.returncode or 0,
+    )
+
+
+async def gather_git_status() -> dict[str, Any]:
+    commands = {
+        "status": "git status",
+        "diff": "git diff",
+        "diff_staged": "git diff --staged",
+        "log": "git log --oneline -10",
+        "current_branch": "git branch --show-current",
+    }
+
+    tasks = {key: run_git_command(cmd) for key, cmd in commands.items()}
+    results = await asyncio.gather(*[tasks[key] for key in commands.keys()])
+
+    output = {}
+    for (key, _), (stdout, stderr, returncode) in zip(commands.items(), results):
+        output[key] = {"stdout": stdout, "stderr": stderr, "returncode": returncode, "success": returncode == 0}
+
+    return output
+
+
+async def main() -> int:
+    try:
+        check_process = await asyncio.create_subprocess_shell(
+            "git rev-parse --git-dir",
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
+        )
+        await check_process.communicate()
+
+        if check_process.returncode != 0:
+            print(json.dumps({"error": "Not a git repository", "cwd": str(Path.cwd())}), file=sys.stderr)
+            return 1
+
+        results = await gather_git_status()
+        print(json.dumps(results, indent=2, ensure_ascii=False))
+        return 0
+
+    except Exception as e:
+        print(json.dumps({"error": str(e), "type": type(e).__name__}), file=sys.stderr)
+        return 1
+
+
+if __name__ == "__main__":
+    sys.exit(asyncio.run(main()))