Initial commit

This commit is contained in:
Zhongwei Li
2025-11-29 18:26:35 +08:00
commit 54e348f7a1
14 changed files with 907 additions and 0 deletions

View File

@@ -0,0 +1,135 @@
#!/usr/bin/env python3
"""PreToolUse hook: show confirmation modal before creating GitHub PR via gh CLI."""
import json
import re
import subprocess
import sys
def parse_gh_pr_create(command: str) -> dict[str, str]:
"""Parse gh pr create command to extract PR parameters.
Args:
command (str): The gh pr create command string
Returns:
(dict): Dictionary with title, body, assignee, reviewer keys
"""
params = {"title": "", "body": "", "assignee": "", "reviewer": ""}
# Extract title (-t or --title)
title_match = re.search(r'(?:-t|--title)\s+["\']([^"\']+)["\']', command)
if title_match:
params["title"] = title_match.group(1)
# Extract body (-b or --body) - handle HEREDOC syntax first, then simple quotes
heredoc_match = re.search(
r'(?:-b|--body)\s+"?\$\(cat\s+<<["\']?(\w+)["\']?\s+(.*?)\s+\1\s*\)"?',
command,
re.DOTALL,
)
if heredoc_match:
params["body"] = heredoc_match.group(2).strip()
else:
body_match = re.search(r'(?:-b|--body)\s+"([^"]+)"', command)
if body_match:
params["body"] = body_match.group(1)
# Extract assignee (-a or --assignee)
assignee_match = re.search(r'(?:-a|--assignee)\s+([^\s]+)', command)
if assignee_match:
params["assignee"] = assignee_match.group(1)
# Extract reviewer (-r or --reviewer)
reviewer_match = re.search(r'(?:-r|--reviewer)\s+([^\s]+)', command)
if reviewer_match:
params["reviewer"] = reviewer_match.group(1)
return params
def resolve_username(assignee: str) -> str:
"""Resolve @me to actual GitHub username.
Args:
assignee (str): Assignee value from command (may be @me)
Returns:
(str): Resolved username or original value
"""
if assignee == "@me":
try:
result = subprocess.run(
["gh", "api", "user", "--jq", ".login"],
capture_output=True,
text=True,
timeout=5,
)
if result.returncode == 0:
return result.stdout.strip()
except (subprocess.TimeoutExpired, FileNotFoundError):
pass
return assignee
def format_confirmation_message(params: dict[str, str]) -> str:
"""Format PR parameters into readable confirmation message.
Args:
params (dict): Dictionary with title, body, assignee, reviewer
Returns:
(str): Formatted confirmation message
"""
# Truncate body if too long
body = params["body"]
if len(body) > 500:
body = body[:500] + "..."
# Resolve assignee
assignee = resolve_username(params["assignee"]) if params["assignee"] else "None"
lines = ["📝 Create Pull Request?", "", f"Title: {params['title']}", ""]
if body:
lines.extend(["Body:", body, ""])
lines.append(f"Assignee: {assignee}")
if params["reviewer"]:
lines.append(f"Reviewer: {params['reviewer']}")
return "\n".join(lines)
try:
input_data = json.load(sys.stdin)
except json.JSONDecodeError as e:
print(f"Error: Invalid JSON input: {e}", file=sys.stderr)
sys.exit(1)
tool_name = input_data.get("tool_name", "")
tool_input = input_data.get("tool_input", {})
command = tool_input.get("command", "")
# Only handle gh pr create commands
if tool_name != "Bash" or not command.strip().startswith("gh pr create"):
sys.exit(0)
# Parse PR parameters
params = parse_gh_pr_create(command)
# Format confirmation message
message = format_confirmation_message(params)
# Return JSON with ask decision
output = {
"hookSpecificOutput": {
"hookEventName": "PreToolUse",
"permissionDecision": "ask",
"permissionDecisionReason": message,
}
}
print(json.dumps(output))
sys.exit(0)

