feat: add PR review and edit tools to CLI and MCP server
This commit is contained in:
+214
-4
@@ -17,11 +17,12 @@ import os
|
||||
import sys
|
||||
import subprocess
|
||||
|
||||
# Auto-execute using the project's local virtual environment Python
|
||||
# Resolve the project root. MCP clients must launch this script directly with
|
||||
# the venv interpreter (venv/bin/python3) — see the config example above. We do
|
||||
# NOT os.execv() to re-point the interpreter: replacing the process after the
|
||||
# client has already wired up the stdio pipes can desync the JSON-RPC transport
|
||||
# (observed with Antigravity/Cascade hosts).
|
||||
PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))
|
||||
venv_python = os.path.join(PROJECT_ROOT, "venv", "bin", "python3")
|
||||
if os.path.exists(venv_python) and sys.executable != venv_python:
|
||||
os.execv(venv_python, [venv_python] + sys.argv)
|
||||
|
||||
# Ensure the project root is on the path so gitea_auth.py can be imported.
|
||||
if PROJECT_ROOT not in sys.path:
|
||||
@@ -208,6 +209,146 @@ def gitea_view_pr(
|
||||
}
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
def gitea_edit_pr(
|
||||
pr_number: int,
|
||||
title: str | None = None,
|
||||
body: str | None = None,
|
||||
state: str | None = None,
|
||||
base: str | None = None,
|
||||
remote: str = "dadeschools",
|
||||
host: str | None = None,
|
||||
org: str | None = None,
|
||||
repo: str | None = None,
|
||||
) -> dict:
|
||||
"""Edit an existing pull request on a Gitea repository.
|
||||
|
||||
Args:
|
||||
pr_number: The pull request index/number (required).
|
||||
title: New PR title.
|
||||
body: New PR description.
|
||||
state: New state — 'open' or 'closed'.
|
||||
base: Target branch name.
|
||||
remote: Known instance — 'dadeschools' or 'prgs'.
|
||||
host: Override the Gitea host.
|
||||
org: Override the owner/organization.
|
||||
repo: Override the repository name.
|
||||
|
||||
Returns:
|
||||
dict with success status and details of the edited PR.
|
||||
"""
|
||||
h, o, r = _resolve(remote, host, org, repo)
|
||||
auth = _auth(h)
|
||||
url = f"{repo_api_url(h, o, r)}/pulls/{pr_number}"
|
||||
payload = {}
|
||||
if title is not None:
|
||||
payload["title"] = title
|
||||
if body is not None:
|
||||
payload["body"] = body
|
||||
if state is not None:
|
||||
payload["state"] = state
|
||||
if base is not None:
|
||||
payload["base"] = base
|
||||
|
||||
if not payload:
|
||||
raise ValueError("At least one field to edit (title, body, state, base) must be provided.")
|
||||
|
||||
data = api_request("PATCH", url, auth, payload)
|
||||
return {
|
||||
"success": True,
|
||||
"number": data["number"],
|
||||
"title": data["title"],
|
||||
"body": data.get("body", ""),
|
||||
"state": data["state"],
|
||||
"url": data["html_url"],
|
||||
}
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
def gitea_get_file(
|
||||
filepath: str,
|
||||
ref: str = "main",
|
||||
remote: str = "dadeschools",
|
||||
host: str | None = None,
|
||||
org: str | None = None,
|
||||
repo: str | None = None,
|
||||
) -> dict:
|
||||
"""Retrieve metadata and content of a file from a Gitea repository.
|
||||
|
||||
Args:
|
||||
filepath: The path to the file in the repository (e.g. 'README.md' or 'src/main.py').
|
||||
ref: The branch, tag, or commit hash to retrieve the file from (default: 'main').
|
||||
remote: Known instance — 'dadeschools' or 'prgs'.
|
||||
host: Override the Gitea host.
|
||||
org: Override the owner/organization.
|
||||
repo: Override the repository name.
|
||||
|
||||
Returns:
|
||||
dict containing 'name', 'path', 'sha', 'size', 'encoding', and 'content' (base64).
|
||||
"""
|
||||
h, o, r = _resolve(remote, host, org, repo)
|
||||
auth = _auth(h)
|
||||
import urllib.parse
|
||||
encoded_path = urllib.parse.quote(filepath, safe="")
|
||||
url = f"{repo_api_url(h, o, r)}/contents/{encoded_path}?ref={ref}"
|
||||
data = api_request("GET", url, auth)
|
||||
return {
|
||||
"name": data.get("name", ""),
|
||||
"path": data.get("path", ""),
|
||||
"sha": data.get("sha", ""),
|
||||
"size": data.get("size", 0),
|
||||
"encoding": data.get("encoding", ""),
|
||||
"content": data.get("content", ""),
|
||||
}
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
def gitea_commit_files(
|
||||
files: list[dict],
|
||||
message: str,
|
||||
branch: str | None = None,
|
||||
new_branch: str | None = None,
|
||||
remote: str = "dadeschools",
|
||||
host: str | None = None,
|
||||
org: str | None = None,
|
||||
repo: str | None = None,
|
||||
) -> dict:
|
||||
"""Commit changes to multiple files in a Gitea repository in a single atomic commit.
|
||||
|
||||
Args:
|
||||
files: List of file operations. Each file dict must contain 'operation' ('create', 'update', 'delete', 'rename'), 'path', and 'content' (base64 encoded for create/update), and optionally 'sha' (required for update/delete) or 'from_path' (for rename).
|
||||
message: The commit message.
|
||||
branch: Optional existing branch to start/commit from.
|
||||
new_branch: Optional new branch name to create for this commit.
|
||||
remote: Known instance — 'dadeschools' or 'prgs'.
|
||||
host: Override the Gitea host.
|
||||
org: Override the owner/organization.
|
||||
repo: Override the repository name.
|
||||
|
||||
Returns:
|
||||
dict with success status and commit/branch information.
|
||||
"""
|
||||
h, o, r = _resolve(remote, host, org, repo)
|
||||
auth = _auth(h)
|
||||
url = f"{repo_api_url(h, o, r)}/contents"
|
||||
|
||||
payload = {
|
||||
"files": files,
|
||||
"message": message,
|
||||
}
|
||||
if branch is not None:
|
||||
payload["branch"] = branch
|
||||
if new_branch is not None:
|
||||
payload["new_branch"] = new_branch
|
||||
|
||||
data = api_request("POST", url, auth, payload)
|
||||
return {
|
||||
"success": True,
|
||||
"commit": data.get("commit", {}).get("sha", ""),
|
||||
"branch": data.get("branch", {}).get("name", ""),
|
||||
}
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
def gitea_merge_pr(
|
||||
pr_number: int,
|
||||
@@ -251,6 +392,75 @@ def gitea_merge_pr(
|
||||
return {"success": True, "message": f"PR #{pr_number} merged via '{do}'."}
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
def gitea_review_pr(
|
||||
pr_number: int,
|
||||
event: str = "APPROVE",
|
||||
body: str = "",
|
||||
merge: bool = False,
|
||||
merge_method: str = "merge",
|
||||
remote: str = "dadeschools",
|
||||
host: str | None = None,
|
||||
org: str | None = None,
|
||||
repo: str | None = None,
|
||||
) -> dict:
|
||||
"""Submit a review on a Gitea pull request and optionally merge it.
|
||||
|
||||
Args:
|
||||
pr_number: The PR number to review.
|
||||
event: Review type — 'APPROVE', 'COMMENT', or 'REQUEST_CHANGES'.
|
||||
body: Review body text / comment.
|
||||
merge: If True and event is 'APPROVE', automatically merge the PR.
|
||||
merge_method: Merge style to use if merging — 'merge', 'squash', or 'rebase'.
|
||||
remote: Known instance — 'dadeschools' or 'prgs'.
|
||||
host: Override the Gitea host.
|
||||
org: Override the owner/organization.
|
||||
repo: Override the repository name.
|
||||
|
||||
Returns:
|
||||
dict with success status and message.
|
||||
"""
|
||||
if event not in ["APPROVE", "COMMENT", "REQUEST_CHANGES"]:
|
||||
raise ValueError(f"Invalid review event: '{event}'. Choose from 'APPROVE', 'COMMENT', 'REQUEST_CHANGES'.")
|
||||
if merge_method not in ["merge", "squash", "rebase"]:
|
||||
raise ValueError(f"Invalid merge method: '{merge_method}'. Choose from 'merge', 'squash', 'rebase'.")
|
||||
|
||||
h, o, r = _resolve(remote, host, org, repo)
|
||||
auth = _auth(h)
|
||||
|
||||
# 1. Fetch PR to get the latest head commit SHA (required for review payload)
|
||||
pr_url = f"{repo_api_url(h, o, r)}/pulls/{pr_number}"
|
||||
pr_data = api_request("GET", pr_url, auth)
|
||||
commit_sha = pr_data.get("head", {}).get("sha")
|
||||
if not commit_sha:
|
||||
raise RuntimeError(f"Could not find head commit SHA for PR #{pr_number}.")
|
||||
|
||||
# 2. Submit the PR review
|
||||
review_url = f"{repo_api_url(h, o, r)}/pulls/{pr_number}/reviews"
|
||||
payload = {
|
||||
"body": body,
|
||||
"event": event,
|
||||
"commit_id": commit_sha
|
||||
}
|
||||
api_request("POST", review_url, auth, payload)
|
||||
msg = f"Successfully submitted review for PR #{pr_number} with event '{event}'."
|
||||
|
||||
# 3. Merge PR if merge is True and event is APPROVE
|
||||
if merge:
|
||||
if event != "APPROVE":
|
||||
msg += " Warning: Skipping merge because review event is not 'APPROVE'."
|
||||
else:
|
||||
merge_url = f"{repo_api_url(h, o, r)}/pulls/{pr_number}/merge"
|
||||
merge_payload = {
|
||||
"Do": merge_method,
|
||||
"force_merge": False
|
||||
}
|
||||
api_request("POST", merge_url, auth, merge_payload)
|
||||
msg += f" Successfully merged PR #{pr_number} using '{merge_method}' method."
|
||||
|
||||
return {"success": True, "message": msg}
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
def gitea_delete_branch(
|
||||
branch: str,
|
||||
|
||||
Reference in New Issue
Block a user