View File

@@ -0,0 +1,162 @@
#!/usr/bin/env python3
"""PreToolUse hook: show confirmation modal before creating git commit."""
import json
import re
import subprocess
import sys
def parse_git_commit_message(command: str) -> dict[str, str]:
"""Parse git commit command to extract commit message.
Args:
command (str): The git commit command string
Returns:
(dict): Dictionary with message and is_amend keys
"""
params = {"message": "", "is_amend": False}
# Check for --amend flag
params["is_amend"] = "--amend" in command
# Try to extract heredoc format: git commit -m "$(cat <<'EOF' ... EOF)"
heredoc_match = re.search(r"<<'EOF'\s*\n(.*?)\nEOF", command, re.DOTALL)
if heredoc_match:
params["message"] = heredoc_match.group(1).strip()
return params
# Try to extract simple -m "message" format
simple_matches = re.findall(r'(?:-m|--message)\s+["\']([^"\']+)["\']', command)
if simple_matches:
# Join multiple -m flags with double newlines
params["message"] = "\n\n".join(simple_matches)
return params
return params
def get_staged_files() -> tuple[list[str], str]:
"""Get list of staged files and diff stats.
Returns:
(tuple): (list of file paths, diff stats string)
"""
try:
# Get list of staged files
files_result = subprocess.run(
["git", "diff", "--cached", "--name-only"],
capture_output=True,
text=True,
timeout=5,
)
# Get diff stats
stats_result = subprocess.run(
["git", "diff", "--cached", "--stat"],
capture_output=True,
text=True,
timeout=5,
)
files = []
if files_result.returncode == 0:
files = [f for f in files_result.stdout.strip().split("\n") if f]
stats = ""
if stats_result.returncode == 0:
# Get last line which contains the summary
stats_lines = stats_result.stdout.strip().split("\n")
if stats_lines:
stats = stats_lines[-1]
return files, stats
except (subprocess.TimeoutExpired, FileNotFoundError):
return [], ""
def format_confirmation_message(message: str, is_amend: bool, files: list[str], stats: str) -> str:
"""Format commit parameters into readable confirmation message.
Args:
message (str): Commit message
is_amend (bool): Whether this is an amend commit
files (list): List of staged file paths
stats (str): Diff statistics string
Returns:
(str): Formatted confirmation message
"""
lines = []
# Header
if is_amend:
lines.append("💾 Amend Previous Commit?")
else:
lines.append("💾 Create Commit?")
lines.append("")
# Commit message
if message:
lines.append("Message:")
lines.append(message)
lines.append("")
# Files
if files:
lines.append(f"Files to be committed ({len(files)}):")
# Show first 15 files, truncate if more
display_files = files[:15]
for f in display_files:
lines.append(f"- {f}")
if len(files) > 15:
lines.append(f"... and {len(files) - 15} more files")
lines.append("")
# Stats
if stats:
lines.append("Stats:")
lines.append(stats)
# Warning if no files staged
if not files:
lines.append("⚠️ No files staged for commit")
return "\n".join(lines)
try:
input_data = json.load(sys.stdin)
except json.JSONDecodeError as e:
print(f"Error: Invalid JSON input: {e}", file=sys.stderr)
sys.exit(1)
tool_name = input_data.get("tool_name", "")
tool_input = input_data.get("tool_input", {})
command = tool_input.get("command", "")
# Only handle git commit commands
if tool_name != "Bash" or not command.strip().startswith("git commit"):
sys.exit(0)
# Parse commit message
params = parse_git_commit_message(command)
# Get staged files and stats
files, stats = get_staged_files()
# Format confirmation message
message = format_confirmation_message(params["message"], params["is_amend"], files, stats)
# Return JSON with ask decision
output = {
"hookSpecificOutput": {
"hookEventName": "PreToolUse",
"permissionDecision": "ask",
"permissionDecisionReason": message,
}
}
print(json.dumps(output))
sys.exit(0)