diff --git a/.env.example b/.env.example index 5463787..4777fac 100644 --- a/.env.example +++ b/.env.example @@ -46,3 +46,12 @@ GITEA_TOKEN_SOURCE=GITEA_TOKEN # profile's values. Leave unset for pure env-based configuration. GITEA_MCP_CONFIG=/Users/jasonwalker/.config/gitea-tools/profiles.json GITEA_MCP_PROFILE=prgs + +# Namespace-scoped active task workspaces (#510). Each MCP namespace uses only +# its own role env var; foreign bindings (e.g. GITEA_AUTHOR_WORKTREE in a +# merger process) are ignored. +# GITEA_AUTHOR_WORKTREE=/path/to/repo/branches/issue-123-work +# GITEA_REVIEWER_WORKTREE=/path/to/repo/branches/review-pr456 +# GITEA_MERGER_WORKTREE=/path/to/repo/branches/merge-pr456 +# GITEA_RECONCILER_WORKTREE=/path/to/repo/branches/reconcile-pr456 +# GITEA_ACTIVE_WORKTREE=/path/to/repo/branches/session-override diff --git a/audit_reconciliation_mode.py b/audit_reconciliation_mode.py new file mode 100644 index 0000000..52d0244 --- /dev/null +++ b/audit_reconciliation_mode.py @@ -0,0 +1,344 @@ +"""Audit vs cleanup phase gates for reconciliation workflows (#419). + +Audit/reconciliation tasks are read-only unless a separate cleanup phase is +explicitly authorized with exact capability proof, safety proof, and +before/after snapshots. Cleanup mutations must be classified in final reports; +audit reports must not claim ``no mutations`` when cleanup occurred. +""" + +from __future__ import annotations + +import re +from typing import Any + +RECONCILE_WORKFLOW_PATH = "workflows/reconcile-landed-pr.md" + +PHASE_AUDIT = "audit" +PHASE_CLEANUP = "cleanup" + +# Tasks that enter audit phase on capability resolution (read-only default). +AUDIT_PHASE_TASKS = frozenset({ + "reconcile-landed-pr", + "reconcile_landed_pr", + "reconcile_issue_claims", + "reconcile_merged_cleanups", +}) + +# Mutation tasks forbidden during audit phase (fail closed). +AUDIT_FORBIDDEN_TASKS = frozenset({ + "delete_branch", + "create_branch", + "push_branch", + "create_pr", + "commit_files", + "gitea_commit_files", + "mark_issue", + "lock_issue", + "claim_issue", + "close_pr", + "close_issue", + "create_issue", + "merge_pr", + "review_pr", + "submit_pr_review", + "comment_pr", + "comment_issue", + "set_issue_labels", +}) + +# Shell/git commands audit phase must not run. +AUDIT_FORBIDDEN_COMMAND_RE = re.compile( + r"(?:^|\s)(?:git\s+(?:push|branch\s+-D|worktree\s+remove)|" + r"gitea_delete_branch|delete_remote_branch)", + re.IGNORECASE, +) + +_NO_MUTATIONS_RE = re.compile( + r"(?:no\s+mutations|mutations\s*:\s*none|no\s+unsafe\s+mutation)", + re.IGNORECASE, +) +_CLEANUP_OCCURRED_RE = re.compile( + r"(?:delete_remote_branch|remove_local_worktree|git\s+branch\s+-D|" + r"git\s+worktree\s+remove|remote branch.*deleted|worktree.*removed|" + r"cleanup\s+phase\s*:\s*(?!none\b)\S)", + re.IGNORECASE, +) +_EXTERNAL_STATE_RE = re.compile( + r"^\s*[-*]?\s*external[- ]state mutations\s*:", + re.IGNORECASE | re.MULTILINE, +) +_GIT_REF_RE = re.compile( + r"^\s*[-*]?\s*git ref mutations\s*:", + re.IGNORECASE | re.MULTILINE, +) +_CLEANUP_MUTATIONS_RE = re.compile( + r"^\s*[-*]?\s*cleanup mutations\s*:", + re.IGNORECASE | re.MULTILINE, +) +_CLEANUP_PHASE_AUTH_RE = re.compile( + r"^\s*[-*]?\s*cleanup phase (?:authorized|authorization)\s*:\s*true", + re.IGNORECASE | re.MULTILINE, +) +_DELETE_CAPABILITY_RE = re.compile( + r"^\s*[-*]?\s*delete.?branch capability(?: proven)?\s*:\s*true", + re.IGNORECASE | re.MULTILINE, +) +_BEFORE_AFTER_RE = re.compile( + r"^\s*[-*]?\s*before/after (?:state )?snapshot\s*:", + re.IGNORECASE | re.MULTILINE, +) +_SAFETY_PROOF_RE = re.compile( + r"^\s*[-*]?\s*(?:branch|worktree) safe to remove\s*:\s*true", + re.IGNORECASE | re.MULTILINE, +) + +_session: dict[str, Any] | None = None + + +def _blank_session() -> dict[str, Any]: + return { + "phase": PHASE_AUDIT, + "entered_from_task": None, + "cleanup_authorized": False, + "cleanup_authorization": {}, + } + + +def current_phase() -> str | None: + """Return active reconciliation phase or None when unset.""" + if not _session: + return None + return _session.get("phase") + + +def active_record() -> dict[str, Any] | None: + """Return a copy of the session record, if any.""" + return dict(_session) if _session else None + + +def clear_phase() -> None: + """Clear reconciliation phase state.""" + global _session + _session = None + + +def enter_audit_phase(task: str) -> dict[str, Any]: + """Enter read-only audit phase for a reconciliation task.""" + global _session + normalized = (task or "").strip().lower() + _session = _blank_session() + _session["entered_from_task"] = normalized + return dict(_session) + + +def authorize_cleanup_phase( + *, + operator_approved: bool = False, + workflow_authorized: bool = False, + delete_capability_proven: bool = False, + safety_proof: dict[str, Any] | None = None, + before_after_snapshot: dict[str, Any] | None = None, +) -> dict[str, Any]: + """Authorize cleanup phase after explicit approval and safety proofs.""" + reasons: list[str] = [] + if not (operator_approved or workflow_authorized): + reasons.append( + "cleanup phase requires operator approval or explicit workflow " + "authorization" + ) + if not delete_capability_proven: + reasons.append( + "cleanup phase requires exact delete_branch capability proof " + "(gitea.branch.delete)" + ) + safety = dict(safety_proof or {}) + if not safety.get("safe_to_delete_remote") and not safety.get( + "safe_to_remove_worktree" + ): + reasons.append( + "cleanup phase requires proof that branch/worktree is safe to remove" + ) + snapshot = dict(before_after_snapshot or {}) + if not snapshot.get("before") or not snapshot.get("after"): + reasons.append( + "cleanup phase requires before/after state snapshot" + ) + + if reasons: + return { + "authorized": False, + "phase": current_phase() or PHASE_AUDIT, + "reasons": reasons, + "safe_next_action": ( + "remain in audit-only mode or supply operator approval, " + "delete_branch capability proof, safety proof, and " + "before/after snapshot before cleanup" + ), + } + + global _session + if _session is None: + _session = _blank_session() + _session["phase"] = PHASE_CLEANUP + _session["cleanup_authorized"] = True + _session["cleanup_authorization"] = { + "operator_approved": operator_approved, + "workflow_authorized": workflow_authorized, + "delete_capability_proven": delete_capability_proven, + "safety_proof": safety, + "before_after_snapshot": snapshot, + } + return { + "authorized": True, + "phase": PHASE_CLEANUP, + "reasons": [], + "cleanup_authorization": dict(_session["cleanup_authorization"]), + "safe_next_action": "proceed with authorized cleanup mutations only", + } + + +def check_audit_task_enters_phase(task: str) -> bool: + """Return whether resolving *task* should enter audit phase.""" + return (task or "").strip().lower() in AUDIT_PHASE_TASKS + + +def check_audit_mutation_allowed(task: str) -> tuple[bool, list[str]]: + """Fail closed when a mutation task runs during audit phase.""" + normalized = (task or "").strip().lower() + phase = current_phase() + if phase != PHASE_AUDIT: + return True, [] + if normalized in AUDIT_FORBIDDEN_TASKS: + return False, [ + f"task '{normalized}' is forbidden in audit-only reconciliation " + "mode: switch to an explicit cleanup phase with operator approval " + "and exact delete_branch capability proof before cleanup mutations" + ] + return True, [] + + +def check_cleanup_execution_allowed() -> tuple[bool, list[str]]: + """Fail closed when cleanup execution is attempted without authorization.""" + phase = current_phase() + if phase == PHASE_CLEANUP and (_session or {}).get("cleanup_authorized"): + return True, [] + if phase is None: + return False, [ + "cleanup execution requires an active reconciliation session; " + "resolve a reconciliation audit task first" + ] + return False, [ + "cleanup execution forbidden in audit-only reconciliation mode; " + "call gitea_authorize_reconciliation_cleanup_phase with operator " + "approval, delete_branch capability proof, safety proof, and " + "before/after snapshot" + ] + + +def classify_cleanup_mutation(action: str) -> str: + """Map a cleanup action to the required mutation ledger category (#419).""" + normalized = (action or "").strip().lower() + if "delete_remote" in normalized or normalized in { + "delete_branch", + "gitea_delete_branch", + }: + return "external-state" + if "branch" in normalized and "delete" in normalized: + return "git-ref" + if "worktree" in normalized or "remove_local" in normalized: + return "cleanup" + return "cleanup" + + +def assess_audit_reconciliation_report(report_text: str) -> dict[str, Any]: + """Validate audit/cleanup reconciliation reports (fail closed).""" + text = report_text or "" + reasons: list[str] = [] + + cleanup_occurred = bool(_CLEANUP_OCCURRED_RE.search(text)) + claims_no_mutations = bool(_NO_MUTATIONS_RE.search(text)) + + if cleanup_occurred and claims_no_mutations: + reasons.append( + "report claims no mutations but documents cleanup mutations; " + "audit-only reports must not perform cleanup and cleanup reports " + "must not claim no mutations" + ) + + if cleanup_occurred: + if not _CLEANUP_PHASE_AUTH_RE.search(text): + reasons.append( + "cleanup mutations reported without " + "'Cleanup phase authorized: true'" + ) + if not _DELETE_CAPABILITY_RE.search(text): + reasons.append( + "cleanup mutations reported without delete_branch capability " + "proof" + ) + if not _BEFORE_AFTER_RE.search(text): + reasons.append( + "cleanup mutations reported without before/after state snapshot" + ) + if not _SAFETY_PROOF_RE.search(text): + reasons.append( + "cleanup mutations reported without branch/worktree safety proof" + ) + + if re.search(r"delete_remote|remote branch.*delet", text, re.I): + if not _EXTERNAL_STATE_RE.search(text): + reasons.append( + "remote branch deletion must be classified under " + "External-state mutations" + ) + if re.search(r"git\s+branch\s+-D|local branch.*delet", text, re.I): + if not _GIT_REF_RE.search(text): + reasons.append( + "local branch deletion must be classified under " + "Git ref mutations" + ) + if re.search(r"worktree.*remov|remove_local_worktree", text, re.I): + if not _CLEANUP_MUTATIONS_RE.search(text): + reasons.append( + "worktree removal must be classified under Cleanup mutations" + ) + + if ( + RECONCILE_WORKFLOW_PATH.replace("workflows/", "") in text + or "reconcile-landed-pr" in text.lower() + ): + if cleanup_occurred and "audit phase" in text.lower(): + if "cleanup phase" not in text.lower(): + reasons.append( + "report mixes audit phase with cleanup mutations without " + "documenting cleanup phase transition" + ) + + proven = not reasons + return { + "proven": proven, + "block": not proven, + "reasons": reasons, + "cleanup_occurred": cleanup_occurred, + "claims_no_mutations": claims_no_mutations, + "safe_next_action": ( + "proceed" + if proven + else "fix audit/cleanup report: separate audit from cleanup phase, " + "classify mutations, and do not claim no mutations after cleanup" + ), + } + + +def assess_audit_command_allowed(command: str) -> tuple[bool, list[str]]: + """Block shell commands that perform cleanup during audit phase.""" + phase = current_phase() + if phase != PHASE_AUDIT: + return True, [] + cmd = (command or "").strip() + if AUDIT_FORBIDDEN_COMMAND_RE.search(cmd): + return False, [ + f"command forbidden in audit-only reconciliation mode: {cmd!r}; " + "authorize cleanup phase before branch/worktree deletion or push" + ] + return True, [] \ No newline at end of file diff --git a/author_mutation_worktree.py b/author_mutation_worktree.py index 29dc05d..dccb94a 100644 --- a/author_mutation_worktree.py +++ b/author_mutation_worktree.py @@ -12,6 +12,8 @@ import subprocess BASE_BRANCHES = frozenset({"master", "main", "dev"}) ACTIVE_WORKTREE_ENV = "GITEA_ACTIVE_WORKTREE" AUTHOR_WORKTREE_ENV = "GITEA_AUTHOR_WORKTREE" +# Author-only: reviewer/merger/reconciler namespaces use role-specific env vars +# via namespace_workspace_binding (#510). def _normalize_path(path: str) -> str: diff --git a/docs/developer-testing-guidelines.md b/docs/developer-testing-guidelines.md index 291ff5d..d85c670 100644 --- a/docs/developer-testing-guidelines.md +++ b/docs/developer-testing-guidelines.md @@ -13,6 +13,25 @@ credentials.** Every test mocks the HTTP client and the keychain/auth lookup. ## 1. Standard test commands +### Canonical runner: `./run-tests.sh` + +The canonical full-validation command is the root-level runner. It invokes the +project virtualenv interpreter and passes any extra arguments straight through +to `pytest`: + +```bash +# Full validation +./run-tests.sh + +# Focused validation (extra args forward to pytest) +./run-tests.sh tests/test_mcp_server.py -q +``` + +`run-tests.sh` runs `venv/bin/python -m pytest "$@"` and fails with a clear +setup message if the virtualenv Python is missing (so a session never silently +falls back to the wrong interpreter). The explicit `venv/bin/python -m pytest` +forms below remain valid and equivalent. + The test suite needs the project virtualenv (it provides the MCP SDK): ```bash diff --git a/docs/issue-acceptance-gate.md b/docs/issue-acceptance-gate.md new file mode 100644 index 0000000..522825b --- /dev/null +++ b/docs/issue-acceptance-gate.md @@ -0,0 +1,52 @@ +# Controller Issue-Acceptance Gate + +A merged PR does not automatically prove an issue is fully satisfied. After +merge, a controller must audit the linked issue against its acceptance criteria +and post a durable handoff before the issue is treated as complete. + +## Workflow position + +1. Author implements the issue and opens a PR. +2. Reviewer reviews the PR. +3. Merger merges the approved PR. +4. **Controller performs issue-acceptance audit.** +5. Controller posts a `## Controller Issue Acceptance` comment with either: + - `STATE: accepted` and checked criteria, or + - a rejection path (`more-work-required`, `needs-tests`, `needs-docs`, etc.) + with `MISSING_WORK` and a paste-ready `NEXT_PROMPT`. + +Gitea may auto-close an issue via `Closes #N` in the PR body. That closure is +merge mechanics only. Controller acceptance is still required before any final +report or queue controller treats the issue as complete. + +## Template + +Use `issue_acceptance_gate.render_controller_acceptance_template()` or the +copy in +[`skills/llm-project-workflow/templates/controller-issue-acceptance.md`](../skills/llm-project-workflow/templates/controller-issue-acceptance.md). + +## Final-report rules + +Final reports must not claim `issue complete` solely because a PR merged. +Either: + +- include a valid `## Controller Issue Acceptance` block with + `STATE: accepted`, or +- explicitly state `controller acceptance pending` and identify the controller + as the next actor. + +`final_report_validator` enforces this through +`issue_acceptance_gate.validate_final_report_issue_acceptance()`. + +## Role boundaries + +- Authors must not mark their own issues accepted. +- Reviewers must not mark issue acceptance unless acting under controller + capability. +- Mergers merge PRs; they do not substitute for controller acceptance. + +## Related + +- #495 — canonical next-action comment templates +- #496 — fail-closed canonical comment validation before posting +- #303 — controller handoff schema for reconciliation workflows \ No newline at end of file diff --git a/docs/llm-workflow-runbooks.md b/docs/llm-workflow-runbooks.md index f71c825..b983a0d 100644 --- a/docs/llm-workflow-runbooks.md +++ b/docs/llm-workflow-runbooks.md @@ -325,6 +325,17 @@ shared state and manual writes can clobber another session's live lease. Use 3. Operator override only when explicitly authorized — record `External-state mutations` and `operator override proof` in the final report. +**Adoption proof in the live lock response (#477):** when `gitea_lock_issue` +adopts an existing own branch, the response carries an `adoption` block with +citable fields — `adoption_decision` (`ADOPT`), `adopted` (`true`), +`adopted_branch`, `adopted_branch_head`, `matcher_summary` (boundary-safe reason +the branch qualified), `competing_branch_check`, and `safe_next_action`. A normal +lock instead returns an `adoption_check` block with `adoption_decision` +(`NO_MATCH`) and `adopted: false`, so a non-adoption response can never be misread +as claiming adoption. Recovery reports should quote the live lock response +`adoption`/`adoption_check` block directly instead of inferring adoption from +separate offline checks. + `gitea_create_pr` rejects lock files that lack sanctioned `lock_provenance` metadata. Final-report validation blocks handoffs that hide lock read/write/delete under `External-state mutations: none` or mix author PR creation with reviewer @@ -366,6 +377,37 @@ explicit control-checkout repair. Portable wording: [`skills/llm-project-workflow/SKILL.md`](../skills/llm-project-workflow/SKILL.md). +### Root checkout guard (#475) + +The MCP server enforces a fail-closed **root checkout guard** before author, +reviewer, and merger mutations when the active workspace is not an isolated +`branches/...` worktree (reconciler close paths remain exempt per #468). + +The guard blocks when the **control checkout** (repository root) is: + +- on a non-stable branch (`master` / `main` / `dev` expected), +- detached HEAD, +- dirty (tracked edits), +- or its `HEAD` does not match `prgs/master` when that ref is available. + +**Remediation (never auto-reset or stash):** + +> Root checkout is not on master. Preserve state, switch root back to master, +> and use `scripts/worktree-review` or the sanctioned issue worktree flow. + +**Recovery after root hijack:** + +1. Preserve any in-progress edits (copy paths, note branch name, or commit on a + rescue branch from a `branches/...` worktree). +2. From the repository root: `git checkout master` (or `main` / `dev` per repo + policy) and `git fetch prgs && git merge --ff-only prgs/master` when safe. +3. Confirm `git status` is clean and `git branch --show-current` is `master`. +4. Resume work only inside `branches/issue--` via `gitea_lock_issue` / + `git worktree add`. + +`branches/...` directories are disposable role worktrees; the root checkout is +the stable orchestration surface only. + ## Shell Spawn Hard-Stop Rule Symptom: a shell tool call returns `exit_code: -1` with empty stdout/stderr. @@ -630,6 +672,33 @@ loop and do **not** substitute WebFetch/Playwright/manual base64. - **Prompt:** `Use any eligible merger profile to merge PR #N if checks pass and it is mergeable. Confirm with "MERGE PR N". Do not force-merge.` +#### Merger lease adoption (#536) + +When review and merge run in **separate sessions**, the merger must **not** +manually seed `reviewer_pr_lease._SESSION_LEASE` or equivalent in-process state. +That ad hoc pattern was used incidentally for PR #493 and PR #421; it is not +canonical proof and is rejected by mutation gates. + +**Canonical merger handoff:** + +1. Confirm a reviewer session holds an active PR lease and posted **APPROVED** + at the current live head (`gitea_get_pr_review_feedback`). +2. In a clean merger worktree under `branches/`, call + `gitea_adopt_merger_pr_lease` with `worktree`, `expected_head_sha`, and + optional `issue_number`. +3. The tool posts durable adoption proof on the PR thread (``) + recording actor, profiles, adopted-from session/comment, adoption reason, and + timestamp, then records sanctioned in-session provenance. +4. Call `gitea_merge_pr` with the same pinned `expected_head_sha`. + +**Forbidden:** Python one-liners or scripts that call `record_session_lease()` +without provenance from `gitea_acquire_reviewer_pr_lease`, +`gitea_adopt_merger_pr_lease`, or `gitea_heartbeat_reviewer_pr_lease`. + +**Same-session review+merge:** the reviewer session may use +`gitea_acquire_reviewer_pr_lease` directly; adoption is only for cross-session +merger handoff. + ### Close the issue after merge / Reconciliation - **Profile:** issue-manager or merger. @@ -812,6 +881,45 @@ scripts/release-tag v0.4.0 --notes-file /tmp/release-notes.md scripts/release-tag v0.4.0 --notes-file /tmp/release-notes.md --push ``` +## Namespace workspace binding (#510) + +Each MCP namespace resolves its **own** active task workspace. Foreign role +worktree environment variables must not poison another namespace's purity +checks. + +| Namespace | Workspace env vars (in priority under `GITEA_ACTIVE_WORKTREE`) | Allowed roots | +|-----------|------------------------------------------------------------------|---------------| +| author | `GITEA_AUTHOR_WORKTREE` | `branches/` worktree only (#274) | +| reviewer | `GITEA_REVIEWER_WORKTREE` | clean `branches/` worktree | +| merger | `GITEA_MERGER_WORKTREE` | clean `branches/` worktree **or** clean control checkout | +| reconciler | `GITEA_RECONCILER_WORKTREE` | clean `branches/` worktree **or** clean control checkout | + +`GITEA_AUTHOR_WORKTREE` is **author-only**. Reviewer, merger, and reconciler +MCP processes ignore it even when it points at a dirty author WIP tree. + +### Safe reconnect / rebind procedure + +When a mutation blocks on workspace binding: + +1. Read the error — it names the **resolved workspace path**, **role + namespace**, and **binding source** (tool arg, env var, or process root). +2. Reconnect or relaunch the correct namespace MCP server from the intended + workspace (or set the role-specific env var before launch). +3. Pass `worktree_path` on reviewer/merger mutation tools when the active + branches/ worktree differs from the MCP process root. +4. **Do not** clean, reset, or discard foreign role worktrees to unblock your + own namespace — that destroys another agent's WIP. + +### CTH guidance for workspace binding blockers + +When posting a Canonical Thread Handoff after a binding blocker: + +- State which namespace was active (author / reviewer / merger / reconciler). +- Quote the resolved workspace path and binding source from the error. +- Name the safe reconnect action (relaunch MCP from `branches/...`, set + `GITEA_*_WORKTREE`, or pass `worktree_path`). +- Explicitly note that foreign worktrees must not be cleaned to unblock. + ## Safety notes - Never place raw tokens or passwords in any LLM MCP config; reference secrets @@ -821,6 +929,7 @@ scripts/release-tag v0.4.0 --notes-file /tmp/release-notes.md --push ## Related documents +- [`issue-acceptance-gate.md`](issue-acceptance-gate.md) — controller issue-acceptance audit after PR merge (#500). - [`../skills/llm-project-workflow/SKILL.md`](../skills/llm-project-workflow/SKILL.md) — portable cross-project LLM workflow skill. - [`gitea-execution-profiles.md`](gitea-execution-profiles.md) — the profile model. - [`gitea-dual-namespace-deployment.md`](gitea-dual-namespace-deployment.md) — static author/reviewer namespace deployment (#139 decision). diff --git a/docs/mcp-menu.md b/docs/mcp-menu.md new file mode 100644 index 0000000..0b028f9 --- /dev/null +++ b/docs/mcp-menu.md @@ -0,0 +1,67 @@ +# MCP operator shell menu + +## Purpose + +`./mcp-menu.sh` is a repository-root terminal menu for onboarding and operating +the Gitea-Tools MCP/Gitea workflow without memorizing every prompt, script path, +or runbook section. + +It is intentionally **safe by default**: status checks and copy-paste workflow +prompts. It does not delete branches, force-push, edit lock files, or bypass +sanctioned MCP tools. + +## How to run + +From the repository root: + +```bash +./mcp-menu.sh +``` + +The script must be executable (`chmod +x mcp-menu.sh`). It uses bash with +`set -euo pipefail`. + +## Safety rules + +- **Read-only by default** — root checkout health is inspection only. +- **No destructive git** — no `git push --force`, branch deletion, or + `--delete` refspecs. +- **No lock-file editing** — issue locks are acquired only through + `gitea_lock_issue`. +- **No raw API bypass** — prompts direct operators to sanctioned MCP tools. +- **Remote mutations require confirmation** — any future menu action that would + mutate remote or server state must be clearly labeled and require explicit + operator confirmation before running. +- **Author work stays under `branches/`** — the root checkout is a stable + control checkout on `master` / `prgs/master`. + +## Menu options + +| Option | Description | +|--------|-------------| +| Project status / root checkout health | Shows cwd, branch, `git status --short --branch`, HEAD SHA, `prgs/master` SHA, and warnings when the root checkout is dirty or off `master`. | +| Author workflow prompts | Ready-to-copy prompts for issue work, conflict-fix sessions, and root checkout recovery. | +| Reviewer workflow prompts | PR review prompt (review-only; no merge). | +| Merger workflow prompts | PR merge prompt (merge gates and explicit approval). | +| Reconciler workflow prompts | Already-landed / closed PR reconciliation prompt. | +| Onboarding new project | Checklist prompt for adding a repository to the MCP workflow. | +| Proxmox deployment placeholder | **Not implemented** — informational message only. | +| Create Proxmox LXC placeholder | **Not implemented** — informational message only. | +| Run tests | Runs `./run-tests.sh` when present; otherwise `venv/bin/python -m pytest`; otherwise fails closed with a clear error. | +| Exit | Quit the menu. | + +## Placeholder-only entries + +**Proxmox deployment** and **Create Proxmox LXC** are placeholders until +dedicated issues implement sanctioned automation. The menu prints a clear +message and does not invoke deploy scripts. + +## Related documentation + +- [`docs/llm-workflow-runbooks.md`](llm-workflow-runbooks.md) — Gitea-specific workflow runbooks +- [`skills/llm-project-workflow/SKILL.md`](../skills/llm-project-workflow/SKILL.md) — portable workflow skill +- [`skills/llm-project-workflow/workflows/`](../skills/llm-project-workflow/workflows/) — canonical task workflows + +## Tests + +Hermetic coverage lives in `tests/test_mcp_menu_script.py`. \ No newline at end of file diff --git a/docs/webui-local-dev.md b/docs/webui-local-dev.md index 0a5b935..96225b6 100644 --- a/docs/webui-local-dev.md +++ b/docs/webui-local-dev.md @@ -46,7 +46,8 @@ Optional environment variables: | `/runtime` | Stub — MCP runtime health (#430) | | `/audit` | Stub — report audit paste (#431) | | `/worktrees` | Stub — hygiene dashboard (#432) | -| `/leases` | Stub — lease visibility (#433) | +| `/leases` | Lease and collision visibility (#433) | +| `/api/leases` | JSON lease/collision export | All routes are GET-only. POST/PUT/PATCH/DELETE return `405` with `read-only-mvp`. @@ -82,8 +83,16 @@ credentials. The UI surfaces pagination proof (returned count, pages fetched, If credentials are missing or the fetch fails, the page shows an explicit error instead of an empty queue (fail closed). +## Lease visibility (#433) + +`/leases` surfaces read-only lease and collision state: local issue lock file, +in-progress claim inventory (#268), reviewer PR lease comments when present +(``, #407), duplicate open PRs per issue (#400), +and duplicate local branches per issue. Links to collision-history backend +issues (#267, #268, #400, #407) are included. No lease acquire/release from UI. + ## Tests ```bash -pytest tests/test_webui_skeleton.py tests/test_webui_project_registry.py tests/test_webui_prompt_library.py tests/test_webui_queue_dashboard.py -q +pytest tests/test_webui_skeleton.py tests/test_webui_project_registry.py tests/test_webui_prompt_library.py tests/test_webui_queue_dashboard.py tests/test_webui_lease_visibility.py -q ``` \ No newline at end of file diff --git a/docs/wiki/MCP-Tools.md b/docs/wiki/MCP-Tools.md index 38af50e..cd46012 100644 --- a/docs/wiki/MCP-Tools.md +++ b/docs/wiki/MCP-Tools.md @@ -24,6 +24,8 @@ - `gitea_dry_run_pr_review` — validation-phase review mechanics. - `gitea_mark_final_review_decision` — mark validation complete. - `gitea_submit_pr_review` / `gitea_review_pr` — gated live review. +- `gitea_acquire_reviewer_pr_lease` / `gitea_heartbeat_reviewer_pr_lease` — per-PR reviewer lease (#407). +- `gitea_adopt_merger_pr_lease` — guarded cross-session merger lease adoption (#536). - `gitea_merge_pr` — gated merge (only merge path). ## Read tools diff --git a/final_report_validator.py b/final_report_validator.py index b2f4c59..42626ab 100644 --- a/final_report_validator.py +++ b/final_report_validator.py @@ -11,7 +11,10 @@ import inspect import re from typing import Any, Callable +import issue_acceptance_gate import issue_lock_provenance +from mcp_native_cleanup_proof import assess_mcp_native_cleanup_proof +from post_merge_cleanup_proof import assess_post_merge_cleanup_proof from review_proofs import ( HANDOFF_HEADING, assess_controller_handoff, @@ -118,6 +121,22 @@ _TARGET_BRANCH_SHA_RE = re.compile( r"target branch sha\s*:\s*[0-9a-f]{40}", re.IGNORECASE, ) +_WORKFLOW_LOAD_HELPER_RE = re.compile( + r"workflow[- ]load helper result\s*:", + re.IGNORECASE, +) +_WORKFLOW_LOAD_HASH_RE = re.compile( + r"workflow[- ]load helper result[\s\S]{0,400}?workflow[_ ]hash\s*:\s*[0-9a-f]{12}", + re.IGNORECASE, +) +_WORKFLOW_LOAD_BOUNDARY_RE = re.compile( + r"workflow[- ]load helper result[\s\S]{0,400}?boundary[_ ]status\s*:\s*(?:clean|violation)", + re.IGNORECASE, +) +_WORKFLOW_FILE_VIEW_NARRATIVE_RE = re.compile( + r"(?:read|viewed|loaded)\s+(?:the\s+)?(?:canonical\s+)?(?:workflow|review-merge-pr\.md)", + re.IGNORECASE, +) _FULL_SHA_RE = re.compile(r"\b[0-9a-f]{40}\b", re.IGNORECASE) _RECONCILE_STALE_FIELDS = ( "pr number opened", @@ -600,6 +619,27 @@ def _rule_conflict_fix_push_proof(report_text: str) -> list[dict[str, str]]: ) +def _rule_worktree_cleanup_audit_proof(report_text: str) -> list[dict[str, str]]: + from worktree_cleanup_audit import assess_cleanup_audit_final_report + + text = report_text or "" + if "cleanup audit" not in text.lower() and "reconciliation table" not in text.lower(): + return [] + result = assess_cleanup_audit_final_report(text) + if result.get("proven"): + return [] + return _findings_from_reasons( + "author.worktree_cleanup_audit_proof", + result.get("reasons") or [], + field="Worktree cleanup audit", + severity="block", + safe_next_action=( + "include reconciliation table counts, disposition rows, and " + "final git worktree list proof" + ), + ) + + def _rule_reviewer_validation_command(report_text: str) -> list[dict[str, str]]: text = report_text or "" if not _BARE_PYTEST_RE.search(text): @@ -954,6 +994,69 @@ def _rule_reconcile_linked_issue_live( return [] +_PR_CLOSE_NEGATIVE = frozenset({"", "none", "n/a", "na", "0", "no", "not closed", "not performed"}) +_ANCESTOR_AFFIRMATIVE_RE = re.compile( + r"ancestor|passed|true|verified|confirmed", re.IGNORECASE +) + + +def _reconciler_pr_close_performed(fields: dict[str, str], lock: dict) -> bool: + """True when the report/session indicates a reconciler PR close happened.""" + if lock.get("pr_closed") is True: + return True + value = (fields.get("prs closed", "") or "").strip().lower() + if value in _PR_CLOSE_NEGATIVE: + return False + # A closed PR is reported by number (e.g. "#99") or an affirmative result. + return bool(re.search(r"#\s*\d+|\b(?:closed|success|done)\b", value)) + + +def _rule_reconcile_close_proof( + report_text: str, + *, + reconciler_close_lock: dict | None = None, +) -> list[dict[str, str]]: + """#306: a reconciler PR close must carry exact proof fields. + + Read-only/comment-only reconciliations are untouched. Once a PR close is + reported (via the ``PRs closed`` field or a session close lock), the + handoff must prove the close capability, ancestor landing, PR close + result, and the linked-issue result — narrative alone fails closed. + """ + fields = _handoff_fields(report_text) + lock = reconciler_close_lock or {} + if not _reconciler_pr_close_performed(fields, lock): + return [] + + missing: list[str] = [] + capabilities = fields.get("capabilities proven", "") + if "gitea.pr.close" not in capabilities.lower(): + missing.append("close capability proof (gitea.pr.close)") + ancestor = fields.get("ancestor proof", "") + if not _ANCESTOR_AFFIRMATIVE_RE.search(ancestor): + missing.append("ancestor proof") + prs_closed = (fields.get("prs closed", "") or "").strip().lower() + if prs_closed in _PR_CLOSE_NEGATIVE and lock.get("pr_closed") is True: + missing.append("PR close result") + linked = fields.get("linked issue live status", "") or fields.get("issues closed", "") + if not linked.strip(): + missing.append("linked issue result") + + if not missing: + return [] + return [ + validator_finding( + "reconcile.close_proof_fields", + "block", + "Reconciler close proof", + "reconciler PR close reported without required proof field(s): " + + ", ".join(missing), + "include close capability proof, ancestor proof, PR close result, " + "and linked issue result in the handoff", + ) + ] + + def _rule_reconcile_pagination_proof(report_text: str) -> list[dict[str, str]]: text = report_text or "" if not _INVENTORY_COMPLETE_RE.search(text): @@ -1040,6 +1143,22 @@ def _rule_shared_manual_lock_pr_override(report_text: str) -> list[dict[str, str ) +def _rule_shared_issue_acceptance_gate(report_text: str) -> list[dict[str, str]]: + result = issue_acceptance_gate.validate_final_report_issue_acceptance(report_text) + if not result.get("applicable") or result.get("valid"): + return [] + return _findings_from_reasons( + "shared.issue_acceptance_gate", + result.get("reasons") or [], + field="Controller acceptance", + severity="block", + safe_next_action=( + "add Controller Issue Acceptance proof or state that controller " + "acceptance is pending; do not claim issue complete from PR merge alone" + ), + ) + + def _rule_shared_author_reviewer_same_run(report_text: str) -> list[dict[str, str]]: result = issue_lock_provenance.assess_author_reviewer_same_run_report(report_text) if result.get("proven"): @@ -1056,6 +1175,70 @@ def _rule_shared_author_reviewer_same_run(report_text: str) -> list[dict[str, st ) +def _rule_reviewer_workflow_load_boundary(report_text: str) -> list[dict[str, str]]: + """#403: require structured workflow-load helper result, not file-view narrative.""" + if not report_text.strip(): + return [] + findings: list[dict[str, str]] = [] + has_helper = bool(_WORKFLOW_LOAD_HELPER_RE.search(report_text)) + has_hash = bool(_WORKFLOW_LOAD_HASH_RE.search(report_text)) + has_boundary = bool(_WORKFLOW_LOAD_BOUNDARY_RE.search(report_text)) + has_narrative_only = bool(_WORKFLOW_FILE_VIEW_NARRATIVE_RE.search(report_text)) + + if has_narrative_only and not has_helper: + findings.append(validator_finding( + "reviewer.workflow_load_boundary", + "block", + "Workflow-load helper result", + ( + "canonical workflow file-view narrative without structured " + "gitea_load_review_workflow helper result" + ), + ( + "include Workflow-load helper result with workflow_hash and " + "boundary_status from gitea_load_review_workflow" + ), + )) + return findings + + if has_helper and (not has_hash or not has_boundary): + missing = [] + if not has_hash: + missing.append("workflow_hash") + if not has_boundary: + missing.append("boundary_status") + findings.append(validator_finding( + "reviewer.workflow_load_boundary", + "block", + "Workflow-load helper result", + ( + "workflow-load helper result incomplete; missing " + + ", ".join(missing) + ), + ( + "copy workflow_load_helper_result fields from " + "gitea_load_review_workflow into the final report" + ), + )) + return findings + + +def _rule_audit_reconciliation_boundary(report_text: str) -> list[dict[str, str]]: + from audit_reconciliation_mode import assess_audit_reconciliation_report + + result = assess_audit_reconciliation_report(report_text) + if result.get("proven"): + return [] + return _findings_from_reasons( + "reconcile.audit_cleanup_boundary", + result.get("reasons") or [], + field="Audit/cleanup phase", + severity="block", + safe_next_action=result.get("safe_next_action") + or "separate audit from authorized cleanup and classify mutations", + ) + + def _rule_reviewer_review_mutation( report_text: str, *, @@ -1075,12 +1258,60 @@ def _rule_reviewer_review_mutation( ) +def _rule_reviewer_mutation_capability_proof(report_text: str) -> list[dict[str, str]]: + from reviewer_mutation_capability_proof import assess_mutation_capability_proof + + result = assess_mutation_capability_proof(report_text) + if not result.get("block"): + return [] + return _findings_from_reasons( + "reviewer.mutation_capability_proof", + result.get("reasons") or [], + field="Capabilities proven", + severity="block", + safe_next_action=result.get("safe_next_action") + or "document exact per-mutation capability proof before each mutation", + ) + + +def _rule_reviewer_post_merge_cleanup_proof(report_text: str) -> list[dict[str, str]]: + result = assess_post_merge_cleanup_proof(report_text) + if not result.get("block"): + return [] + return _findings_from_reasons( + "reviewer.post_merge_cleanup_proof", + result.get("reasons") or [], + field="Cleanup status", + severity="block", + safe_next_action=result.get("safe_next_action") + or "report CLEANUP_SKIPPED with blocker or full cleanup checklist", + ) + + +def _rule_shared_mcp_native_cleanup_proof(report_text: str) -> list[dict[str, str]]: + result = assess_mcp_native_cleanup_proof(report_text) + if not result.get("block"): + return [] + return _findings_from_reasons( + "shared.mcp_native_cleanup_proof", + result.get("reasons") or [], + field="Cleanup mutations", + severity="block", + safe_next_action=result.get("safe_next_action") + or "use authorized reconciler MCP cleanup tools; never raw scripts", + ) + + _SHARED_ISSUE_LOCK_RULES = ( _rule_shared_issue_lock_external_state, _rule_shared_manual_lock_pr_override, _rule_shared_author_reviewer_same_run, ) +_SHARED_CLEANUP_PROOF_RULES = ( + _rule_shared_mcp_native_cleanup_proof, +) + _RULES_BY_TASK: dict[str, list[Callable[..., list[dict[str, str]]]]] = { "review_pr": [ _rule_shared_controller_handoff, @@ -1103,8 +1334,12 @@ _RULES_BY_TASK: dict[str, list[Callable[..., list[dict[str, str]]]]] = { _rule_reviewer_already_landed_eligible, _rule_reviewer_already_landed_state, _rule_reviewer_target_branch_freshness, + _rule_reviewer_workflow_load_boundary, _rule_reviewer_mutation_ledger, _rule_reviewer_review_mutation, + _rule_reviewer_mutation_capability_proof, + _rule_reviewer_post_merge_cleanup_proof, + *_SHARED_CLEANUP_PROOF_RULES, _rule_reviewer_stale_head_proof, ], "reconcile_already_landed": [ @@ -1112,13 +1347,16 @@ _RULES_BY_TASK: dict[str, list[Callable[..., list[dict[str, str]]]]] = { _rule_shared_state_handoff_next_action, _rule_shared_email_disclosure, *_SHARED_ISSUE_LOCK_RULES, + *_SHARED_CLEANUP_PROOF_RULES, _rule_reconcile_stale_author_fields, _rule_reconcile_eligible_reviewed, _rule_reconcile_linked_issue_live, + _rule_reconcile_close_proof, _rule_reconcile_pagination_proof, _rule_reviewer_git_fetch_readonly, _rule_reviewer_legacy_workspace_mutations, _rule_reviewer_vague_mutations_none, + _rule_audit_reconciliation_boundary, ], "author_issue": [ _rule_shared_controller_handoff, @@ -1132,8 +1370,10 @@ _RULES_BY_TASK: dict[str, list[Callable[..., list[dict[str, str]]]]] = { _rule_shared_state_handoff_next_action, _rule_shared_email_disclosure, *_SHARED_ISSUE_LOCK_RULES, + _rule_shared_issue_acceptance_gate, _rule_reviewer_vague_mutations_none, _rule_conflict_fix_push_proof, + _rule_worktree_cleanup_audit_proof, ], "issue_filing": [ _rule_shared_controller_handoff, @@ -1216,6 +1456,7 @@ def assess_final_report_validator( issue_filing_lock: dict | None = None, session_pr_opened: bool = False, validation_session: dict | None = None, + reconciler_close_lock: dict | None = None, ) -> dict[str, Any]: """Validate final-report text against task-specific proof rules (#327). @@ -1272,6 +1513,7 @@ def assess_final_report_validator( "local_edits": local_edits, "session_pr_opened": session_pr_opened, "validation_session": validation_session, + "reconciler_close_lock": reconciler_close_lock, } for rule in _RULES_BY_TASK.get(normalized_kind, ()): diff --git a/gitea_mcp_server.py b/gitea_mcp_server.py index b5681e9..ed650b5 100644 --- a/gitea_mcp_server.py +++ b/gitea_mcp_server.py @@ -165,6 +165,7 @@ _preflight_capability_called = False _preflight_whoami_violation = False _preflight_capability_violation = False _preflight_resolved_role = None +_preflight_resolved_task: str | None = None _process_start_porcelain: str | None = None _preflight_whoami_baseline_porcelain: str | None = None _preflight_capability_baseline_porcelain: str | None = None @@ -174,12 +175,98 @@ _preflight_reviewer_violation_files: list[str] = [] ACTIVE_WORKTREE_ENV = "GITEA_ACTIVE_WORKTREE" AUTHOR_WORKTREE_ENV = "GITEA_AUTHOR_WORKTREE" +REVIEWER_WORKTREE_ENV = "GITEA_REVIEWER_WORKTREE" +MERGER_WORKTREE_ENV = "GITEA_MERGER_WORKTREE" +RECONCILER_WORKTREE_ENV = "GITEA_RECONCILER_WORKTREE" + +import namespace_workspace_binding as nwb # noqa: E402 def _preflight_in_test_mode() -> bool: return "pytest" in sys.modules or "unittest" in sys.modules +def _reviewer_session_worktree() -> str | None: + import reviewer_pr_lease as _reviewer_pr_lease + + session = _reviewer_pr_lease.get_session_lease() + if not session: + return None + worktree = (session.get("worktree") or "").strip() + return worktree or None + + +def _effective_workspace_role() -> str: + """Resolve the namespace key used for workspace binding (#510).""" + profile = get_profile() + role = _preflight_resolved_role + if not role: + role = _role_kind( + profile.get("allowed_operations") or [], + profile.get("forbidden_operations") or [], + ) + return nwb.normalize_role_kind( + role, + profile_name=profile.get("profile_name"), + ) + + +def _actual_profile_role() -> str: + """Resolve the workspace role from the ACTIVE PROFILE alone (#540). + + Unlike :func:`_effective_workspace_role`, this never consults + ``_preflight_resolved_role``. A task capability whose ``required_role_kind`` + is ``author`` (e.g. ``comment_issue``) stamps ``_preflight_resolved_role = + "author"``; the #274/#475 role exemptions must key off the real profile + identity so that stamp cannot poison a genuine reviewer/merger/reconciler + session into being treated as an author. Author profiles still classify as + ``author`` here, so author blocking is preserved. + """ + profile = get_profile() + role = _role_kind( + profile.get("allowed_operations") or [], + profile.get("forbidden_operations") or [], + ) + return nwb.normalize_role_kind( + role, + profile_name=profile.get("profile_name"), + ) + + +def _resolve_preflight_workspace_path(worktree_path: str | None = None) -> str: + """Resolve the namespace-scoped workspace root inspected by pre-flight guards.""" + role = _effective_workspace_role() + workspace, _source = nwb.resolve_namespace_workspace( + role_kind=role, + worktree_path=worktree_path, + process_project_root=PROJECT_ROOT, + session_lease_worktree=( + _reviewer_session_worktree() if role in {"reviewer", "merger"} else None + ), + profile_name=get_profile().get("profile_name"), + ) + return workspace + + +def _resolve_namespace_mutation_context(worktree_path: str | None = None) -> dict: + """Canonical namespace workspace + repository root for guards (#460/#510).""" + role = _effective_workspace_role() + return nwb.resolve_namespace_mutation_context( + role_kind=role, + worktree_path=worktree_path, + process_project_root=PROJECT_ROOT, + session_lease_worktree=( + _reviewer_session_worktree() if role in {"reviewer", "merger"} else None + ), + profile_name=get_profile().get("profile_name"), + ) + + +def _resolve_author_mutation_context(worktree_path: str | None = None) -> dict: + """Backward-compatible alias for namespace workspace context.""" + return _resolve_namespace_mutation_context(worktree_path) + + def _ensure_process_start_porcelain() -> str: """Capture the shared-worktree baseline once per MCP process (#252).""" global _process_start_porcelain @@ -188,26 +275,6 @@ def _ensure_process_start_porcelain() -> str: return _process_start_porcelain -def _resolve_preflight_workspace_path(worktree_path: str | None = None) -> str: - """Resolve the workspace root inspected by pre-flight guards.""" - return author_mutation_worktree.resolve_mutation_workspace( - worktree_path, - PROJECT_ROOT, - active_worktree_env=os.environ.get(ACTIVE_WORKTREE_ENV), - author_worktree_env=os.environ.get(AUTHOR_WORKTREE_ENV), - ) - - -def _resolve_author_mutation_context(worktree_path: str | None = None) -> dict: - """Canonical workspace + repository root for runtime_context and guards (#460).""" - return author_mutation_worktree.resolve_author_mutation_context( - worktree_path, - PROJECT_ROOT, - active_worktree_env=os.environ.get(ACTIVE_WORKTREE_ENV), - author_worktree_env=os.environ.get(AUTHOR_WORKTREE_ENV), - ) - - def _get_git_root(path: str) -> str | None: try: res = subprocess.run( @@ -275,7 +342,7 @@ def _format_preflight_files(files: list[str]) -> str: def _preflight_workspace_details(worktree_path: str | None, dirty_files: list[str]) -> dict: - ctx = _resolve_author_mutation_context(worktree_path) + ctx = _resolve_namespace_mutation_context(worktree_path) workspace = ctx["workspace_path"] inspected_root = _get_git_root(workspace) process_root = ctx["process_project_root"] @@ -293,6 +360,9 @@ def _preflight_workspace_details(worktree_path: str | None, dirty_files: list[st "dirty_files": list(dirty_files), "dirty_scope": dirty_scope, "workspace_roots_aligned": ctx["roots_aligned"], + "workspace_role_kind": ctx.get("workspace_role_kind"), + "workspace_binding_source": ctx.get("workspace_binding_source"), + "ignored_bindings": list(ctx.get("ignored_bindings") or []), } if not ctx["roots_aligned"]: details["workspace_root_mismatch"] = ( @@ -303,13 +373,19 @@ def _preflight_workspace_details(worktree_path: str | None, dirty_files: list[st def _format_preflight_workspace_details(details: dict) -> str: - return ( - f"MCP server process root: {details.get('mcp_server_process_root')}; " - f"active task workspace root: {details.get('active_task_workspace_root')}; " - f"inspected git root: {details.get('inspected_git_root')}; " - f"dirty files: {_format_preflight_files(details.get('dirty_files') or [])}; " - f"dirty scope: {details.get('dirty_scope')}" - ) + parts = [ + f"MCP server process root: {details.get('mcp_server_process_root')}", + f"active task workspace root: {details.get('active_task_workspace_root')}", + f"workspace role: {details.get('workspace_role_kind')}", + f"binding source: {details.get('workspace_binding_source')}", + f"inspected git root: {details.get('inspected_git_root')}", + f"dirty files: {_format_preflight_files(details.get('dirty_files') or [])}", + f"dirty scope: {details.get('dirty_scope')}", + ] + ignored = details.get("ignored_bindings") or [] + if ignored: + parts.append(f"ignored foreign bindings: {'; '.join(ignored)}") + return "; ".join(parts) def assess_preflight_status(worktree_path: str | None = None) -> dict: @@ -332,6 +408,25 @@ def assess_preflight_status(worktree_path: str | None = None) -> dict: "Active task workspace has tracked file edits before mutation " f"({_format_preflight_workspace_details(workspace_details)})" ) + role = _effective_workspace_role() + if role in nwb.NON_AUTHOR_ROLES: + binding = nwb.assess_metadata_only_worktree_binding( + role_kind=role, + declared_worktree_path=worktree_path, + mutation_workspace=_resolve_preflight_workspace_path(worktree_path), + process_project_root=PROJECT_ROOT, + profile_name=get_profile().get("profile_name"), + ) + if binding.get("block"): + reasons.append(binding["reasons"][0]) + reasons.append( + nwb.format_namespace_workspace_binding_error( + role_kind=role, + workspace_path=binding["mutation_workspace"], + binding_source="MCP server process root (default)", + reasons=binding.get("reasons"), + ) + ) return { "preflight_ready": not reasons, "preflight_block_reasons": reasons, @@ -379,11 +474,30 @@ def assess_preflight_status(worktree_path: str | None = None) -> dict: } -def record_preflight_check(type_name: str, resolved_role: str | None = None): +def _clear_preflight_capability_state() -> None: + """Drop resolved capability proof (consumed by a mutation or fresh resolve).""" + global _preflight_capability_called, _preflight_capability_violation + global _preflight_resolved_task + global _preflight_capability_baseline_porcelain, _preflight_capability_violation_files + global _preflight_reviewer_violation_files + + _preflight_capability_called = False + _preflight_capability_violation = False + _preflight_capability_violation_files = [] + _preflight_capability_baseline_porcelain = None + _preflight_resolved_task = None + _preflight_reviewer_violation_files = [] + + +def record_preflight_check( + type_name: str, + resolved_role: str | None = None, + resolved_task: str | None = None, +): """Record a pre-flight check (whoami or capability) with session-scoped deltas.""" global _preflight_whoami_called, _preflight_capability_called global _preflight_whoami_violation, _preflight_capability_violation - global _preflight_resolved_role + global _preflight_resolved_role, _preflight_resolved_task global _preflight_whoami_baseline_porcelain, _preflight_capability_baseline_porcelain global _preflight_whoami_violation_files, _preflight_capability_violation_files global _preflight_reviewer_violation_files @@ -391,13 +505,24 @@ def record_preflight_check(type_name: str, resolved_role: str | None = None): current = _get_workspace_porcelain() if type_name == "whoami": - # Fresh whoami restarts the capability step and re-evaluates violations - # instead of replaying a sticky record (#252). - _preflight_capability_called = False - _preflight_capability_violation = False - _preflight_capability_violation_files = [] - _preflight_capability_baseline_porcelain = None - _preflight_reviewer_violation_files = [] + # Re-evaluate whoami violations instead of replaying sticky state (#252). + # Interleaved read-only whoami must not clear a valid capability proof (#469). + saved_capability = None + preserve_capability = ( + _preflight_capability_called + and _preflight_whoami_called + and not _preflight_whoami_violation + ) + if preserve_capability: + saved_capability = ( + _preflight_capability_violation, + list(_preflight_capability_violation_files), + _preflight_capability_baseline_porcelain, + _preflight_resolved_role, + _preflight_resolved_task, + ) + else: + _clear_preflight_capability_state() process_start = _ensure_process_start_porcelain() whoami_delta = _new_tracked_changes_since(process_start, current) @@ -405,6 +530,20 @@ def record_preflight_check(type_name: str, resolved_role: str | None = None): _preflight_whoami_violation_files = whoami_delta _preflight_whoami_baseline_porcelain = current _preflight_whoami_called = True + + if ( + preserve_capability + and saved_capability is not None + and not whoami_delta + ): + ( + _preflight_capability_violation, + _preflight_capability_violation_files, + _preflight_capability_baseline_porcelain, + _preflight_resolved_role, + _preflight_resolved_task, + ) = saved_capability + _preflight_capability_called = True elif type_name == "capability": baseline = _preflight_whoami_baseline_porcelain or "" capability_delta = _new_tracked_changes_since(baseline, current) @@ -414,13 +553,31 @@ def record_preflight_check(type_name: str, resolved_role: str | None = None): _preflight_capability_called = True if resolved_role: _preflight_resolved_role = resolved_role + if resolved_task: + _preflight_resolved_task = resolved_task def _enforce_branches_only_author_mutation(worktree_path: str | None = None) -> None: - """#274: author mutations must run from a branches/ session worktree.""" - if _preflight_resolved_role == "reviewer": + """#274: author file/branch mutations must run from a branches/ worktree. + + Reviewer, merger, and reconciler roles are exempt: reconciler ``close_pr`` + is a Gitea metadata mutation and must not require ``GITEA_AUTHOR_WORKTREE`` + (#468). Non-author namespaces use dedicated workspace env vars (#510). + + The exemption honours BOTH the effective workspace role and the actual + profile role (#540). ``comment_issue`` preflight stamps + ``_preflight_resolved_role = "author"`` (its ``required_role_kind``), which + would otherwise poison :func:`_effective_workspace_role` into classifying a + genuine reconciler as an author and defeat this exemption. Keying off the + real profile role as well preserves the exemption without weakening author + blocking — an actual author profile classifies as ``author`` in both. + """ + if ( + _effective_workspace_role() in nwb.NON_AUTHOR_ROLES + or _actual_profile_role() in nwb.NON_AUTHOR_ROLES + ): return - ctx = _resolve_author_mutation_context(worktree_path) + ctx = _resolve_namespace_mutation_context(worktree_path) workspace = ctx["workspace_path"] git_state = issue_lock_worktree.read_worktree_git_state(workspace) assessment = author_mutation_worktree.assess_author_mutation_worktree( @@ -434,7 +591,11 @@ def _enforce_branches_only_author_mutation(worktree_path: str | None = None) -> ) -def verify_preflight_purity(remote: str | None = None, worktree_path: str | None = None): +def verify_preflight_purity( + remote: str | None = None, + worktree_path: str | None = None, + task: str | None = None, +): """Verify that identity and capability were verified prior to session edits.""" global _preflight_reviewer_violation_files @@ -453,12 +614,23 @@ def verify_preflight_purity(remote: str | None = None, worktree_path: str | None raise RuntimeError( "Pre-flight order violation: Task capability (gitea_resolve_task_capability) has not been resolved (fail closed)" ) + if ( + task is not None + and _preflight_resolved_task is not None + and task != _preflight_resolved_task + ): + raise RuntimeError( + "Pre-flight task mismatch: " + f"resolved '{_preflight_resolved_task}' but mutation requires " + f"'{task}' (fail closed)" + ) - ctx = _resolve_author_mutation_context(worktree_path) + ctx = _resolve_namespace_mutation_context(worktree_path) workspace = ctx["workspace_path"] canonical_root = ctx["canonical_repo_root"] process_root = ctx["process_project_root"] real_workspace = os.path.realpath(workspace) + role = ctx.get("workspace_role_kind") or _effective_workspace_role() if real_workspace != process_root: if not _preflight_in_test_mode(): @@ -475,11 +647,15 @@ def verify_preflight_purity(remote: str | None = None, worktree_path: str | None dirty_files = sorted(_parse_porcelain_entries(_get_workspace_porcelain(workspace))) if dirty_files: - details = _preflight_workspace_details(workspace, dirty_files) raise RuntimeError( - "Pre-flight order violation: Active task workspace has tracked " - "file edits before mutation (fail closed). " - f"{_format_preflight_workspace_details(details)}" + nwb.format_namespace_workspace_binding_error( + role_kind=role, + workspace_path=workspace, + binding_source=ctx.get("workspace_binding_source") + or "unknown binding source", + dirty_files=dirty_files, + ignored_bindings=ctx.get("ignored_bindings"), + ) ) else: if _preflight_whoami_violation: @@ -495,19 +671,115 @@ def verify_preflight_purity(remote: str | None = None, worktree_path: str | None f"{_format_preflight_files(_preflight_capability_violation_files)}" ) - if _preflight_resolved_role == "reviewer": + if role in {"reviewer", "merger"}: current = _get_workspace_porcelain() baseline = _preflight_capability_baseline_porcelain or "" reviewer_delta = _new_tracked_changes_since(baseline, current) _preflight_reviewer_violation_files = reviewer_delta if reviewer_delta: raise RuntimeError( - "Reviewer role violation: Reviewer profile is forbidden from modifying " + f"{role.title()} role violation: profile is forbidden from modifying " "tracked workspace files (fail closed). Offending files: " f"{_format_preflight_files(reviewer_delta)}" ) + _enforce_root_checkout_guard(worktree_path) _enforce_branches_only_author_mutation(worktree_path) + _clear_preflight_capability_state() + + +def _verify_role_mutation_workspace( + remote: str | None = None, + *, + worktree_path: str | None = None, + worktree: str | None = None, + task: str | None = None, +) -> str: + """Bind reviewer/merger mutations to the active namespace workspace (#510).""" + # Check running runtimes to prevent stale mutations + try: + if "PYTEST_CURRENT_TEST" not in os.environ or "GITEA_FORCE_MCP_RUNTIME_CHECK" in os.environ: + config = gitea_config.load_config() + required_permission = task_capability_map.required_permission(task) if task else None + matching_profiles = [] + if required_permission and config and "profiles" in config: + for p_name, p_data in config["profiles"].items(): + p_allowed = p_data.get("allowed_operations") or [] + p_forbidden = p_data.get("forbidden_operations") or [] + p_allowed_n = [] + for op in p_allowed: + try: + p_allowed_n.append(gitea_config.normalize_operation(op)) + except Exception: + pass + p_forbidden_n = [] + for op in p_forbidden: + try: + p_forbidden_n.append(gitea_config.normalize_operation(op)) + except Exception: + pass + ok, _ = gitea_config.check_operation(required_permission, p_allowed_n, p_forbidden_n) + if ok: + matching_profiles.append(p_name) + runtime_reasons = _check_mcp_runtimes_diagnostics(task or "unknown", matching_profiles) + if runtime_reasons: + raise RuntimeError("; ".join(runtime_reasons)) + except Exception as exc: + if "stale-runtime:" in str(exc): + raise RuntimeError(str(exc)) + pass + + role = _effective_workspace_role() + git_state = issue_lock_worktree.read_worktree_git_state( + _resolve_preflight_workspace_path(worktree_path) + ) + assessment = nwb.assess_namespace_mutation_workspace( + role_kind=role, + worktree_path=worktree_path, + worktree=worktree, + process_project_root=PROJECT_ROOT, + session_lease_worktree=( + _reviewer_session_worktree() if role in {"reviewer", "merger"} else None + ), + profile_name=get_profile().get("profile_name"), + current_branch=git_state.get("current_branch"), + ) + if assessment["block"]: + raise RuntimeError( + nwb.format_namespace_workspace_binding_error( + role_kind=role, + workspace_path=assessment["mutation_workspace"], + binding_source=assessment.get("workspace_binding_source") + or "unknown binding source", + reasons=assessment.get("reasons"), + ignored_bindings=assessment.get("ignored_bindings"), + ) + ) + resolved = assessment["mutation_workspace"] + verify_preflight_purity(remote, worktree_path=resolved, task=task) + return resolved + + +def _enforce_root_checkout_guard(worktree_path: str | None = None) -> None: + """#475: fail closed when the stable control checkout is contaminated.""" + ctx = _resolve_author_mutation_context(worktree_path) + canonical_root = ctx["canonical_repo_root"] + workspace = ctx["workspace_path"] + git_state = issue_lock_worktree.read_worktree_git_state(canonical_root) + remote_master_sha = root_checkout_guard.resolve_remote_master_sha(canonical_root) + assessment = root_checkout_guard.assess_root_checkout_guard( + workspace_path=workspace, + canonical_repo_root=canonical_root, + current_branch=git_state.get("current_branch"), + head_sha=git_state.get("head_sha"), + porcelain_status=git_state.get("porcelain_status") or "", + remote_master_sha=remote_master_sha, + resolved_role=_preflight_resolved_role, + actual_role=_actual_profile_role(), + ) + if assessment["block"]: + raise RuntimeError(root_checkout_guard.format_root_checkout_guard_error(assessment)) + from mcp.server.fastmcp import FastMCP # noqa: E402 @@ -530,6 +802,9 @@ import role_session_router # noqa: E402 import role_namespace_gate # noqa: E402 import task_capability_map # noqa: E402 import review_proofs # noqa: E402 +import review_workflow_boundary # noqa: E402 +import review_workflow_load # noqa: E402 +import mcp_session_state # noqa: E402 import agent_temp_artifacts import issue_lock_worktree # noqa: E402 import issue_lock_provenance # noqa: E402 @@ -539,15 +814,21 @@ import stacked_pr_support # noqa: E402 import merge_approval_gate # noqa: E402 import already_landed_reconcile # noqa: E402 import author_mutation_worktree # noqa: E402 +import root_checkout_guard # noqa: E402 +import remote_repo_guard # noqa: E402 import issue_claim_heartbeat # noqa: E402 import issue_work_duplicate_gate # noqa: E402 import reviewer_pr_lease # noqa: E402 +import merger_lease_adoption # noqa: E402 import merged_cleanup_reconcile # noqa: E402 +import worktree_cleanup_audit # noqa: E402 import reconciler_profile # noqa: E402 import reconciliation_workflow # noqa: E402 +import audit_reconciliation_mode # noqa: E402 import review_merge_state_machine # noqa: E402 import pr_work_lease # noqa: E402 import native_mcp_preference # noqa: E402 +import worktree_cleanup_audit # noqa: E402 # Keyed issue-lock storage (#443): per remote/org/repo/issue files under @@ -949,11 +1230,49 @@ def _resolve(remote: str, host: str | None, org: str | None, repo: str | None): if remote not in REMOTES: raise ValueError(f"Unknown remote '{remote}'. Choose from: {list(REMOTES)}") profile = REMOTES[remote] - return ( - host or profile["host"], - org or profile["org"], - repo or profile["repo"], + resolved_host = host or profile["host"] + resolved_org = org or profile["org"] + resolved_repo = repo or profile["repo"] + _enforce_remote_repo_guard( + remote, + resolved_org, + resolved_repo, + org_explicit=org is not None, + repo_explicit=repo is not None, ) + return (resolved_host, resolved_org, resolved_repo) + + +def _enforce_remote_repo_guard( + remote: str, + resolved_org: str, + resolved_repo: str, + *, + org_explicit: bool, + repo_explicit: bool, +) -> None: + """Fail closed on a remote/repo mismatch vs. the local git remote (#530). + + Best-effort: bypassed under pytest unless ``GITEA_FORCE_REMOTE_REPO_CHECK`` is + set, so the unit suite (which calls tools with bare remotes against mocked APIs) + is unaffected. In production it protects every read/lookup/mutation tool because + they all resolve targets through :func:`_resolve`. + """ + if "pytest" in sys.modules and not os.environ.get( + "GITEA_FORCE_REMOTE_REPO_CHECK" + ): + return + local_remote_url = _local_git_remote_url(remote) + assessment = remote_repo_guard.assess_remote_repo_match( + remote=remote, + resolved_org=resolved_org, + resolved_repo=resolved_repo, + local_remote_url=local_remote_url, + org_explicit=org_explicit, + repo_explicit=repo_explicit, + ) + if assessment["block"]: + raise RuntimeError(remote_repo_guard.format_remote_repo_guard_error(assessment)) def _auth(host: str) -> str: @@ -1223,7 +1542,7 @@ def gitea_create_issue( ) if blocked: return blocked - verify_preflight_purity(remote, worktree_path=worktree_path) + verify_preflight_purity(remote, worktree_path=worktree_path, task="create_issue") base = repo_api_url(h, o, r) open_issues = api_get_all(f"{base}/issues?state=open&type=issues", auth) closed_issues = api_get_all( @@ -1262,7 +1581,6 @@ def gitea_create_issue( return _with_optional_url({"number": data["number"]}, data.get("html_url")) -@mcp.tool() def _list_open_pulls(h: str, o: str, r: str, auth: str) -> list[dict]: """Fetch all OPEN pull requests for a repo (used for stacked-base proof, #484).""" try: @@ -1273,6 +1591,7 @@ def _list_open_pulls(h: str, o: str, r: str, auth: str) -> list[dict]: ) +@mcp.tool() def gitea_lock_issue( issue_number: int, branch_name: str, @@ -1356,7 +1675,7 @@ def gitea_lock_issue( ) else: git_state = issue_lock_worktree.read_worktree_git_state(resolved_worktree) - verify_preflight_purity(remote, worktree_path=resolved_worktree) + verify_preflight_purity(remote, worktree_path=resolved_worktree, task="lock_issue") lock_assessment = issue_lock_worktree.assess_issue_lock_worktree( worktree_path=resolved_worktree, current_branch=git_state.get("current_branch"), @@ -1484,6 +1803,14 @@ def gitea_lock_issue( f"Adopted existing branch '{branch_name}' and locked issue " f"#{issue_number} for recovery (fail-closed check complete)." ) + else: + # #477 AC2: normal (no-adoption) lock responses carry explicit, + # adoption-free proof metadata so they stay clear and cannot be + # misread as claiming a branch was adopted. + result["adoption_check"] = issue_lock_adoption.build_non_adoption_lock_proof( + issue_number=issue_number, + branch_name=branch_name, + ) if agent_artifacts: result["warnings"] = [ "Agent temp artifacts at repo root (delete before implementation): " @@ -1572,7 +1899,7 @@ def gitea_create_pr( ) if blocked: return blocked - verify_preflight_purity(remote, worktree_path=worktree_path) + verify_preflight_purity(remote, worktree_path=worktree_path, task="create_pr") h, o, r = _resolve(remote, host, org, repo) # ── Issue Lock Validation (Issue #194 / #196 / #443) ── @@ -2154,44 +2481,127 @@ _REVIEW_ACTIONS = { _TERMINAL_REVIEW_ACTIONS = frozenset({"approve", "request_changes"}) -# In-process only (#211): never persist to /tmp — host-global files are -# spoofable by any local process and go stale across sessions. +# Durable across MCP daemon process pools (#559), but never under /tmp (#211): +# host-global temp files are spoofable. Persistence uses the user-private +# cache directory from mcp_session_state (mode 0o700 / files 0o600), keyed by +# remote + profile identity with TTL. _REVIEW_DECISION_LOCK: dict | None = None +def _decision_lock_binding(lock: dict | None = None) -> dict: + """Resolve key fields for durable decision-lock storage.""" + profile = get_profile() + profile_name = (profile.get("profile_name") or "").strip() + env_lock = (os.environ.get(SESSION_PROFILE_LOCK_ENV) or "").strip() + stored_lock = ((lock or {}).get("session_profile_lock") or "").strip() + session_lock = env_lock or stored_lock or profile_name + remote = ((lock or {}).get("remote") or "").strip() or None + org = ((lock or {}).get("org") or (lock or {}).get("ready_org") or "").strip() or None + repo = ((lock or {}).get("repo") or (lock or {}).get("ready_repo") or "").strip() or None + return { + "session_profile": profile_name, + "session_profile_lock": session_lock, + "profile_identity": mcp_session_state.current_profile_identity( + profile_name=profile_name, + session_profile_lock=session_lock, + ), + "remote": remote, + "org": org, + "repo": repo, + } + + def _load_review_decision_lock(): + """Load decision lock from memory, falling back to durable session state.""" global _REVIEW_DECISION_LOCK + if _REVIEW_DECISION_LOCK is not None: + return _REVIEW_DECISION_LOCK + binding = _decision_lock_binding() + # Profile-keyed durable file; remote/org/repo validated from payload (#559). + durable = mcp_session_state.load_state( + kind=mcp_session_state.KIND_DECISION_LOCK, + profile_identity=binding.get("profile_identity"), + ) + if durable is not None: + _REVIEW_DECISION_LOCK = dict(durable) return _REVIEW_DECISION_LOCK def _save_review_decision_lock(data): + """Persist decision lock to memory + durable shared state (#559).""" global _REVIEW_DECISION_LOCK - _REVIEW_DECISION_LOCK = data + if data is None: + binding = _decision_lock_binding(_REVIEW_DECISION_LOCK) + mcp_session_state.clear_state( + kind=mcp_session_state.KIND_DECISION_LOCK, + profile_identity=binding.get("profile_identity"), + ) + _REVIEW_DECISION_LOCK = None + return + payload = dict(data) + binding = _decision_lock_binding(payload) + payload.setdefault("session_pid", os.getpid()) + payload["writer_pid"] = os.getpid() + payload["session_profile"] = binding["session_profile"] or payload.get( + "session_profile" + ) + payload["session_profile_lock"] = binding["session_profile_lock"] + payload["profile_identity"] = binding["profile_identity"] + if binding.get("remote") and not payload.get("remote"): + payload["remote"] = binding["remote"] + persisted = mcp_session_state.save_state( + kind=mcp_session_state.KIND_DECISION_LOCK, + payload=payload, + remote=payload.get("remote"), + org=payload.get("org") or payload.get("ready_org"), + repo=payload.get("repo") or payload.get("ready_repo"), + profile_identity=payload.get("profile_identity"), + ) + _REVIEW_DECISION_LOCK = dict(persisted or payload) def _review_decision_session_reasons(lock: dict | None) -> list[str]: - """Reject locks that do not belong to this MCP process/session.""" + """Reject locks that do not belong to this MCP session identity (#559). + + Different daemon PIDs in the same IDE session pool are allowed when the + profile identity and remote match. Spoofed/stale locks still fail closed. + """ if lock is None: return [] reasons = [] - if lock.get("session_pid") != os.getpid(): - reasons.append( - "review decision lock was created in a different process " - "(fail closed)" - ) env_lock = (os.environ.get(SESSION_PROFILE_LOCK_ENV) or "").strip() - stored_lock = (lock.get("session_profile_lock") or "").strip() + stored_lock = (lock.get("session_profile_lock") or lock.get("profile_identity") or "").strip() if env_lock and stored_lock and env_lock != stored_lock: reasons.append( "review decision lock session profile lock mismatch (fail closed)" ) + # TTL / identity checks from durable envelope fields. + for reason in mcp_session_state.identity_match_reasons( + lock, + remote=lock.get("remote"), + org=lock.get("org") or lock.get("ready_org"), + repo=lock.get("repo") or lock.get("ready_repo"), + profile_identity=env_lock or stored_lock, + ): + if "profile identity mismatch" in reason or "expired" in reason or "future" in reason or "missing recorded_at" in reason: + reasons.append(reason) return reasons -def init_review_decision_lock(remote: str | None, task: str | None): +def init_review_decision_lock(remote: str | None, task: str | None, force: bool = True): """Seed read-only-until-ready state for reviewer PR review tasks.""" if task != "review_pr": return + if not force: + lock = _load_review_decision_lock() + if lock is not None: + env_lock = (os.environ.get(SESSION_PROFILE_LOCK_ENV) or "").strip() + stored_lock = (lock.get("session_profile_lock") or "").strip() + same_remote = lock.get("remote") == remote + same_profile = (not env_lock or not stored_lock or env_lock == stored_lock) + if same_remote and same_profile and not _review_decision_session_reasons(lock): + return + review_workflow_load.clear_review_workflow_load() profile = get_profile() profile_name = (profile.get("profile_name") or "").strip() session_lock = ( @@ -2205,6 +2615,10 @@ def init_review_decision_lock(remote: str | None, task: str | None): "session_pid": os.getpid(), "session_profile": profile_name, "session_profile_lock": session_lock, + "profile_identity": mcp_session_state.current_profile_identity( + profile_name=profile_name, + session_profile_lock=session_lock, + ), "final_review_decision_ready": False, "ready_pr_number": None, "ready_action": None, @@ -2218,6 +2632,11 @@ def init_review_decision_lock(remote: str | None, task: str | None): }) +def _review_workflow_load_gate_reasons() -> list[str]: + """Fail closed when canonical review workflow was not loaded (#389).""" + return review_workflow_load.review_workflow_load_blockers(PROJECT_ROOT) + + def check_review_decision_gate( pr_number: int, action: str, @@ -2228,7 +2647,10 @@ def check_review_decision_gate( repo: str | None = None, ) -> list[str]: """Fail closed unless validation completed and the final decision is ready.""" - reasons = [] + reasons = list(_review_workflow_load_gate_reasons()) + if reasons: + reasons.extend(review_workflow_load.recovery_handoff_without_replay()) + return reasons lock = _load_review_decision_lock() if lock is None: reasons.append( @@ -2575,6 +2997,85 @@ def gitea_get_pr_review_feedback( } +def _fetch_pr_lease_comments_safe( + pr_number: int, + *, + remote: str, + host: str | None, + org: str | None, + repo: str | None, + limit: int = 100, + require_open: bool = False, +) -> dict: + """Fetch PR thread comments with structured fail-closed errors (#519).""" + h, o, r = _resolve(remote, host, org, repo) + auth = _auth(h) + resolved_repo = f"{o}/{r}" + pr_url = f"{repo_api_url(h, o, r)}/pulls/{pr_number}" + try: + pr = api_request("GET", pr_url, auth) + except RuntimeError as exc: + return { + "success": False, + "comments": [], + "reasons": [ + "PR lookup failed before conflict-fix push assessment " + f"(pr_number={pr_number}, repo={resolved_repo}, remote={remote}): " + f"{_redact(str(exc))}" + ], + "pr_lookup": "failed", + "resolved_repo": resolved_repo, + "remote": remote, + "pr_number": pr_number, + } + pr_state = (pr.get("state") or "").strip().lower() + if require_open and pr_state != "open": + return { + "success": False, + "comments": [], + "reasons": [ + f"PR #{pr_number} on {resolved_repo} is not open " + f"(state={pr_state or 'unknown'})" + ], + "pr_lookup": "not_open", + "resolved_repo": resolved_repo, + "remote": remote, + "pr_number": pr_number, + "head_sha": (pr.get("head") or {}).get("sha"), + } + api = f"{repo_api_url(h, o, r)}/issues/{pr_number}/comments" + try: + comments = api_request("GET", api, auth) + except RuntimeError as exc: + return { + "success": False, + "comments": [], + "reasons": [ + "PR comment fetch failed during conflict-fix push assessment " + f"(pr_number={pr_number}, issue_index={pr_number}, " + f"repo={resolved_repo}, remote={remote}): " + f"{_redact(str(exc))}" + ], + "pr_lookup": "ok", + "resolved_repo": resolved_repo, + "remote": remote, + "pr_number": pr_number, + "head_sha": (pr.get("head") or {}).get("sha"), + } + if not isinstance(comments, list): + comments = [] + return { + "success": True, + "comments": list(comments[:limit]), + "reasons": [], + "pr_lookup": "ok", + "resolved_repo": resolved_repo, + "remote": remote, + "pr_number": pr_number, + "head_sha": (pr.get("head") or {}).get("sha"), + } + + def _list_pr_lease_comments( pr_number: int, *, @@ -2584,11 +3085,24 @@ def _list_pr_lease_comments( repo: str | None, limit: int = 100, ) -> list[dict]: - """Fetch PR/issue thread comments used for reviewer/conflict-fix leases.""" + """Fetch PR/issue thread comments used for reviewer/conflict-fix leases. + + Intentionally does **not** pre-fetch the PR via GET /pulls/{n}. Shared + callers (merge approval feedback, reviewer lease gates) depend on a single + comments GET so mock sequences and fail-open #485 non-list handling stay + stable. Structured PR-lookup failures belong only to + :func:`_fetch_pr_lease_comments_safe` used by conflict-fix push assessment + (#519). + """ h, o, r = _resolve(remote, host, org, repo) auth = _auth(h) api = f"{repo_api_url(h, o, r)}/issues/{pr_number}/comments" - comments = api_request("GET", api, auth) or [] + comments = api_request("GET", api, auth) + # Fail safe to no lease comments when the API returns a non-list payload + # (e.g. an error object such as an HTTP 401 body): lease state can only be + # proven from real comment entries, never inferred from an error shape (#485). + if not isinstance(comments, list): + return [] return list(comments[:limit]) @@ -2631,10 +3145,14 @@ def _evaluate_pr_review_submission( *, live: bool, final_review_decision_ready: bool = False, + worktree_path: str | None = None, ) -> dict: """Shared gate chain for live submit and dry-run review tools.""" - verify_preflight_purity(remote) + _verify_role_mutation_workspace( + remote, worktree_path=worktree_path, task="review_pr" + ) action = (action or "").strip().lower() + workflow_blockers = _review_workflow_load_gate_reasons() if live else [] result = { "requested_action": action, "performed": False, @@ -2650,6 +3168,10 @@ def _evaluate_pr_review_submission( "reasons": [], } reasons = result["reasons"] + if workflow_blockers: + reasons.extend(workflow_blockers) + reasons.extend(review_workflow_load.recovery_handoff_without_replay()) + return result if action not in _REVIEW_ACTIONS: reasons.append( @@ -2868,6 +3390,13 @@ def gitea_mark_final_review_decision( } org = resolved_org repo = resolved_repo + workflow_blockers = _review_workflow_load_gate_reasons() + if workflow_blockers: + return { + "marked_ready": False, + "reasons": workflow_blockers + ( + review_workflow_load.recovery_handoff_without_replay()), + } hard_stop = terminal_review_hard_stop_reasons(pr_number, "mark_ready") if hard_stop: return {"marked_ready": False, "reasons": hard_stop} @@ -3033,6 +3562,7 @@ def gitea_dry_run_pr_review( host: str | None = None, org: str | None = None, repo: str | None = None, + worktree_path: str | None = None, ) -> dict: """Validate review submission mechanics without a live PR mutation.""" return _evaluate_pr_review_submission( @@ -3045,6 +3575,7 @@ def gitea_dry_run_pr_review( org=org, repo=repo, live=False, + worktree_path=worktree_path, ) @@ -3060,6 +3591,7 @@ def gitea_submit_pr_review( org: str | None = None, repo: str | None = None, final_review_decision_ready: bool = False, + worktree_path: str | None = None, ) -> dict: """Gated PR review mutation: comment findings, request changes, or approve. @@ -3078,6 +3610,7 @@ def gitea_submit_pr_review( repo=repo, live=True, final_review_decision_ready=final_review_decision_ready, + worktree_path=worktree_path, ) @@ -3140,12 +3673,12 @@ def gitea_edit_pr( if not payload: raise ValueError("At least one field to edit (title, body, state, base) must be provided.") - verify_preflight_purity(remote) + closing = payload.get("state") == "closed" + verify_preflight_purity(remote, task="close_pr" if closing else None) # PR closure is a first-class capability, distinct from retitling or # rebasing edits (#216). Gate BEFORE auth/API setup so a blocked close # never touches the network. - closing = payload.get("state") == "closed" if closing: gate_reasons = _profile_operation_gate("gitea.pr.close") if gate_reasons: @@ -3390,7 +3923,7 @@ def gitea_commit_files( branch="", ) - verify_preflight_purity(remote) + verify_preflight_purity(remote, task="commit_files") processed_files, source_proofs = _prepare_commit_payload_files(files) h, o, r = _resolve(remote, host, org, repo) @@ -3438,6 +3971,7 @@ def gitea_merge_pr( host: str | None = None, org: str | None = None, repo: str | None = None, + worktree_path: str | None = None, ) -> dict: """Gated merge of a Gitea pull request (#16). @@ -3484,6 +4018,10 @@ def gitea_merge_pr( host: Override the Gitea host. org: Override the owner/organization. repo: Override the repository name. + worktree_path: Merger workspace path under ``branches/`` or clean + control checkout; defaults to ``GITEA_MERGER_WORKTREE``, + ``GITEA_ACTIVE_WORKTREE``, or the MCP server process root. Ignores + foreign ``GITEA_AUTHOR_WORKTREE`` bindings (#510). Returns: dict describing the attempt: performed, authenticated user, profile @@ -3491,7 +4029,10 @@ def gitea_merge_pr( reasons/gates passed or blocked, and merge result / merge commit if available. Never secrets. """ - verify_preflight_purity(remote) + _verify_role_mutation_workspace( + remote, worktree_path=worktree_path, task="merge_pr" + ) + workflow_blockers = _review_workflow_load_gate_reasons() do = (do or "").strip().lower() result = { "performed": False, @@ -3509,6 +4050,10 @@ def gitea_merge_pr( "reasons": [], } reasons = result["reasons"] + if workflow_blockers: + reasons.extend(workflow_blockers) + reasons.extend(review_workflow_load.recovery_handoff_without_replay()) + return result # Gate 1 — valid merge method (no API call on a bad method). if do not in _MERGE_METHODS: @@ -4020,7 +4565,19 @@ def gitea_delete_branch( "permission_report": _permission_block_report("gitea.branch.delete"), } - verify_preflight_purity(remote) + audit_allowed, audit_reasons = ( + audit_reconciliation_mode.check_audit_mutation_allowed("delete_branch") + ) + if not audit_allowed: + return { + "success": False, + "performed": False, + "required_permission": "gitea.branch.delete", + "reasons": audit_reasons, + "audit_phase": audit_reconciliation_mode.current_phase(), + } + + verify_preflight_purity(remote, task="delete_branch") h, o, r = _resolve(remote, host, org, repo) auth = _auth(h) import urllib.parse @@ -4051,6 +4608,59 @@ def _remote_branch_exists(h: str, o: str, r: str, auth: str, branch: str) -> boo raise +@mcp.tool() +def gitea_capture_branches_worktree_snapshot( + open_pr_branches: list[str] | None = None, + active_lock_branch: str | None = None, + leased_paths: list[str] | None = None, + worktree_path: str | None = None, +) -> dict: + """Read-only: capture ``branches/`` and worktree audit snapshot (#404).""" + read_block = _profile_operation_gate("gitea.read") + if read_block: + return { + "success": False, + "reasons": read_block, + "permission_report": _permission_block_report("gitea.read"), + } + root = PROJECT_ROOT + if worktree_path: + root = os.path.realpath(os.path.abspath(worktree_path)) + git_root = _get_git_root(root) + if git_root: + root = git_root + snapshot = worktree_cleanup_audit.capture_branches_worktree_snapshot( + root, + open_pr_branches=open_pr_branches, + active_lock_branch=active_lock_branch, + leased_paths=leased_paths, + ) + return {"success": True, "snapshot": snapshot} + + +@mcp.tool() +def gitea_assess_worktree_cleanup_integrity( + before_snapshot: dict, + after_snapshot: dict, + removals: list[dict] | None = None, + explained_missing: dict[str, str] | None = None, +) -> dict: + """Read-only: reconcile cleanup before/after snapshots (#404).""" + read_block = _profile_operation_gate("gitea.read") + if read_block: + return { + "integrity_passed": False, + "reasons": read_block, + "permission_report": _permission_block_report("gitea.read"), + } + return worktree_cleanup_audit.assess_worktree_cleanup_integrity( + before=before_snapshot, + after=after_snapshot, + removals=removals, + explained_missing=explained_missing, + ) + + @mcp.tool() def gitea_reconcile_merged_cleanups( dry_run: bool = True, @@ -4089,6 +4699,18 @@ def gitea_reconcile_merged_cleanups( "execute_confirmed must be True when dry_run=False (fail closed)" ) + if not dry_run: + exec_allowed, exec_reasons = ( + audit_reconciliation_mode.check_cleanup_execution_allowed() + ) + if not exec_allowed: + return { + "success": False, + "performed": False, + "reasons": exec_reasons, + "audit_phase": audit_reconciliation_mode.current_phase(), + } + h, o, r = _resolve(remote, host, org, repo) auth = _auth(h) base = repo_api_url(h, o, r) @@ -4129,7 +4751,7 @@ def gitea_reconcile_merged_cleanups( report["executed"] = False return {"success": True, "performed": False, **report} - verify_preflight_purity(remote) + verify_preflight_purity(remote, task="reconcile_merged_cleanups") actions: list[dict] = [] for entry in report.get("entries") or []: head_branch = entry.get("head_branch") or "" @@ -4171,6 +4793,53 @@ def gitea_reconcile_merged_cleanups( return {"success": True, "performed": True, **report} +@mcp.tool() +def gitea_authorize_reconciliation_cleanup_phase( + operator_approved: bool = False, + workflow_authorized: bool = False, + delete_capability_proven: bool = False, + safe_to_delete_remote: bool = False, + safe_to_remove_worktree: bool = False, + before_state: str = "", + after_state: str = "", +) -> dict: + """Authorize cleanup phase after audit-only reconciliation (#419). + + Requires operator or workflow approval, exact delete_branch capability proof, + branch/worktree safety proof, and before/after state snapshots. Audit phase + forbids branch deletion, worktree removal, pushes, and issue/PR mutations. + """ + delete_gate = _profile_operation_gate("gitea.branch.delete") + capability_ok = not bool(delete_gate) + if delete_capability_proven and delete_gate: + return { + "authorized": False, + "performed": False, + "delete_capability_verified": False, + "reasons": [ + "delete_capability_proven=true but active profile lacks " + "gitea.branch.delete", + ] + delete_gate, + "audit_phase": audit_reconciliation_mode.current_phase(), + } + result = audit_reconciliation_mode.authorize_cleanup_phase( + operator_approved=operator_approved, + workflow_authorized=workflow_authorized, + delete_capability_proven=delete_capability_proven and capability_ok, + safety_proof={ + "safe_to_delete_remote": safe_to_delete_remote, + "safe_to_remove_worktree": safe_to_remove_worktree, + }, + before_after_snapshot={ + "before": (before_state or "").strip(), + "after": (after_state or "").strip(), + }, + ) + result["performed"] = bool(result.get("authorized")) + result["delete_capability_verified"] = capability_ok + return result + + @mcp.tool() def gitea_assess_already_landed_reconciliation( pr_number: int, @@ -4343,6 +5012,90 @@ def gitea_scan_already_landed_open_prs( } +@mcp.tool() +def gitea_audit_worktree_cleanup( + remote: str = "dadeschools", + host: str | None = None, + org: str | None = None, + repo: str | None = None, + ttl_hours: float = worktree_cleanup_audit.DEFAULT_TTL_HOURS, +) -> dict: + """Read-only: classify every session-owned worktree under ``branches/`` (#401). + + Audits the local ``branches/`` directory, classifying each worktree as + active open PR, active issue work, dirty local, clean stale removable, + detached review leftover, or unsafe/unknown. Open PR branch heads are + fetched live so a worktree tied to an open PR is never marked removable; + the active issue-lock branch is read from the local lock file and treated + as active work. Deletes nothing and mutates no Gitea state. + + Fails closed if the live open-PR list cannot be fetched: without it, + removability cannot be proven, so no candidates are returned. + + Args: + remote: Known instance — 'dadeschools' or 'prgs'. + host: Override the Gitea host. + org: Override the owner/organization. + repo: Override the repository name. + ttl_hours: Age (hours) after which a clean issue/conflict-fix + worktree becomes stale-removable (default from + GITEA_WORKTREE_TTL_HOURS). + + Returns: + dict with per-worktree classifications, counts, removable + candidates, and the ``git worktree list`` verification proof. + """ + read_block = _profile_operation_gate("gitea.read") + if read_block: + return { + "success": False, + "performed": False, + "reasons": read_block, + "permission_report": _permission_block_report("gitea.read"), + } + + try: + h, o, r = _resolve(remote, host, org, repo) + auth = _auth(h) + open_prs = api_get_all(f"{repo_api_url(h, o, r)}/pulls?state=open", auth) + except Exception as exc: + return { + "success": False, + "performed": False, + "open_pr_state_verified": False, + "reasons": [ + "could not fetch live open PRs; removability unverified " + f"(fail closed): {_redact(str(exc))}" + ], + } + + open_pr_branches = { + str((pr.get("head") or {}).get("ref")) + for pr in open_prs + if (pr.get("head") or {}).get("ref") + } + + active_issue_branches: set[str] = set() + lock = merged_cleanup_reconcile.read_issue_lock(ISSUE_LOCK_FILE) + if lock and lock.get("branch_name"): + active_issue_branches.add(str(lock["branch_name"]).strip()) + + report = worktree_cleanup_audit.audit_branches_directory( + PROJECT_ROOT, + open_pr_branches=open_pr_branches, + active_issue_branches=active_issue_branches, + now=datetime.now(timezone.utc), + ttl_hours=ttl_hours, + ) + return { + "success": True, + "performed": False, + "open_pr_state_verified": True, + "task_mode": "work-issue", + **report, + } + + @mcp.tool() def gitea_reconcile_already_landed_pr( pr_number: int, @@ -4443,7 +5196,7 @@ def gitea_reconcile_already_landed_pr( ) return result - verify_preflight_purity(remote) + verify_preflight_purity(remote, task="reconcile_already_landed_pr") if post_comment and comment_body.strip(): comment_block = _profile_operation_gate("gitea.pr.comment") @@ -4546,7 +5299,7 @@ def gitea_close_issue( task_capability_map.required_permission("close_issue")) if blocked: return blocked - verify_preflight_purity(remote) + verify_preflight_purity(remote, task="close_issue") h, o, r = _resolve(remote, host, org, repo) auth = _auth(h) url = f"{repo_api_url(h, o, r)}/issues/{issue_number}" @@ -4974,7 +5727,7 @@ def gitea_acquire_reviewer_pr_lease( "permission_report": _permission_block_report("gitea.pr.comment"), } - verify_preflight_purity(remote) + _verify_role_mutation_workspace(remote, worktree=worktree, task="review_pr") h, o, r = _resolve(remote, host, org, repo) auth = _auth(h) profile = get_profile() @@ -4984,6 +5737,13 @@ def gitea_acquire_reviewer_pr_lease( comments = _fetch_pr_comments( pr_number, remote=remote, host=host, org=org, repo=repo) + # Refuse merge-oriented lease acquisition/adoption on an already-merged or + # closed PR: the lease is moot and adopting it for merge work is unsafe (#515). + pr_live = api_request( + "GET", f"{repo_api_url(h, o, r)}/pulls/{pr_number}", auth) or {} + pr_merged_or_closed = bool( + pr_live.get("merged") or pr_live.get("merged_at") + ) or (str(pr_live.get("state") or "").strip().lower() == "closed") assessment = reviewer_pr_lease.assess_acquire_lease( comments, pr_number=pr_number, @@ -4996,6 +5756,7 @@ def gitea_acquire_reviewer_pr_lease( candidate_head=candidate_head, target_branch=target_branch, target_branch_sha=target_branch_sha, + pr_merged_or_closed=pr_merged_or_closed, ) if not assessment.get("acquire_allowed"): return { @@ -5003,6 +5764,7 @@ def gitea_acquire_reviewer_pr_lease( "acquired": False, "reasons": assessment.get("reasons") or [], "existing_lease": assessment.get("existing_lease"), + "post_merge_moot": assessment.get("post_merge_moot", False), } body = assessment["lease_body"] @@ -5031,7 +5793,10 @@ def gitea_acquire_reviewer_pr_lease( "target_branch_sha": target_branch_sha, "repo": repo_label, "comment_id": posted.get("id"), - }) + }, lease_provenance=merger_lease_adoption.build_lease_provenance( + source=merger_lease_adoption.SOURCE_ACQUIRE, + comment_id=posted.get("id"), + )) return { "success": True, "acquired": True, @@ -5043,6 +5808,175 @@ def gitea_acquire_reviewer_pr_lease( } +@mcp.tool() +def gitea_adopt_merger_pr_lease( + pr_number: int, + worktree: str, + expected_head_sha: str, + issue_number: int | None = None, + session_id: str | None = None, + remote: str = "dadeschools", + host: str | None = None, + org: str | None = None, + repo: str | None = None, +) -> dict: + """Adopt an active reviewer PR lease for a merger session (#536). + + Merger sessions must not manually seed ``reviewer_pr_lease._SESSION_LEASE``. + This tool posts durable adoption proof on the PR thread and records + sanctioned in-session lease provenance so ``gitea_merge_pr`` can proceed + after a separate reviewer session held the lease. + + Args: + pr_number: Open PR to adopt. + worktree: Merger workspace under ``branches/``. + expected_head_sha: Approved head the merger will pin during merge. + issue_number: Optional linked issue for the adoption record. + session_id: Optional merger session id (generated when omitted). + remote: Known instance — 'dadeschools' or 'prgs'. + host: Override the Gitea host. + org: Override the owner/organization. + repo: Override the repository name. + + Returns: + dict with adoption proof, session lease, and block reasons when the + guarded preconditions are not met. + """ + read_block = _profile_operation_gate("gitea.read") + if read_block: + return { + "success": False, + "adopted": False, + "reasons": read_block, + "permission_report": _permission_block_report("gitea.read"), + } + comment_block = _profile_operation_gate("gitea.pr.comment") + if comment_block: + return { + "success": False, + "adopted": False, + "reasons": comment_block, + "permission_report": _permission_block_report("gitea.pr.comment"), + } + merge_block = _profile_operation_gate("gitea.pr.merge") + if merge_block: + return { + "success": False, + "adopted": False, + "reasons": merge_block, + "permission_report": _permission_block_report("gitea.pr.merge"), + } + + _verify_role_mutation_workspace( + remote, worktree=worktree, task="adopt_merger_pr_lease" + ) + h, o, r = _resolve(remote, host, org, repo) + auth = _auth(h) + profile = get_profile() + profile_name = profile.get("profile_name") or "unknown" + identity = _authenticated_username(remote) or profile.get("username") or "" + sid = (session_id or "").strip() or reviewer_pr_lease.new_session_id() + repo_label = f"{o}/{r}" + + pr_live = api_request( + "GET", f"{repo_api_url(h, o, r)}/pulls/{pr_number}", auth) or {} + pr_state = (pr_live.get("state") or "").strip().lower() + pr_open = pr_state == "open" and not pr_live.get("merged") + live_head = (pr_live.get("head", {}) or {}).get("sha") or pr_live.get( + "head_sha" + ) + + feedback = gitea_get_pr_review_feedback( + pr_number=pr_number, + remote=remote, + host=host, + org=org, + repo=repo, + ) + approval_at_head = bool(feedback.get("approval_at_current_head")) + + comments = _fetch_pr_comments( + pr_number, remote=remote, host=host, org=org, repo=repo) + assessment = merger_lease_adoption.assess_adopt_merger_lease( + comments, + pr_number=pr_number, + adopter_identity=identity, + adopter_profile=profile_name, + adopter_session_id=sid, + repo=repo_label, + issue_number=issue_number, + worktree=worktree, + expected_head_sha=expected_head_sha, + live_head_sha=live_head, + approval_at_head=approval_at_head, + pr_open=pr_open, + ) + if not assessment.get("adopt_allowed"): + return { + "success": False, + "adopted": False, + "pr_number": pr_number, + "reasons": assessment.get("reasons") or [], + "active_lease": assessment.get("active_lease"), + "expected_head_sha": expected_head_sha, + "live_head_sha": live_head, + } + + active = assessment.get("active_lease") or {} + body = assessment["adoption_body"] + comment_url = f"{repo_api_url(h, o, r)}/issues/{pr_number}/comments" + with _audited( + "comment_pr", + host=h, + remote=remote, + org=o, + repo=r, + pr_number=pr_number, + request_metadata={"source": "adopt_merger_pr_lease"}, + ): + posted = api_request("POST", comment_url, auth, {"body": body}) + + provenance = merger_lease_adoption.build_lease_provenance( + source=merger_lease_adoption.SOURCE_ADOPT, + comment_id=posted.get("id"), + adopted_from_session_id=active.get("session_id"), + adopted_from_profile=active.get("profile"), + adopted_from_reviewer_identity=active.get("reviewer_identity"), + adoption_reason=merger_lease_adoption.DEFAULT_ADOPTION_REASON, + ) + session_lease = reviewer_pr_lease.record_session_lease({ + "pr_number": pr_number, + "issue_number": issue_number or active.get("issue_number"), + "session_id": sid, + "reviewer_identity": identity, + "profile": profile_name, + "worktree": worktree, + "phase": "adopted", + "candidate_head": live_head, + "target_branch": active.get("target_branch") or "master", + "target_branch_sha": active.get("target_branch_sha"), + "repo": repo_label, + "comment_id": posted.get("id"), + }, lease_provenance=provenance) + + return { + "success": True, + "adopted": True, + "pr_number": pr_number, + "session_id": sid, + "adoption_comment_id": posted.get("id"), + "adopted_from_session_id": active.get("session_id"), + "adopted_from_profile": active.get("profile"), + "adopted_from_reviewer_identity": active.get("reviewer_identity"), + "adoption_reason": merger_lease_adoption.DEFAULT_ADOPTION_REASON, + "expected_head_sha": expected_head_sha, + "live_head_sha": live_head, + "session_lease": session_lease, + "lease_provenance": provenance, + "reasons": [], + } + + @mcp.tool() def gitea_heartbeat_reviewer_pr_lease( pr_number: int, @@ -5075,7 +6009,7 @@ def gitea_heartbeat_reviewer_pr_lease( ], } - verify_preflight_purity(remote) + verify_preflight_purity(remote, task="review_pr") h, o, r = _resolve(remote, host, org, repo) auth = _auth(h) body = reviewer_pr_lease.format_lease_body( @@ -5103,6 +6037,17 @@ def gitea_heartbeat_reviewer_pr_lease( ): posted = api_request("POST", comment_url, auth, {"body": body}) + prior_provenance = (session.get("lease_provenance") or {}).copy() + heartbeat_provenance = merger_lease_adoption.build_lease_provenance( + source=merger_lease_adoption.SOURCE_HEARTBEAT, + comment_id=posted.get("id"), + adopted_from_session_id=prior_provenance.get("adopted_from_session_id"), + adopted_from_profile=prior_provenance.get("adopted_from_profile"), + adopted_from_reviewer_identity=prior_provenance.get( + "adopted_from_reviewer_identity" + ), + adoption_reason=prior_provenance.get("adoption_reason"), + ) updated = reviewer_pr_lease.record_session_lease({ **session, "phase": phase, @@ -5110,7 +6055,8 @@ def gitea_heartbeat_reviewer_pr_lease( "candidate_head": candidate_head or session.get("candidate_head"), "target_branch_sha": target_branch_sha or session.get("target_branch_sha"), "last_comment_id": posted.get("id"), - }) + "comment_id": posted.get("id"), + }, lease_provenance=heartbeat_provenance) return { "success": True, "posted": True, @@ -5151,6 +6097,238 @@ def gitea_assess_reviewer_pr_lease( } +@mcp.tool() +def gitea_cleanup_post_merge_moot_lease( + pr_number: int, + apply: bool = False, + remote: str = "dadeschools", + host: str | None = None, + org: str | None = None, + repo: str | None = None, +) -> dict: + """Safely resolve a moot reviewer lease left on an ALREADY-MERGED/closed PR (#515). + + Read-first and fail-safe. This tool never merges and never adopts a lease. + It only acts when the live PR state is merged/closed, and it never steals or + force-cleans an active *foreign* lease while the PR is still open. When + ``apply`` is true and a lease is still active on a merged/closed PR, it posts + a terminal ``phase: released`` lease marker (``blocker: post-merge-moot``) — + an append-only comment that neutralises the moot lease without deleting any + other session's comment. + + Args: + pr_number: The PR whose lingering lease to assess/clean. + apply: When false (default) report only (read-only). When true, post the + terminal released marker if — and only if — cleanup is allowed. + remote: Known instance — 'dadeschools' or 'prgs'. + host: Override the Gitea host. + org: Override the owner/organization. + repo: Override the repository name. + + Returns: + dict reporting PR merged/closed state, merge_commit_sha, linked-issue + closure state, whether the lease is moot, whether cleanup was performed + or skipped (and why), and ``no_merge_or_adoption`` True — this path never + merges or adopts. + """ + read_block = _profile_operation_gate("gitea.read") + if read_block: + return { + "success": False, + "cleanup_performed": False, + "no_merge_or_adoption": True, + "reasons": read_block, + "permission_report": _permission_block_report("gitea.read"), + } + h, o, r = _resolve(remote, host, org, repo) + auth = _auth(h) + pr_live = api_request( + "GET", f"{repo_api_url(h, o, r)}/pulls/{pr_number}", auth) or {} + comments = _fetch_pr_comments( + pr_number, remote=remote, host=host, org=org, repo=repo) + assessment = reviewer_pr_lease.assess_post_merge_moot_lease( + comments, + pr_number=pr_number, + pr_merged=bool(pr_live.get("merged") or pr_live.get("merged_at")), + pr_state=pr_live.get("state"), + merge_commit_sha=pr_live.get("merge_commit_sha"), + ) + + # Best-effort linked-issue closure state for the report. + active = assessment.get("active_lease") or {} + issue_no = active.get("issue_number") + linked_issue_state = None + if issue_no: + try: + issue = api_request( + "GET", f"{repo_api_url(h, o, r)}/issues/{issue_no}", auth) or {} + linked_issue_state = issue.get("state") + except Exception: + linked_issue_state = None + + report = { + "success": True, + "pr_number": pr_number, + "pr_state": assessment.get("pr_state"), + "pr_merged_or_closed": assessment.get("pr_merged_or_closed"), + "merge_commit_sha": assessment.get("merge_commit_sha"), + "linked_issue_number": issue_no, + "linked_issue_state": linked_issue_state, + "lease_moot": assessment.get("is_moot"), + "active_lease": assessment.get("active_lease"), + "cleanup_allowed": assessment.get("cleanup_allowed"), + "cleanup_performed": False, + "no_merge_or_adoption": True, + "mode": "apply" if apply else "read_only", + "reasons": assessment.get("reasons") or [], + } + + if not apply: + return report + if not assessment.get("cleanup_allowed"): + report["cleanup_skipped_reason"] = ( + assessment.get("reasons") or ["cleanup not allowed"] + ) + return report + + comment_block = _profile_operation_gate("gitea.pr.comment") + if comment_block: + report["success"] = False + report["reasons"] = comment_block + report["permission_report"] = _permission_block_report("gitea.pr.comment") + return report + + verify_preflight_purity(remote) + body = assessment["release_body"] + comment_url = f"{repo_api_url(h, o, r)}/issues/{pr_number}/comments" + with _audited( + "comment_pr", + host=h, + remote=remote, + org=o, + repo=r, + pr_number=pr_number, + request_metadata={"source": "cleanup_post_merge_moot_lease"}, + ): + posted = api_request("POST", comment_url, auth, {"body": body}) + + report["cleanup_performed"] = True + report["released_comment_id"] = posted.get("id") + return report + + +@mcp.tool() +def gitea_release_reviewer_pr_lease( + pr_number: int, + worktree: str | None = None, + remote: str = "dadeschools", + host: str | None = None, + org: str | None = None, + repo: str | None = None, +) -> dict: + """Safely release an active reviewer lease owned by the current session on an open PR. + + This provides a canonical way to clean up reviewer leases when a review fails + before submission. + + Args: + pr_number: The PR number whose reviewer lease to release. + worktree: Optional worktree path (resolves automatically if not supplied). + remote: Known instance — 'dadeschools' or 'prgs'. + host: Override the Gitea host. + org: Override the owner/organization. + repo: Override the repository name. + + Returns: + dict reporting release status. + """ + _verify_role_mutation_workspace( + remote, worktree=worktree, task="review_pr" + ) + h, o, r = _resolve(remote, host, org, repo) + auth = _auth(h) + + comments = _fetch_pr_comments( + pr_number, remote=remote, host=host, org=org, repo=repo + ) + active = reviewer_pr_lease.find_active_reviewer_lease( + comments, pr_number=pr_number + ) + + if not active: + return { + "success": True, + "released": False, + "reasons": ["no active reviewer lease found on PR"], + } + + identity = _authenticated_username(remote) or "" + session = reviewer_pr_lease.get_session_lease() or {} + session_id = session.get("session_id") + + owner_identity = active.get("reviewer_identity") + owner_session_id = active.get("session_id") + + # Authorize release: identity matches, session_id matches, or lease is expired/reclaimable + authorized = False + reasons = [] + if owner_identity and identity and owner_identity == identity: + authorized = True + elif owner_session_id and session_id and owner_session_id == session_id: + authorized = True + else: + freshness = reviewer_pr_lease.classify_lease_freshness(active) + if freshness in {"expired", "reclaimable"}: + authorized = True + + if not authorized: + return { + "success": False, + "released": False, + "reasons": [ + f"unauthorized to release active lease: owned by {owner_identity} " + f"(session {owner_session_id})" + ], + } + + # Format and post release comment + body = reviewer_pr_lease.format_lease_body( + repo=active.get("repo") or f"{o}/{r}", + pr_number=pr_number, + issue_number=active.get("issue_number"), + reviewer_identity=owner_identity or identity, + profile=active.get("profile") or "reviewer", + session_id=owner_session_id or session_id or "unknown", + worktree=active.get("worktree") or "", + phase="released", + candidate_head=active.get("candidate_head"), + target_branch=active.get("target_branch") or "master", + target_branch_sha=active.get("target_branch_sha"), + blocker="manual-release", + ) + + comment_url = f"{repo_api_url(h, o, r)}/issues/{pr_number}/comments" + with _audited( + "comment_pr", + host=h, + remote=remote, + org=o, + repo=r, + pr_number=pr_number, + request_metadata={"source": "release_reviewer_pr_lease"}, + ): + posted = api_request("POST", comment_url, auth, {"body": body}) + + reviewer_pr_lease.clear_session_lease() + + return { + "success": True, + "released": True, + "comment_id": posted.get("id"), + "reasons": [], + } + + @mcp.tool() def gitea_list_issue_comments( issue_number: int, @@ -5192,7 +6370,12 @@ def gitea_list_issue_comments( h, o, r = _resolve(remote, host, org, repo) auth = _auth(h) api = f"{repo_api_url(h, o, r)}/issues/{issue_number}/comments" - comments = api_request("GET", api, auth) or [] + comments = api_request("GET", api, auth) + # Fail safe to no comments when the API returns a non-list payload (e.g. an + # error object such as an HTTP 401 body) so listing never crashes on a + # malformed response (#485). + if not isinstance(comments, list): + comments = [] reveal = _reveal_endpoints() out = [] for c in comments[:limit]: @@ -5217,6 +6400,7 @@ def gitea_create_issue_comment( host: str | None = None, org: str | None = None, repo: str | None = None, + worktree_path: str | None = None, ) -> dict: """Post a markdown comment to a Gitea issue's discussion thread. @@ -5238,6 +6422,7 @@ def gitea_create_issue_comment( host: Override the Gitea host. org: Override the owner/organization. repo: Override the repository name. + worktree_path: Optional path to verify branches-only guard. Returns: dict with 'success', 'comment_id', and 'issue_number' ('url' only @@ -5246,7 +6431,7 @@ def gitea_create_issue_comment( (permission blocks also carry a structured 'permission_report', #142). """ - verify_preflight_purity(remote) + verify_preflight_purity(remote, worktree_path=worktree_path, task="comment_issue") gate_reasons = _profile_operation_gate("gitea.issue.comment") reasons = list(gate_reasons) if not (body or "").strip(): @@ -5545,6 +6730,8 @@ _PROJECT_SKILLS = { "steps": [ "Resolve task first: gitea_resolve_task_capability(task='review_pr') " "to confirm reviewer namespace and avoid author-profile blocks.", + "Load canonical workflow proof with gitea_load_review_workflow " + "before any review/merge mutation (#389).", "Verify reviewer identity with gitea_whoami; the PR author " "must be a different user.", "Reconcile live queue state FIRST (do not trust prior handoffs): " @@ -6468,6 +7655,8 @@ def gitea_get_runtime_context( ), "role_kind": _role_kind(allowed, forbidden), "shell_health": native_mcp_preference.shell_health_status(), + "workflow_load_proof": review_workflow_load.workflow_load_status( + PROJECT_ROOT), } if reveal and h: @@ -6476,6 +7665,80 @@ def gitea_get_runtime_context( return result +@mcp.tool() +def gitea_record_pre_review_command( + command: str, + cwd: str | None = None, + classification: str | None = None, +) -> dict: + """Classify and record a command executed before workflow load (#403). + + Read-only with respect to Gitea API. Pre-review inventory/diagnostic commands + may be recorded as allowed; boundary violations block reviewer mutations. + """ + recorded = review_workflow_boundary.record_pre_review_command( + command, + cwd=cwd, + project_root=PROJECT_ROOT, + classification=classification, + ) + boundary_state = review_workflow_boundary.assess_boundary_status(PROJECT_ROOT) + return { + "success": True, + "recorded": recorded, + "boundary_status": boundary_state.get("boundary_status"), + "boundary_clean": boundary_state.get("boundary_clean"), + "reasons": list(boundary_state.get("reasons") or []), + } + + +@mcp.tool() +def gitea_load_review_workflow( + prompt_text: str | None = None, +) -> dict: + """Load and record canonical review-merge workflow proof for this session (#389, #403). + + Read-only with respect to Gitea API; records in-process workflow source/hash + proof and session boundary state required before reviewer review or merge + mutations. + """ + try: + recorded = review_workflow_load.record_review_workflow_load( + PROJECT_ROOT, prompt_text=prompt_text) + except OSError as exc: + return { + "success": False, + "loaded": False, + "reasons": [str(exc)], + "recovery_handoff": review_workflow_load.recovery_handoff_without_replay(), + } + boundary_reasons = review_workflow_boundary.boundary_blockers(PROJECT_ROOT) + helper = review_workflow_boundary.workflow_load_helper_result( + recorded, PROJECT_ROOT) + return { + "success": not boundary_reasons, + "loaded": True, + "workflow_source": recorded["workflow_source"], + "task_mode": recorded["task_mode"], + "workflow_hash": recorded["workflow_hash"], + "workflow_version": recorded["workflow_version"], + "final_report_schema_path": recorded["final_report_schema_path"], + "final_report_schema_hash": recorded["final_report_schema_hash"], + "prompt_conflicts_with_workflow": recorded[ + "prompt_conflicts_with_workflow"], + "prompt_conflict_reasons": recorded.get("prompt_conflict_reasons") or [], + "workflow_load_proof_present": True, + "boundary_status": recorded.get("boundary_status"), + "boundary_clean": recorded.get("boundary_clean"), + "workflow_load_helper_result": helper, + "reasons": boundary_reasons, + "recovery_handoff": ( + review_workflow_load.recovery_handoff_without_replay() + if boundary_reasons else [] + ), + } + + @mcp.tool() def gitea_list_profiles() -> dict: """Read-only: list all Gitea MCP profiles with redacted metadata. @@ -6780,7 +8043,7 @@ def gitea_mark_issue( task_capability_map.required_permission("mark_issue")) if blocked: return blocked - verify_preflight_purity(remote, worktree_path=worktree_path) + verify_preflight_purity(remote, worktree_path=worktree_path, task="mark_issue") h, o, r = _resolve(remote, host, org, repo) auth = _auth(h) base = repo_api_url(h, o, r) @@ -6858,7 +8121,7 @@ def gitea_post_heartbeat( task_capability_map.required_permission("post_heartbeat")) if blocked: return blocked - verify_preflight_purity(remote) + verify_preflight_purity(remote, task="post_heartbeat") active_profile = profile or get_profile().get("profile_name") body = issue_claim_heartbeat.format_heartbeat_body( kind="progress", @@ -6967,22 +8230,39 @@ def gitea_assess_conflict_fix_push( "reasons": read_block, "permission_report": _permission_block_report("gitea.read"), } - comments = _list_pr_lease_comments( + fetched = _fetch_pr_lease_comments_safe( pr_number, remote=remote, host=host, org=org, repo=repo, + require_open=True, ) - return pr_work_lease.assess_conflict_fix_push( + if not fetched["success"]: + return { + "push_allowed": False, + "block": True, + "reasons": fetched["reasons"], + "pr_lookup": fetched.get("pr_lookup"), + "resolved_repo": fetched.get("resolved_repo"), + "remote": fetched.get("remote"), + "pr_number": pr_number, + "assessment_failed": True, + } + assessment = pr_work_lease.assess_conflict_fix_push( pr_number=pr_number, - comments=comments, + comments=fetched["comments"], branch_head_before=branch_head_before, branch_head_after=branch_head_after, worktree_path=worktree_path, push_cwd=push_cwd, is_fast_forward=is_fast_forward, ) + assessment["pr_lookup"] = fetched.get("pr_lookup") + assessment["resolved_repo"] = fetched.get("resolved_repo") + assessment["remote"] = fetched.get("remote") + assessment["live_pr_head_sha"] = fetched.get("head_sha") + return assessment @mcp.tool() @@ -7101,7 +8381,7 @@ def gitea_cleanup_stale_claims( task_capability_map.required_permission("cleanup_stale_claims")) if blocked: return blocked - verify_preflight_purity(remote) + verify_preflight_purity(remote, task="cleanup_stale_claims") h, o, r = _resolve(remote, host, org, repo) auth = _auth(h) @@ -7255,7 +8535,7 @@ def gitea_set_issue_labels( task_capability_map.required_permission("set_issue_labels")) if blocked: return blocked - verify_preflight_purity(remote) + verify_preflight_purity(remote, task="set_issue_labels") h, o, r = _resolve(remote, host, org, repo) auth = _auth(h) base = repo_api_url(h, o, r) @@ -7433,6 +8713,102 @@ def gitea_route_task_session( ) +def _check_mcp_runtimes_diagnostics(task: str, matching_profiles: list[str]) -> list[str]: + """Check running runtimes and return errors if they are missing or stale.""" + import subprocess + import re + from datetime import datetime + + reasons = [] + code_path = os.path.join(PROJECT_ROOT, "gitea_mcp_server.py") + if not os.path.exists(code_path): + return reasons + + code_mtime = datetime.fromtimestamp(os.path.getmtime(code_path)) + + try: + proc = subprocess.run( + ["ps", "-o", "pid,lstart,command", "-ax"], + capture_output=True, text=True, check=True + ) + except Exception as exc: + return [f"stale-runtime: failed to list running processes: {exc}"] + + self_pid = os.getpid() + self_stale = False + + running_profiles = {} + for line in proc.stdout.splitlines()[1:]: + line = line.strip() + if not line or "mcp_server.py" not in line: + continue + + parts = line.split(None, 6) + if len(parts) < 7: + continue + pid_str = parts[0] + lstart_str = " ".join(parts[1:6]) + + try: + pid = int(pid_str) + start_time = datetime.strptime(lstart_str, "%a %b %d %H:%M:%S %Y") + except Exception: + continue + + try: + env_proc = subprocess.run( + ["ps", "eww", str(pid)], + capture_output=True, text=True, check=True + ) + env_out = env_proc.stdout + except Exception: + continue + + profile = "gitea-default" + match = re.search(r'\bGITEA_MCP_PROFILE=([^\s]+)', env_out) + if match: + profile = match.group(1) + + is_stale = start_time < code_mtime + if pid == self_pid and is_stale: + self_stale = True + + if profile not in running_profiles or start_time > running_profiles[profile]["start_time"]: + running_profiles[profile] = { + "pid": pid, + "start_time": start_time, + "is_stale": is_stale + } + + if self_stale: + reasons.append( + "stale-runtime: The active Gitea MCP server process is stale (running code from before changes were merged). " + "Please fully restart the Gitea MCP server (e.g. touch /Users/jasonwalker/.gemini/config/mcp_config.json) and retry." + ) + + if matching_profiles: + any_running = False + any_fresh = False + for mp in matching_profiles: + if mp in running_profiles: + any_running = True + if not running_profiles[mp]["is_stale"]: + any_fresh = True + break + if not any_running: + reasons.append( + f"stale-runtime: None of the matching profiles for task '{task}' ({matching_profiles}) are running in the OS. " + "Please restart the MCP server to ensure they are spawned." + ) + elif not any_fresh: + reasons.append( + f"stale-runtime: All matching profiles for task '{task}' ({matching_profiles}) are running but stale. " + "Please fully restart the Gitea MCP server and retry." + ) + + return reasons + + @mcp.tool() def gitea_resolve_task_capability( task: str, @@ -7492,7 +8868,7 @@ def gitea_resolve_task_capability( "exact_safe_next_action": next_safe_action, } - record_preflight_check("capability", required_role) + record_preflight_check("capability", required_role, resolved_task=task) # Try automatic dispatch switching _ensure_matching_profile(required_permission, required_role, remote, host) @@ -7543,6 +8919,16 @@ def gitea_resolve_task_capability( configured = len(matching_profiles) > 0 available_in_session = allowed_in_current_session + if "PYTEST_CURRENT_TEST" not in os.environ or "GITEA_FORCE_MCP_RUNTIME_CHECK" in os.environ: + runtime_reasons = _check_mcp_runtimes_diagnostics(task, matching_profiles) + if runtime_reasons: + restart_required = True + reason_msg = "; ".join(runtime_reasons) + next_safe_action = ( + "stale-runtime: Gitea MCP runtime conflict or missing process detected. " + "Please fully restart the Gitea MCP server and retry." + ) + if not allowed_in_current_session: if configured and switching: restart_required = True @@ -7620,6 +9006,7 @@ def gitea_resolve_task_capability( init_review_decision_lock( remote if remote in REMOTES else None, task, + force=False, ) record_mutation_authority(profile["profile_name"], username, remote if remote in REMOTES else None, task) @@ -7644,7 +9031,20 @@ def gitea_resolve_task_capability( } if reason_msg: result["reason"] = reason_msg + if task in ("review_pr", "merge_pr"): + result["workflow_load_proof"] = review_workflow_load.workflow_load_status( + PROJECT_ROOT) + if not result["workflow_load_proof"].get("workflow_load_valid"): + guidance = ( + "Call gitea_load_review_workflow before any reviewer review " + "or merge mutation." + ) + if guidance not in task_role_guidance: + task_role_guidance.append(guidance) role_session_router.sync_route_from_capability(result) + if audit_reconciliation_mode.check_audit_task_enters_phase(task): + phase_record = audit_reconciliation_mode.enter_audit_phase(task) + result["reconciliation_phase"] = phase_record.get("phase") was_terminal = capability_stop_terminal.is_active() terminal = capability_stop_terminal.sync_from_capability_result(result) if terminal: diff --git a/issue_acceptance_gate.py b/issue_acceptance_gate.py new file mode 100644 index 0000000..cd731c2 --- /dev/null +++ b/issue_acceptance_gate.py @@ -0,0 +1,300 @@ +"""Controller issue-acceptance gate helpers (#500). + +Pure validation for controller acceptance comments and final-report claims +that an issue is complete. Does not post comments or close issues. +""" + +from __future__ import annotations + +import re + +ACCEPTANCE_HEADING = "controller issue acceptance" + +REQUIRED_FIELDS = ( + "STATE", + "WHO_IS_NEXT", + "NEXT_ACTION", + "NEXT_PROMPT", + "ISSUE", + "MERGED_PR", + "MERGE_COMMIT", + "ACCEPTANCE_CRITERIA_CHECKED", + "VALIDATION_REVIEWED", + "CONTROLLER_DECISION", + "WHY", +) + +ACCEPTED_STATES = frozenset({"accepted"}) +REJECTION_STATES = frozenset({ + "more-work-required", + "more_work_required", + "needs-tests", + "needs_tests", + "needs-docs", + "needs_docs", + "needs-feature-enhancement", + "needs_feature_enhancement", + "needs-follow-up-issue", + "needs_follow_up_issue", + "blocked", +}) + +ALLOWED_NEXT_ACTORS = frozenset({ + "controller", + "author", + "reviewer", + "merger", + "reconciler", + "user", +}) + +_FIELD_RE = re.compile(r"^\s*(?:[-*]\s*)?([A-Z][A-Z0-9_ ]+)\s*:\s*(.*)$") +_FULL_SHA_RE = re.compile(r"\b[0-9a-f]{40}\b", re.IGNORECASE) +_ISSUE_REF_RE = re.compile(r"#\d+") +_PR_REF_RE = re.compile(r"#\d+") +_CHECKED_ITEM_RE = re.compile(r"\[[xX]\]") +_UNCHECKED_ITEM_RE = re.compile(r"\[[\s]\]") + +_CLAIMS_ISSUE_COMPLETE_RE = re.compile( + r"\bissue\s+(?:is\s+)?(?:complete|completed|accepted|closed\s+as\s+complete|fully\s+satisfied)\b|" + r"\bissue\s+acceptance\s*:\s*accepted\b|" + r"\bcontroller\s+acceptance\s*:\s*(?:accepted|complete)\b", + re.IGNORECASE, +) +_MERGE_ONLY_COMPLETE_RE = re.compile( + r"(?:pr\s+merged|merged\s+pr|merge\s+result\s*:\s*merged).{0,120}" + r"(?:issue\s+(?:is\s+)?(?:complete|closed|accepted)|issue\s+complete)", + re.IGNORECASE | re.DOTALL, +) +_CLAIMS_CONTROLLER_ACCEPTANCE_RE = re.compile( + r"controller\s+issue\s+acceptance|controller\s+acceptance\s+(?:posted|complete|pending)", + re.IGNORECASE, +) +_PENDING_ACCEPTANCE_RE = re.compile( + r"controller\s+acceptance\s+(?:pending|required|not\s+(?:yet\s+)?(?:performed|complete))", + re.IGNORECASE, +) + + +def render_controller_acceptance_template() -> str: + """Return the canonical controller issue-acceptance comment template.""" + return """## Controller Issue Acceptance + +STATE: + + +WHO_IS_NEXT: + + +NEXT_ACTION: + + +NEXT_PROMPT: + + +ISSUE: +#... + +MERGED_PR: +#... + +MERGE_COMMIT: +<40-character SHA> + +ACCEPTANCE_CRITERIA_CHECKED: +- [x] ... +- [ ] ... + +VALIDATION_REVIEWED: + + +CONTROLLER_DECISION: + + +WHY: + + +MISSING_WORK: + + +FOLLOW_UP_ISSUES: + + +BLOCKERS: + + +LAST_UPDATED_BY: + +""" + + +def extract_acceptance_fields(text: str | None) -> dict[str, str]: + """Return upper-case labeled fields from a controller acceptance block.""" + fields: dict[str, str] = {} + current_key: str | None = None + for line in (text or "").splitlines(): + match = _FIELD_RE.match(line) + if match: + current_key = match.group(1).strip().upper().replace(" ", "_") + fields[current_key] = match.group(2).strip() + continue + stripped = line.strip() + if current_key and stripped: + existing = fields.get(current_key, "") + fields[current_key] = ( + f"{existing}\n{stripped}" if existing else stripped + ) + return fields + + +def contains_acceptance_block(text: str | None) -> bool: + return ACCEPTANCE_HEADING in (text or "").lower() + + +def _empty_or_placeholder(value: str | None) -> bool: + value = (value or "").strip().lower() + return not value or value in {"none", "n/a", "unknown", "tbd", "<...>", "..."} + + +def _normalize_state(value: str | None) -> str: + return (value or "").strip().lower().replace(" ", "_").replace("-", "_") + + +def validate_controller_acceptance_comment(text: str | None) -> dict: + """Validate a controller issue-acceptance comment.""" + body = text or "" + if not contains_acceptance_block(body): + return { + "valid": False, + "fields": {}, + "reasons": ["missing Controller Issue Acceptance heading"], + } + + fields = extract_acceptance_fields(body) + reasons: list[str] = [] + + for field in REQUIRED_FIELDS: + if _empty_or_placeholder(fields.get(field)): + reasons.append(f"missing required controller acceptance field: {field}") + + state = _normalize_state(fields.get("STATE")) + if state and state not in ACCEPTED_STATES and state not in REJECTION_STATES: + reasons.append( + "STATE must be accepted or a rejection path " + "(more-work-required, needs-tests, needs-docs, " + "needs-feature-enhancement, needs-follow-up-issue, blocked)" + ) + + actor = (fields.get("WHO_IS_NEXT") or "").strip().lower() + if actor and actor not in ALLOWED_NEXT_ACTORS: + reasons.append( + "WHO_IS_NEXT must be one of: " + + ", ".join(sorted(ALLOWED_NEXT_ACTORS)) + ) + + if not _ISSUE_REF_RE.search(fields.get("ISSUE") or ""): + reasons.append("ISSUE must cite an issue number (#N)") + if not _PR_REF_RE.search(fields.get("MERGED_PR") or ""): + reasons.append("MERGED_PR must cite a merged PR number (#N)") + if not _FULL_SHA_RE.search(fields.get("MERGE_COMMIT") or ""): + reasons.append("MERGE_COMMIT must include a full 40-character SHA") + + criteria = fields.get("ACCEPTANCE_CRITERIA_CHECKED") or "" + if not _CHECKED_ITEM_RE.search(criteria) and not _UNCHECKED_ITEM_RE.search(criteria): + reasons.append( + "ACCEPTANCE_CRITERIA_CHECKED must list checked/unchecked criteria items" + ) + + decision = (fields.get("CONTROLLER_DECISION") or "").strip().lower() + if state in ACCEPTED_STATES: + if decision not in {"accepted", "accept"}: + reasons.append("accepted STATE requires CONTROLLER_DECISION: accepted") + if not _CHECKED_ITEM_RE.search(criteria): + reasons.append( + "accepted STATE requires at least one checked acceptance criterion" + ) + if _empty_or_placeholder(fields.get("WHY")): + reasons.append("accepted STATE requires WHY with acceptance rationale") + elif state in REJECTION_STATES: + if decision not in {"rejected", "reject", "more_work_required"}: + reasons.append( + "rejection STATE requires CONTROLLER_DECISION: rejected" + ) + if _empty_or_placeholder(fields.get("NEXT_PROMPT")): + reasons.append( + "rejection STATE requires a paste-ready NEXT_PROMPT for the next actor" + ) + missing = fields.get("MISSING_WORK") or "" + if _empty_or_placeholder(missing): + reasons.append( + "rejection STATE requires MISSING_WORK describing what is still needed" + ) + + return { + "valid": not reasons, + "fields": fields, + "reasons": reasons, + } + + +def claims_issue_complete(text: str | None) -> bool: + """Return True when text claims an issue is complete/accepted.""" + return bool(_CLAIMS_ISSUE_COMPLETE_RE.search(text or "")) + + +def claims_merge_only_issue_complete(text: str | None) -> bool: + """Return True when text treats PR merge as issue completion.""" + return bool(_MERGE_ONLY_COMPLETE_RE.search(text or "")) + + +def claims_controller_acceptance_update(text: str | None) -> bool: + return bool(_CLAIMS_CONTROLLER_ACCEPTANCE_RE.search(text or "")) + + +def notes_controller_acceptance_pending(text: str | None) -> bool: + return bool(_PENDING_ACCEPTANCE_RE.search(text or "")) + + +def validate_final_report_issue_acceptance(report_text: str | None) -> dict: + """Validate issue-completion and controller-acceptance claims in final reports.""" + text = report_text or "" + reasons: list[str] = [] + + complete_claim = claims_issue_complete(text) + merge_only = claims_merge_only_issue_complete(text) + acceptance_claim = claims_controller_acceptance_update(text) + pending_noted = notes_controller_acceptance_pending(text) + has_block = contains_acceptance_block(text) + + if merge_only and not (has_block and validate_controller_acceptance_comment(text)["valid"]): + reasons.append( + "final report treats PR merge as issue completion without controller acceptance proof" + ) + + if complete_claim and not pending_noted: + if not has_block: + reasons.append( + "final report claims issue complete but includes no Controller Issue Acceptance block" + ) + else: + result = validate_controller_acceptance_comment(text) + if not result["valid"]: + reasons.extend(result["reasons"]) + else: + state = _normalize_state(result["fields"].get("STATE")) + if state not in ACCEPTED_STATES: + reasons.append( + "final report claims issue complete but controller STATE is not accepted" + ) + + if acceptance_claim and has_block: + result = validate_controller_acceptance_comment(text) + if not result["valid"]: + reasons.extend(result["reasons"]) + + applicable = complete_claim or merge_only or acceptance_claim or pending_noted + return { + "applicable": applicable, + "valid": not reasons, + "reasons": reasons, + } \ No newline at end of file diff --git a/issue_lock_adoption.py b/issue_lock_adoption.py index 8f8a8d6..81a5a92 100644 --- a/issue_lock_adoption.py +++ b/issue_lock_adoption.py @@ -20,6 +20,43 @@ ADOPT = "adopt_existing_branch" BLOCK_COMPETING = "block_competing_branch" NO_MATCH = "no_matching_branch" +# Citable decision labels aligned with the ``assess_own_branch_adoption`` +# outcomes, surfaced verbatim in the live ``gitea_lock_issue`` response so +# recovery reports (#473-style) can quote the lock tool output directly +# instead of inferring adoption from separate offline checks (#477). +DECISION_LABELS = { + ADOPT: "ADOPT", + BLOCK_COMPETING: "BLOCK_COMPETING", + NO_MATCH: "NO_MATCH", +} + +_SAFE_NEXT_ACTIONS = { + ADOPT: ( + "Own existing branch adopted for lock recovery; proceed to " + "gitea_create_pr for this issue and cite this adoption proof." + ), + BLOCK_COMPETING: ( + "Competing same-issue branch(es) exist; resolve branch ownership " + "before locking. No adoption performed (fail closed)." + ), + NO_MATCH: ( + "No existing branch carries this issue marker; normal lock path " + "applied. No adoption performed." + ), +} + + +def decision_label(outcome: str) -> str: + """Map an ``assess_own_branch_adoption`` outcome to its citable label.""" + return DECISION_LABELS.get(outcome, "UNKNOWN") + + +def safe_next_action(outcome: str) -> str: + """Return the safe next action string for an adoption *outcome*.""" + return _SAFE_NEXT_ACTIONS.get( + outcome, "Unknown adoption outcome; treat as fail closed." + ) + def _branch_name(entry) -> str: if isinstance(entry, dict): @@ -128,6 +165,42 @@ def assess_own_branch_adoption( } +def _matcher_summary(issue_number: int, assessment: dict) -> str: + """Explain, citably, why the assessed branch did or did not qualify. + + Names the numeric word-boundary rule so reports can show that + ``issue-42`` was not matched inside ``issue-420`` (#440 / #477 AC3). + """ + outcome = assessment.get("outcome") + matched = assessment.get("matched_branch") + competing = assessment.get("competing_branches") or [] + if outcome == ADOPT and matched: + return ( + f"branch '{matched}' exactly matches the issue-{int(issue_number)} " + f"marker (numeric word-boundary; 'issue-{int(issue_number)}' is not " + f"matched inside 'issue-{int(issue_number)}0')" + ) + if outcome == BLOCK_COMPETING: + return ( + f"competing same-issue branch(es) {competing} carry the " + f"issue-{int(issue_number)} marker but are not the requested " + f"branch; ownership is ambiguous (fail closed)" + ) + return ( + f"no existing branch carries the issue-{int(issue_number)} marker " + f"under the numeric word-boundary rule" + ) + + +def _competing_branch_check(assessment: dict) -> dict: + """Structured competing-branch verdict for the proof block.""" + competing = list(assessment.get("competing_branches") or []) + return { + "result": "blocked" if competing else "clear", + "competing_branches": competing, + } + + def build_adoption_proof( *, issue_number: int, @@ -143,7 +216,18 @@ def build_adoption_proof( Requirement #4: adoption results must carry issue number, branch name, branch head commit, adoption reason, no-existing-PR proof, no-competing- live-lock proof, and lock file path/status. + + #477: additionally surface explicit, citable adoption-proof fields tied to + the ``assess_own_branch_adoption`` outcome (``adoption_decision``, + ``adopted``, ``adopted_branch``, ``adopted_branch_head``, + ``matcher_summary``, ``competing_branch_check``, ``safe_next_action``) so a + recovery session can quote the live lock response directly. The explicit + fields are populated for any outcome; ``adopted_branch`` / + ``adopted_branch_head`` are set only when the outcome is ADOPT so a + non-adoption proof can never be misread as claiming adoption. """ + outcome = assessment.get("outcome") + adopted = outcome == ADOPT return { "issue_number": issue_number, "branch_name": branch_name, @@ -153,4 +237,36 @@ def build_adoption_proof( "no_competing_live_lock_proof": bool(competing_lock_checked), "lock_file_path": lock_file_path, "lock_file_status": lock_file_status, + # Explicit citable fields (#477). + "adoption_decision": decision_label(outcome), + "adopted": adopted, + "adopted_branch": branch_name if adopted else None, + "adopted_branch_head": assessment.get("matched_head_sha") if adopted else None, + "matcher_summary": _matcher_summary(issue_number, assessment), + "competing_branch_check": _competing_branch_check(assessment), + "safe_next_action": safe_next_action(outcome), + } + + +def build_non_adoption_lock_proof(*, issue_number: int, branch_name: str) -> dict: + """Safe, adoption-free proof metadata for a normal (NO_MATCH) lock. + + Requirement #477 AC2: non-adoption lock responses must stay clear and must + not imply adoption. This returns explicit ``adopted: False`` metadata with + the ``NO_MATCH`` decision so a normal lock response can carry citable proof + without ever asserting a branch was adopted. + """ + return { + "issue_number": issue_number, + "branch_name": branch_name, + "adoption_decision": DECISION_LABELS[NO_MATCH], + "adopted": False, + "adopted_branch": None, + "adopted_branch_head": None, + "matcher_summary": ( + f"no existing branch carries the issue-{int(issue_number)} marker; " + f"normal lock path (no adoption)" + ), + "competing_branch_check": {"result": "clear", "competing_branches": []}, + "safe_next_action": safe_next_action(NO_MATCH), } \ No newline at end of file diff --git a/mcp-menu.sh b/mcp-menu.sh new file mode 100755 index 0000000..4d124f2 --- /dev/null +++ b/mcp-menu.sh @@ -0,0 +1,232 @@ +#!/usr/bin/env bash +# mcp-menu.sh — Repository-root operator menu for MCP/Gitea workflow onboarding. +# +# Safe by default: read-only status and copy-paste prompts unless an action is +# explicitly labeled and confirmed. No branch deletion, force-push, lock-file +# editing, or raw API bypass. +set -euo pipefail + +SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" +REPO_ROOT="$SCRIPT_DIR" + +pause() { + read -r -p "Press Enter to return to the menu..." +} + +print_banner() { + printf '\n=== Gitea-Tools MCP Operator Menu ===\n' + printf 'Repository: %s\n' "$REPO_ROOT" + printf 'Safe by default — destructive actions require explicit confirmation.\n\n' +} + +show_root_checkout_health() { + printf '\n--- Project status / root checkout health ---\n\n' + printf 'Current directory: %s\n' "$(pwd)" + local branch head_sha prgs_master_sha dirty + branch="$(git -C "$REPO_ROOT" branch --show-current 2>/dev/null || true)" + if [[ -z "$branch" ]]; then + branch="(detached HEAD)" + fi + printf 'Current branch: %s\n' "$branch" + printf '\nGit status (short, branch):\n' + git -C "$REPO_ROOT" status --short --branch || true + head_sha="$(git -C "$REPO_ROOT" rev-parse HEAD 2>/dev/null || echo 'unknown')" + printf '\nHEAD SHA: %s\n' "$head_sha" + if git -C "$REPO_ROOT" rev-parse --verify prgs/master >/dev/null 2>&1; then + prgs_master_sha="$(git -C "$REPO_ROOT" rev-parse prgs/master)" + printf 'prgs/master SHA: %s\n' "$prgs_master_sha" + if [[ "$head_sha" != "$prgs_master_sha" ]]; then + printf '\nWARNING: root checkout HEAD does not match prgs/master.\n' + printf 'Keep the stable control checkout on master/prgs/master.\n' + fi + else + printf 'prgs/master SHA: unavailable (remote ref not fetched)\n' + fi + if [[ -n "$(git -C "$REPO_ROOT" status --porcelain 2>/dev/null || true)" ]]; then + printf '\nWARNING: root checkout has uncommitted changes (dirty).\n' + printf 'Author mutations belong in a session worktree under branches/.\n' + fi + if [[ "$branch" != "master" && "$branch" != "main" && "$branch" != "dev" ]]; then + printf '\nWARNING: root checkout is not on a stable base branch (master/main/dev).\n' + printf 'Return to master before using the control checkout.\n' + fi + pause +} + +print_prompt_block() { + local title="$1" + local body="$2" + printf '\n--- %s ---\n\n' "$title" + printf '%s\n' "$body" + printf '\n(Copy the prompt above into your LLM session.)\n' + pause +} + +show_author_prompts() { + while true; do + printf '\n--- Author workflow prompts ---\n' + printf ' 1) Work issue (author/coder)\n' + printf ' 2) Conflict-fix author session\n' + printf ' 3) Root checkout recovery session\n' + printf ' 0) Back\n' + read -r -p 'Choice: ' choice + case "$choice" in + 1) + print_prompt_block "Author — work issue" \ +"You are the AUTHOR session for /. + +Goal: implement issue # only. + +Workflow: +1. Preflight: prove identity, work_issue/create_pr capability, clean session worktree under branches/. +2. gitea_lock_issue for issue # and branch feat/issue--. +3. Implement in the locked worktree only — never mutate the root control checkout. +4. Validate, commit, push, gitea_create_pr. Final report with issue, branch, SHA, PR, tests, mutation ledger." + ;; + 2) + print_prompt_block "Author — conflict-fix session" \ +"You are the AUTHOR session for / in conflict-fix mode. + +Goal: resolve merge conflicts on PR #

/ branch only. + +Workflow: +1. Preflight: prove author identity and exact push/commit capability for the locked PR branch. +2. Confirm conflict-fix lease and stale-head protection before pushing. +3. Work only in the session-owned worktree under branches/ — never the root checkout. +4. Rebase or merge target branch, run tests, push, update PR. No force-push without explicit operator approval. +5. Final report: conflict resolution proof, new HEAD SHA, tests, mutation ledger." + ;; + 3) + print_prompt_block "Author — root checkout recovery" \ +"You are a RECOVERY session for /. + +Goal: restore the stable root control checkout to clean master/prgs/master. + +Workflow: +1. Inspect root checkout: branch, git status --short --branch, HEAD vs prgs/master. +2. Do not implement features from the root checkout. Stash or move work to branches/ first. +3. Return root to master (or main/dev per project policy) matching prgs/master with no dirty tracked files. +4. Report before/after branch, SHA, dirty state, and safe next action for author worktree creation." + ;; + 0) return ;; + *) printf 'Invalid choice.\n' ;; + esac + done +} + +show_reviewer_prompts() { + print_prompt_block "Reviewer — PR review" \ +"You are the REVIEWER session for /. + +Goal: review PR #

only — do not merge unless explicitly switched to merger mode. + +Workflow: +1. Load canonical workflow: skills/llm-project-workflow/workflows/review-merge-pr.md +2. Preflight: prove reviewer identity, review_pr capability, clean review worktree. +3. gitea_view_pr, validate scope, run required checks in the correct worktree. +4. gitea_review_pr with approve, request-changes, or comment as warranted. +5. Final report: PR head SHA, verdict, validation evidence, mutation ledger. No merge in reviewer-only runs." +} + +show_merger_prompts() { + print_prompt_block "Merger — PR merge" \ +"You are the MERGER session for /. + +Goal: merge PR #

only after every gate passes. + +Workflow: +1. Load canonical workflow: skills/llm-project-workflow/workflows/review-merge-pr.md +2. Preflight: prove merger identity and exact merge_pr capability for the current PR head SHA. +3. Confirm approval pins the current head SHA; re-validate if the branch moved. +4. gitea_merge_pr only on explicit operator approval after all gates pass. +5. Final report: merged SHA, cleanup handoff, mutation ledger." +} + +show_reconciler_prompts() { + print_prompt_block "Reconciler — already-landed / closed PR cleanup" \ +"You are the RECONCILER session for /. + +Goal: reconcile already-landed open PRs — close or comment only when exact capability is proven. + +Workflow: +1. Load canonical workflow: skills/llm-project-workflow/workflows/reconcile-landed-pr.md +2. Preflight: prove reconciler identity and gitea.pr.close (or authorized close) capability. +3. Do not review, merge, implement code, or create branches. +4. gitea_scan_already_landed_open_prs / gitea_reconcile_already_landed_pr as appropriate. +5. Final report: PR numbers handled, close proof, mutation ledger." +} + +show_onboarding_prompt() { + print_prompt_block "Onboarding — new project to MCP workflow" \ +"Onboard / into the MCP Control Plane workflow. + +Checklist: +1. Prove identity and task capability via gitea_whoami and gitea_resolve_task_capability. +2. Configure separate MCP namespaces/profiles: author, reviewer, merger/reconciler as needed. +3. Register gitea-tools (and jenkins-mcp / glitchtip-mcp if applicable) in the client MCP config. +4. Copy skills/llm-project-workflow/SKILL.md guidance into the target repo or ECC install. +5. Verify gitea_get_runtime_context, gitea_lock_issue, and worktree rules under branches/. +6. Run ./mcp-menu.sh for day-to-day prompts; use docs/mcp-menu.md and docs/llm-workflow-runbooks.md. + +Canonical router: skills/llm-project-workflow/SKILL.md" +} + +show_proxmox_placeholder() { + printf '\n--- Proxmox deployment (placeholder) ---\n\n' + printf 'Push this project to Proxmox — TODO / issue-backed\n' + printf 'Create Proxmox LXC — TODO / issue-backed\n\n' + printf 'These actions are NOT implemented yet.\n' + printf 'Track deployment automation in dedicated Gitea issues before enabling here.\n' + printf 'This menu will not run deploy scripts until sanctioned tooling exists.\n' + pause +} + +run_tests() { + printf '\n--- Run tests ---\n\n' + if [[ -x "$REPO_ROOT/run-tests.sh" ]]; then + printf 'Running ./run-tests.sh ...\n\n' + (cd "$REPO_ROOT" && ./run-tests.sh) + pause + return + fi + if [[ -x "$REPO_ROOT/venv/bin/python" ]]; then + printf 'run-tests.sh not found; falling back to venv/bin/python -m pytest\n\n' + (cd "$REPO_ROOT" && ./venv/bin/python -m pytest) + pause + return + fi + printf 'ERROR: No test runner available (fail closed).\n' >&2 + printf 'Expected ./run-tests.sh or venv/bin/python for pytest fallback.\n' >&2 + exit 1 +} + +main_menu() { + while true; do + print_banner + printf ' 1) Project status / root checkout health\n' + printf ' 2) Author workflow prompts\n' + printf ' 3) Reviewer workflow prompts\n' + printf ' 4) Merger workflow prompts\n' + printf ' 5) Reconciler workflow prompts\n' + printf ' 6) Onboarding new project to this MCP workflow\n' + printf ' 7) Proxmox deployment menu placeholder\n' + printf ' 8) Create Proxmox LXC placeholder\n' + printf ' 9) Run tests\n' + printf ' 0) Exit\n' + read -r -p 'Choice: ' choice + case "$choice" in + 1) show_root_checkout_health ;; + 2) show_author_prompts ;; + 3) show_reviewer_prompts ;; + 4) show_merger_prompts ;; + 5) show_reconciler_prompts ;; + 6) show_onboarding_prompt ;; + 7|8) show_proxmox_placeholder ;; + 9) run_tests ;; + 0) printf 'Goodbye.\n'; exit 0 ;; + *) printf 'Invalid choice.\n'; pause ;; + esac + done +} + +main_menu \ No newline at end of file diff --git a/mcp_native_cleanup_proof.py b/mcp_native_cleanup_proof.py new file mode 100644 index 0000000..70f2a5e --- /dev/null +++ b/mcp_native_cleanup_proof.py @@ -0,0 +1,359 @@ +"""MCP-native post-merge cleanup proof verifier (#517). + +Post-merge cleanup of leases, comments, branches, and worktrees must be +proven through explicit MCP tools (or approved helpers), never raw scripts or +ad hoc git/API commands. Merger/reviewer sessions must hand cleanup to a +reconciler profile with the right capability proof. +""" + +from __future__ import annotations + +import re +from typing import Any + +AUTHORIZED_CLEANUP_TOOLS = frozenset({ + "gitea_cleanup_post_merge_moot_lease", + "gitea_reconcile_merged_cleanups", + "gitea_delete_branch", + "gitea_cleanup_merged_pr_branch", + "gitea_audit_worktree_cleanup", + "gitea_capture_branches_worktree_snapshot", + "gitea_assess_worktree_cleanup_integrity", + "gitea_authorize_reconciliation_cleanup_phase", +}) + +_RAW_BRANCH_DELETE_PATTERNS = ( + re.compile(r"\bgit(?:\s+-C\s+\S+)?\s+branch\s+-[dD]\b[^\n\r]*", re.I), + re.compile(r"\bgit(?:\s+-C\s+\S+)?\s+push\b[^\n\r]*\s--delete\b[^\n\r]*", re.I), + re.compile(r"\bgit(?:\s+-C\s+\S+)?\s+push\b[^\n\r]*\s:[^\s`]+", re.I), +) + +_RAW_COMMENT_DELETE_PATTERNS = ( + re.compile( + r"(?:curl|wget|httpie)\b[^\n\r]*\b(?:DELETE|delete)\b[^\n\r]*" + r"(?:/comments/|issues/\d+/comments)", + re.I, + ), + re.compile( + r"\bDELETE\b[^\n\r]*/repos/[^\n\r]*/issues/\d+/comments/\d+", + re.I, + ), + re.compile( + r"\b(?:delete_issue_comment|remove_issue_comment|purge_comments?)\b", + re.I, + ), + re.compile( + r"\b(?:raw|ad hoc|adhoc)\b[^\n\r]{0,40}\bcomment\b[^\n\r]{0,40}\bdelete", + re.I, + ), +) + +_CLEANUP_MUTATIONS_RE = re.compile( + r"^\s*[-*]?\s*cleanup mutations\s*:\s*(.+)$", + re.IGNORECASE | re.MULTILINE, +) +_GIT_REF_MUTATIONS_RE = re.compile( + r"^\s*[-*]?\s*git ref mutations\s*:\s*(.+)$", + re.IGNORECASE | re.MULTILINE, +) +_WORKTREE_MUTATIONS_RE = re.compile( + r"^\s*[-*]?\s*worktree mutations\s*:\s*(.+)$", + re.IGNORECASE | re.MULTILINE, +) +_MERGE_MUTATIONS_RE = re.compile( + r"^\s*[-*]?\s*merge mutations\s*:\s*(.+)$", + re.IGNORECASE | re.MULTILINE, +) +_NONE_LEDGER_VALUES = frozenset({"", "none", "n/a"}) +_MUTATION_LEDGER_PATTERNS = ( + _CLEANUP_MUTATIONS_RE, + _GIT_REF_MUTATIONS_RE, + _WORKTREE_MUTATIONS_RE, +) +_RAW_WORKTREE_REMOVE_PATTERNS = ( + re.compile(r"\bgit(?:\s+-C\s+\S+)?\s+worktree\s+remove\b", re.I), +) +_AUTHORIZED_TOOL_RE = re.compile( + r"\b(" + "|".join(re.escape(t) for t in sorted(AUTHORIZED_CLEANUP_TOOLS)) + r")\b", + re.IGNORECASE, +) +_RECONCILER_CAPABILITY_RE = re.compile( + r"(?:reconciler|gitea\.branch\.delete|delete_branch|reconcile_merged_cleanups)", + re.IGNORECASE, +) +_MERGER_CLEANUP_ROLE_RE = re.compile( + r"(?:merger|reviewer).{0,80}(?:deleted|removed|cleaned).{0,80}" + r"(?:branch|worktree|comment|lease)", + re.IGNORECASE, +) +_HANDOFF_TO_RECONCILER_RE = re.compile( + r"(?:hand(?:ed)? off|defer(?:red)?|next actor).{0,60}reconciler", + re.IGNORECASE, +) + + +def _ledger_field_body(text: str, pattern: re.Pattern[str]) -> str | None: + match = pattern.search(text) + if not match: + return None + return (match.group(1) or "").strip() + + +def _is_none_ledger_value(value: str) -> bool: + return value.strip().lower() in _NONE_LEDGER_VALUES + + +def scoped_cleanup_mutation_ledger_text(text: str | None) -> str: + """Return mutation-ledger bodies scoped to cleanup enforcement (#517).""" + text = text or "" + parts: list[str] = [] + for pattern in _MUTATION_LEDGER_PATTERNS: + body = _ledger_field_body(text, pattern) + if body is not None and not _is_none_ledger_value(body): + parts.append(body) + return "\n".join(parts) + + +def _git_ref_cleanup_claimed(body: str) -> bool: + return bool(raw_branch_delete_commands(body)) + + +def _worktree_cleanup_claimed(body: str) -> bool: + for pattern in _RAW_WORKTREE_REMOVE_PATTERNS: + if pattern.search(body): + return True + return bool( + re.search( + r"\b(?:removed|deleted)\b[^\n]{0,40}\bworktree\b", + body, + re.IGNORECASE, + ) + ) + + +def _cleanup_claiming_ledger_fields(text: str) -> list[tuple[str, str]]: + claiming: list[tuple[str, str]] = [] + cleanup_body = _ledger_field_body(text, _CLEANUP_MUTATIONS_RE) + if cleanup_body is not None and not _is_none_ledger_value(cleanup_body): + claiming.append(("Cleanup mutations", cleanup_body)) + + git_ref_body = _ledger_field_body(text, _GIT_REF_MUTATIONS_RE) + if git_ref_body is not None and not _is_none_ledger_value(git_ref_body): + if _git_ref_cleanup_claimed(git_ref_body): + claiming.append(("Git ref mutations", git_ref_body)) + + worktree_body = _ledger_field_body(text, _WORKTREE_MUTATIONS_RE) + if worktree_body is not None and not _is_none_ledger_value(worktree_body): + if _worktree_cleanup_claimed(worktree_body): + claiming.append(("Worktree mutations", worktree_body)) + return claiming + + +def raw_branch_delete_commands(text: str | None) -> list[str]: + """Return raw git branch-delete commands cited in *text*.""" + if not text: + return [] + commands: list[str] = [] + for pattern in _RAW_BRANCH_DELETE_PATTERNS: + commands.extend(match.group(0).strip("` ") for match in pattern.finditer(text)) + return list(dict.fromkeys(commands)) + + +def raw_comment_delete_commands(text: str | None) -> list[str]: + """Return raw comment-deletion commands/scripts cited in *text*.""" + if not text: + return [] + commands: list[str] = [] + for pattern in _RAW_COMMENT_DELETE_PATTERNS: + commands.extend(match.group(0).strip("` ") for match in pattern.finditer(text)) + return list(dict.fromkeys(commands)) + + +def assess_raw_branch_delete_report(text: str | None) -> dict[str, Any]: + """Fail closed when mutation ledgers cite raw git branch deletion.""" + commands = raw_branch_delete_commands(scoped_cleanup_mutation_ledger_text(text)) + reasons = [ + ( + "raw git branch deletion bypasses MCP branch.delete cleanup gates: " + f"{command}" + ) + for command in commands + ] + return { + "proven": not reasons, + "block": bool(reasons), + "commands": commands, + "reasons": reasons, + "safe_next_action": ( + "use gitea_delete_branch, gitea_reconcile_merged_cleanups, or another " + "approved MCP cleanup helper with explicit branch.delete capability" + if reasons + else "proceed" + ), + } + + +def assess_raw_comment_delete_report(text: str | None) -> dict[str, Any]: + """Fail closed when mutation ledgers cite raw comment deletion.""" + commands = raw_comment_delete_commands(scoped_cleanup_mutation_ledger_text(text)) + reasons = [ + ( + "raw comment deletion bypasses MCP lease cleanup gates: " + f"{command}" + ) + for command in commands + ] + return { + "proven": not reasons, + "block": bool(reasons), + "commands": commands, + "reasons": reasons, + "safe_next_action": ( + "use gitea_cleanup_post_merge_moot_lease (append-only release comment) " + "or hand cleanup to a reconciler session; never delete lease comments" + if reasons + else "proceed" + ), + } + + +def assess_merge_cleanup_mutation_separation(text: str | None) -> dict[str, Any]: + """Require merge mutations and cleanup mutations in separate ledger fields.""" + text = text or "" + merge_match = _MERGE_MUTATIONS_RE.search(text) + cleanup_match = _CLEANUP_MUTATIONS_RE.search(text) + reasons: list[str] = [] + + if cleanup_match and not merge_match: + combined = re.search( + r"merge.{0,40}cleanup mutations|cleanup.{0,40}merge mutations", + text, + re.IGNORECASE, + ) + if combined: + reasons.append( + "merge and cleanup mutations must use separate 'Merge mutations' " + "and 'Cleanup mutations' ledger fields" + ) + + if merge_match and cleanup_match: + merge_val = (merge_match.group(1) or "").strip().lower() + cleanup_val = (cleanup_match.group(1) or "").strip().lower() + if merge_val == cleanup_val and merge_val not in {"", "none"}: + reasons.append( + "merge mutations and cleanup mutations must not duplicate the " + "same ledger entry" + ) + + return { + "proven": not reasons, + "block": bool(reasons), + "reasons": reasons, + "safe_next_action": ( + "split merge mutations (gitea_merge_pr) from cleanup mutations " + "(reconciler MCP tools) in the controller handoff" + if reasons + else "proceed" + ), + } + + +def assess_authorized_reconciler_cleanup_path(text: str | None) -> dict[str, Any]: + """Validate cleanup claims cite authorized MCP tools and reconciler capability.""" + text = text or "" + claiming = _cleanup_claiming_ledger_fields(text) + if not claiming: + return { + "proven": True, + "block": False, + "reasons": [], + "safe_next_action": "proceed", + } + + reasons: list[str] = [] + for field_name, body in claiming: + if not _AUTHORIZED_TOOL_RE.search(body): + reasons.append( + f"{field_name} cleanup must name an authorized MCP cleanup tool " + f"({', '.join(sorted(AUTHORIZED_CLEANUP_TOOLS))})" + ) + if not _RECONCILER_CAPABILITY_RE.search(text): + reasons.append( + "post-merge cleanup requires reconciler capability proof " + "(reconciler profile, gitea.branch.delete, or reconcile_merged_cleanups)" + ) + + return { + "proven": not reasons, + "block": bool(reasons), + "reasons": reasons, + "safe_next_action": ( + "hand cleanup to a prgs-reconciler session and cite the exact MCP tool " + "plus delete_branch/reconcile_merged_cleanups capability proof" + if reasons + else "proceed" + ), + } + + +def assess_merger_cleanup_handoff_guidance(text: str | None) -> dict[str, Any]: + """Merger sessions that performed cleanup must hand off to reconciler.""" + text = text or "" + if not _MERGER_CLEANUP_ROLE_RE.search(text): + return { + "proven": True, + "block": False, + "reasons": [], + "safe_next_action": "proceed", + } + if _HANDOFF_TO_RECONCILER_RE.search(text): + return { + "proven": True, + "block": False, + "reasons": [], + "safe_next_action": "proceed", + } + return { + "proven": False, + "block": True, + "reasons": [ + "merger/reviewer session performed cleanup but did not hand off to " + "reconciler for MCP-native cleanup" + ], + "safe_next_action": ( + "merger sessions must end with cleanup handed to prgs-reconciler; " + "do not perform ad hoc branch/comment/worktree cleanup as merger" + ), + } + + +def assess_mcp_native_cleanup_proof(report_text: str | None) -> dict[str, Any]: + """Composite #517 verifier for MCP-native post-merge cleanup proof.""" + text = report_text or "" + checks = ( + assess_raw_branch_delete_report(text), + assess_raw_comment_delete_report(text), + assess_merge_cleanup_mutation_separation(text), + assess_authorized_reconciler_cleanup_path(text), + assess_merger_cleanup_handoff_guidance(text), + ) + reasons: list[str] = [] + safe_next = "proceed" + for result in checks: + reasons.extend(result.get("reasons") or []) + if result.get("block") and result.get("safe_next_action"): + safe_next = result["safe_next_action"] + + block = bool(reasons) + return { + "proven": not block, + "block": block, + "reasons": reasons, + "safe_next_action": safe_next, + "raw_branch_commands": raw_branch_delete_commands( + scoped_cleanup_mutation_ledger_text(text) + ), + "raw_comment_commands": raw_comment_delete_commands( + scoped_cleanup_mutation_ledger_text(text) + ), + } \ No newline at end of file diff --git a/mcp_session_state.py b/mcp_session_state.py new file mode 100644 index 0000000..3f93784 --- /dev/null +++ b/mcp_session_state.py @@ -0,0 +1,391 @@ +"""Durable MCP session validation state shared across daemon process pools (#559). + +The IDE often routes sequential MCP tool calls to different daemon processes. +Session-scoped proofs (workflow load, review decision lock) must therefore +survive process boundaries while remaining fail-closed against spoofing. + +Security notes (extends #211): +- Never store under host-global ``/tmp`` (world-writable, spoofable). +- Default root is ``~/.cache/gitea-tools/session-state`` (mode ``0o700``). +- Files are written atomically with mode ``0o600``. +- Records are keyed by remote + org + repo + profile identity, not by PID. +- TTL prevents indefinitely stale reuse across unrelated sessions. +""" + +from __future__ import annotations + +import fcntl +import json +import os +import re +import tempfile +from contextlib import contextmanager +from datetime import datetime, timedelta, timezone +from typing import Any + +STATE_DIR_ENV = "GITEA_MCP_SESSION_STATE_DIR" +DEFAULT_STATE_DIR = os.path.expanduser("~/.cache/gitea-tools/session-state") +TTL_HOURS_ENV = "GITEA_MCP_SESSION_STATE_TTL_HOURS" +DEFAULT_TTL_HOURS = 4.0 + +KIND_WORKFLOW_LOAD = "review_workflow_load" +KIND_DECISION_LOCK = "review_decision_lock" + +_SAFE_SEGMENT_RE = re.compile(r"[^A-Za-z0-9._+-]+") +SESSION_PROFILE_LOCK_ENV = "GITEA_SESSION_PROFILE_LOCK" + + +def default_state_dir() -> str: + raw = (os.environ.get(STATE_DIR_ENV) or DEFAULT_STATE_DIR).strip() + return raw or DEFAULT_STATE_DIR + + +def ttl_hours() -> float: + raw = (os.environ.get(TTL_HOURS_ENV) or "").strip() + if not raw: + return DEFAULT_TTL_HOURS + try: + value = float(raw) + except ValueError: + return DEFAULT_TTL_HOURS + return value if value > 0 else DEFAULT_TTL_HOURS + + +def _sanitize_segment(value: str) -> str: + text = (value or "").strip() + if not text: + return "_" + return _SAFE_SEGMENT_RE.sub("_", text) + + +def current_profile_identity( + profile_name: str | None = None, + session_profile_lock: str | None = None, + profile_identity: str | None = None, +) -> str: + """Resolve the session profile identity used as the durable key.""" + env_lock = (os.environ.get(SESSION_PROFILE_LOCK_ENV) or "").strip() + explicit = (profile_identity or session_profile_lock or "").strip() + lock = (explicit or env_lock or "").strip() + name = (profile_name or "").strip() + return lock or name or "unknown-profile" + + +def state_key( + *, + kind: str, + remote: str | None = None, + org: str | None = None, + repo: str | None = None, + profile_identity: str | None = None, +) -> str: + """Build durable filename key. + + Session proofs are one-active-per-profile (workflow load / decision lock), + so the primary key is kind + profile identity. Remote/org/repo are stored + inside the payload and validated on load (#559), which lets a later daemon + process recover state without already knowing the remote argument. + """ + # Keep remote/org/repo parameters for API stability / future kinds; they are + # intentionally not part of the filename for session-scoped proofs. + _ = (remote, org, repo) + return "-".join( + _sanitize_segment(part) + for part in ( + kind, + profile_identity or "unknown-profile", + ) + ) + + +def state_file_path( + *, + kind: str, + remote: str | None = None, + org: str | None = None, + repo: str | None = None, + profile_identity: str | None = None, + state_dir: str | None = None, +) -> str: + root = (state_dir or default_state_dir()).strip() + name = state_key( + kind=kind, + remote=remote, + org=org, + repo=repo, + profile_identity=profile_identity, + ) + return os.path.join(root, f"{name}.json") + + +def _ensure_state_dir(state_dir: str | None = None) -> str: + root = (state_dir or default_state_dir()).strip() + os.makedirs(root, mode=0o700, exist_ok=True) + return root + + +def _now_utc() -> datetime: + return datetime.now(timezone.utc) + + +def _parse_iso(value: str | None) -> datetime | None: + text = (value or "").strip() + if not text: + return None + if text.endswith("Z"): + text = text[:-1] + "+00:00" + try: + parsed = datetime.fromisoformat(text) + except ValueError: + return None + if parsed.tzinfo is None: + return parsed.replace(tzinfo=timezone.utc) + return parsed.astimezone(timezone.utc) + + +@contextmanager +def _exclusive_file_lock(lock_path: str): + os.makedirs(os.path.dirname(lock_path) or ".", exist_ok=True) + fd = os.open(lock_path, os.O_CREAT | os.O_RDWR, 0o600) + try: + fcntl.flock(fd, fcntl.LOCK_EX) + yield fd + finally: + try: + fcntl.flock(fd, fcntl.LOCK_UN) + finally: + os.close(fd) + + +def _read_json(path: str) -> dict[str, Any] | None: + if not path or not os.path.exists(path): + return None + try: + with open(path, encoding="utf-8") as handle: + data = json.load(handle) + except (OSError, json.JSONDecodeError): + return None + return data if isinstance(data, dict) else None + + +def _write_json(path: str, data: dict[str, Any]) -> None: + parent = os.path.dirname(path) or "." + os.makedirs(parent, mode=0o700, exist_ok=True) + payload = json.dumps(data, indent=2, sort_keys=True) + "\n" + fd, temp_path = tempfile.mkstemp(prefix=".session-", suffix=".json", dir=parent) + try: + with os.fdopen(fd, "w", encoding="utf-8") as handle: + handle.write(payload) + handle.flush() + os.fsync(handle.fileno()) + os.chmod(temp_path, 0o600) + os.replace(temp_path, path) + try: + os.chmod(path, 0o600) + except OSError: + pass + finally: + if os.path.exists(temp_path): + try: + os.remove(temp_path) + except OSError: + pass + + +def identity_match_reasons( + record: dict[str, Any] | None, + *, + remote: str | None = None, + org: str | None = None, + repo: str | None = None, + profile_identity: str | None = None, +) -> list[str]: + """Return fail-closed reasons when durable record identity does not match.""" + if record is None: + return [] + reasons: list[str] = [] + expected_profile = current_profile_identity(profile_identity=profile_identity) + stored_profile = ( + (record.get("session_profile_lock") or record.get("profile_identity") or "") + .strip() + ) + if stored_profile and expected_profile and stored_profile != expected_profile: + if expected_profile != "unknown-profile": + reasons.append( + "session state profile identity mismatch " + f"(stored={stored_profile!r}, active={expected_profile!r}; fail closed)" + ) + + for field, expected in ( + ("remote", remote), + ("org", org), + ("repo", repo), + ): + want = (expected or "").strip() + have = (str(record.get(field) or "")).strip() + if want and have and want != have: + reasons.append( + f"session state {field} mismatch " + f"(stored={have!r}, expected={want!r}; fail closed)" + ) + + recorded_at = _parse_iso(record.get("recorded_at") or record.get("updated_at")) + if recorded_at is None: + reasons.append("session state missing recorded_at timestamp (fail closed)") + else: + age = _now_utc() - recorded_at + if age > timedelta(hours=ttl_hours()): + reasons.append( + f"session state expired after {ttl_hours():g}h (fail closed)" + ) + if age < timedelta(0): + reasons.append("session state recorded_at is in the future (fail closed)") + return reasons + + +def load_state( + *, + kind: str, + remote: str | None = None, + org: str | None = None, + repo: str | None = None, + profile_identity: str | None = None, + state_dir: str | None = None, +) -> dict[str, Any] | None: + """Load durable state payload when identity checks pass.""" + profile = current_profile_identity(profile_identity=profile_identity) + path = state_file_path( + kind=kind, + remote=remote, + org=org, + repo=repo, + profile_identity=profile, + state_dir=state_dir, + ) + lock_path = f"{path}.lock" + with _exclusive_file_lock(lock_path): + envelope = _read_json(path) + if not envelope: + return None + payload = envelope.get("payload") + if not isinstance(payload, dict): + return None + # Identity fields live on both envelope and payload for convenience. + merged = dict(payload) + for key in ( + "kind", + "remote", + "org", + "repo", + "profile_identity", + "session_profile_lock", + "recorded_at", + "updated_at", + "writer_pid", + ): + if key in envelope and key not in merged: + merged[key] = envelope[key] + reasons = identity_match_reasons( + merged, + remote=remote, + org=org, + repo=repo, + profile_identity=profile, + ) + if reasons: + return None + return merged + + +def save_state( + *, + kind: str, + payload: dict[str, Any] | None, + remote: str | None = None, + org: str | None = None, + repo: str | None = None, + profile_identity: str | None = None, + state_dir: str | None = None, +) -> dict[str, Any] | None: + """Persist or clear durable state for the given session identity key.""" + profile = current_profile_identity( + profile_name=payload.get("session_profile") if payload else None, + session_profile_lock=( + (payload or {}).get("session_profile_lock") or profile_identity + ), + ) + # Prefer explicit args over payload fields for key location. + key_remote = remote if remote is not None else (payload or {}).get("remote") + key_org = org if org is not None else (payload or {}).get("org") + key_repo = repo if repo is not None else (payload or {}).get("repo") + + root = _ensure_state_dir(state_dir) + path = state_file_path( + kind=kind, + remote=key_remote, + org=key_org, + repo=key_repo, + profile_identity=profile, + state_dir=root, + ) + lock_path = f"{path}.lock" + with _exclusive_file_lock(lock_path): + if payload is None: + for candidate in (path, lock_path): + if os.path.exists(candidate): + try: + os.remove(candidate) + except OSError: + pass + return None + + now = _now_utc().isoformat().replace("+00:00", "Z") + body = dict(payload) + body.setdefault("session_pid", os.getpid()) + body["writer_pid"] = os.getpid() + body["profile_identity"] = profile + if not (body.get("session_profile_lock") or "").strip(): + body["session_profile_lock"] = profile + body["recorded_at"] = body.get("recorded_at") or now + body["updated_at"] = now + if key_remote is not None: + body["remote"] = key_remote + if key_org is not None: + body["org"] = key_org + if key_repo is not None: + body["repo"] = key_repo + + envelope = { + "kind": kind, + "remote": key_remote, + "org": key_org, + "repo": key_repo, + "profile_identity": profile, + "session_profile_lock": body.get("session_profile_lock"), + "recorded_at": body["recorded_at"], + "updated_at": body["updated_at"], + "writer_pid": body["writer_pid"], + "payload": body, + } + _write_json(path, envelope) + return dict(body) + + +def clear_state( + *, + kind: str, + remote: str | None = None, + org: str | None = None, + repo: str | None = None, + profile_identity: str | None = None, + state_dir: str | None = None, +) -> None: + save_state( + kind=kind, + payload=None, + remote=remote, + org=org, + repo=repo, + profile_identity=profile_identity, + state_dir=state_dir, + ) diff --git a/merger_lease_adoption.py b/merger_lease_adoption.py new file mode 100644 index 0000000..8c05de1 --- /dev/null +++ b/merger_lease_adoption.py @@ -0,0 +1,253 @@ +"""Guarded merger adoption of an existing reviewer PR lease (#536). + +Replaces manual in-process ``_SESSION_LEASE`` seeding with an auditable, +comment-backed adoption path for merger sessions that inherit a reviewer lease +after formal approval at the current PR head. +""" + +from __future__ import annotations + +from datetime import datetime, timezone +from typing import Any + +import reviewer_pr_lease as leases + +ADOPTION_MARKER = "" +DEFAULT_ADOPTION_REASON = "merger-handoff-approved-head" + +SOURCE_ADOPT = "gitea_adopt_merger_pr_lease" +SOURCE_ACQUIRE = "gitea_acquire_reviewer_pr_lease" +SOURCE_HEARTBEAT = "gitea_heartbeat_reviewer_pr_lease" + +SANCTIONED_PROVENANCE_SOURCES = frozenset({ + SOURCE_ADOPT, + SOURCE_ACQUIRE, + SOURCE_HEARTBEAT, +}) + +_MERGER_ADOPTABLE_FRESHNESS = frozenset({"active", "stale_warning"}) + + +def format_adoption_body( + *, + repo: str, + pr_number: int, + issue_number: int | None, + adopter_identity: str, + adopter_profile: str, + adopter_session_id: str, + worktree: str, + candidate_head: str | None, + target_branch: str, + target_branch_sha: str | None, + adopted_from_session_id: str, + adopted_from_profile: str, + adopted_from_reviewer_identity: str, + adopted_from_comment_id: int | None, + adoption_reason: str = DEFAULT_ADOPTION_REASON, + adopted_at: datetime | None = None, +) -> str: + """Render a durable merger adoption proof comment.""" + adopted_at = adopted_at or datetime.now(timezone.utc) + adopted_text = adopted_at.astimezone(timezone.utc).replace(microsecond=0).isoformat().replace( + "+00:00", "Z" + ) + lease_body = leases.format_lease_body( + repo=repo, + pr_number=pr_number, + issue_number=issue_number, + reviewer_identity=adopter_identity, + profile=adopter_profile, + session_id=adopter_session_id, + worktree=worktree, + phase="adopted", + candidate_head=candidate_head, + target_branch=target_branch, + target_branch_sha=target_branch_sha, + last_activity=adopted_at, + blocker="none", + ) + lines = [ + ADOPTION_MARKER, + f"adopted_at: {adopted_text}", + f"adopted_by_identity: {adopter_identity}", + f"adopted_by_profile: {adopter_profile}", + f"adopted_from_session_id: {adopted_from_session_id}", + f"adopted_from_profile: {adopted_from_profile}", + f"adopted_from_reviewer_identity: {adopted_from_reviewer_identity}", + f"adopted_from_comment_id: {adopted_from_comment_id or 'none'}", + f"adoption_reason: {adoption_reason}", + lease_body, + ] + return "\n".join(lines) + + +def is_adoption_comment(body: str) -> bool: + return ADOPTION_MARKER in (body or "") + + +def build_lease_provenance( + *, + source: str, + comment_id: int | None = None, + adopted_from_session_id: str | None = None, + adopted_from_profile: str | None = None, + adopted_from_reviewer_identity: str | None = None, + adoption_reason: str | None = None, + recorded_at: datetime | None = None, +) -> dict[str, Any]: + recorded_at = recorded_at or datetime.now(timezone.utc) + recorded_text = recorded_at.astimezone(timezone.utc).replace(microsecond=0).isoformat().replace( + "+00:00", "Z" + ) + proof = { + "source": source, + "recorded_at": recorded_text, + } + if comment_id is not None: + proof["comment_id"] = comment_id + if adopted_from_session_id: + proof["adopted_from_session_id"] = adopted_from_session_id + if adopted_from_profile: + proof["adopted_from_profile"] = adopted_from_profile + if adopted_from_reviewer_identity: + proof["adopted_from_reviewer_identity"] = adopted_from_reviewer_identity + if adoption_reason: + proof["adoption_reason"] = adoption_reason + return proof + + +def is_sanctioned_session_lease(session: dict[str, Any] | None) -> bool: + if not session: + return False + provenance = session.get("lease_provenance") or {} + source = (provenance.get("source") or "").strip() + if source not in SANCTIONED_PROVENANCE_SOURCES: + return False + if source == SOURCE_ADOPT: + return bool(provenance.get("comment_id")) and bool( + provenance.get("adopted_from_session_id") + ) + if source in {SOURCE_ACQUIRE, SOURCE_HEARTBEAT}: + return bool(session.get("comment_id") or provenance.get("comment_id")) + return False + + +def assess_adopt_merger_lease( + comments: list[dict], + *, + pr_number: int, + adopter_identity: str, + adopter_profile: str, + adopter_session_id: str, + repo: str, + issue_number: int | None, + worktree: str, + expected_head_sha: str | None, + live_head_sha: str | None, + approval_at_head: bool, + pr_open: bool = True, + now: datetime | None = None, +) -> dict[str, Any]: + """Decide whether a merger may adopt the active reviewer lease (#536).""" + now = now or datetime.now(timezone.utc) + reasons: list[str] = [] + pinned = leases._normalize_sha(expected_head_sha) + live = leases._normalize_sha(live_head_sha) + + if not pr_open: + reasons.append( + f"PR #{pr_number} is not open; merger lease adoption is only for open PRs" + ) + if not approval_at_head: + reasons.append( + "formal APPROVED review at the current PR head is required before " + "merger lease adoption (fail closed)" + ) + if not pinned: + reasons.append("expected_head_sha is required for merger lease adoption") + if not live: + reasons.append("live PR head SHA unavailable (fail closed)") + elif pinned and live and pinned != live: + reasons.append( + "expected_head_sha does not match live PR head; refresh approval before adoption" + ) + + active = leases.find_active_reviewer_lease( + comments, pr_number=pr_number, now=now + ) + if not active: + reasons.append( + f"no active reviewer lease on PR #{pr_number} to adopt; reviewer " + "must acquire first" + ) + else: + owner_session = (active.get("session_id") or "").strip() + freshness = active.get("freshness") or leases.classify_lease_freshness( + active, now=now + ) + if freshness not in _MERGER_ADOPTABLE_FRESHNESS: + reasons.append( + f"active reviewer lease freshness is '{freshness}'; explicit " + "reclaim is not implemented (fail closed)" + ) + lease_head = active.get("candidate_head") + if lease_head and live and lease_head != live: + reasons.append( + "active reviewer lease candidate_head differs from live PR head" + ) + if owner_session and owner_session == adopter_session_id: + # Same session already holds the thread lease — allow idempotent adopt + # only when the newest entry is not yet an adoption by this session. + entries = leases._lease_entries(comments, pr_number=pr_number) + newest = entries[-1] if entries else None + if newest and (newest.get("phase") or "") == "adopted": + reasons.append( + "merger session already recorded an adoption lease on this PR" + ) + elif owner_session and owner_session != adopter_session_id: + # Cross-session merger handoff: adopt the foreign reviewer lease. + pass + + if not (adopter_identity or "").strip(): + reasons.append("adopter identity required") + if not (adopter_session_id or "").strip(): + reasons.append("adopter session_id required") + if not (worktree or "").strip(): + reasons.append("merger worktree path required") + if "merger" not in (adopter_profile or "").lower(): + reasons.append( + f"profile '{adopter_profile}' is not a merger profile; adoption is " + "merger-only (fail closed)" + ) + + adopt_allowed = not reasons + adoption_body = None + if adopt_allowed and active: + adoption_body = format_adoption_body( + repo=repo, + pr_number=pr_number, + issue_number=issue_number or active.get("issue_number"), + adopter_identity=adopter_identity, + adopter_profile=adopter_profile, + adopter_session_id=adopter_session_id, + worktree=worktree, + candidate_head=live, + target_branch=active.get("target_branch") or "master", + target_branch_sha=active.get("target_branch_sha"), + adopted_from_session_id=active.get("session_id") or "", + adopted_from_profile=active.get("profile") or "unknown", + adopted_from_reviewer_identity=active.get("reviewer_identity") or "", + adopted_from_comment_id=active.get("comment_id"), + adopted_at=now, + ) + + return { + "adopt_allowed": adopt_allowed, + "reasons": reasons, + "active_lease": active, + "adoption_body": adoption_body, + "adopter_session_id": adopter_session_id, + "expected_head_sha": pinned, + "live_head_sha": live, + } \ No newline at end of file diff --git a/namespace_workspace_binding.py b/namespace_workspace_binding.py new file mode 100644 index 0000000..4831907 --- /dev/null +++ b/namespace_workspace_binding.py @@ -0,0 +1,307 @@ +"""Namespace-scoped MCP workspace binding (#510). + +Each role namespace (author, reviewer, merger, reconciler) resolves its own +active task workspace. Foreign role worktree environment variables must not +poison workspace purity checks in another namespace. +""" + +from __future__ import annotations + +import os + +import author_mutation_worktree as amw + +ACTIVE_WORKTREE_ENV = amw.ACTIVE_WORKTREE_ENV +AUTHOR_WORKTREE_ENV = amw.AUTHOR_WORKTREE_ENV +REVIEWER_WORKTREE_ENV = "GITEA_REVIEWER_WORKTREE" +MERGER_WORKTREE_ENV = "GITEA_MERGER_WORKTREE" +RECONCILER_WORKTREE_ENV = "GITEA_RECONCILER_WORKTREE" + +ROLE_WORKTREE_ENVS: dict[str, str] = { + "author": AUTHOR_WORKTREE_ENV, + "reviewer": REVIEWER_WORKTREE_ENV, + "merger": MERGER_WORKTREE_ENV, + "reconciler": RECONCILER_WORKTREE_ENV, +} + +NON_AUTHOR_ROLES = frozenset({"reviewer", "merger", "reconciler"}) + + +def normalize_role_kind( + role_kind: str | None, + *, + profile_name: str | None = None, +) -> str: + """Map profile/task role to a workspace namespace key.""" + role = (role_kind or "author").strip().lower() + profile = (profile_name or "").strip().lower() + if role == "reviewer" and "merger" in profile: + return "merger" + if role in ROLE_WORKTREE_ENVS: + return role + return "author" + + +def _env_value(env: dict[str, str] | os._Environ, key: str) -> str | None: + text = (env.get(key) or "").strip() + return text or None + + +def resolve_namespace_workspace( + *, + role_kind: str, + worktree_path: str | None = None, + worktree: str | None = None, + process_project_root: str, + env: dict[str, str] | os._Environ | None = None, + session_lease_worktree: str | None = None, + profile_name: str | None = None, +) -> tuple[str, str]: + """Return ``(resolved_path, binding_source)`` for *role_kind*.""" + env_map = env if env is not None else os.environ + role = normalize_role_kind(role_kind, profile_name=profile_name) + role_env_key = ROLE_WORKTREE_ENVS[role] + + for candidate, source in ( + (worktree_path, "worktree_path argument"), + (worktree, "worktree argument"), + (_env_value(env_map, ACTIVE_WORKTREE_ENV), f"{ACTIVE_WORKTREE_ENV} environment variable"), + (_env_value(env_map, role_env_key), f"{role_env_key} environment variable"), + (session_lease_worktree if role in {"reviewer", "merger"} else None, + "reviewer PR lease worktree"), + ): + text = (candidate or "").strip() + if text: + return os.path.realpath(os.path.abspath(text)), source + + return os.path.realpath(process_project_root), "MCP server process root (default)" + + +def resolve_namespace_mutation_context( + *, + role_kind: str, + worktree_path: str | None, + process_project_root: str, + env: dict[str, str] | os._Environ | None = None, + session_lease_worktree: str | None = None, + worktree: str | None = None, + profile_name: str | None = None, +) -> dict: + """Shared workspace resolution for runtime_context and mutation guards.""" + workspace, binding_source = resolve_namespace_workspace( + role_kind=role_kind, + worktree_path=worktree_path, + worktree=worktree, + process_project_root=process_project_root, + env=env, + session_lease_worktree=session_lease_worktree, + profile_name=profile_name, + ) + process_root = os.path.realpath(process_project_root) + role = normalize_role_kind(role_kind, profile_name=profile_name) + pollution = assess_foreign_role_worktree_pollution( + role_kind=role, + resolved_workspace=workspace, + binding_source=binding_source, + env=env, + profile_name=profile_name, + ) + canonical_root = amw.resolve_canonical_repo_root(process_root, process_root) + return { + "workspace_path": workspace, + "workspace_binding_source": binding_source, + "workspace_role_kind": role, + "ignored_bindings": pollution.get("ignored_bindings") or [], + "process_project_root": process_root, + "canonical_repo_root": canonical_root, + "roots_aligned": canonical_root == process_root, + } + + +def assess_foreign_role_worktree_pollution( + *, + role_kind: str, + resolved_workspace: str, + binding_source: str, + env: dict[str, str] | os._Environ | None = None, + profile_name: str | None = None, +) -> dict: + """Detect when a foreign role env would have hijacked workspace binding.""" + env_map = env if env is not None else os.environ + role = normalize_role_kind(role_kind, profile_name=profile_name) + if role == "author": + return {"would_pollute": False, "ignored_bindings": []} + + ignored: list[str] = [] + author_path = _env_value(env_map, AUTHOR_WORKTREE_ENV) + if author_path: + author_real = os.path.realpath(os.path.abspath(author_path)) + resolved_real = os.path.realpath(resolved_workspace) + if author_real != resolved_real and binding_source != f"{AUTHOR_WORKTREE_ENV} environment variable": + ignored.append( + f"{AUTHOR_WORKTREE_ENV}={author_real} (ignored for {role} namespace)" + ) + return { + "would_pollute": bool(ignored), + "ignored_bindings": ignored, + } + + +def assess_metadata_only_worktree_binding( + *, + role_kind: str, + declared_worktree_path: str | None, + mutation_workspace: str, + process_project_root: str, + profile_name: str | None = None, +) -> dict: + """Fail closed when declared worktree_path would not redirect mutations.""" + declared = (declared_worktree_path or "").strip() + process_root = os.path.realpath(process_project_root) + mutation_root = os.path.realpath(mutation_workspace) + role = normalize_role_kind(role_kind, profile_name=profile_name) + if not declared: + return {"block": False, "reasons": [], "metadata_only": False} + + declared_root = os.path.realpath(os.path.abspath(declared)) + if declared_root == mutation_root: + return {"block": False, "reasons": [], "metadata_only": False} + + if declared_root != process_root and mutation_root == process_root: + return { + "block": True, + "metadata_only": True, + "reasons": [ + f"worktree_path is metadata-only for {role} mutations: preflight " + f"inspected '{declared_root}' but mutation tools would still " + f"validate MCP server process root '{process_root}'" + ], + "declared_worktree_path": declared_root, + "mutation_workspace": mutation_root, + "process_project_root": process_root, + } + + return {"block": False, "reasons": [], "metadata_only": False} + + +def format_namespace_workspace_binding_error( + *, + role_kind: str, + workspace_path: str, + binding_source: str, + reasons: list[str] | None = None, + ignored_bindings: list[str] | None = None, + dirty_files: list[str] | None = None, +) -> str: + """Canonical error when namespace workspace binding blocks mutations.""" + role = normalize_role_kind(role_kind) + workspace = os.path.realpath(workspace_path) + parts = [ + f"Namespace workspace binding blocked ({role} namespace, #510): " + f"resolved workspace '{workspace}' via {binding_source}." + ] + if ignored_bindings: + parts.append( + "Foreign role bindings ignored: " + "; ".join(ignored_bindings) + "." + ) + if dirty_files: + parts.append( + "Dirty tracked files in active task workspace: " + + ", ".join(dirty_files) + + "." + ) + if reasons: + parts.append("Details: " + "; ".join(reasons) + ".") + parts.append( + "Remediation: reconnect or relaunch the MCP server from a clean dedicated " + f"branches/ {role} worktree, set " + f"{ROLE_WORKTREE_ENVS.get(role, ACTIVE_WORKTREE_ENV)} or {ACTIVE_WORKTREE_ENV} " + "to that path, or pass worktree_path on mutation tools. Do not clean or " + "reset foreign role worktrees to unblock this namespace." + ) + return " ".join(parts) + + +def assess_namespace_mutation_workspace( + *, + role_kind: str, + worktree_path: str | None, + worktree: str | None, + process_project_root: str, + env: dict[str, str] | os._Environ | None = None, + session_lease_worktree: str | None = None, + profile_name: str | None = None, + current_branch: str | None = None, +) -> dict: + """Evaluate namespace workspace binding before preflight/mutation.""" + ctx = resolve_namespace_mutation_context( + role_kind=role_kind, + worktree_path=worktree_path, + worktree=worktree, + process_project_root=process_project_root, + env=env, + session_lease_worktree=session_lease_worktree, + profile_name=profile_name, + ) + mutation_workspace = ctx["workspace_path"] + binding_source = ctx["workspace_binding_source"] + role = ctx["workspace_role_kind"] + process_root = ctx["process_project_root"] + + metadata = assess_metadata_only_worktree_binding( + role_kind=role, + declared_worktree_path=worktree_path, + mutation_workspace=mutation_workspace, + process_project_root=process_root, + profile_name=profile_name, + ) + pollution = assess_foreign_role_worktree_pollution( + role_kind=role, + resolved_workspace=mutation_workspace, + binding_source=binding_source, + env=env, + profile_name=profile_name, + ) + + reasons = list(metadata.get("reasons") or []) + if role == "author": + branches = amw.assess_author_mutation_worktree( + workspace_path=mutation_workspace, + project_root=ctx["canonical_repo_root"], + current_branch=current_branch, + ) + if branches["block"]: + reasons.extend(branches["reasons"]) + elif ( + role == "reviewer" + and mutation_workspace == process_root + and not amw.is_path_under_branches(mutation_workspace, ctx["canonical_repo_root"]) + ): + reasons.append( + f"{role} mutation blocked: workspace is the stable control checkout; " + f"create or reconnect to a session-owned worktree under branches/ " + f"or set {ROLE_WORKTREE_ENVS[role]} / {ACTIVE_WORKTREE_ENV}" + ) + elif ( + role in {"reviewer", "merger"} + and mutation_workspace != process_root + and not amw.is_path_under_branches(mutation_workspace, ctx["canonical_repo_root"]) + ): + reasons.append( + f"{role} mutation blocked: workspace '{mutation_workspace}' is not under " + f"'{ctx['canonical_repo_root']}/branches/'" + ) + + block = bool(reasons) + return { + "block": block, + "reasons": reasons, + "mutation_workspace": mutation_workspace, + "workspace_binding_source": binding_source, + "workspace_role_kind": role, + "process_project_root": process_root, + "canonical_repo_root": ctx["canonical_repo_root"], + "metadata_only": metadata.get("metadata_only", False), + "declared_worktree_path": metadata.get("declared_worktree_path"), + "ignored_bindings": pollution.get("ignored_bindings") or [], + } \ No newline at end of file diff --git a/post_merge_cleanup_proof.py b/post_merge_cleanup_proof.py new file mode 100644 index 0000000..6562a04 --- /dev/null +++ b/post_merge_cleanup_proof.py @@ -0,0 +1,239 @@ +"""Post-merge cleanup proof verifier for reviewer final reports (#402).""" + +from __future__ import annotations + +import re +from typing import Any + +CLEANUP_SKIPPED = "CLEANUP_SKIPPED" +CLEANUP_PERFORMED = "CLEANUP_PERFORMED" + +_CLEANUP_SECTION_HINT = re.compile( + r"(?:cleanup (?:status|result|mutations)|post-merge cleanup|" + r"gitea_delete_branch|remote branch.*deleted|worktree remove)", + re.IGNORECASE, +) +_CLEANUP_SKIPPED_RE = re.compile(r"\bCLEANUP_SKIPPED\b", re.IGNORECASE) +_CLEANUP_BLOCKER_RE = re.compile( + r"(?:cleanup blocker|cleanup skip(?:ped)? reason)\s*:\s*(.+)", + re.IGNORECASE, +) +_REMOTE_DELETE_CLAIM_RE = re.compile( + r"(?:gitea_delete_branch|remote (?:head )?branch (?:was )?deleted|" + r"deleted remote branch|delete_branch)", + re.IGNORECASE, +) +_WORKTREE_REMOVE_CLAIM_RE = re.compile( + r"(?:git worktree remove|worktree (?:was )?removed|removed (?:local )?worktree|" + r"worktree cleanup performed)", + re.IGNORECASE, +) +_DELETE_CAPABILITY_RE = re.compile( + r"(?:delete[- ]branch capability resolved|gitea\.branch\.delete)\s*:\s*" + r".*(?:gitea\.branch\.delete|delete_branch).*(?:resolved|allowed|proven)|" + r"gitea\.branch\.delete\s+(?:resolved|allowed|proven)", + re.IGNORECASE, +) +_DELETE_TASK_RE = re.compile( + r"(?:delete_branch|cleanup_branch|reconcile_merged_cleanups)", + re.IGNORECASE, +) +_MERGE_RESULT_RE = re.compile( + r"merge result\s*:\s*(?:merged|success|performed)", + re.IGNORECASE, +) +_MERGE_COMMIT_SHA_RE = re.compile( + r"(?:merge commit sha|merged commit sha|merge commit)\s*:\s*([0-9a-f]{7,40})", + re.IGNORECASE, +) +_PR_HEAD_BRANCH_RE = re.compile( + r"(?:merged pr head branch|pr head branch|deleted branch)\s*:\s*(\S+)", + re.IGNORECASE, +) +_BRANCH_NOT_PROTECTED_RE = re.compile( + r"branch (?:is )?not protected|branch protection\s*:\s*(?:none|false|no)", + re.IGNORECASE, +) +_OPEN_PR_INVENTORY_RE = re.compile( + r"(?:no other open pr(?:\s+references)?(?:\s+\S+)?|open pr inventory proof|" + r"open pr references).*(?:none|zero|0|clear|inventory complete)|" + r"no other open pr references branch", + re.IGNORECASE, +) +_ACTIVE_CLAIM_LEASE_RE = re.compile( + r"(?:no active (?:heartbeat|claim|lease)|" + r"(?:active )?(?:heartbeat|claim|lease)(?:/(?:claim|lease))*\s*:\s*none)", + re.IGNORECASE, +) +_SESSION_OWNED_WORKTREE_RE = re.compile( + r"(?:removed worktree path|cleanup worktree path|session-owned worktree)\s*:\s*(\S+)", + re.IGNORECASE, +) +_BRANCHES_PATH_RE = re.compile(r"\bbranches/", re.IGNORECASE) +_CLEAN_TRACKED_RE = re.compile( + r"(?:pre-removal tracked state|tracked state before removal)\s*:\s*clean", + re.IGNORECASE, +) +_CLEAN_UNTRACKED_RE = re.compile( + r"(?:pre-removal untracked state|untracked state before removal)\s*:\s*clean", + re.IGNORECASE, +) +_WORKTREE_LIST_AFTER_RE = re.compile( + r"(?:git worktree list after|post-removal worktree list|worktree list after)", + re.IGNORECASE, +) +_WRONG_BRANCH_RE = re.compile( + r"deleted branch (?:does not match|!=|differs from) (?:merged )?pr head", + re.IGNORECASE, +) + + +def _claims_remote_delete(text: str) -> bool: + return bool(_REMOTE_DELETE_CLAIM_RE.search(text)) + + +def _claims_worktree_remove(text: str) -> bool: + return bool(_WORKTREE_REMOVE_CLAIM_RE.search(text)) + + +def _branch_safety_fields_present(text: str) -> list[str]: + missing: list[str] = [] + if not _DELETE_CAPABILITY_RE.search(text): + missing.append("delete-branch capability resolved (gitea.branch.delete)") + if not _DELETE_TASK_RE.search(text): + missing.append("delete-branch task named (delete_branch or cleanup)") + if not _MERGE_RESULT_RE.search(text): + missing.append("merge result: merged") + if not _MERGE_COMMIT_SHA_RE.search(text): + missing.append("merge commit SHA") + if not _PR_HEAD_BRANCH_RE.search(text): + missing.append("merged PR head branch / deleted branch name") + if not _BRANCH_NOT_PROTECTED_RE.search(text): + missing.append("branch not protected proof") + if not _OPEN_PR_INVENTORY_RE.search(text): + missing.append("open PR inventory proof (no other PR references branch)") + if not _ACTIVE_CLAIM_LEASE_RE.search(text): + missing.append("no active heartbeat/claim/lease proof") + return missing + + +def _worktree_cleanup_fields_present(text: str) -> list[str]: + missing: list[str] = [] + match = _SESSION_OWNED_WORKTREE_RE.search(text) + path = match.group(1).strip() if match else "" + if not path: + missing.append("session-owned worktree path") + elif not _BRANCHES_PATH_RE.search(path.replace("\\", "/")): + missing.append("worktree path under branches/") + if not _CLEAN_TRACKED_RE.search(text): + missing.append("pre-removal tracked state: clean") + if not _CLEAN_UNTRACKED_RE.search(text): + missing.append("pre-removal untracked state: clean") + if not _WORKTREE_LIST_AFTER_RE.search(text): + missing.append("git worktree list after removal") + return missing + + +def assess_post_merge_cleanup_proof( + report_text: str, + *, + cleanup_session: dict | None = None, +) -> dict[str, Any]: + """Validate post-merge cleanup claims carry safety-gate proof (#402).""" + text = report_text or "" + session = dict(cleanup_session or {}) + reasons: list[str] = [] + + if _CLEANUP_SKIPPED_RE.search(text) or session.get("outcome") == CLEANUP_SKIPPED: + blocker = (session.get("blocker") or "").strip() + if not blocker: + match = _CLEANUP_BLOCKER_RE.search(text) + blocker = match.group(1).strip() if match else "" + if blocker.upper() == CLEANUP_SKIPPED: + blocker = "" + if not blocker: + reasons.append( + "CLEANUP_SKIPPED requires exact cleanup blocker reason (#402)" + ) + return { + "block": bool(reasons), + "proven": not reasons, + "outcome": CLEANUP_SKIPPED, + "remote_delete_claimed": False, + "worktree_remove_claimed": False, + "reasons": reasons, + "safe_next_action": ( + "report CLEANUP_SKIPPED with exact blocker; do not claim performed cleanup" + if reasons + else "proceed" + ), + } + + if not _CLEANUP_SECTION_HINT.search(text) and not session.get("cleanup_claimed"): + return { + "block": False, + "proven": True, + "outcome": None, + "remote_delete_claimed": False, + "worktree_remove_claimed": False, + "reasons": [], + "safe_next_action": "proceed", + } + + remote_delete = bool( + session.get("remote_delete_claimed") or _claims_remote_delete(text) + ) + worktree_remove = bool( + session.get("worktree_remove_claimed") or _claims_worktree_remove(text) + ) + + if _WRONG_BRANCH_RE.search(text): + reasons.append( + "cleanup report claims deleted branch that is not the merged PR head branch" + ) + + if remote_delete: + reasons.extend( + f"remote branch deletion missing {field}" + for field in _branch_safety_fields_present(text) + ) + + if worktree_remove: + reasons.extend( + f"worktree removal missing {field}" + for field in _worktree_cleanup_fields_present(text) + ) + + if (remote_delete or worktree_remove) and not (remote_delete or worktree_remove): + pass + + if not remote_delete and not worktree_remove: + cleanup_mutations = re.search( + r"cleanup mutations\s*:\s*(?!none\b)\S", + text, + re.IGNORECASE, + ) + if cleanup_mutations: + reasons.append( + "cleanup mutations reported without post-merge cleanup proof checklist" + ) + + outcome = CLEANUP_PERFORMED if (remote_delete or worktree_remove) and not reasons else None + if remote_delete or worktree_remove: + outcome = CLEANUP_PERFORMED if not reasons else "CLEANUP_CLAIMED_UNPROVEN" + + block = bool(reasons) + return { + "block": block, + "proven": not block, + "outcome": outcome, + "remote_delete_claimed": remote_delete, + "worktree_remove_claimed": worktree_remove, + "reasons": reasons, + "safe_next_action": ( + "report CLEANUP_SKIPPED with exact blocker or include the full cleanup " + "checklist before claiming remote delete or worktree removal" + if reasons + else "proceed" + ), + } \ No newline at end of file diff --git a/remote_repo_guard.py b/remote_repo_guard.py new file mode 100644 index 0000000..f4938bd --- /dev/null +++ b/remote_repo_guard.py @@ -0,0 +1,98 @@ +"""Remote/repo mismatch guard (#530). + +Bare ``remote`` names resolve to a default ``org``/``repo`` via the ``REMOTES`` +table in :mod:`gitea_auth`. For the ``prgs`` instance the hardcoded default repo +is ``Timesheet``, but the tools in this project operate on +``Scaled-Tech-Consulting/Gitea-Tools``. When a session runs inside a Gitea-Tools +worktree and calls a tool with a bare ``remote=prgs`` (no explicit ``org``/``repo``), +the resolved target silently points at the wrong repository, producing false 404s +and risking mutation of a different repo. + +This module provides a pure assessment that compares the MCP-resolved ``org/repo`` +against the local git remote URL and fails closed on a genuine mismatch, unless the +caller supplied explicit ``org``/``repo`` (in which case their intent is authoritative) +or the local remote URL is unavailable (best-effort corroboration only). +""" + +from __future__ import annotations + +REMEDIATION = ( + "Pass explicit org= and repo= matching the local git remote, " + "e.g. org=Scaled-Tech-Consulting repo=Gitea-Tools." +) + + +def assess_remote_repo_match( + *, + remote: str, + resolved_org: str, + resolved_repo: str, + local_remote_url: str | None, + org_explicit: bool, + repo_explicit: bool, +) -> dict: + """Fail closed when the resolved org/repo disagrees with the local git remote. + + The guard is intentionally conservative: + + * When the caller passed both ``org`` and ``repo`` explicitly, their intent is + authoritative and the guard never blocks. + * When the local git remote URL is unavailable (``None``/empty), corroboration + is impossible, so the guard does not block (best-effort only). + * Otherwise, the resolved ``org/repo`` slug must appear in the local remote URL + (case-insensitive); if it does not, the guard blocks. + """ + reasons: list[str] = [] + + if org_explicit and repo_explicit: + return _assessment(True, reasons, remote, resolved_org, resolved_repo, local_remote_url) + + url = (local_remote_url or "").strip() + if not url: + return _assessment(True, reasons, remote, resolved_org, resolved_repo, local_remote_url) + + expected_slug = f"{resolved_org}/{resolved_repo}".lower() + if expected_slug in url.lower(): + return _assessment(True, reasons, remote, resolved_org, resolved_repo, local_remote_url) + + reasons.append( + f"MCP-resolved repository '{resolved_org}/{resolved_repo}' for remote " + f"'{remote}' does not match the local git remote URL '{url}'" + ) + return _assessment(False, reasons, remote, resolved_org, resolved_repo, local_remote_url) + + +def format_remote_repo_guard_error(assessment: dict) -> str: + """Single RuntimeError message for the MCP resolver gate.""" + reasons = "; ".join( + assessment.get("reasons") or ["remote/repo resolution mismatch"] + ) + resolved = ( + f"{assessment.get('resolved_org')}/{assessment.get('resolved_repo')}" + ) + local = assessment.get("local_remote_url") or "(unknown)" + return ( + f"Remote/repo guard (#530): {reasons}. " + f"Resolved target: {resolved}; local git remote: {local}. " + f"{REMEDIATION}" + ) + + +def _assessment( + proven: bool, + reasons: list[str], + remote: str, + resolved_org: str, + resolved_repo: str, + local_remote_url: str | None, +) -> dict: + return { + "proven": proven, + "block": not proven, + "reasons": reasons, + "remote": remote, + "resolved_org": resolved_org, + "resolved_repo": resolved_repo, + "local_remote_url": local_remote_url, + "remediation": REMEDIATION, + } diff --git a/review_proofs.py b/review_proofs.py index febbcd0..194ce9e 100644 --- a/review_proofs.py +++ b/review_proofs.py @@ -5115,6 +5115,15 @@ def assess_pr_queue_cleanup_report(report_text: str | None) -> dict: return _assess(report_text or "") +def assess_audit_reconciliation_report(report_text: str | None) -> dict: + """#419: validate audit vs cleanup reconciliation report boundaries.""" + from audit_reconciliation_mode import ( + assess_audit_reconciliation_report as _assess, + ) + + return _assess(report_text or "") + + _GATE_PASSED_VALUE = re.compile(r"\bpassed\b", re.I) _NOT_APPLICABLE_VALUE = re.compile( @@ -5608,3 +5617,12 @@ def assess_proof_backed_handoff_report(report_text, **kwargs): from reviewer_proof_backed_handoff import assess_proof_backed_handoff_report as _assess return _assess(report_text, **kwargs) + + +def assess_mutation_capability_proof(report_text, **kwargs): + """#405: exact per-mutation capability proof in reviewer final reports.""" + from reviewer_mutation_capability_proof import ( + assess_mutation_capability_proof as _assess, + ) + + return _assess(report_text, **kwargs) diff --git a/review_workflow_boundary.py b/review_workflow_boundary.py new file mode 100644 index 0000000..f96aaab --- /dev/null +++ b/review_workflow_boundary.py @@ -0,0 +1,259 @@ +"""Reviewer session boundary tracking for workflow-load gate (#403). + +Pre-review commands executed before ``gitea_load_review_workflow`` must be +classified. Boundary violations block downstream reviewer mutations even when +workflow hash proof is present. +""" + +from __future__ import annotations + +import os +import re +from typing import Any + +CLASSIFICATION_READ_ONLY_INVENTORY = "read_only_inventory" +CLASSIFICATION_DIAGNOSTIC = "diagnostic" +CLASSIFICATION_BOUNDARY_VIOLATION = "boundary_violation" +CLASSIFICATION_UNCLASSIFIED = "unclassified" + +ALLOWED_CLASSIFICATIONS = frozenset({ + CLASSIFICATION_READ_ONLY_INVENTORY, + CLASSIFICATION_DIAGNOSTIC, + CLASSIFICATION_BOUNDARY_VIOLATION, + CLASSIFICATION_UNCLASSIFIED, +}) + +_PRE_REVIEW_COMMANDS: list[dict[str, Any]] = [] + +_READ_ONLY_INVENTORY_PATTERNS = ( + re.compile( + r"\bgitea[_-](?:list|view|whoami|get[-_]|resolve[-_]task|check[-_]pr|route[-_]task)", + re.I, + ), + re.compile(r"\bgit\s+(?:fetch|remote\s+update|branch\s+-a|log|show|rev-parse)\b", re.I), + re.compile(r"\bgit\s+status\b", re.I), + re.compile(r"\bgit\s+worktree\s+list\b", re.I), +) + +_DIAGNOSTIC_PATTERNS = ( + re.compile(r"\bgit\s+diff(?:\s+--stat)?\b", re.I), + re.compile(r"\bwhich\s+pytest\b", re.I), + re.compile(r"\bpytest\s+--version\b", re.I), +) + +_BOUNDARY_VIOLATION_PATTERNS: tuple[tuple[re.Pattern[str], str], ...] = ( + (re.compile(r"\b(?:pytest|python\s+-m\s+pytest|python\s+-m\s+unittest)\b", re.I), + "validation command before workflow load"), + (re.compile(r"\bprofiles\.json\b", re.I), "local profile config inspection"), + (re.compile(r"\bgitea-mcp(?:\.v2-contexts)?\.json\b", re.I), + "local Gitea MCP config inspection"), + (re.compile(r"\b\.env(?:\.|$|\b)", re.I), "credential file inspection"), + (re.compile(r"\bkeychain\b", re.I), "credential store inspection"), + (re.compile(r"\bpkill\b", re.I), "MCP repair activity"), + (re.compile(r"\b(?:edit|write|modify).{0,40}\bmcp\b", re.I), + "MCP config exploration"), + (re.compile(r"\bgit\s+(?:add|commit|reset|clean|checkout|merge|rebase|push)\b", re.I), + "git mutation before workflow load"), +) + + +def clear_pre_review_commands() -> None: + """Test helper and session reset.""" + global _PRE_REVIEW_COMMANDS + _PRE_REVIEW_COMMANDS = [] + + +def pre_review_commands() -> list[dict[str, Any]]: + """Return a shallow copy of recorded pre-review commands.""" + return [dict(entry) for entry in _PRE_REVIEW_COMMANDS] + + +def _normalize_path(path: str | None) -> str: + return os.path.realpath(os.path.abspath((path or "").strip() or os.getcwd())) + + +def is_main_checkout_path(cwd: str | None, project_root: str | None) -> bool: + """True when *cwd* is the stable control checkout (not under branches/).""" + if not project_root: + return False + root = _normalize_path(project_root) + path = _normalize_path(cwd) + if path != root: + return False + marker = f"{os.sep}branches{os.sep}" + return marker not in path + + +def classify_pre_review_command( + command: str, + *, + cwd: str | None = None, + project_root: str | None = None, +) -> dict[str, Any]: + """Classify a command executed before workflow load.""" + text = (command or "").strip() + path = _normalize_path(cwd) + root = _normalize_path(project_root) if project_root else None + reasons: list[str] = [] + + for pattern, label in _BOUNDARY_VIOLATION_PATTERNS: + if pattern.search(text): + if label.startswith("validation") and root and not is_main_checkout_path(path, root): + continue + if label.startswith("git mutation") and root and not is_main_checkout_path(path, root): + continue + reasons.append(label) + return { + "command": text, + "cwd": path, + "classification": CLASSIFICATION_BOUNDARY_VIOLATION, + "reasons": reasons, + } + + for pattern in _READ_ONLY_INVENTORY_PATTERNS: + if pattern.search(text): + return { + "command": text, + "cwd": path, + "classification": CLASSIFICATION_READ_ONLY_INVENTORY, + "reasons": [], + } + + for pattern in _DIAGNOSTIC_PATTERNS: + if pattern.search(text): + return { + "command": text, + "cwd": path, + "classification": CLASSIFICATION_DIAGNOSTIC, + "reasons": [], + } + + if root and is_main_checkout_path(path, root): + if re.search(r"\b(?:cat|head|less|read)\b", text, re.I): + if re.search(r"workflow|skill|runbook", text, re.I): + return { + "command": text, + "cwd": path, + "classification": CLASSIFICATION_BOUNDARY_VIOLATION, + "reasons": [ + "canonical workflow viewed as local file without " + "gitea_load_review_workflow (narrative load is not proof)" + ], + } + + return { + "command": text, + "cwd": path, + "classification": CLASSIFICATION_UNCLASSIFIED, + "reasons": [ + "pre-review command not classified; record via " + "gitea_record_pre_review_command before workflow load" + ], + } + + +def record_pre_review_command( + command: str, + *, + cwd: str | None = None, + project_root: str | None = None, + classification: str | None = None, +) -> dict[str, Any]: + """Record and classify a pre-review command for the current session.""" + assessed = classify_pre_review_command( + command, cwd=cwd, project_root=project_root) + if classification: + if classification not in ALLOWED_CLASSIFICATIONS: + assessed["classification"] = CLASSIFICATION_UNCLASSIFIED + assessed["reasons"] = [ + f"unknown classification '{classification}'; fail closed" + ] + else: + assessed["classification"] = classification + assessed["reasons"] = [] + entry = { + **assessed, + "session_pid": os.getpid(), + } + _PRE_REVIEW_COMMANDS.append(entry) + return dict(entry) + + +def assess_boundary_status(project_root: str | None = None) -> dict[str, Any]: + """Summarize pre-review boundary state for session proof and reports.""" + violations = [ + entry for entry in _PRE_REVIEW_COMMANDS + if entry.get("classification") == CLASSIFICATION_BOUNDARY_VIOLATION + ] + unclassified = [ + entry for entry in _PRE_REVIEW_COMMANDS + if entry.get("classification") == CLASSIFICATION_UNCLASSIFIED + ] + reasons: list[str] = [] + for entry in violations: + reasons.extend(entry.get("reasons") or [ + f"boundary violation: {entry.get('command', '')[:80]}" + ]) + for entry in unclassified: + reasons.extend(entry.get("reasons") or [ + "unclassified pre-review command blocks reviewer mutations" + ]) + + clean = not reasons + return { + "boundary_status": "clean" if clean else "violation", + "boundary_clean": clean, + "pre_review_command_count": len(_PRE_REVIEW_COMMANDS), + "boundary_violation_count": len(violations), + "unclassified_command_count": len(unclassified), + "violations": [ + { + "command": v.get("command"), + "cwd": v.get("cwd"), + "reasons": list(v.get("reasons") or []), + } + for v in violations + ], + "reasons": reasons, + } + + +def boundary_blockers(project_root: str | None = None) -> list[str]: + """Reasons reviewer mutations must fail closed due to boundary state.""" + status = assess_boundary_status(project_root) + if status.get("boundary_clean"): + return [] + return list(status.get("reasons") or [ + "reviewer session boundary violation before workflow load" + ]) + + +def workflow_load_helper_result( + load: dict | None, + project_root: str | None = None, +) -> dict[str, Any]: + """Structured helper result for final reports (#403).""" + boundary = assess_boundary_status(project_root) + if load is None: + return { + "workflow_load_proof_present": False, + "workflow_source": None, + "workflow_hash": None, + "final_report_schema_hash": None, + "boundary_status": boundary.get("boundary_status"), + "boundary_clean": False, + "reasons": [ + "gitea_load_review_workflow helper result missing from report" + ], + } + return { + "workflow_load_proof_present": True, + "workflow_source": load.get("workflow_source"), + "workflow_hash": load.get("workflow_hash"), + "final_report_schema_path": load.get("final_report_schema_path"), + "final_report_schema_hash": load.get("final_report_schema_hash"), + "boundary_status": load.get("boundary_status", boundary.get("boundary_status")), + "boundary_clean": bool(load.get("boundary_clean", boundary.get("boundary_clean"))), + "pre_review_command_count": boundary.get("pre_review_command_count"), + "reasons": [], + } \ No newline at end of file diff --git a/review_workflow_load.py b/review_workflow_load.py new file mode 100644 index 0000000..8deb622 --- /dev/null +++ b/review_workflow_load.py @@ -0,0 +1,321 @@ +"""Canonical review-merge workflow load proof for reviewer mutations (#389, #403, #559).""" + +from __future__ import annotations + +import hashlib +import os +import re +from pathlib import Path + +import mcp_session_state +import review_workflow_boundary as boundary + +WORKFLOW_REL_PATH = ( + "skills/llm-project-workflow/workflows/review-merge-pr.md" +) +SCHEMA_REL_PATH = ( + "skills/llm-project-workflow/schemas/review-merge-final-report.md" +) +TASK_MODE = "review-merge-pr" +LOAD_TOOL_NAME = "gitea_load_review_workflow" + +_REVIEW_WORKFLOW_LOAD: dict | None = None + + +def compute_content_hash(text: str) -> str: + """Short deterministic hash for workflow/schema version proof.""" + return hashlib.sha256((text or "").encode("utf-8")).hexdigest()[:12] + + +def _read_text(path: Path) -> str: + return path.read_text(encoding="utf-8") + + +def _canonical_paths(project_root: str) -> tuple[Path, Path]: + root = Path(project_root) + workflow = root / WORKFLOW_REL_PATH + schema = root / SCHEMA_REL_PATH + if not workflow.is_file(): + raise FileNotFoundError(f"canonical workflow missing: {workflow}") + if not schema.is_file(): + raise FileNotFoundError(f"final report schema missing: {schema}") + return workflow, schema + + +def build_canonical_workflow_metadata( + project_root: str, + *, + prompt_text: str | None = None, +) -> dict: + """Load workflow + schema from disk and compute proof metadata.""" + workflow_path, schema_path = _canonical_paths(project_root) + workflow_text = _read_text(workflow_path) + schema_text = _read_text(schema_path) + workflow_hash = compute_content_hash(workflow_text) + schema_hash = compute_content_hash(schema_text) + conflict, conflict_reasons = assess_prompt_conflict(prompt_text) + return { + "workflow_source": WORKFLOW_REL_PATH, + "workflow_path": str(workflow_path), + "task_mode": TASK_MODE, + "workflow_hash": workflow_hash, + "workflow_version": workflow_hash, + "final_report_schema_path": SCHEMA_REL_PATH, + "final_report_schema_hash": schema_hash, + "prompt_conflicts_with_workflow": conflict, + "prompt_conflict_reasons": conflict_reasons, + "load_tool": LOAD_TOOL_NAME, + } + + +def assess_prompt_conflict(prompt_text: str | None) -> tuple[bool, list[str]]: + """Detect obvious task-mode conflicts between prompt and review workflow.""" + if not (prompt_text or "").strip(): + return False, [] + text = prompt_text.lower() + reasons: list[str] = [] + conflicting = ( + (r"\bwork[- ]issue\b", "work-issue author mode"), + (r"\bcreate[- ]issue\b", "create-issue mode"), + (r"\bauthor/coder\b", "author/coder mode"), + (r"\breconcile[- ]landed\b", "reconcile-landed mode"), + ) + for pattern, label in conflicting: + if re.search(pattern, text): + reasons.append( + f"active prompt appears to request {label} while loading " + f"{TASK_MODE} workflow" + ) + return bool(reasons), reasons + + +def _session_binding_fields() -> dict: + """Capture profile identity used to share state across daemon processes.""" + env_lock = (os.environ.get(mcp_session_state.SESSION_PROFILE_LOCK_ENV) or "").strip() + profile_name = (os.environ.get("GITEA_MCP_PROFILE") or "").strip() + remote = (os.environ.get("GITEA_MCP_REMOTE") or "").strip() or None + identity = mcp_session_state.current_profile_identity( + profile_name=profile_name, + session_profile_lock=env_lock, + ) + return { + "session_profile": profile_name or identity, + "session_profile_lock": env_lock or identity, + "profile_identity": identity, + "remote": remote, + } + + +def _persist_workflow_load(record: dict | None) -> dict | None: + """Write durable workflow-load proof (or clear it).""" + binding = _session_binding_fields() + if record is None: + mcp_session_state.clear_state( + kind=mcp_session_state.KIND_WORKFLOW_LOAD, + remote=binding.get("remote"), + profile_identity=binding.get("profile_identity"), + ) + return None + payload = dict(record) + payload.update({ + k: v for k, v in binding.items() if v is not None + }) + return mcp_session_state.save_state( + kind=mcp_session_state.KIND_WORKFLOW_LOAD, + payload=payload, + remote=payload.get("remote"), + org=payload.get("org"), + repo=payload.get("repo"), + profile_identity=payload.get("profile_identity"), + ) + + +def _load_durable_workflow_load() -> dict | None: + binding = _session_binding_fields() + return mcp_session_state.load_state( + kind=mcp_session_state.KIND_WORKFLOW_LOAD, + remote=binding.get("remote"), + profile_identity=binding.get("profile_identity"), + ) + + +def _active_workflow_load() -> dict | None: + """Prefer in-process cache; fall back to durable shared state (#559).""" + global _REVIEW_WORKFLOW_LOAD + if _REVIEW_WORKFLOW_LOAD is not None: + return _REVIEW_WORKFLOW_LOAD + durable = _load_durable_workflow_load() + if durable is not None: + _REVIEW_WORKFLOW_LOAD = dict(durable) + return _REVIEW_WORKFLOW_LOAD + + +def record_review_workflow_load( + project_root: str, + *, + prompt_text: str | None = None, +) -> dict: + """Record workflow load proof for the current MCP session (durable + memory).""" + global _REVIEW_WORKFLOW_LOAD + meta = build_canonical_workflow_metadata( + project_root, prompt_text=prompt_text) + boundary_state = boundary.assess_boundary_status(project_root) + binding = _session_binding_fields() + record = { + **meta, + "session_pid": os.getpid(), + "loaded": True, + "boundary_status": boundary_state.get("boundary_status"), + "boundary_clean": boundary_state.get("boundary_clean"), + "pre_review_command_count": boundary_state.get("pre_review_command_count"), + "boundary_violation_count": boundary_state.get("boundary_violation_count"), + "boundary_reasons": list(boundary_state.get("reasons") or []), + **{k: v for k, v in binding.items() if v is not None}, + } + persisted = _persist_workflow_load(record) + _REVIEW_WORKFLOW_LOAD = dict(persisted or record) + return dict(_REVIEW_WORKFLOW_LOAD) + + +def clear_review_workflow_load() -> None: + """Test helper and review_pr session reset.""" + global _REVIEW_WORKFLOW_LOAD + _REVIEW_WORKFLOW_LOAD = None + _persist_workflow_load(None) + boundary.clear_pre_review_commands() + + +def workflow_load_status(project_root: str | None = None) -> dict: + """Non-throwing status for capability/runtime reports.""" + load = _active_workflow_load() + if load is None: + return { + "workflow_load_proof_present": False, + "workflow_load_valid": False, + "workflow_source": None, + "workflow_hash": None, + "final_report_schema_path": SCHEMA_REL_PATH, + "reasons": [ + f"{LOAD_TOOL_NAME} has not been called in this session " + "(fail closed for reviewer mutations)" + ], + } + reasons = _session_validation_reasons(load, project_root) + boundary_reasons = boundary.boundary_blockers(project_root) + if boundary_reasons: + reasons = list(reasons) + boundary_reasons + return { + "workflow_load_proof_present": True, + "workflow_load_valid": not reasons, + "workflow_source": load.get("workflow_source"), + "workflow_hash": load.get("workflow_hash"), + "task_mode": load.get("task_mode"), + "final_report_schema_path": load.get("final_report_schema_path"), + "final_report_schema_hash": load.get("final_report_schema_hash"), + "prompt_conflicts_with_workflow": load.get( + "prompt_conflicts_with_workflow"), + "session_pid": load.get("session_pid"), + "writer_pid": load.get("writer_pid"), + "profile_identity": load.get("profile_identity"), + "boundary_status": load.get("boundary_status"), + "boundary_clean": load.get("boundary_clean"), + "workflow_load_helper_result": boundary.workflow_load_helper_result( + load, project_root), + "reasons": reasons, + } + + +def _session_validation_reasons( + load: dict, + project_root: str | None, +) -> list[str]: + """Validate durable/in-memory load proof for this session identity (#559).""" + reasons: list[str] = [] + # Cross-process daemon pools are allowed when profile identity matches. + # Reject only when the stored profile identity conflicts with this process. + binding = _session_binding_fields() + stored_identity = ( + load.get("session_profile_lock") + or load.get("profile_identity") + or "" + ).strip() + active_identity = (binding.get("profile_identity") or "").strip() + if ( + stored_identity + and active_identity + and stored_identity != active_identity + and active_identity != "unknown-profile" + and stored_identity != "unknown-profile" + ): + reasons.append( + "workflow load proof profile identity mismatch " + f"(stored={stored_identity!r}, active={active_identity!r}; fail closed)" + ) + return reasons + + # Expired durable records are treated as absent. + identity_reasons = mcp_session_state.identity_match_reasons( + load, + remote=binding.get("remote") or load.get("remote"), + org=load.get("org"), + repo=load.get("repo"), + profile_identity=active_identity or stored_identity, + ) + # Filter out remote-mismatch noise when remote was not bound at load time. + for reason in identity_reasons: + if "missing recorded_at" in reason or "expired" in reason or "future" in reason: + reasons.append(reason) + elif "profile identity mismatch" in reason: + reasons.append(reason) + if reasons: + return reasons + + if load.get("prompt_conflicts_with_workflow"): + reasons.extend(load.get("prompt_conflict_reasons") or [ + "active prompt conflicts with loaded review-merge workflow" + ]) + if project_root: + try: + current = build_canonical_workflow_metadata(project_root) + except OSError as exc: + reasons.append(f"cannot re-verify workflow hash: {exc}") + return reasons + if current["workflow_hash"] != load.get("workflow_hash"): + reasons.append( + "stored workflow hash is stale; reload via " + f"{LOAD_TOOL_NAME} (fail closed)" + ) + if current["final_report_schema_hash"] != load.get( + "final_report_schema_hash"): + reasons.append( + "stored final-report schema hash is stale; reload via " + f"{LOAD_TOOL_NAME} (fail closed)" + ) + return reasons + + +def review_workflow_load_blockers( + project_root: str | None = None, +) -> list[str]: + """Reasons reviewer mutations must fail closed.""" + boundary_reasons = boundary.boundary_blockers(project_root) + load = _active_workflow_load() + if boundary_reasons and load is None: + return boundary_reasons + status = workflow_load_status(project_root) + if not status.get("workflow_load_proof_present"): + return list(status.get("reasons") or []) + boundary_reasons + if not status.get("workflow_load_valid"): + return list(status.get("reasons") or []) + return [] + + +def recovery_handoff_without_replay() -> list[str]: + """Safe next-step lines that must not include approve/merge replay.""" + return [ + "Reload the canonical workflow via gitea_load_review_workflow, then " + "rerun the full review-merge workflow from inventory.", + "Do not call gitea_submit_pr_review, gitea_mark_final_review_decision, " + "or gitea_merge_pr until workflow-load proof is present.", + "Do not include approve/merge replay commands in the recovery handoff.", + ] diff --git a/reviewer_mutation_capability_proof.py b/reviewer_mutation_capability_proof.py new file mode 100644 index 0000000..1488738 --- /dev/null +++ b/reviewer_mutation_capability_proof.py @@ -0,0 +1,129 @@ +"""Exact per-mutation capability proof verifier for reviewer reports (#405). + +A reviewer final report may prove ``review_pr`` capability and then also merge +a PR or delete a remote branch. Merge and branch deletion are separate +mutations that require their own exact capability proof — a nearby capability +must never authorize a different operation. This verifier requires a +mutation-capability table pairing every performed mutation with the exact +task/permission resolved *before* that mutation. +""" + +from __future__ import annotations + +import re + +# Mutations this verifier tracks, with the exact capability tokens that +# authorize each. A row for the mutation must cite one of its own tokens; +# tokens from a different mutation (a "nearby capability") never count. +_REVIEW_TOKENS = ("review_pr", "gitea.pr.review", "gitea.pr.approve", + "gitea.pr.request_changes", "request_changes_pr", "approve_pr") +_MERGE_TOKENS = ("merge_pr", "gitea.pr.merge") +_DELETE_TOKENS = ("delete_branch", "gitea.branch.delete") + +# Detect that a mutation was actually performed (not merely mentioned as a +# non-goal or skipped). +_MERGE_PERFORMED = re.compile( + r"(?:gitea_merge_pr\b(?![^\n]*\b(?:not called|skipped|blocked)\b)|" + r"^\s*[-*]?\s*merge result\s*:\s*merged\b|" + r"\bpr merged\b|\bmerge commit\s*(?:sha)?\s*[:=]?\s*[0-9a-f]{7,})", + re.IGNORECASE | re.MULTILINE, +) +_DELETE_PERFORMED = re.compile( + r"(?:gitea_delete_branch\b(?![^\n]*\b(?:not called|skipped|blocked)\b)|" + r"^\s*[-*]?\s*(?:remote )?branch deleted\s*:|" + r"\bdeleted (?:the )?(?:remote )?branch\b|" + r"^\s*[-*]?\s*branch deletion\s*:\s*(?!skipped|none|not)\S)", + re.IGNORECASE | re.MULTILINE, +) +_REVIEW_PERFORMED = re.compile( + r"(?:gitea_submit_pr_review\b|gitea_mark_final_review_decision\b|" + r"^\s*[-*]?\s*review (?:decision|verdict|mutation)\s*:\s*" + r"(?:approved|request[_ ]changes)\b|\breview submitted\b)", + re.IGNORECASE | re.MULTILINE, +) + +# Post-hoc proof: capability resolved *after* the mutation is never valid. +_POST_HOC = re.compile( + r"capabilit(?:y|ies)\s+(?:resolved|proven|checked)\s+(?:after|post[- ])\s*" + r"(?:the\s+)?(?:merge|deletion|delete|mutation|review)", + re.IGNORECASE, +) + +# The report must carry an explicit mutation-capability table. +_TABLE_MARKER = re.compile( + r"mutation[- ]capability(?:\s+table)?|capability[- ]per[- ]mutation", + re.IGNORECASE, +) + + +def _tokens_present(text: str, tokens: tuple[str, ...]) -> bool: + low = text.lower() + return any(tok.lower() in low for tok in tokens) + + +def assess_mutation_capability_proof(report_text: str) -> dict: + """Validate exact per-mutation capability proof in a reviewer report. + + Returns ``{proven, block, reasons, safe_next_action}``. A report that + performs no mutation beyond an ordinary review passes only when its + review capability is cited; merge/delete each demand their own exact + capability row. Fail closed on nearby-capability substitution, a + missing table, missing rows, or post-hoc proof. + """ + text = report_text or "" + reasons: list[str] = [] + + merged = bool(_MERGE_PERFORMED.search(text)) + deleted = bool(_DELETE_PERFORMED.search(text)) + reviewed = bool(_REVIEW_PERFORMED.search(text)) + + extra_mutation = merged or deleted + + if _POST_HOC.search(text): + reasons.append( + "capability proof recorded after the mutation; exact capability " + "must be resolved before each mutation" + ) + + # A review-only report needs its review capability cited; no table required. + if reviewed and not _tokens_present(text, _REVIEW_TOKENS): + reasons.append( + "review mutation performed without exact review capability proof " + "(review_pr / gitea.pr.review)" + ) + + if extra_mutation and not _TABLE_MARKER.search(text): + reasons.append( + "mutation beyond review performed without a mutation-capability " + "table (mutation, exact task/capability, result, order-before)" + ) + + if merged: + if not _tokens_present(text, _MERGE_TOKENS): + reasons.append( + "merge performed without exact merge capability proof " + "(merge_pr / gitea.pr.merge); nearby review_pr does not " + "authorize merge" + ) + + if deleted: + if not _tokens_present(text, _DELETE_TOKENS): + reasons.append( + "branch deletion performed without exact delete capability " + "proof (delete_branch / gitea.branch.delete); nearby " + "merge_pr does not authorize branch deletion" + ) + + proven = not reasons + return { + "proven": proven, + "block": not proven, + "reasons": reasons, + "safe_next_action": ( + "proceed" + if proven + else "add a mutation-capability table with the exact resolved " + "task/permission and pre-mutation order for every mutation; " + "skip any mutation whose exact capability is unproven" + ), + } diff --git a/reviewer_pr_lease.py b/reviewer_pr_lease.py index 6e28be5..f55390b 100644 --- a/reviewer_pr_lease.py +++ b/reviewer_pr_lease.py @@ -23,6 +23,7 @@ _ACTIVE_PHASES = frozenset({ "approved", "request-changes", "merging", + "adopted", }) DEFAULT_LEASE_TTL_MINUTES = 120 @@ -217,12 +218,24 @@ def assess_acquire_lease( candidate_head: str | None, target_branch: str, target_branch_sha: str | None, + pr_merged_or_closed: bool = False, now: datetime | None = None, ) -> dict[str, Any]: - """Fail closed when another session holds an active lease.""" + """Fail closed when another session holds an active lease. + + When *pr_merged_or_closed* is true the PR has already merged/closed, so any + reviewer-lease acquisition or adoption for merge work is moot: fail closed + with a ``post_merge_moot`` reason and never mint a lease body (#515). + """ now = now or datetime.now(timezone.utc) reasons: list[str] = [] existing = find_active_reviewer_lease(comments, pr_number=pr_number, now=now) + post_merge_moot = bool(pr_merged_or_closed) + if post_merge_moot: + reasons.append( + f"post_merge_moot: PR #{pr_number} is already merged/closed; reviewer " + "lease adoption for merge is moot (fail closed)" + ) if existing: owner_session = (existing.get("session_id") or "").strip() freshness = existing.get("freshness") or classify_lease_freshness(existing, now=now) @@ -270,12 +283,120 @@ def assess_acquire_lease( "existing_lease": existing, "lease_body": body, "session_id": session_id, + "post_merge_moot": post_merge_moot, } -def record_session_lease(lease: dict[str, Any]) -> dict[str, Any]: +def assess_post_merge_moot_lease( + comments: list[dict], + *, + pr_number: int, + pr_merged: bool = False, + pr_state: str | None = None, + merge_commit_sha: str | None = None, + now: datetime | None = None, +) -> dict[str, Any]: + """Assess a reviewer lease left lingering on an already-merged/closed PR (#515). + + Read-first and fail-safe: + + - Only treats a lease as moot when the live PR state is merged/closed. + - Never proposes touching an *active* lease while the PR is still open + (``cleanup_allowed`` stays false and a refusal reason is returned). + - When the PR is merged/closed and a lease is still active, ``cleanup_allowed`` + is true and a terminal ``phase: released`` lease body (``blocker: + post-merge-moot``) is provided so the moot lease can be neutralised by an + append-only comment — never by deleting a foreign session's comment, and + never by adopting or merging. + + Posting the released body makes that lease terminal, so a subsequent call + finds no active lease and reports nothing left to clean (idempotent). + """ + now = now or datetime.now(timezone.utc) + merged_or_closed = bool(pr_merged) or ( + str(pr_state or "").strip().lower() == "closed" + ) + active = find_active_reviewer_lease(comments, pr_number=pr_number, now=now) + # The newest lease comment is authoritative: once a terminal marker + # (released/done/blocked) is the latest entry, the lease is resolved even if + # an earlier non-terminal comment from the same session still lingers. This + # keeps post-merge cleanup idempotent. + entries = _lease_entries(comments, pr_number=pr_number) + newest = entries[-1] if entries else None + newest_terminal = bool(newest) and ( + (newest.get("phase") or "").strip().lower() in _TERMINAL_PHASES + ) + reasons: list[str] = [] + cleanup_allowed = False + release_body: str | None = None + is_moot = bool(active) and merged_or_closed and not newest_terminal + + if not merged_or_closed: + if active: + reasons.append( + f"PR #{pr_number} is still open; refusing to touch active reviewer " + "lease (fail closed)" + ) + else: + reasons.append( + f"PR #{pr_number} is still open; no post-merge lease cleanup applicable" + ) + elif newest_terminal: + reasons.append( + f"PR #{pr_number} reviewer lease already released/terminal; nothing to clean" + ) + elif active: + cleanup_allowed = True + release_body = format_lease_body( + repo=active.get("repo") or "", + pr_number=pr_number, + issue_number=active.get("issue_number"), + reviewer_identity=active.get("reviewer_identity") or "", + profile=active.get("profile") or "unknown", + session_id=active.get("session_id") or "", + worktree=active.get("worktree") or "", + phase="released", + candidate_head=active.get("candidate_head"), + target_branch=active.get("target_branch") or "master", + target_branch_sha=active.get("target_branch_sha"), + last_activity=now, + blocker="post-merge-moot", + ) + else: + reasons.append( + f"PR #{pr_number} is merged/closed but no active reviewer lease remains; " + "nothing to clean" + ) + + return { + "pr_number": pr_number, + "pr_state": pr_state, + "pr_merged_or_closed": merged_or_closed, + "merge_commit_sha": merge_commit_sha, + "active_lease": active, + "is_moot": is_moot, + "cleanup_allowed": cleanup_allowed, + "release_body": release_body, + "reasons": reasons, + } + + +def record_session_lease( + lease: dict[str, Any], + *, + lease_provenance: dict[str, Any] | None = None, +) -> dict[str, Any]: + """Record the in-session lease mirror for mutation gates. + + *lease_provenance* must be supplied by sanctioned MCP tools (#536). Bare + manual seeding without provenance cannot satisfy merger/reviewer mutation + gates. + """ global _SESSION_LEASE - _SESSION_LEASE = dict(lease) + stored = dict(lease) + if lease_provenance: + stored["lease_provenance"] = dict(lease_provenance) + _SESSION_LEASE = stored return dict(_SESSION_LEASE) @@ -308,13 +429,25 @@ def assess_mutation_lease_gate( if not session: reasons.append( f"no in-session reviewer lease recorded; acquire via " - f"gitea_acquire_reviewer_pr_lease before {mutation}" + f"gitea_acquire_reviewer_pr_lease or adopt via " + f"gitea_adopt_merger_pr_lease before {mutation}" ) - elif session.get("pr_number") != pr_number: + else: + import merger_lease_adoption as mla + + if not mla.is_sanctioned_session_lease(session): + reasons.append( + "in-session lease lacks sanctioned provenance; manual " + "_SESSION_LEASE seeding is not canonical proof — use " + "gitea_acquire_reviewer_pr_lease or gitea_adopt_merger_pr_lease" + ) + if session and session.get("pr_number") != pr_number: reasons.append( f"session lease is for PR #{session.get('pr_number')}, not #{pr_number}" ) - elif (session.get("session_id") or "") != (session_id or session.get("session_id")): + elif session and (session.get("session_id") or "") != ( + session_id or session.get("session_id") + ): reasons.append("session lease session_id mismatch (fail closed)") if active: diff --git a/root_checkout_guard.py b/root_checkout_guard.py new file mode 100644 index 0000000..03af9e7 --- /dev/null +++ b/root_checkout_guard.py @@ -0,0 +1,148 @@ +"""Root checkout guard (#475). + +The project root checkout is the stable control checkout on master/prgs/master. +Author/reviewer/merge flows must fail closed when the control checkout is +contaminated (wrong branch, detached HEAD, dirty, or HEAD behind/ahead of +prgs/master). Isolated ``branches/...`` worktrees remain allowed. +""" + +from __future__ import annotations + +import os +import subprocess + +from author_mutation_worktree import is_path_under_branches +from reviewer_worktree import parse_dirty_tracked_files + +REMEDIATION = ( + "Root checkout is not on master. Preserve state, switch root back to master, " + "and use scripts/worktree-review or the sanctioned issue worktree flow." +) + +BASE_BRANCHES = frozenset({"master", "main", "dev"}) +REMOTE_MASTER_REFS = ("prgs/master", "refs/remotes/prgs/master") + + +def resolve_remote_master_sha( + canonical_repo_root: str, + *, + remote_refs: tuple[str, ...] | None = None, +) -> str | None: + """Return the commit SHA for the tracking master ref when available.""" + root = (canonical_repo_root or "").strip() + if not root: + return None + for ref in remote_refs or REMOTE_MASTER_REFS: + res = subprocess.run( + ["git", "-C", root, "rev-parse", "--verify", ref], + capture_output=True, + text=True, + check=False, + ) + if res.returncode == 0: + sha = (res.stdout or "").strip() + if sha: + return sha + return None + + +resolve_tracking_master_sha = resolve_remote_master_sha + + +def assess_root_checkout_guard( + *, + workspace_path: str, + canonical_repo_root: str, + current_branch: str | None, + head_sha: str | None, + porcelain_status: str, + remote_master_sha: str | None, + resolved_role: str | None = None, + actual_role: str | None = None, +) -> dict: + """Fail closed when the control checkout is not clean master/prgs/master. + + ``resolved_role`` is the preflight-resolved *task* role and ``actual_role`` + is the *active profile* role (#540). The reconciler exemption honours either + signal so a ``comment_issue`` preflight (which stamps the task role as + ``author``) cannot strip a genuine reconciler of its exemption. An actual + author profile classifies as ``author`` in both signals, so author blocking + on a contaminated control checkout is preserved. + + Merger *strictness* (a merger must not be auto-exempted by working from a + ``branches/`` worktree) stays keyed on the resolved task role: merge + operations resolve their own task role, and widening the merger test with + ``actual_role`` would wrongly subject a merger operating from its clean + workspace under a non-merge task to full control-checkout checks. + """ + reasons: list[str] = [] + root = os.path.realpath(canonical_repo_root) + workspace = os.path.realpath(workspace_path) + branch = (current_branch or "").strip() + dirty_files = parse_dirty_tracked_files(porcelain_status) + + if resolved_role == "reconciler" or actual_role == "reconciler": + return _assessment(True, [], root, workspace, branch, head_sha, dirty_files) + + if resolved_role != "merger" and is_path_under_branches(workspace, root): + return _assessment(True, [], root, workspace, branch, head_sha, dirty_files) + + if dirty_files: + reasons.append( + "control checkout has tracked local edits before role work " + f"(dirty files: {', '.join(dirty_files)})" + ) + + if not branch: + reasons.append("control checkout is detached HEAD; expected branch 'master'") + elif branch not in BASE_BRANCHES: + reasons.append( + f"control checkout branch '{branch}' is not a stable base branch " + f"({'/'.join(sorted(BASE_BRANCHES))})" + ) + + if remote_master_sha and head_sha and head_sha != remote_master_sha: + reasons.append( + "control checkout HEAD does not match prgs/master " + f"(HEAD {head_sha[:12]}, prgs/master {remote_master_sha[:12]})" + ) + + proven = not reasons + return _assessment(proven, reasons, root, workspace, branch or None, head_sha, dirty_files) + + +def format_root_checkout_guard_error(assessment: dict) -> str: + """Single RuntimeError message for MCP preflight gates.""" + root = assessment.get("canonical_repo_root") or "(unknown)" + workspace = assessment.get("workspace_path") or "(unknown)" + reasons = "; ".join(assessment.get("reasons") or ["unknown root checkout violation"]) + return ( + f"Root checkout guard (#475): {reasons}. " + f"canonical repository root: {root}; workspace: {workspace}. " + f"{REMEDIATION}" + ) + + +def _assessment( + proven: bool, + reasons: list[str], + canonical_repo_root: str, + workspace_path: str, + current_branch: str | None, + head_sha: str | None, + dirty_files: list[str], +) -> dict: + return { + "proven": proven, + "block": not proven, + "reasons": reasons, + "canonical_repo_root": canonical_repo_root, + "workspace_path": workspace_path, + "current_branch": current_branch, + "head_sha": head_sha, + "dirty_files": dirty_files, + "remediation": REMEDIATION, + } + + +assess_root_checkout = assess_root_checkout_guard \ No newline at end of file diff --git a/run-tests.sh b/run-tests.sh new file mode 100755 index 0000000..6930108 --- /dev/null +++ b/run-tests.sh @@ -0,0 +1,13 @@ +#!/usr/bin/env bash +set -euo pipefail + +ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" +PYTHON="$ROOT_DIR/venv/bin/python" + +if [[ ! -x "$PYTHON" ]]; then + echo "ERROR: expected virtualenv Python at $PYTHON" >&2 + echo "Create the venv first, then run: venv/bin/python -m pytest" >&2 + exit 1 +fi + +exec "$PYTHON" -m pytest "$@" diff --git a/skills/llm-project-workflow/schemas/review-merge-final-report.md b/skills/llm-project-workflow/schemas/review-merge-final-report.md index 9eb3592..6ef00bc 100644 --- a/skills/llm-project-workflow/schemas/review-merge-final-report.md +++ b/skills/llm-project-workflow/schemas/review-merge-final-report.md @@ -63,8 +63,14 @@ Do not use legacy fields: `Pinned reviewed head`, `Scratch worktree used`, - Current status: - Safe next action: - Safety statement: +- Workflow-load helper result: ``` +The **Workflow-load helper result** field must carry structured output from +`gitea_load_review_workflow` (workflow_hash, final_report_schema_hash, +boundary_status). Narrative claims that workflow files were viewed locally are +not sufficient (#403). + ### Already-landed handoff overrides When eligibility class is `ALREADY_LANDED_RECONCILE_REQUIRED`: diff --git a/skills/llm-project-workflow/templates/controller-issue-acceptance.md b/skills/llm-project-workflow/templates/controller-issue-acceptance.md new file mode 100644 index 0000000..148ef30 --- /dev/null +++ b/skills/llm-project-workflow/templates/controller-issue-acceptance.md @@ -0,0 +1,68 @@ +# Controller issue-acceptance prompt + +Use after a PR merges when auditing whether the linked issue is truly complete. + +```text +Audit issue # against its acceptance criteria after merged PR #. +Post a Controller Issue Acceptance comment with checked criteria, validation +reviewed, controller decision, next actor, and paste-ready next prompt. +Do not mark the issue accepted unless every required criterion is satisfied. +``` + +## Comment template + +```text +## Controller Issue Acceptance + +STATE: + + +WHO_IS_NEXT: + + +NEXT_ACTION: + + +NEXT_PROMPT: + + +ISSUE: +#... + +MERGED_PR: +#... + +MERGE_COMMIT: +<40-character SHA> + +ACCEPTANCE_CRITERIA_CHECKED: +- [x] ... +- [ ] ... + +VALIDATION_REVIEWED: + + +CONTROLLER_DECISION: + + +WHY: + + +MISSING_WORK: + + +FOLLOW_UP_ISSUES: + + +BLOCKERS: + + +LAST_UPDATED_BY: + +``` + +## Rejection paths + +When rejecting completion, `STATE` must name the gap (`needs-tests`, +`needs-docs`, `more-work-required`, etc.), `MISSING_WORK` must be explicit, +and `NEXT_PROMPT` must be ready for the next author session. \ No newline at end of file diff --git a/skills/llm-project-workflow/templates/merge-pr.md b/skills/llm-project-workflow/templates/merge-pr.md index ae99544..515ae10 100644 --- a/skills/llm-project-workflow/templates/merge-pr.md +++ b/skills/llm-project-workflow/templates/merge-pr.md @@ -10,6 +10,10 @@ Load the canonical workflow first: Final report schema: `schemas/review-merge-final-report.md`. Rules (llm-project-workflow): +- Repository targeting (#530): pass explicit `remote=`, `org=`, and `repo=` on + every gitea-tools call (e.g. `remote=prgs org=Scaled-Tech-Consulting + repo=Gitea-Tools`). A bare `remote=prgs` can resolve to the wrong default repo + and is blocked when it disagrees with the local git remote URL. - Only an eligible, NON-author reviewer merges. If authenticated user == PR author → STOP. - Do not merge unless the PR is open, mergeable, and its checks/review pass. @@ -41,7 +45,15 @@ Steps: 8. Confirm remote master now contains the merge commit (or the expected changes if squash merged). *Note: Gitea PR "closed" state is NOT equivalent to "merged". Do not assume a closed PR succeeded without verifying the actual landed changes.* -Then run the cleanup template (worktree-cleanup.md): +Post-merge cleanup (#517): merger sessions must NOT perform ad hoc cleanup. +- Record merge mutations separately from cleanup mutations in the controller handoff. +- Hand cleanup to a `prgs-reconciler` session — never raw `git branch -d`, + `git push --delete`, curl/API comment deletion, or local scripts. +- Reconciler cleanup must cite authorized MCP tools + (`gitea_reconcile_merged_cleanups`, `gitea_cleanup_post_merge_moot_lease`, + `gitea_delete_branch`, etc.) plus `gitea.branch.delete` capability proof. + +Then run the cleanup template (worktree-cleanup.md) in a reconciler session: - Verify expected file/commit presence on master (post-merge file-presence verification): - Run: git fetch --prune; git checkout master; git pull master --ff-only - Verify that the expected files added/modified in the PR are present on master (or absent if deleted). diff --git a/skills/llm-project-workflow/templates/post-merge-cleanup-handoff.md b/skills/llm-project-workflow/templates/post-merge-cleanup-handoff.md new file mode 100644 index 0000000..37a0ba6 --- /dev/null +++ b/skills/llm-project-workflow/templates/post-merge-cleanup-handoff.md @@ -0,0 +1,27 @@ +# Template: post-merge cleanup (reconciler only) + +Copy, fill the `<...>` fields, and paste as the task prompt. Run only after merge +is confirmed on remote master. Merger sessions must hand off here — never perform +this cleanup inline (#517). + +```text +Task: MCP-native post-merge cleanup for PR # / issue #. + +Rules: +- Active profile must be reconciler (`prgs-reconciler`) with `gitea.branch.delete`. +- Use MCP tools only — no raw git branch delete, no API comment deletion scripts. +- Record cleanup mutations separately from merge mutations in the handoff. + +Steps: +1. `gitea_resolve_task_capability(task="reconcile_merged_cleanups", remote=prgs)` +2. Confirm PR # merged on /master. +3. `gitea_cleanup_post_merge_moot_lease` if a reviewer lease remains (append-only). +4. `gitea_reconcile_merged_cleanups` for branch/worktree cleanup with dry-run first. +5. Report authorized cleanup tools used and reconciler capability proof. + +Handoff ledger (required fields): +- Merge mutations: (none — merger already recorded gitea_merge_pr) +- Cleanup mutations: list exact MCP tools invoked +- Reconciler capability: profile + gitea.branch.delete proof +- Next actor: controller acceptance or none +``` \ No newline at end of file diff --git a/skills/llm-project-workflow/templates/review-pr.md b/skills/llm-project-workflow/templates/review-pr.md index 391f962..a959634 100644 --- a/skills/llm-project-workflow/templates/review-pr.md +++ b/skills/llm-project-workflow/templates/review-pr.md @@ -35,6 +35,10 @@ Load the canonical workflow first: Final report schema: `schemas/review-merge-final-report.md`. Rules (llm-project-workflow): +- Repository targeting (#530): pass explicit `remote=`, `org=`, and `repo=` on + every gitea-tools call (e.g. `remote=prgs org=Scaled-Tech-Consulting + repo=Gitea-Tools`). A bare `remote=prgs` can resolve to the wrong default repo + and is blocked when it disagrees with the local git remote URL. - Review in a SEPARATE detached review worktree, never the author's folder. - Worktree safety (#233): before checkout, diff, validation, review, or merge, report the starting worktree path and whether it was dirty. If unrelated diff --git a/skills/llm-project-workflow/templates/start-issue.md b/skills/llm-project-workflow/templates/start-issue.md index 4b8c159..be09053 100644 --- a/skills/llm-project-workflow/templates/start-issue.md +++ b/skills/llm-project-workflow/templates/start-issue.md @@ -15,6 +15,10 @@ Rules (llm-project-workflow): - Work only in an isolated branch worktree under branches/. The main checkout is orchestration/status only. - Do not self-review or self-merge. +- Repository targeting (#530): pass explicit `remote=`, `org=`, and `repo=` on + every gitea-tools call (e.g. `remote=prgs org=Scaled-Tech-Consulting + repo=Gitea-Tools`). A bare `remote=prgs` can resolve to the wrong default repo + and is blocked when it disagrees with the local git remote URL. Steps: 0. Work Selection Rule — before any claim, branch, or file edits, acquire or diff --git a/skills/llm-project-workflow/templates/worktree-cleanup.md b/skills/llm-project-workflow/templates/worktree-cleanup.md index a9ffa95..4884525 100644 --- a/skills/llm-project-workflow/templates/worktree-cleanup.md +++ b/skills/llm-project-workflow/templates/worktree-cleanup.md @@ -26,3 +26,43 @@ Steps: Handoff: merge confirmed, issue closed, branch+worktree removed, checkout clean. ``` + +## Branches cleanup audit integrity (#404) + +Any bulk or multi-path cleanup under `branches/` must capture auditable before/after +identity for every initial directory and registered worktree. Use +`worktree_cleanup_audit.capture_cleanup_snapshot` before and after cleanup, record +every intentional removal in a removal log (path, method, order, timestamp, +pre-removal proof), then run `reconcile_cleanup_audit` and +`assess_cleanup_audit_integrity`. + +The cleanup report must include a reconciliation table: + +* initial count +* removed count +* preserved count +* missing-unexplained count +* final count + +Fail closed when: + +* a preserved (active PR, dirty, claim/lease, or unsafe) worktree disappears without + a removal log entry or explicit explanation +* the removal log omits a removed clean-stale path +* final counts do not reconcile with initial minus removed + +If another session removes or mutates a worktree during cleanup, record the path +under explained missing entries — never treat silent disappearance as success. + +## Bulk `branches/` cleanup audit (#404) + +Before removing multiple session-owned worktrees: + +1. Call `gitea_capture_branches_worktree_snapshot` and record the before snapshot. +2. Remove only paths classified as `clean_stale_removable` with explicit per-path proof. +3. Log every removal with path, method, and timestamp/order. +4. Capture an after snapshot with the same tool. +5. Call `gitea_assess_worktree_cleanup_integrity` with before, after, and the removal log. +6. Fail closed when any protected path (active PR, dirty, claim/lease) disappears + without an explained state transition. +7. Final report must include the reconciliation table and `git worktree list` proof. diff --git a/skills/llm-project-workflow/workflows/reconcile-landed-pr.md b/skills/llm-project-workflow/workflows/reconcile-landed-pr.md index 10ba60d..c58f3fe 100644 --- a/skills/llm-project-workflow/workflows/reconcile-landed-pr.md +++ b/skills/llm-project-workflow/workflows/reconcile-landed-pr.md @@ -304,6 +304,40 @@ If any required mutation capability is missing: * include safe next action (profile switch, human close, or dedicated reconciler profile) +## 15A. Audit vs cleanup phase (#419) + +Reconciliation audits are **read-only** unless a separate cleanup phase is +explicitly authorized. + +**Audit phase forbids** (``audit_reconciliation_mode.check_audit_mutation_allowed`` +fails closed): + +* ``gitea_delete_branch`` +* ``git branch -D`` +* ``git worktree remove`` +* pushes +* issue/PR mutations +* file edits + +Dry-run merged-cleanup reconciliation (``gitea_reconcile_merged_cleanups`` with +``dry_run=True``) stays in audit phase. Execution requires: + +1. Operator approval or workflow authorization +2. Exact ``delete_branch`` capability proof (``gitea.branch.delete``) +3. Proof branch/worktree is safe to remove +4. Before/after state snapshot + +Call ``gitea_authorize_reconciliation_cleanup_phase`` before any cleanup +mutation. Final reports must not claim ``no mutations`` if cleanup occurred. +Classify cleanup mutations as: + +* remote branch deletion → **External-state mutations** +* local branch deletion → **Git ref mutations** +* worktree removal → **Cleanup mutations** + +``audit_reconciliation_mode.assess_audit_reconciliation_report`` validates +these boundaries in final reports. + ## 16. Mutation classification Use precise mutation categories in the final report: @@ -341,6 +375,24 @@ Include: * confirmation that no normal review, approval, request-changes, or merge was performed +## 18A. Reconciler close proof is enforced (#306) + +When a reconciler run closes a PR, the final-report validator +(`final_report_validator` rule `reconcile.close_proof_fields`) **blocks** the +handoff unless it carries all four close proofs. The prompt is guidance; the +MCP validator is the authority. + +A close is detected from `PRs closed: #` (or a session close lock). Once a +close is reported, the handoff must include: + +* `Capabilities proven:` naming `gitea.pr.close` — the exact close capability +* `Ancestor proof:` — the landed/ancestor proof for the closed PR +* `PRs closed:` — the PR close result (the closed PR number) +* `Linked issue live status:` (or `Issues closed:`) — the linked-issue result + +Comment-only and blocked reconciliations (no PR close) are unaffected: the rule +returns no finding when nothing was closed. + ## 19. Local artifact and report consistency rule Do not create local walkthrough, notes, markdown, JSON, or report artifacts diff --git a/skills/llm-project-workflow/workflows/review-merge-pr.md b/skills/llm-project-workflow/workflows/review-merge-pr.md index 1c4e9be..5a90ea5 100644 --- a/skills/llm-project-workflow/workflows/review-merge-pr.md +++ b/skills/llm-project-workflow/workflows/review-merge-pr.md @@ -36,6 +36,44 @@ If available, load it first and report: If the canonical workflow cannot be loaded and the project requires it, stop and produce a recovery handoff only. +## 0A. Workflow-load and session boundary anchor (#403) + +The MCP gate is the authority — not local file viewing. + +Before any reviewer mutation: + +1. Record pre-review commands with `gitea_record_pre_review_command` when they + are not automatically classified (inventory/diagnostic commands may be + recorded explicitly for proof). +2. Call `gitea_load_review_workflow` to establish workflow hash proof **and** + session boundary state in the same in-process session proof. +3. Do not claim the workflow was loaded by reading + `skills/llm-project-workflow/workflows/review-merge-pr.md` as a local file; + that narrative does not satisfy the validator. + +Allowed before workflow load (classify as `read_only_inventory` or +`diagnostic`): + +* `gitea_whoami`, `gitea_resolve_task_capability`, `gitea_list_prs`, + `gitea_view_pr`, `gitea_get_runtime_context` +* `git fetch` / `git remote update` for inventory +* `git status`, `git worktree list` (read-only) + +Boundary violations (block downstream reviewer mutations even after load): + +* validation commands (`pytest`, `python -m unittest`) in the main checkout +* local profile/credential/config inspection (`profiles.json`, `gitea-mcp.json`, + `.env`, keychain dumps) +* MCP repair (`pkill`, MCP config edits) +* git mutations before workflow load + +Final reports must include a structured **Workflow-load helper result** block +copied from `gitea_load_review_workflow`, including at minimum: + +* `workflow_hash` +* `final_report_schema_hash` +* `boundary_status` (`clean` or `violation`) + ## 1. Start with live identity, profile, runtime, and capability checks Prove: @@ -796,6 +834,21 @@ inventory before continuing. Final reports must include lease session id, acquisition proof, heartbeat status, and release/blocked status. +## 26B-1. Merger lease adoption (#536) + +When review and merge are **different sessions**, the merger must adopt the +reviewer's lease — never manually seed `reviewer_pr_lease._SESSION_LEASE`. + +Before merge in a merger-only session: + +1. Confirm `approval_at_current_head` via `gitea_get_pr_review_feedback`. +2. Call `gitea_adopt_merger_pr_lease` from a clean merger worktree under + `branches/`, passing `expected_head_sha` pinned to the approved head. +3. Quote the adoption comment id and `adopted_from_session_id` in the handoff. +4. Proceed to `gitea_merge_pr` only after adoption succeeds. + +Manual in-process lease seeding is rejected by mutation gates (fail closed). + ## 26C. Conflict-fix lease and stale-head protection (#399) Before validating, approving, or merging a PR: @@ -871,6 +924,16 @@ Confirm: Clean only the session-owned `branches/` review worktree if the project workflow explicitly allows cleanup. +Review, baseline, and merge-simulation worktrees created during this run are +transient and are removed automatically at successful completion once they are +clean, carry no open PR, and hold no active lease (#401). Use +`gitea_audit_worktree_cleanup` (read-only) to classify `branches/` entries; only +`clean_stale_removable` and `detached_review_leftover` may be removed, one-by-one, +after `git worktree list` proof plus per-worktree proof of path, branch/HEAD, +clean/dirty status, no active PR/lease, and the removal result. Dirty, +active-PR, active-issue, and leased worktrees are never deleted automatically; a +failed removal must be reported with the leftover path and reason. + Do not delete or mutate unrelated branches/worktrees. Do not touch the main checkout except to update the stable branch after merge if explicitly allowed by the workflow. @@ -881,6 +944,61 @@ Do not update the main checkout if merge failed, was blocked, or produced reconc If any local artifact is created after final cleanup, run and report a new final status check. +## 28A. Post-merge cleanup proof checklist (#402) + +Successful tool execution is not proof that cleanup was authorized. Before claiming remote branch deletion or local worktree removal, the final report must carry the full safety checklist below. If any gate is missing, report `CLEANUP_SKIPPED` with the exact blocker — never perform cleanup and never claim it was performed. + +### Remote branch deletion checklist + +When `gitea_delete_branch` (or equivalent) deletes the merged PR head branch, report: + +* Delete-branch capability resolved: name the task (`delete_branch` / `cleanup_branch` / `reconcile_merged_cleanups`) and permission (`gitea.branch.delete`) with resolver proof before the delete call +* Merge result: merged +* Merge commit SHA: full 40-character SHA +* Merged PR head branch / deleted branch: exact branch name (must match) +* Branch protection: none / branch is not protected +* Open PR inventory proof: no other open PR references the branch +* Active heartbeat/claim/lease: none + +### Local worktree removal checklist + +When removing session-owned review/simulation worktrees under `branches/`, report: + +* Session-owned worktree path: exact path under `branches/` +* Pre-removal tracked state: clean +* Pre-removal untracked state: clean +* Git worktree list after removal: command output or equivalent proof + +### Skipped cleanup + +If any gate fails, report: + +* Cleanup outcome: `CLEANUP_SKIPPED` +* Cleanup blocker: exact missing gate (for example `gitea.branch.delete capability not resolved`) + +Skipped cleanup with an exact blocker passes validation. Performed-cleanup claims without the checklist fail validation. + +## 28B. MCP-native cleanup only (#517) + +Post-merge cleanup of leases, comments, branches, and worktrees must go through +explicit MCP tools — never raw git, curl/API scripts, or ad hoc helper scripts. + +Merger and reviewer sessions must **not** perform cleanup inline. Record: + +* **Merge mutations** — only `gitea_merge_pr` (or review mutations for review-only runs) +* **Cleanup mutations** — only authorized reconciler MCP tools, cited by exact tool name + +Authorized cleanup tools include `gitea_reconcile_merged_cleanups`, +`gitea_cleanup_post_merge_moot_lease`, `gitea_delete_branch`, and +`gitea_cleanup_merged_pr_branch` (when available). Lease cleanup uses +append-only release comments — never delete another session's lease comment. + +Hand cleanup to a `prgs-reconciler` session with `gitea.branch.delete` capability +proof. Raw `git branch -d`, `git push --delete`, and comment-deletion API calls +are blocked in final-report validation. + +Template: `templates/post-merge-cleanup-handoff.md`. + ## 29. Recovery handoff rules If blocked, produce a recovery handoff with: @@ -979,6 +1097,26 @@ Use precise wording: Do not collapse review, merge, cleanup, or external-state mutations into vague wording. +## 31B. Mutation-capability table (#405) + +Every performed mutation requires exact capability proof resolved **before** that +mutation executes. Nearby capabilities never authorize a different operation — +`review_pr` does not authorize `merge_pr`, and `merge_pr` does not authorize +`delete_branch` / `gitea.branch.delete`. + +When any mutation beyond a bare review occurs (merge, branch delete, issue +close/comment, etc.), the final report must include a **mutation-capability table** +with one row per performed mutation: + +* mutation (tool/action name) +* exact task/capability resolved (for example `merge_pr` / `gitea.pr.merge`) +* result +* order/timestamp proof that capability was resolved before the mutation + +If exact capability proof is missing, skip the mutation or stop the workflow — +never claim a performed mutation without its row. Post-hoc capability proof after +the mutation fails validation. + ## 31A. Local artifact and report consistency rule Do not create local walkthrough, notes, markdown, JSON, or report artifacts during reviewer runs unless the canonical workflow or operator explicitly requires it. diff --git a/skills/llm-project-workflow/workflows/work-issue.md b/skills/llm-project-workflow/workflows/work-issue.md index 6766523..c7f5747 100644 --- a/skills/llm-project-workflow/workflows/work-issue.md +++ b/skills/llm-project-workflow/workflows/work-issue.md @@ -626,9 +626,14 @@ When pushing to an existing PR branch to resolve merge conflicts: * session worktree path * push cwd * whether the push is fast-forward -3. Do not push when a reviewer holds an active lease on the same PR. -4. Do not force-push. -5. Do not push from the main checkout or wrong cwd. + * explicit `remote`, `org`, and `repo` when not using defaults +3. If assessment returns `assessment_failed: true` or `pr_lookup: failed`, stop + and produce a recovery handoff with the structured `reasons` and + `resolved_repo` fields — do not treat an MCP HTTP 500 as proof the push was + unsafe or safe (#519). +4. Do not push when a reviewer holds an active lease on the same PR. +5. Do not force-push. +6. Do not push from the main checkout or wrong cwd. Conflict-fix final reports must state: @@ -699,6 +704,39 @@ Do not update the main checkout unless the canonical workflow explicitly allows Any cleanup is a mutation and must be reported. +### 22A. Session-owned worktree cleanup and TTL (#401) + +Every session-owned worktree created under `branches/` has ownership metadata: +path, workflow type, issue number, PR number, branch/head SHA, creator +identity/profile, created timestamp, last-used timestamp, and cleanup +eligibility. + +Cleanup is classification-driven. `gitea_audit_worktree_cleanup` (read-only) +classifies every `branches/` entry as exactly one of: + +* `active_open_pr` — branch has an open PR; never auto-removed. +* `active_issue_work` — active claim/lease or fresh issue worktree; never + auto-removed. +* `dirty_local_worktree` — uncommitted changes; never auto-removed. +* `clean_stale_removable` — clean, no PR, no lease; removable. +* `detached_review_leftover` — clean detached review/baseline/merge-simulation + worktree; removable. +* `unsafe_unknown` — protected base checkout or unknown workflow type; never + auto-removed. + +Only `clean_stale_removable` and `detached_review_leftover` may be removed, and +only one-by-one after `git worktree list` proof plus per-worktree proof of: +worktree path, branch/HEAD, clean/dirty status, no active PR/lease, and the +removal result. Review, baseline, and merge-simulation worktrees are removed +automatically at successful workflow completion; issue/conflict-fix worktrees +are removed only after their TTL (`GITEA_WORKTREE_TTL_HOURS`, default 24h) +expires and no lock/lease is held. + +Dirty, active-PR, active-issue, and leased worktrees are never deleted +automatically. If a removal fails, the final report must list the leftover +worktree path and the reason. Include the `git worktree list` output as final +cleanup verification. + ## 23. Recovery handoff rules If blocked, produce a recovery handoff with: diff --git a/task_capability_map.py b/task_capability_map.py index 1d40a78..c86ce2a 100644 --- a/task_capability_map.py +++ b/task_capability_map.py @@ -68,6 +68,10 @@ TASK_CAPABILITY_MAP: dict[str, dict[str, str]] = { "permission": "gitea.pr.merge", "role": "reviewer", }, + "adopt_merger_pr_lease": { + "permission": "gitea.pr.comment", + "role": "reviewer", + }, "blind_pr_queue_review": { "permission": "gitea.pr.review", "role": "reviewer", @@ -104,6 +108,10 @@ TASK_CAPABILITY_MAP: dict[str, dict[str, str]] = { "permission": "gitea.read", "role": "author", }, + "reconciliation_cleanup": { + "permission": "gitea.branch.delete", + "role": "author", + }, "work_issue": { "permission": "gitea.pr.create", "role": "author", diff --git a/test_mcp_conn.py b/test_mcp_conn.py new file mode 100644 index 0000000..7481ce9 --- /dev/null +++ b/test_mcp_conn.py @@ -0,0 +1,129 @@ +#!/usr/bin/env python3 +"""Live health-check script to verify Gitea MCP namespace connections. + +Spawns the MCP server processes as defined in the IDE's global config, +performs the JSON-RPC handshake, and queries the tools list to verify +that the connection is fully operational and doesn't return EOF. +""" +import json +import os +import subprocess +import sys + +def test_connection(name, config): + print(f"Testing MCP connection for '{name}'...") + command = config.get("command") + args = config.get("args", []) + env = config.get("env", {}) + + # Merge current environment + run_env = os.environ.copy() + run_env.update(env) + + # Spawn subprocess + try: + proc = subprocess.Popen( + [command] + args, + stdin=subprocess.PIPE, + stdout=subprocess.PIPE, + stderr=subprocess.PIPE, + text=True, + bufsize=1, + env=run_env + ) + except Exception as e: + print(f" [FAIL] Failed to spawn process: {e}") + return False + + # Send initialize request + init_req = { + "jsonrpc": "2.0", + "method": "initialize", + "params": { + "protocolVersion": "2024-11-05", + "capabilities": {}, + "clientInfo": {"name": "healthcheck", "version": "1.0"} + }, + "id": 1 + } + + try: + proc.stdin.write(json.dumps(init_req) + "\n") + proc.stdin.flush() + + # Read response + line = proc.stdout.readline() + if not line: + stderr_content = proc.stderr.read() + print(f" [FAIL] Received EOF from process. Stderr:\n{stderr_content}") + proc.terminate() + return False + + print(f" [OK] Received initialize response: {line.strip()[:150]}...") + + # Send initialized notification + init_notif = { + "jsonrpc": "2.0", + "method": "notifications/initialized" + } + proc.stdin.write(json.dumps(init_notif) + "\n") + proc.stdin.flush() + + # Send tools/list request + list_req = { + "jsonrpc": "2.0", + "method": "tools/list", + "params": {}, + "id": 2 + } + proc.stdin.write(json.dumps(list_req) + "\n") + proc.stdin.flush() + + line = proc.stdout.readline() + if not line: + print(" [FAIL] Received EOF on tools/list request.") + proc.terminate() + return False + + res = json.loads(line) + if "error" in res: + print(f" [FAIL] Server returned error: {res['error']}") + proc.terminate() + return False + + tools = res.get("result", {}).get("tools", []) + tool_names = [t.get("name") for t in tools] + print(f" [OK] Successfully retrieved {len(tool_names)} tools: {tool_names[:5]}...") + + proc.terminate() + return True + except Exception as e: + print(f" [FAIL] Error during handshake: {e}") + proc.terminate() + return False + +def main(): + config_path = "/Users/jasonwalker/.gemini/config/mcp_config.json" + try: + with open(config_path) as f: + mcp_config = json.load(f) + except Exception as e: + print(f"Failed to load mcp_config.json: {e}") + sys.exit(1) + + servers = mcp_config.get("mcpServers", {}) + failed = False + for name in ["gitea-author", "gitea-reviewer"]: + if name in servers: + if not test_connection(name, servers[name]): + failed = True + else: + print(f"Server '{name}' not found in mcp_config.json") + + if failed: + sys.exit(1) + else: + print("All Gitea MCP connection tests passed!") + +if __name__ == "__main__": + main() diff --git a/tests/conftest.py b/tests/conftest.py index 7bbf78c..1c34986 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -18,14 +18,25 @@ def _reset_mutation_authority(monkeypatch): explicitly. """ monkeypatch.delenv("GITEA_SESSION_PROFILE_LOCK", raising=False) + # Isolate durable session-state files so tests never share host cache (#559). + import tempfile + + _state_tmp = tempfile.TemporaryDirectory(prefix="gitea-session-state-") + monkeypatch.setenv("GITEA_MCP_SESSION_STATE_DIR", _state_tmp.name) try: import mcp_server except Exception: + _state_tmp.cleanup() yield return monkeypatch.setattr(mcp_server, "_MUTATION_AUTHORITY", None) monkeypatch.setattr(mcp_server, "_IDENTITY_CACHE", {}) monkeypatch.setattr(mcp_server, "_REVIEW_DECISION_LOCK", None) + try: + import review_workflow_load + review_workflow_load._REVIEW_WORKFLOW_LOAD = None + except Exception: + pass try: import capability_stop_terminal capability_stop_terminal.clear() @@ -37,3 +48,13 @@ def _reset_mutation_authority(monkeypatch): capability_stop_terminal.clear() except Exception: pass + try: + import review_workflow_load + review_workflow_load._REVIEW_WORKFLOW_LOAD = None + except Exception: + pass + try: + mcp_server._REVIEW_DECISION_LOCK = None + except Exception: + pass + _state_tmp.cleanup() diff --git a/tests/test_assess_conflict_fix_push_tool.py b/tests/test_assess_conflict_fix_push_tool.py new file mode 100644 index 0000000..36166cd --- /dev/null +++ b/tests/test_assess_conflict_fix_push_tool.py @@ -0,0 +1,139 @@ +"""Regression tests for gitea_assess_conflict_fix_push structured failures (#519).""" + +import os +import sys +import unittest +from unittest.mock import patch + +sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent)) + +from mcp_server import ( # noqa: E402 + _fetch_pr_lease_comments_safe, + gitea_assess_conflict_fix_push, +) + +FAKE_AUTH = "Basic dGVzdDp0ZXN0" +AUTHOR_ENV = { + "GITEA_PROFILE_NAME": "prgs-author", + "GITEA_ALLOWED_OPERATIONS": "gitea.read,gitea.branch.push,gitea.pr.create", +} + +HEAD_BEFORE = "dad1dc8d5108ab01ed83065334116d7425a4471c" +HEAD_AFTER = "3f3d6cb35d0fe225dcb236247f2a2d0ec193fa35" +OPEN_PR = { + "number": 508, + "state": "open", + "head": {"sha": HEAD_AFTER}, +} + + +class TestFetchPrLeaseCommentsSafe(unittest.TestCase): + @patch("mcp_server.api_request") + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_pr_lookup_404_returns_structured_failure(self, _auth, mock_api): + mock_api.side_effect = RuntimeError( + 'HTTP 404: {"message":"issue does not exist","index":508}' + ) + result = _fetch_pr_lease_comments_safe( + 508, remote="prgs", host=None, org=None, repo=None) + self.assertFalse(result["success"]) + self.assertEqual(result["comments"], []) + self.assertTrue(result["reasons"]) + self.assertIn("PR lookup failed", result["reasons"][0]) + self.assertEqual(result["pr_lookup"], "failed") + + @patch("mcp_server.api_request") + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_comment_fetch_404_after_pr_lookup(self, _auth, mock_api): + def _api(method, url, auth, *args, **kwargs): + if "/pulls/" in url: + return OPEN_PR + raise RuntimeError( + 'HTTP 404: {"message":"issue does not exist","index":508}' + ) + + mock_api.side_effect = _api + result = _fetch_pr_lease_comments_safe( + 508, remote="prgs", host=None, org=None, repo=None) + self.assertFalse(result["success"]) + self.assertIn("comment fetch failed", result["reasons"][0].lower()) + self.assertEqual(result["pr_lookup"], "ok") + + @patch("mcp_server.api_request") + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_success_returns_comments_and_head_sha(self, _auth, mock_api): + comment = {"id": 1, "body": ""} + + def _api(method, url, auth, *args, **kwargs): + if "/pulls/" in url: + return OPEN_PR + return [comment] + + mock_api.side_effect = _api + result = _fetch_pr_lease_comments_safe( + 508, remote="prgs", host=None, org=None, repo=None) + self.assertTrue(result["success"]) + self.assertEqual(result["comments"], [comment]) + self.assertEqual(result["head_sha"], OPEN_PR["head"]["sha"]) + + +class TestAssessConflictFixPushTool(unittest.TestCase): + @patch("mcp_server._fetch_pr_lease_comments_safe") + @patch("mcp_server._profile_operation_gate", return_value=None) + def test_returns_structured_block_when_pr_lookup_fails( + self, _gate, mock_fetch, + ): + mock_fetch.return_value = { + "success": False, + "comments": [], + "reasons": ["PR lookup failed"], + "pr_lookup": "failed", + "resolved_repo": "Scaled-Tech-Consulting/Gitea-Tools", + "remote": "prgs", + "pr_number": 508, + } + with patch.dict(os.environ, AUTHOR_ENV, clear=True): + result = gitea_assess_conflict_fix_push( + pr_number=508, + branch_head_before=HEAD_BEFORE, + branch_head_after=HEAD_AFTER, + worktree_path="/proj/branches/fix-pr508", + push_cwd="/proj/branches/fix-pr508", + is_fast_forward=True, + remote="prgs", + ) + self.assertFalse(result["push_allowed"]) + self.assertTrue(result.get("assessment_failed")) + self.assertEqual(result["pr_lookup"], "failed") + + @patch("mcp_server._fetch_pr_lease_comments_safe") + @patch("mcp_server._profile_operation_gate", return_value=None) + def test_valid_push_assessment_when_pr_and_comments_resolve( + self, _gate, mock_fetch, + ): + mock_fetch.return_value = { + "success": True, + "comments": [], + "reasons": [], + "pr_lookup": "ok", + "resolved_repo": "Scaled-Tech-Consulting/Gitea-Tools", + "remote": "prgs", + "pr_number": 508, + "head_sha": HEAD_AFTER, + } + with patch.dict(os.environ, AUTHOR_ENV, clear=True): + result = gitea_assess_conflict_fix_push( + pr_number=508, + branch_head_before=HEAD_BEFORE, + branch_head_after=HEAD_AFTER, + worktree_path="/proj/branches/fix-pr508", + push_cwd="/proj/branches/fix-pr508", + is_fast_forward=True, + remote="prgs", + ) + self.assertTrue(result["push_allowed"]) + self.assertEqual(result.get("live_pr_head_sha"), HEAD_AFTER) + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_audit.py b/tests/test_audit.py index b233b15..a44885a 100644 --- a/tests/test_audit.py +++ b/tests/test_audit.py @@ -157,6 +157,7 @@ class _AuditWiringBase(unittest.TestCase): def tearDown(self): mcp_server._IDENTITY_CACHE.clear() + mcp_server.review_workflow_load.clear_review_workflow_load() self._dir.cleanup() def _env(self, **extra): @@ -291,12 +292,15 @@ class TestGatedToolAudit(_AuditWiringBase): def setUp(self): super().setUp() - from mcp_server import init_review_decision_lock - from tests.test_mcp_server import _install_owned_reviewer_lease + from tests.test_mcp_server import _init_reviewer_session, _install_owned_reviewer_lease import reviewer_pr_lease + import review_workflow_boundary + import review_workflow_load - # init_review_decision_lock clears any prior session lease (#407). - init_review_decision_lock("prgs", "review_pr") + # Session init clears any prior session lease (#407) and loads workflow (#389). + _init_reviewer_session("prgs") + self.addCleanup(review_workflow_load.clear_review_workflow_load) + self.addCleanup(review_workflow_boundary.clear_pre_review_commands) self._lease_patch = _install_owned_reviewer_lease(8) self._lease_patch.start() self._auth_identity_patch = patch( @@ -337,6 +341,7 @@ class TestGatedToolAudit(_AuditWiringBase): env = self._env(GITEA_PROFILE_NAME="gitea-merger", GITEA_ALLOWED_OPERATIONS="read,merge") with patch.dict(os.environ, env, clear=True): + mcp_server.gitea_load_review_workflow() r = gitea_merge_pr(pr_number=8, confirmation="MERGE PR 8", expected_head_sha="abc123", remote="prgs") self.assertTrue(r["performed"]) @@ -356,6 +361,7 @@ class TestGatedToolAudit(_AuditWiringBase): env = self._env(GITEA_PROFILE_NAME="gitea-merger", GITEA_ALLOWED_OPERATIONS="read,merge") with patch.dict(os.environ, env, clear=True): + mcp_server.gitea_load_review_workflow() r = gitea_merge_pr(pr_number=8, confirmation="MERGE PR 8", remote="prgs") self.assertFalse(r["performed"]) recs = self._records() diff --git a/tests/test_audit_reconciliation_mode.py b/tests/test_audit_reconciliation_mode.py new file mode 100644 index 0000000..a117fab --- /dev/null +++ b/tests/test_audit_reconciliation_mode.py @@ -0,0 +1,291 @@ +"""Tests for audit vs cleanup reconciliation mode (#419).""" +import os +import sys +import unittest +from unittest.mock import patch + +sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent)) + +import audit_reconciliation_mode as arm +import mcp_server +from audit_reconciliation_mode import ( + AUDIT_FORBIDDEN_TASKS, + PHASE_AUDIT, + PHASE_CLEANUP, + assess_audit_command_allowed, + assess_audit_reconciliation_report, + authorize_cleanup_phase, + check_audit_mutation_allowed, + check_cleanup_execution_allowed, + classify_cleanup_mutation, + clear_phase, + enter_audit_phase, +) +from final_report_validator import assess_final_report_validator +from review_proofs import assess_audit_reconciliation_report as proofs_assess +from task_capability_map import required_permission, required_role + +DELETE_PROFILE = { + "profile_name": "prgs-author-delete", + "allowed_operations": ["gitea.read", "gitea.branch.delete"], + "forbidden_operations": [], + "audit_label": "prgs-author-delete", +} + +READ_PROFILE = { + "profile_name": "prgs-author", + "allowed_operations": ["gitea.read", "gitea.issue.comment"], + "forbidden_operations": ["gitea.branch.delete"], + "audit_label": "prgs-author", +} + +READ_ENV = { + "GITEA_MCP_CONFIG": os.path.join( + os.path.dirname(__file__), "..", "profiles.json" + ), + "GITEA_MCP_PROFILE": "prgs-author", +} + + +def _authorize_cleanup(**kwargs): + defaults = { + "operator_approved": True, + "delete_capability_proven": True, + "safety_proof": {"safe_to_delete_remote": True}, + "before_after_snapshot": { + "before": "remote branch exists", + "after": "remote branch absent", + }, + } + defaults.update(kwargs) + return authorize_cleanup_phase(**defaults) + + +def _cleanup_report(**overrides): + base = ( + "Task mode: reconcile-landed-pr\n" + "Workflow source: workflows/reconcile-landed-pr.md\n" + "Audit phase: read-only assessment\n" + "Cleanup phase authorized: true\n" + "Delete-branch capability proven: true\n" + "Branch safe to remove: true\n" + "Before/after state snapshot: remote branch feat/x present → absent\n" + "External-state mutations: deleted remote branch feat/x\n" + "Git ref mutations: none\n" + "Cleanup mutations: removed worktree branches/feat-x\n" + ) + for key, value in overrides.items(): + base = base.replace(key, value) + return base + + +class TestAuditPhaseGates(unittest.TestCase): + def setUp(self): + clear_phase() + enter_audit_phase("reconcile-landed-pr") + + def tearDown(self): + clear_phase() + + def test_audit_blocks_delete_branch_task(self): + allowed, reasons = check_audit_mutation_allowed("delete_branch") + self.assertFalse(allowed) + self.assertTrue(reasons) + + def test_audit_blocks_worktree_shell_commands(self): + allowed, reasons = assess_audit_command_allowed( + "git worktree remove branches/feat-x" + ) + self.assertFalse(allowed) + self.assertTrue(reasons) + + def test_audit_blocks_local_branch_delete_command(self): + allowed, reasons = assess_audit_command_allowed("git branch -D feat/x") + self.assertFalse(allowed) + + def test_audit_allows_read_tasks(self): + allowed, _ = check_audit_mutation_allowed("reconcile_landed_pr") + self.assertTrue(allowed) + + def test_forbidden_set_covers_issue_and_pr_mutations(self): + for task in ("close_pr", "comment_issue", "commit_files", "push_branch"): + self.assertIn(task, AUDIT_FORBIDDEN_TASKS) + + +class TestCleanupAuthorization(unittest.TestCase): + def setUp(self): + clear_phase() + enter_audit_phase("reconcile_merged_cleanups") + + def tearDown(self): + clear_phase() + + def test_cleanup_without_approval_blocked(self): + result = authorize_cleanup_phase( + delete_capability_proven=True, + safety_proof={"safe_to_delete_remote": True}, + before_after_snapshot={"before": "a", "after": "b"}, + ) + self.assertFalse(result["authorized"]) + + def test_cleanup_without_capability_proof_blocked(self): + result = _authorize_cleanup(delete_capability_proven=False) + self.assertFalse(result["authorized"]) + + def test_cleanup_without_snapshot_blocked(self): + result = _authorize_cleanup(before_after_snapshot={"before": "", "after": ""}) + self.assertFalse(result["authorized"]) + + def test_authorized_cleanup_switches_phase(self): + result = _authorize_cleanup() + self.assertTrue(result["authorized"]) + self.assertEqual(result["phase"], PHASE_CLEANUP) + + def test_cleanup_execution_allowed_only_after_authorization(self): + self.assertFalse(check_cleanup_execution_allowed()[0]) + _authorize_cleanup() + self.assertTrue(check_cleanup_execution_allowed()[0]) + + +class TestReportVerifier(unittest.TestCase): + def test_false_no_mutations_after_cleanup_blocked(self): + report = ( + "Task mode: reconcile-landed-pr\n" + "No mutations performed.\n" + "delete_remote_branch feat/dup\n" + ) + result = assess_audit_reconciliation_report(report) + self.assertFalse(result["proven"]) + self.assertIn("no mutations", result["reasons"][0].lower()) + + def test_cleanup_without_authorization_fields_blocked(self): + report = ( + "Task mode: reconcile-landed-pr\n" + "remove_local_worktree branches/feat-x\n" + ) + result = assess_audit_reconciliation_report(report) + self.assertFalse(result["proven"]) + + def test_authorized_cleanup_report_passes(self): + result = assess_audit_reconciliation_report(_cleanup_report()) + self.assertTrue(result["proven"]) + + def test_mutation_classification_enforced(self): + report = ( + "Task mode: reconcile-landed-pr\n" + "Cleanup phase authorized: true\n" + "Delete-branch capability proven: true\n" + "Branch safe to remove: true\n" + "Before/after state snapshot: present\n" + "remove_local_worktree branches/feat-x\n" + ) + result = assess_audit_reconciliation_report(report) + self.assertFalse(result["proven"]) + + def test_proofs_export_matches_module(self): + report = "No mutations performed.\ndelete_remote_branch feat/dup" + self.assertEqual( + proofs_assess(report)["proven"], + assess_audit_reconciliation_report(report)["proven"], + ) + + def test_final_report_validator_includes_boundary_rule(self): + report = "No mutations performed.\ndelete_remote_branch feat/dup" + result = assess_final_report_validator( + report_text=report, + task_kind="reconcile_already_landed", + ) + self.assertTrue(result["blocked"]) + rule_ids = [f["rule_id"] for f in result["findings"]] + self.assertIn("reconcile.audit_cleanup_boundary", rule_ids) + + +class TestMutationClassification(unittest.TestCase): + def test_remote_delete_is_external_state(self): + self.assertEqual( + classify_cleanup_mutation("delete_remote_branch"), + "external-state", + ) + + def test_worktree_remove_is_cleanup(self): + self.assertEqual( + classify_cleanup_mutation("remove_local_worktree"), + "cleanup", + ) + + +class TestMcpGates(unittest.TestCase): + def setUp(self): + clear_phase() + enter_audit_phase("reconcile_merged_cleanups") + self.mock_api = patch("mcp_server.api_request").start() + self.mock_auth = patch( + "mcp_server.get_auth_header", return_value="token test" + ).start() + + def tearDown(self): + patch.stopall() + clear_phase() + mcp_server._IDENTITY_CACHE.clear() + + @patch.dict(os.environ, READ_ENV, clear=True) + @patch("mcp_server.get_profile", return_value=DELETE_PROFILE) + def test_delete_branch_blocked_in_audit_phase(self, _profile): + mcp_server.record_preflight_check("whoami") + mcp_server.record_preflight_check("capability", resolved_role="author") + result = mcp_server.gitea_delete_branch(branch="feat/dup", remote="prgs") + self.assertFalse(result["success"]) + self.assertEqual(result["audit_phase"], PHASE_AUDIT) + self.mock_api.assert_not_called() + + @patch.dict(os.environ, READ_ENV, clear=True) + @patch("mcp_server.get_profile", return_value=DELETE_PROFILE) + def test_reconcile_execute_blocked_without_cleanup_auth(self, _profile): + with self.assertRaises(ValueError): + mcp_server.gitea_reconcile_merged_cleanups( + dry_run=False, + execute_confirmed=False, + remote="prgs", + ) + + @patch.dict(os.environ, READ_ENV, clear=True) + @patch("mcp_server.get_profile", return_value=READ_PROFILE) + def test_cleanup_auth_fails_without_delete_capability(self, _profile): + result = mcp_server.gitea_authorize_reconciliation_cleanup_phase( + operator_approved=True, + delete_capability_proven=True, + safe_to_delete_remote=True, + before_state="exists", + after_state="gone", + ) + self.assertFalse(result["authorized"]) + self.assertFalse(result["delete_capability_verified"]) + + @patch.dict(os.environ, READ_ENV, clear=True) + @patch("mcp_server.get_profile", return_value=DELETE_PROFILE) + def test_delete_branch_allowed_after_cleanup_authorization(self, _profile): + mcp_server.gitea_authorize_reconciliation_cleanup_phase( + operator_approved=True, + delete_capability_proven=True, + safe_to_delete_remote=True, + before_state="exists", + after_state="gone", + ) + mcp_server.record_preflight_check("whoami") + mcp_server.record_preflight_check("capability", resolved_role="author") + self.mock_api.return_value = {} + result = mcp_server.gitea_delete_branch(branch="feat/dup", remote="prgs") + self.assertTrue(result["success"]) + + +class TestTaskCapabilityMap(unittest.TestCase): + def test_reconciliation_cleanup_maps_delete_permission(self): + self.assertEqual( + required_permission("reconciliation_cleanup"), + "gitea.branch.delete", + ) + self.assertEqual(required_role("reconciliation_cleanup"), "author") + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_create_issue_workspace_guard.py b/tests/test_create_issue_workspace_guard.py index 8d1d1f6..7c8a2e8 100644 --- a/tests/test_create_issue_workspace_guard.py +++ b/tests/test_create_issue_workspace_guard.py @@ -38,8 +38,18 @@ class TestCreateIssueWorkspaceGuard(unittest.TestCase): @patch("gitea_mcp_server.role_session_router.check_author_mutation_after_reviewer_stop", return_value=(True, [])) @patch("gitea_mcp_server.api_request") @patch("gitea_mcp_server.api_get_all", return_value=[]) - @patch("gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", return_value={"current_branch": "feat/issue-1"}) - def test_create_issue_stable_checkout_rejected(self, _git, _get_all, mock_api, _role, _ns, _prof, _auth): + @patch("gitea_mcp_server.root_checkout_guard.resolve_remote_master_sha", return_value="a" * 40) + @patch( + "gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", + return_value={ + "current_branch": "master", + "head_sha": "a" * 40, + "porcelain_status": "", + }, + ) + def test_create_issue_stable_checkout_rejected( + self, _git, _remote_sha, _get_all, mock_api, _role, _ns, _prof, _auth, + ): # Without worktree_path/env hints, workspace resolves to PROJECT_ROOT. When that # path is the stable control checkout (not under branches/), mutation must fail. with patch.object(srv, "PROJECT_ROOT", CONTROL_CHECKOUT_ROOT): diff --git a/tests/test_final_report_validator.py b/tests/test_final_report_validator.py index 0de0a62..4701458 100644 --- a/tests/test_final_report_validator.py +++ b/tests/test_final_report_validator.py @@ -504,6 +504,81 @@ class TestCanonicalReconcileSchema(unittest.TestCase): ) +class TestReconcilerCloseProof(unittest.TestCase): + """Issue #306: a reconciler PR close must carry its proof fields.""" + + def _closed_report(self, **kwargs): + # A report that claims PR #99 was closed via the reconciler path. + report = ( + _reconcile_handoff(**kwargs) + .replace( + "- Capabilities proven: gitea.read, gitea.pr.comment", + "- Capabilities proven: gitea.read, gitea.pr.comment, gitea.pr.close", + ) + .replace("- Missing capabilities: gitea.pr.close", "- Missing capabilities: none") + .replace("- PRs closed: none", "- PRs closed: #99") + .replace( + "- Blocker: missing gitea.pr.close capability", "- Blocker: none" + ) + ) + return report + + def test_full_proof_close_passes(self): + result = assess_final_report_validator( + self._closed_report(), "reconcile_already_landed" + ) + self.assertEqual(result["grade"], "A") + self.assertFalse( + any(f["rule_id"] == "reconcile.close_proof_fields" for f in result["findings"]) + ) + + def test_close_without_capability_proof_blocks(self): + # Claims PRs closed: #99 but never proves gitea.pr.close capability. + report = _reconcile_handoff().replace("- PRs closed: none", "- PRs closed: #99") + result = assess_final_report_validator(report, "reconcile_already_landed") + self.assertTrue(result["blocked"]) + self.assertTrue( + any(f["rule_id"] == "reconcile.close_proof_fields" for f in result["findings"]) + ) + + def test_close_without_ancestor_proof_blocks(self): + report = self._closed_report(drop=("Ancestor proof",)) + result = assess_final_report_validator(report, "reconcile_already_landed") + self.assertTrue(result["blocked"]) + self.assertTrue( + any(f["rule_id"] == "reconcile.close_proof_fields" for f in result["findings"]) + ) + + def test_close_without_linked_issue_result_blocks(self): + report = self._closed_report(drop=("Linked issue live status", "Issues closed")) + result = assess_final_report_validator(report, "reconcile_already_landed") + self.assertTrue(result["blocked"]) + self.assertTrue( + any(f["rule_id"] == "reconcile.close_proof_fields" for f in result["findings"]) + ) + + def test_close_lock_requires_proof_even_if_text_says_none(self): + # Session lock proves a PR was closed; the report omits close proof. + result = assess_final_report_validator( + _reconcile_handoff(), + "reconcile_already_landed", + reconciler_close_lock={"pr_closed": True}, + ) + self.assertTrue(result["blocked"]) + self.assertTrue( + any(f["rule_id"] == "reconcile.close_proof_fields" for f in result["findings"]) + ) + + def test_comment_only_reconcile_needs_no_close_proof(self): + result = assess_final_report_validator( + _reconcile_handoff(), "reconcile_already_landed" + ) + self.assertEqual(result["grade"], "A") + self.assertFalse( + any(f["rule_id"] == "reconcile.close_proof_fields" for f in result["findings"]) + ) + + class TestEntryPoint(unittest.TestCase): def test_unknown_task_kind_blocks(self): result = assess_final_report_validator("report", "unknown_mode") diff --git a/tests/test_issue_540_comment_role_poison.py b/tests/test_issue_540_comment_role_poison.py new file mode 100644 index 0000000..6cc6122 --- /dev/null +++ b/tests/test_issue_540_comment_role_poison.py @@ -0,0 +1,271 @@ +"""Regression tests for #540: comment_issue preflight must not poison the +actual reconciler role for #274 branch-only / #475 root-checkout exemptions. + +`resolve_task_capability("comment_issue")` stamps +``_preflight_resolved_role = "author"`` because ``comment_issue`` has +``required_role_kind = author``. Before the fix, ``_effective_workspace_role`` +preferred that stamp and a genuine ``prgs-reconciler`` session was treated as an +author inside the #274 branch-only mutation guard and the #475 root checkout +guard, blocking ``gitea_create_issue_comment`` from the control checkout. + +The fix keys the role exemptions off the *actual profile role* as well, so the +exemption survives the poisoned task role while author profiles stay blocked. +""" +from __future__ import annotations + +import os +import sys +import unittest +from pathlib import Path +from unittest.mock import patch + +sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) + +import gitea_mcp_server as srv # noqa: E402 +import root_checkout_guard as rcg # noqa: E402 + +FAKE_AUTH = "token test" +CONTROL_CHECKOUT_ROOT = str(Path(__file__).resolve().parents[3]) +MASTER_SHA = "a" * 40 +OTHER_SHA = "b" * 40 + +RECONCILER_PROFILE = { + "profile_name": "prgs-reconciler", + "allowed_operations": [ + "gitea.read", + "gitea.pr.close", + "gitea.pr.comment", + "gitea.issue.comment", + ], + "forbidden_operations": [ + "gitea.pr.approve", + "gitea.pr.merge", + "gitea.pr.create", + "gitea.branch.push", + "gitea.repo.commit", + ], + "audit_label": "prgs-reconciler", +} + +AUTHOR_PROFILE = { + "profile_name": "prgs-author", + "allowed_operations": [ + "gitea.read", + "gitea.issue.comment", + "gitea.issue.create", + "gitea.pr.create", + "gitea.branch.push", + "gitea.repo.commit", + ], + "forbidden_operations": [ + "gitea.pr.approve", + "gitea.pr.merge", + ], + "audit_label": "prgs-author", +} + +REVIEWER_PROFILE = { + "profile_name": "prgs-reviewer", + "allowed_operations": ["gitea.read", "gitea.pr.approve", "gitea.pr.merge"], + "forbidden_operations": ["gitea.pr.create", "gitea.branch.push"], + "audit_label": "prgs-reviewer", +} + + +class TestActualProfileRole(unittest.TestCase): + """`_actual_profile_role` ignores the poisoned preflight task role (#540).""" + + def tearDown(self): + srv._preflight_resolved_role = None + + @patch("gitea_mcp_server.get_profile", return_value=RECONCILER_PROFILE) + def test_reconciler_profile_role_survives_author_task_stamp(self, _profile): + srv._preflight_resolved_role = "author" # comment_issue poison + self.assertEqual(srv._actual_profile_role(), "reconciler") + # _effective_workspace_role is still poisoned to author (unchanged #510)... + self.assertEqual(srv._effective_workspace_role(), "author") + + @patch("gitea_mcp_server.get_profile", return_value=AUTHOR_PROFILE) + def test_author_profile_role_is_author(self, _profile): + srv._preflight_resolved_role = "author" + self.assertEqual(srv._actual_profile_role(), "author") + + @patch("gitea_mcp_server.get_profile", return_value=REVIEWER_PROFILE) + def test_reviewer_profile_role_is_reviewer(self, _profile): + srv._preflight_resolved_role = "author" + self.assertEqual(srv._actual_profile_role(), "reviewer") + + +class TestBranchesOnlyExemptionRealRole(unittest.TestCase): + """#274 branches-only exemption keys off the actual profile role (#540).""" + + def tearDown(self): + srv._preflight_resolved_role = None + + @patch("gitea_mcp_server.get_profile", return_value=RECONCILER_PROFILE) + def test_reconciler_exempt_despite_author_task_stamp(self, _profile): + srv._preflight_resolved_role = "author" # poison + # Must return without raising and without resolving an author worktree. + with patch("gitea_mcp_server._resolve_namespace_mutation_context") as ctx: + srv._enforce_branches_only_author_mutation() + ctx.assert_not_called() + + @patch("gitea_mcp_server.get_profile", return_value=AUTHOR_PROFILE) + def test_author_not_exempt(self, _profile): + srv._preflight_resolved_role = "author" + sentinel = RuntimeError("author-mutation-guard-reached") + + def _blow_up(*_a, **_k): + raise sentinel + + # Author is not exempt: the guard proceeds to resolve/assess the + # workspace (proven by reaching the patched context resolver). + with patch( + "gitea_mcp_server._resolve_namespace_mutation_context", + side_effect=_blow_up, + ): + with self.assertRaises(RuntimeError) as raised: + srv._enforce_branches_only_author_mutation() + self.assertIs(raised.exception, sentinel) + + +class TestRootCheckoutGuardRealRole(unittest.TestCase): + """#475 root guard honours the actual profile role too (#540).""" + + def _assess(self, **kwargs): + defaults = { + "workspace_path": CONTROL_CHECKOUT_ROOT, + "canonical_repo_root": CONTROL_CHECKOUT_ROOT, + "current_branch": "feat/some-branch", + "head_sha": OTHER_SHA, + "porcelain_status": " M gitea_mcp_server.py\n", + "remote_master_sha": MASTER_SHA, + "resolved_role": "author", # poisoned task role + } + defaults.update(kwargs) + return rcg.assess_root_checkout_guard(**defaults) + + def test_actual_reconciler_exempt_despite_poisoned_resolved_author(self): + result = self._assess(actual_role="reconciler") + self.assertTrue(result["proven"]) + self.assertFalse(result["block"]) + + def test_actual_author_still_blocked_on_contaminated_root(self): + result = self._assess(actual_role="author") + self.assertTrue(result["block"]) + + def test_merger_strictness_stays_keyed_on_resolved_task_role(self): + # When the merge task resolves the merger role, the branches/ auto + # exemption is denied and a clean control checkout is required. + blocked = self._assess( + workspace_path=f"{CONTROL_CHECKOUT_ROOT}/branches/review-pr-1", + current_branch="review-pr-1", + porcelain_status="", + head_sha=OTHER_SHA, + resolved_role="merger", + ) + self.assertTrue(blocked["block"]) + + def test_actual_merger_does_not_over_tighten_non_merge_task(self): + # A merger profile whose current task did NOT resolve to merger keeps + # the branches/ workspace exemption (regression guard for #540): the + # actual_role signal must not force merger strictness here. + result = self._assess( + workspace_path=f"{CONTROL_CHECKOUT_ROOT}/branches/review-pr-1", + current_branch="review-pr-1", + porcelain_status="", + head_sha=OTHER_SHA, + resolved_role="reviewer", + actual_role="merger", + ) + self.assertTrue(result["proven"]) + self.assertFalse(result["block"]) + + def test_backward_compatible_without_actual_role(self): + # No actual_role supplied: behaviour falls back to resolved_role. + result = self._assess(resolved_role="reconciler") + self.assertTrue(result["proven"]) + + +class TestReconcilerCommentThroughCanonicalPath(unittest.TestCase): + """Integration: reconciler comment survives the poisoned author task role.""" + + def setUp(self): + srv._preflight_whoami_called = True + srv._preflight_capability_called = True + srv._preflight_whoami_violation = False + srv._preflight_capability_violation = False + srv._preflight_capability_baseline_porcelain = "" + self._orig_in_test = srv._preflight_in_test_mode + srv._preflight_in_test_mode = lambda: False + + def tearDown(self): + srv._preflight_in_test_mode = self._orig_in_test + srv._preflight_resolved_role = None + + @patch("gitea_mcp_server._get_workspace_porcelain", return_value="") + @patch("gitea_mcp_server._auth", return_value=FAKE_AUTH) + @patch("gitea_mcp_server.api_request") + @patch( + "gitea_mcp_server.root_checkout_guard.resolve_remote_master_sha", + return_value=MASTER_SHA, + ) + @patch( + "gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", + return_value={ + "current_branch": "master", + "head_sha": MASTER_SHA, + "porcelain_status": "", + }, + ) + @patch("gitea_mcp_server.get_profile", return_value=RECONCILER_PROFILE) + def test_reconciler_comment_from_control_checkout_succeeds( + self, _profile, _git, _remote_sha, mock_api, _auth, _porcelain + ): + srv._preflight_resolved_role = "author" # comment_issue poison + mock_api.return_value = {"id": 9001, "html_url": "https://x/y"} + with patch.object(srv, "PROJECT_ROOT", CONTROL_CHECKOUT_ROOT): + with patch.dict(os.environ, {}, clear=False): + os.environ.pop("GITEA_AUTHOR_WORKTREE", None) + os.environ.pop("GITEA_ACTIVE_WORKTREE", None) + os.environ.pop("GITEA_RECONCILER_WORKTREE", None) + result = srv.gitea_create_issue_comment( + 515, "canonical reconciler audit", remote="prgs" + ) + self.assertTrue(result["success"]) + self.assertEqual(result["comment_id"], 9001) + mock_api.assert_called_once() + + @patch("gitea_mcp_server._get_workspace_porcelain", return_value="") + @patch("gitea_mcp_server._auth", return_value=FAKE_AUTH) + @patch("gitea_mcp_server.api_request") + @patch( + "gitea_mcp_server.root_checkout_guard.resolve_remote_master_sha", + return_value=MASTER_SHA, + ) + @patch( + "gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", + return_value={ + "current_branch": "master", + "head_sha": MASTER_SHA, + "porcelain_status": "", + }, + ) + @patch("gitea_mcp_server.get_profile", return_value=AUTHOR_PROFILE) + def test_author_comment_from_control_checkout_blocked( + self, _profile, _git, _remote_sha, mock_api, _auth, _porcelain + ): + srv._preflight_resolved_role = "author" + with patch.object(srv, "PROJECT_ROOT", CONTROL_CHECKOUT_ROOT): + with patch.dict(os.environ, {}, clear=False): + os.environ.pop("GITEA_AUTHOR_WORKTREE", None) + os.environ.pop("GITEA_ACTIVE_WORKTREE", None) + with self.assertRaises(RuntimeError): + srv.gitea_create_issue_comment( + 515, "author note", remote="prgs" + ) + mock_api.assert_not_called() + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/test_issue_acceptance_gate.py b/tests/test_issue_acceptance_gate.py new file mode 100644 index 0000000..ead154d --- /dev/null +++ b/tests/test_issue_acceptance_gate.py @@ -0,0 +1,183 @@ +"""Tests for controller issue-acceptance gate (#500).""" +import sys +import unittest +from pathlib import Path + +sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) + +import issue_acceptance_gate # noqa: E402 +from final_report_validator import assess_final_report_validator # noqa: E402 + + +def _accepted_comment(**overrides): + lines = [ + "## Controller Issue Acceptance", + "", + "STATE:", + "accepted", + "", + "WHO_IS_NEXT:", + "controller", + "", + "NEXT_ACTION:", + "Close the tracker follow-up after verification.", + "", + "NEXT_PROMPT:", + "Verify deployment checklist for issue #500 and post acceptance.", + "", + "ISSUE:", + "#500", + "", + "MERGED_PR:", + "#503", + "", + "MERGE_COMMIT:", + "0fdc8f582026b72a229d59a172c0a63ac4aaeaf9", + "", + "ACCEPTANCE_CRITERIA_CHECKED:", + "- [x] documentation added", + "- [x] validator tests added", + "", + "VALIDATION_REVIEWED:", + "pytest tests/test_issue_acceptance_gate.py -q", + "", + "CONTROLLER_DECISION:", + "accepted", + "", + "WHY:", + "All acceptance criteria satisfied with proof.", + "", + "MISSING_WORK:", + "none", + "", + "FOLLOW_UP_ISSUES:", + "none", + "", + "BLOCKERS:", + "none", + "", + "LAST_UPDATED_BY:", + "jcwalker3 / prgs-controller / 2026-07-08", + ] + text = "\n".join(lines) + for key, value in overrides.items(): + text = text.replace(f"{key}:\n", f"{key}:\n{value}\n", 1) + return text + + +def _rejection_comment(state="needs-tests"): + text = _accepted_comment() + text = text.replace("STATE:\naccepted", f"STATE:\n{state}") + text = text.replace( + "CONTROLLER_DECISION:\naccepted", + "CONTROLLER_DECISION:\nrejected", + ) + text = text.replace( + "ACCEPTANCE_CRITERIA_CHECKED:\n- [x] documentation added\n- [x] validator tests added", + "ACCEPTANCE_CRITERIA_CHECKED:\n- [ ] regression tests for rejection paths", + ) + text = text.replace( + "MISSING_WORK:\nnone", + "MISSING_WORK:\nAdd regression tests for controller rejection paths.", + ) + text = text.replace( + "NEXT_PROMPT:\nVerify deployment checklist for issue #500 and post acceptance.", + "NEXT_PROMPT:\nImplement the missing tests for issue #500 and reopen the PR.", + ) + return text + + +class TestControllerAcceptanceComment(unittest.TestCase): + def test_accepted_comment_valid(self): + result = issue_acceptance_gate.validate_controller_acceptance_comment( + _accepted_comment() + ) + self.assertTrue(result["valid"], result["reasons"]) + + def test_missing_who_is_next_rejected(self): + text = _accepted_comment().replace("WHO_IS_NEXT:\ncontroller\n", "WHO_IS_NEXT:\n\n") + result = issue_acceptance_gate.validate_controller_acceptance_comment(text) + self.assertFalse(result["valid"]) + self.assertTrue(any("WHO_IS_NEXT" in r for r in result["reasons"])) + + def test_missing_next_prompt_on_rejection(self): + text = _rejection_comment() + text = text.replace( + "NEXT_PROMPT:\nImplement the missing tests for issue #500 and reopen the PR.\n", + "NEXT_PROMPT:\n\n", + ) + result = issue_acceptance_gate.validate_controller_acceptance_comment(text) + self.assertFalse(result["valid"]) + self.assertTrue(any("NEXT_PROMPT" in r for r in result["reasons"])) + + def test_vague_merge_only_completion_detected(self): + text = "PR merged. Issue complete." + self.assertTrue(issue_acceptance_gate.claims_merge_only_issue_complete(text)) + + def test_rejection_paths_require_missing_work(self): + for state in ( + "more-work-required", + "needs-tests", + "needs-docs", + "needs-feature-enhancement", + "needs-follow-up-issue", + ): + text = _rejection_comment(state=state).replace( + "MISSING_WORK:\nAdd regression tests for controller rejection paths.\n", + "MISSING_WORK:\n\n", + ) + result = issue_acceptance_gate.validate_controller_acceptance_comment(text) + self.assertFalse(result["valid"], state) + self.assertTrue(any("MISSING_WORK" in r for r in result["reasons"])) + + def test_accepted_requires_checked_criteria(self): + text = _accepted_comment().replace( + "ACCEPTANCE_CRITERIA_CHECKED:\n- [x] documentation added\n- [x] validator tests added\n", + "ACCEPTANCE_CRITERIA_CHECKED:\n- [ ] documentation added\n", + ) + result = issue_acceptance_gate.validate_controller_acceptance_comment(text) + self.assertFalse(result["valid"]) + self.assertTrue(any("checked acceptance criterion" in r for r in result["reasons"])) + + +class TestFinalReportIntegration(unittest.TestCase): + def test_merge_only_complete_blocks_work_issue_report(self): + report = ( + "## Controller Handoff\n\n" + "- Task: work issue #500\n" + "- Merge result: merged\n" + "- Current status: PR merged; issue complete\n" + ) + result = assess_final_report_validator(report, "work_issue") + self.assertTrue( + any( + f["rule_id"] == "shared.issue_acceptance_gate" + for f in result["findings"] + ), + result["findings"], + ) + + def test_pending_acceptance_allowed(self): + report = ( + "## Controller Handoff\n\n" + "- Task: work issue #500\n" + "- Merge result: merged\n" + "- Current status: controller acceptance pending\n" + ) + result = assess_final_report_validator(report, "work_issue") + self.assertFalse( + any( + f["rule_id"] == "shared.issue_acceptance_gate" + for f in result["findings"] + ), + result["findings"], + ) + + def test_accepted_block_allows_complete_claim(self): + report = _accepted_comment() + "\n\nIssue is complete." + gate = issue_acceptance_gate.validate_final_report_issue_acceptance(report) + self.assertTrue(gate["valid"], gate["reasons"]) + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_issue_comment_workspace_guard.py b/tests/test_issue_comment_workspace_guard.py new file mode 100644 index 0000000..5359786 --- /dev/null +++ b/tests/test_issue_comment_workspace_guard.py @@ -0,0 +1,257 @@ +import os +import sys +import unittest +from pathlib import Path +from unittest.mock import MagicMock, patch + +sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) + +import gitea_mcp_server as srv + + +FAKE_AUTH = {"Authorization": "token test-token"} +CONTROL_CHECKOUT_ROOT = str(Path(__file__).resolve().parents[3]) + + +class TestIssueCommentWorkspaceGuard(unittest.TestCase): + AUTHOR_ENV = { + "GITEA_PROFILE_NAME": "gitea-author", + "GITEA_ALLOWED_OPERATIONS": "gitea.read,gitea.issue.comment", + } + + def setUp(self): + srv._preflight_whoami_called = True + srv._preflight_capability_called = True + srv._preflight_resolved_role = "author" + srv._preflight_resolved_task = "comment_issue" + srv._preflight_whoami_violation = False + srv._preflight_capability_violation = False + srv._preflight_reviewer_violation_files = [] + + self._orig_in_test = srv._preflight_in_test_mode + srv._preflight_in_test_mode = lambda: False + self.addCleanup(self._restore_preflight_mode) + + def _restore_preflight_mode(self): + srv._preflight_in_test_mode = self._orig_in_test + + def _git_state(self, valid_worktree: str): + def side_effect(path): + real = os.path.realpath(path) + if real == os.path.realpath(CONTROL_CHECKOUT_ROOT): + return { + "current_branch": "master", + "head_sha": "a" * 40, + "porcelain_status": "", + } + if real == os.path.realpath(valid_worktree): + return { + "current_branch": "fix/issue-560-issue-comment-worktree-path", + "head_sha": "b" * 40, + "porcelain_status": "", + } + return { + "current_branch": "other", + "head_sha": "c" * 40, + "porcelain_status": "", + } + + return side_effect + + def _subprocess(self, valid_worktree: str, outside_worktree: str | None = None): + def side_effect(cmd, *args, **kwargs): + result = MagicMock(returncode=0, stdout="") + cwd = "" + if isinstance(cmd, list) and "-C" in cmd: + cwd = os.path.realpath(cmd[cmd.index("-C") + 1]) + + if isinstance(cmd, list) and "--show-toplevel" in cmd: + if cwd == os.path.realpath(valid_worktree): + result.stdout = f"{valid_worktree}\n" + elif outside_worktree and cwd == os.path.realpath(outside_worktree): + result.stdout = f"{outside_worktree}\n" + else: + result.stdout = f"{CONTROL_CHECKOUT_ROOT}\n" + return result + + if isinstance(cmd, list) and "--git-common-dir" in cmd: + if cwd == os.path.realpath(valid_worktree): + result.stdout = f"{CONTROL_CHECKOUT_ROOT}/.git\n" + elif outside_worktree and cwd == os.path.realpath(outside_worktree): + result.stdout = "/tmp/other-repo/.git\n" + else: + result.stdout = f"{CONTROL_CHECKOUT_ROOT}/.git\n" + return result + + return result + + return side_effect + + @patch("gitea_mcp_server._auth", return_value=FAKE_AUTH) + @patch("gitea_mcp_server.api_request") + @patch("gitea_mcp_server.root_checkout_guard.resolve_remote_master_sha", return_value="a" * 40) + def test_comment_rejects_control_checkout_without_worktree_path( + self, _remote_sha, mock_api, _auth + ): + valid_worktree = os.path.join( + CONTROL_CHECKOUT_ROOT, + "branches", + "issue-560-issue-comment-worktree-path", + ) + mock_api.return_value = {"id": 42} + with patch.object(srv, "PROJECT_ROOT", CONTROL_CHECKOUT_ROOT): + with patch( + "gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", + side_effect=self._git_state(valid_worktree), + ): + with self.assertRaises(RuntimeError) as ctx: + srv.gitea_create_issue_comment( + issue_number=557, + body="canonical evidence", + remote="prgs", + ) + self.assertIn("stable control checkout", str(ctx.exception)) + mock_api.assert_not_called() + + @patch("gitea_mcp_server._auth", return_value=FAKE_AUTH) + @patch("gitea_mcp_server.api_request") + @patch("gitea_mcp_server.root_checkout_guard.resolve_remote_master_sha", return_value="a" * 40) + @patch("os.path.isdir", return_value=True) + @patch("os.path.exists", return_value=True) + def test_comment_accepts_valid_branches_worktree_path_with_explicit_repo( + self, _exists, _isdir, _remote_sha, mock_api, _auth + ): + valid_worktree = os.path.join( + CONTROL_CHECKOUT_ROOT, + "branches", + "issue-560-issue-comment-worktree-path", + ) + mock_api.return_value = {"id": 43} + with patch.object(srv, "PROJECT_ROOT", CONTROL_CHECKOUT_ROOT): + with patch("gitea_mcp_server._get_workspace_porcelain", return_value=""): + with patch( + "gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", + side_effect=self._git_state(valid_worktree), + ): + with patch( + "gitea_mcp_server.subprocess.run", + side_effect=self._subprocess(valid_worktree), + ): + with patch.dict(os.environ, self.AUTHOR_ENV, clear=True): + result = srv.gitea_create_issue_comment( + issue_number=557, + body="canonical evidence", + remote="prgs", + org="Scaled-Tech-Consulting", + repo="Gitea-Tools", + worktree_path=valid_worktree, + ) + + self.assertTrue(result["success"]) + self.assertTrue(result["performed"]) + self.assertEqual(result["comment_id"], 43) + method, url, _auth_arg, payload = mock_api.call_args[0] + self.assertEqual(method, "POST") + self.assertIn("/repos/Scaled-Tech-Consulting/Gitea-Tools/issues/557/comments", url) + self.assertEqual(payload, {"body": "canonical evidence"}) + + @patch("gitea_mcp_server._auth", return_value=FAKE_AUTH) + @patch("gitea_mcp_server.api_request") + @patch("gitea_mcp_server.root_checkout_guard.resolve_remote_master_sha", return_value="a" * 40) + @patch("os.path.isdir", return_value=True) + @patch("os.path.exists", return_value=True) + def test_comment_rejects_outside_repo_worktree_path( + self, _exists, _isdir, _remote_sha, mock_api, _auth + ): + valid_worktree = os.path.join( + CONTROL_CHECKOUT_ROOT, + "branches", + "issue-560-issue-comment-worktree-path", + ) + outside_worktree = "/tmp/not-gitea-tools-worktree" + with patch.object(srv, "PROJECT_ROOT", CONTROL_CHECKOUT_ROOT): + with patch("gitea_mcp_server._get_workspace_porcelain", return_value=""): + with patch( + "gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", + side_effect=self._git_state(valid_worktree), + ): + with patch( + "gitea_mcp_server.subprocess.run", + side_effect=self._subprocess(valid_worktree, outside_worktree), + ): + with patch.dict(os.environ, self.AUTHOR_ENV, clear=True): + with self.assertRaises(RuntimeError) as ctx: + srv.gitea_create_issue_comment( + issue_number=557, + body="canonical evidence", + remote="prgs", + worktree_path=outside_worktree, + ) + self.assertIn("does not belong to the target repository", str(ctx.exception)) + mock_api.assert_not_called() + + @patch("gitea_mcp_server._auth", return_value=FAKE_AUTH) + @patch("gitea_mcp_server.api_request") + def test_comment_still_requires_capability_preflight(self, mock_api, _auth): + srv._preflight_capability_called = False + with patch.object(srv, "PROJECT_ROOT", CONTROL_CHECKOUT_ROOT): + with self.assertRaises(RuntimeError) as ctx: + srv.gitea_create_issue_comment( + issue_number=557, + body="canonical evidence", + remote="prgs", + worktree_path=os.path.join( + CONTROL_CHECKOUT_ROOT, + "branches", + "issue-560-issue-comment-worktree-path", + ), + ) + self.assertIn("Task capability", str(ctx.exception)) + mock_api.assert_not_called() + + @patch("gitea_mcp_server._auth", return_value=FAKE_AUTH) + @patch("gitea_mcp_server.api_request") + @patch("gitea_mcp_server.root_checkout_guard.resolve_remote_master_sha", return_value="a" * 40) + @patch("os.path.isdir", return_value=True) + @patch("os.path.exists", return_value=True) + def test_comment_still_requires_issue_comment_permission( + self, _exists, _isdir, _remote_sha, mock_api, _auth + ): + valid_worktree = os.path.join( + CONTROL_CHECKOUT_ROOT, + "branches", + "issue-560-issue-comment-worktree-path", + ) + denied_env = { + "GITEA_PROFILE_NAME": "gitea-author", + "GITEA_ALLOWED_OPERATIONS": "gitea.read,gitea.pr.comment", + } + with patch.object(srv, "PROJECT_ROOT", CONTROL_CHECKOUT_ROOT): + with patch("gitea_mcp_server._get_workspace_porcelain", return_value=""): + with patch( + "gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", + side_effect=self._git_state(valid_worktree), + ): + with patch( + "gitea_mcp_server.subprocess.run", + side_effect=self._subprocess(valid_worktree), + ): + with patch.dict(os.environ, denied_env, clear=True): + result = srv.gitea_create_issue_comment( + issue_number=557, + body="canonical evidence", + remote="prgs", + worktree_path=valid_worktree, + ) + self.assertFalse(result["success"]) + self.assertFalse(result["performed"]) + self.assertIn("permission_report", result) + self.assertEqual( + result["permission_report"]["missing_permission"], + "gitea.issue.comment", + ) + mock_api.assert_not_called() + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/test_issue_lock_adoption.py b/tests/test_issue_lock_adoption.py index a579cbc..270248e 100644 --- a/tests/test_issue_lock_adoption.py +++ b/tests/test_issue_lock_adoption.py @@ -11,6 +11,7 @@ from issue_lock_adoption import ( # noqa: E402 NO_MATCH, assess_own_branch_adoption, build_adoption_proof, + build_non_adoption_lock_proof, ) REQ = "feat/issue-420-server-code-parity" @@ -134,5 +135,93 @@ class TestBuildAdoptionProof(unittest.TestCase): self.assertTrue(proof["no_competing_live_lock_proof"]) +class TestExplicitAdoptionProofFields(unittest.TestCase): + """#477: explicit, citable adoption-proof fields for all outcomes.""" + + def _proof(self, assessment, branch): + return build_adoption_proof( + issue_number=420, + branch_name=branch, + assessment=assessment, + open_pr_checked=True, + competing_lock_checked=True, + lock_file_path="/tmp/example-lock.json", + lock_file_status="written", + ) + + def test_adopt_proof_exposes_explicit_fields(self): + assessment = assess_own_branch_adoption( + issue_number=420, + requested_branch=REQ, + existing_branches=[{"name": REQ, "commit_sha": "934688a"}], + ) + proof = self._proof(assessment, REQ) + self.assertEqual(proof["adoption_decision"], "ADOPT") + self.assertTrue(proof["adopted"]) + self.assertEqual(proof["adopted_branch"], REQ) + self.assertEqual(proof["adopted_branch_head"], "934688a") + self.assertEqual(proof["competing_branch_check"]["result"], "clear") + self.assertEqual(proof["competing_branch_check"]["competing_branches"], []) + self.assertIn("gitea_create_pr", proof["safe_next_action"]) + self.assertIn("exactly matches", proof["matcher_summary"]) + + def test_block_proof_reports_competing_and_does_not_claim_adoption(self): + assessment = assess_own_branch_adoption( + issue_number=420, + requested_branch=REQ, + existing_branches=[{"name": "feat/issue-420-rogue"}], + ) + proof = self._proof(assessment, REQ) + self.assertEqual(proof["adoption_decision"], "BLOCK_COMPETING") + self.assertFalse(proof["adopted"]) + self.assertIsNone(proof["adopted_branch"]) + self.assertIsNone(proof["adopted_branch_head"]) + self.assertEqual(proof["competing_branch_check"]["result"], "blocked") + self.assertIn( + "feat/issue-420-rogue", + proof["competing_branch_check"]["competing_branches"], + ) + self.assertIn("fail closed", proof["safe_next_action"]) + + def test_no_match_proof_does_not_claim_adoption(self): + assessment = assess_own_branch_adoption( + issue_number=420, + requested_branch=REQ, + existing_branches=[{"name": "feat/issue-999-unrelated"}], + ) + proof = self._proof(assessment, REQ) + self.assertEqual(proof["adoption_decision"], "NO_MATCH") + self.assertFalse(proof["adopted"]) + self.assertIsNone(proof["adopted_branch"]) + self.assertEqual(proof["competing_branch_check"]["result"], "clear") + + def test_substring_collision_stays_boundary_safe(self): + # issue-42 must not adopt/claim against an issue-420 branch (#440/#477). + own = "feat/issue-42-widget" + assessment = assess_own_branch_adoption( + issue_number=42, + requested_branch=own, + existing_branches=[ + {"name": own, "commit_sha": "abc1234"}, + {"name": "feat/issue-420-server-code-parity"}, + ], + ) + proof = self._proof(assessment, own) + self.assertEqual(proof["adoption_decision"], "ADOPT") + self.assertEqual(proof["adopted_branch"], own) + self.assertEqual(proof["competing_branch_check"]["competing_branches"], []) + + def test_non_adoption_lock_proof_is_adoption_free(self): + proof = build_non_adoption_lock_proof( + issue_number=196, branch_name="feat/issue-196-mutations" + ) + self.assertEqual(proof["adoption_decision"], "NO_MATCH") + self.assertFalse(proof["adopted"]) + self.assertIsNone(proof["adopted_branch"]) + self.assertIsNone(proof["adopted_branch_head"]) + self.assertEqual(proof["competing_branch_check"]["result"], "clear") + self.assertIn("no adoption", proof["safe_next_action"].lower()) + + if __name__ == "__main__": unittest.main() \ No newline at end of file diff --git a/tests/test_llm_agent_sha.py b/tests/test_llm_agent_sha.py index 4d74dfa..2751d9d 100644 --- a/tests/test_llm_agent_sha.py +++ b/tests/test_llm_agent_sha.py @@ -22,6 +22,7 @@ from unittest.mock import patch sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent)) +import mcp_server # noqa: E402 from mcp_server import ( # noqa: E402 gitea_check_pr_eligibility, gitea_merge_pr, diff --git a/tests/test_llm_workflow_split.py b/tests/test_llm_workflow_split.py index a5e5e33..5e997aa 100644 --- a/tests/test_llm_workflow_split.py +++ b/tests/test_llm_workflow_split.py @@ -87,6 +87,14 @@ def test_reconcile_landed_workflow_contract(): assert "PARTIAL_RECONCILE_COMMENT_THEN_STOP" in text assert "RECOVERY_HANDOFF_ONLY" in text assert "resolve_partial_reconciliation_plan" in text + assert "check_audit_mutation_allowed" in text + assert "gitea_authorize_reconciliation_cleanup_phase" in text + + +def test_audit_reconciliation_verifier_exported(): + from review_proofs import assess_audit_reconciliation_report + + assert callable(assess_audit_reconciliation_report) def test_create_issue_workflow_contract(): diff --git a/tests/test_lock_issue_mcp_registration.py b/tests/test_lock_issue_mcp_registration.py new file mode 100644 index 0000000..4b20763 --- /dev/null +++ b/tests/test_lock_issue_mcp_registration.py @@ -0,0 +1,127 @@ +"""Regression tests for gitea_lock_issue MCP tool registration (#521).""" + +from __future__ import annotations + +import os +import tempfile +import unittest +from unittest.mock import patch + +import issue_lock_store +import mcp_server +from mcp_server import gitea_create_pr, gitea_lock_issue + +FAKE_AUTH = "Basic dGVzdDp0ZXN0" + +ISSUE_WRITE_ENV = { + "GITEA_ALLOWED_OPERATIONS": ( + "gitea.issue.create,gitea.issue.close,gitea.issue.comment" + ), +} + +CREATE_PR_ENV = { + "GITEA_PROFILE_NAME": "author-test", + "GITEA_ALLOWED_OPERATIONS": ( + "gitea.read,gitea.pr.create,gitea.branch.push," + "gitea.issue.create,gitea.issue.close,gitea.issue.comment" + ), + "GITEA_FORBIDDEN_OPERATIONS": ( + "gitea.pr.approve,gitea.pr.merge,gitea.pr.review" + ), +} + + +def _clean_master_git_state_for_lock(): + return { + "current_branch": "master", + "porcelain_status": "", + "base_equivalent": True, + "inspected_git_root": "/scratch/wt", + "base_branch": "origin/master", + } + + +def _registered_tool_names() -> set[str]: + manager = mcp_server.mcp._tool_manager + tools = getattr(manager, "_tools", None) or {} + return set(tools.keys()) + + +class TestLockIssueMcpRegistration(unittest.TestCase): + def test_gitea_lock_issue_registered_as_public_mcp_tool(self): + names = _registered_tool_names() + self.assertIn("gitea_lock_issue", names) + + def test_internal_list_open_pulls_not_exposed_as_mcp_tool(self): + names = _registered_tool_names() + self.assertNotIn("_list_open_pulls", names) + + +class TestCreatePrLockRegistrationFlow(unittest.TestCase): + def setUp(self): + self._lock_dir = tempfile.TemporaryDirectory() + self._env_patcher = patch.dict( + os.environ, + { + **CREATE_PR_ENV, + "GITEA_ISSUE_LOCK_DIR": self._lock_dir.name, + }, + clear=True, + ) + self._env_patcher.start() + self._dup_fetcher_patcher = patch( + "mcp_server.issue_duplicate_context_fetcher", + return_value=([], [], {"status": "not_claimed"}), + ) + self._dup_fetcher_patcher.start() + + def tearDown(self): + self._dup_fetcher_patcher.stop() + self._env_patcher.stop() + self._lock_dir.cleanup() + + @patch("mcp_server.role_session_router.check_author_mutation_after_reviewer_stop", + return_value=(True, [])) + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_create_pr_still_fails_closed_without_issue_lock(self, _auth, _role): + with self.assertRaises(RuntimeError) as ctx: + gitea_create_pr( + title="feat: X Closes #521", + head="feat/issue-521-lock-issue-registration", + remote="prgs", + ) + self.assertIn("Issue lock is missing", str(ctx.exception)) + + @patch("mcp_server.api_request") + @patch( + "mcp_server.issue_lock_worktree.read_worktree_git_state", + return_value=_clean_master_git_state_for_lock(), + ) + @patch("mcp_server.api_get_all", return_value=[]) + @patch("mcp_server.role_session_router.check_author_mutation_after_reviewer_stop", + return_value=(True, [])) + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_create_pr_proceeds_after_valid_issue_lock( + self, _auth, _role, _api, _git_state, mock_api_request + ): + worktree = os.path.realpath(os.getcwd()) + mock_api_request.return_value = {"number": 521, "html_url": "https://example/pr/521"} + lock_res = gitea_lock_issue( + issue_number=521, + branch_name="feat/issue-521-lock-issue-registration", + remote="prgs", + worktree_path=worktree, + ) + self.assertTrue(lock_res["success"]) + lock_path = lock_res["lock_file_path"] + self.assertTrue(os.path.exists(lock_path)) + lock = issue_lock_store.read_lock_file(lock_path) + self.assertEqual(lock["issue_number"], 521) + + res = gitea_create_pr( + title="fix: restore lock tool registration Closes #521", + head="feat/issue-521-lock-issue-registration", + remote="prgs", + worktree_path=worktree, + ) + self.assertEqual(res["number"], 521) \ No newline at end of file diff --git a/tests/test_mcp_menu_script.py b/tests/test_mcp_menu_script.py new file mode 100644 index 0000000..a447648 --- /dev/null +++ b/tests/test_mcp_menu_script.py @@ -0,0 +1,124 @@ +"""Hermetic tests for repository-root MCP operator menu (#478).""" +import os +import stat +import unittest +from pathlib import Path + +REPO_ROOT = Path(__file__).resolve().parent.parent +SCRIPT = REPO_ROOT / "mcp-menu.sh" +DOCS = REPO_ROOT / "docs" / "mcp-menu.md" + +REQUIRED_MENU_LABELS = ( + "Project status / root checkout health", + "Author workflow prompts", + "Reviewer workflow prompts", + "Merger workflow prompts", + "Reconciler workflow prompts", + "Onboarding new project to this MCP workflow", + "Proxmox deployment menu placeholder", + "Create Proxmox LXC placeholder", + "Run tests", + "Exit", +) + +DANGEROUS_PATTERNS = ( + "git push --force", + "git push -f", + "--force-with-lease", + "delete-branch", + "gitea_delete_branch", + "issue-locks", + "issue_lock_store", + "curl ", + "wget ", +) + + +class TestMcpMenuScript(unittest.TestCase): + def setUp(self): + self.assertTrue(SCRIPT.is_file(), "mcp-menu.sh must exist at repo root") + self.content = SCRIPT.read_text(encoding="utf-8") + + def test_script_exists_at_repo_root(self): + self.assertEqual(SCRIPT.name, "mcp-menu.sh") + self.assertEqual(SCRIPT.parent, REPO_ROOT) + + def test_executable_bit_is_set(self): + mode = SCRIPT.stat().st_mode + self.assertTrue(mode & stat.S_IXUSR, "mcp-menu.sh must be executable by owner") + + def test_shebang(self): + first_line = self.content.splitlines()[0] + self.assertEqual(first_line, "#!/usr/bin/env bash") + + def test_uses_set_euo_pipefail(self): + self.assertIn("set -euo pipefail", self.content) + + def test_no_dangerous_commands(self): + lowered = self.content.lower() + for pattern in DANGEROUS_PATTERNS: + with self.subTest(pattern=pattern): + self.assertNotIn(pattern.lower(), lowered) + + def test_no_branch_deletion_verbs(self): + for token in ("git branch -D", "git branch -d", "push :refs"): + with self.subTest(token=token): + self.assertNotIn(token, self.content) + + def test_contains_required_menu_labels(self): + for label in REQUIRED_MENU_LABELS: + with self.subTest(label=label): + self.assertIn(label, self.content) + + def test_run_tests_prefers_run_tests_sh(self): + self.assertIn('run-tests.sh', self.content) + run_tests_idx = self.content.index("run_tests()") + run_tests_body = self.content[run_tests_idx : run_tests_idx + 800] + pytest_idx = run_tests_body.find("pytest") + run_tests_sh_idx = run_tests_body.find("run-tests.sh") + self.assertGreater(pytest_idx, 0) + self.assertGreater(run_tests_sh_idx, 0) + self.assertLess(run_tests_sh_idx, pytest_idx) + + def test_run_tests_fail_closed_without_runner(self): + self.assertIn("fail closed", self.content.lower()) + self.assertIn("exit 1", self.content) + + def test_proxmox_entries_are_placeholders(self): + self.assertIn("TODO / issue-backed", self.content) + self.assertIn("NOT implemented", self.content) + + def test_root_health_shows_required_fields(self): + health_fn = self._extract_function("show_root_checkout_health") + for snippet in ( + "status --short --branch", + "rev-parse HEAD", + "prgs/master", + "WARNING: root checkout", + ): + with self.subTest(snippet=snippet): + self.assertIn(snippet, health_fn) + + def test_docs_mention_mcp_menu_sh(self): + self.assertTrue(DOCS.is_file(), "docs/mcp-menu.md must exist") + docs_text = DOCS.read_text(encoding="utf-8") + self.assertIn("./mcp-menu.sh", docs_text) + self.assertIn("placeholder", docs_text.lower()) + + def _extract_function(self, name: str) -> str: + marker = f"{name}() {{" + start = self.content.index(marker) + depth = 0 + for idx in range(start, len(self.content)): + char = self.content[idx] + if char == "{": + depth += 1 + elif char == "}": + depth -= 1 + if depth == 0: + return self.content[start : idx + 1] + self.fail(f"Could not parse function {name}") + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_mcp_native_cleanup_proof.py b/tests/test_mcp_native_cleanup_proof.py new file mode 100644 index 0000000..d965b0d --- /dev/null +++ b/tests/test_mcp_native_cleanup_proof.py @@ -0,0 +1,251 @@ +"""Tests for MCP-native post-merge cleanup proof (#517).""" +import sys +import unittest +from pathlib import Path + +sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) + +from final_report_validator import assess_final_report_validator # noqa: E402 +from mcp_native_cleanup_proof import ( # noqa: E402 + assess_authorized_reconciler_cleanup_path, + assess_mcp_native_cleanup_proof, + assess_merger_cleanup_handoff_guidance, + assess_raw_branch_delete_report, + assess_raw_comment_delete_report, +) + + +def _authorized_cleanup_report(**overrides): + fields = { + "Merge mutations": "gitea_merge_pr on PR #100", + "Cleanup mutations": ( + "gitea_reconcile_merged_cleanups deleted remote branch; " + "gitea_cleanup_post_merge_moot_lease released lease" + ), + "Reconciler capability": "prgs-reconciler / gitea.branch.delete resolved", + "Next actor": "reconciler session complete", + } + fields.update(overrides) + lines = ["## Controller Handoff", ""] + lines.extend(f"- {key}: {value}" for key, value in fields.items()) + return "\n".join(lines) + + +class TestRawBranchDeleteBlocked(unittest.TestCase): + def test_git_branch_d_blocked(self): + report = "Cleanup mutations: git branch -d feat/issue-1-x" + result = assess_raw_branch_delete_report(report) + self.assertTrue(result["block"]) + self.assertIn("git branch -d", result["commands"][0]) + + def test_git_push_delete_in_ledger_blocked(self): + report = "- Git ref mutations: git push origin --delete feat/x" + result = assess_raw_branch_delete_report(report) + self.assertTrue(result["block"]) + + def test_narrative_git_push_delete_not_blocked(self): + report = "\n".join([ + "## Validation", + "Workflow §28B forbids raw git push --delete for cleanup.", + "- Cleanup mutations: none", + "- Git ref mutations: none", + "- Worktree mutations: none", + ]) + result = assess_raw_branch_delete_report(report) + self.assertFalse(result["block"]) + + def test_authorized_mcp_path_passes(self): + report = _authorized_cleanup_report() + result = assess_raw_branch_delete_report(report) + self.assertFalse(result["block"]) + + +class TestRawCommentDeleteBlocked(unittest.TestCase): + def test_curl_delete_comment_in_ledger_blocked(self): + report = ( + "- Cleanup mutations: curl -X DELETE https://gitea.example/api/v1/repos/o/r/" + "issues/9/comments/42" + ) + result = assess_raw_comment_delete_report(report) + self.assertTrue(result["block"]) + + def test_narrative_comment_delete_not_blocked(self): + report = "\n".join([ + "Files reviewed: post_merge_cleanup_proof.py (raw comment delete patterns)", + "Validation note: never use delete_issue_comment for lease cleanup.", + "- Cleanup mutations: none", + "- Git ref mutations: none", + "- Worktree mutations: none", + ]) + result = assess_raw_comment_delete_report(report) + self.assertFalse(result["block"]) + + def test_delete_issue_comment_helper_in_ledger_blocked(self): + report = "- Cleanup mutations: delete_issue_comment purged lease comments" + result = assess_raw_comment_delete_report(report) + self.assertTrue(result["block"]) + + def test_moot_lease_append_only_passes(self): + report = _authorized_cleanup_report() + result = assess_raw_comment_delete_report(report) + self.assertFalse(result["block"]) + + +class TestAuthorizedReconcilerPath(unittest.TestCase): + def test_cleanup_without_mcp_tool_blocked(self): + report = "\n".join([ + "## Controller Handoff", + "- Cleanup mutations: removed branch manually after merge", + "- Reconciler capability: prgs-reconciler", + ]) + result = assess_authorized_reconciler_cleanup_path(report) + self.assertTrue(result["block"]) + self.assertTrue(any("authorized MCP" in r for r in result["reasons"])) + + def test_authorized_cleanup_succeeds(self): + result = assess_authorized_reconciler_cleanup_path(_authorized_cleanup_report()) + self.assertFalse(result["block"]) + + def test_git_ref_cleanup_without_mcp_tool_blocked(self): + report = "\n".join([ + "## Controller Handoff", + "- Cleanup mutations: none", + "- Git ref mutations: git branch -D feat/issue-517-test", + "- Reconciler capability: prgs-reconciler", + ]) + result = assess_authorized_reconciler_cleanup_path(report) + self.assertTrue(result["block"]) + self.assertTrue( + any("Git ref mutations cleanup must name" in r for r in result["reasons"]) + ) + + def test_worktree_cleanup_without_mcp_tool_blocked(self): + report = "\n".join([ + "## Controller Handoff", + "- Cleanup mutations: none", + "- Worktree mutations: git worktree remove branches/review-pr542", + "- Reconciler capability: prgs-reconciler", + ]) + result = assess_authorized_reconciler_cleanup_path(report) + self.assertTrue(result["block"]) + self.assertTrue( + any("Worktree mutations cleanup must name" in r for r in result["reasons"]) + ) + + def test_git_fetch_in_git_ref_mutations_not_cleanup_claim(self): + report = "\n".join([ + "## Controller Handoff", + "- Cleanup mutations: none", + "- Git ref mutations: git fetch prgs master", + ]) + result = assess_authorized_reconciler_cleanup_path(report) + self.assertFalse(result["block"]) + + +class TestMergerHandoffGuidance(unittest.TestCase): + def test_merger_cleanup_without_handoff_blocked(self): + report = "Merger deleted remote branch and removed worktree after merge" + result = assess_merger_cleanup_handoff_guidance(report) + self.assertTrue(result["block"]) + + def test_merger_cleanup_with_reconciler_handoff_passes(self): + report = ( + "Merger deleted remote branch; next actor: reconciler for worktree audit" + ) + result = assess_merger_cleanup_handoff_guidance(report) + self.assertFalse(result["block"]) + + +class TestCompositeVerifier(unittest.TestCase): + def test_full_authorized_report_passes(self): + result = assess_mcp_native_cleanup_proof(_authorized_cleanup_report()) + self.assertFalse(result["block"], result["reasons"]) + + def test_raw_bypasses_fail_composite(self): + report = "\n".join([ + "## Controller Handoff", + "- Merge mutations: gitea_merge_pr", + "- Cleanup mutations: git branch -D feat/x; curl -X DELETE .../comments/1", + ]) + result = assess_mcp_native_cleanup_proof(report) + self.assertTrue(result["block"]) + self.assertGreaterEqual(len(result["reasons"]), 2) + + def test_narrative_cleanup_mentions_pass_when_ledgers_none(self): + report = "\n".join([ + "## Review summary", + "Validated workflow §28B MCP-native cleanup guidance.", + "Files reviewed describe raw git branch -d and delete_issue_comment blocks.", + "", + "## Controller Handoff", + "- Cleanup mutations: none", + "- Git ref mutations: none", + "- Worktree mutations: none", + ]) + result = assess_mcp_native_cleanup_proof(report) + self.assertFalse(result["block"], result["reasons"]) + + def test_git_ref_cleanup_bypass_blocked_in_composite(self): + report = "\n".join([ + "## Controller Handoff", + "- Cleanup mutations: none", + "- Git ref mutations: git branch -D feat/issue-517-test", + "- Reconciler capability: prgs-reconciler", + ]) + result = assess_mcp_native_cleanup_proof(report) + self.assertTrue(result["block"]) + self.assertTrue( + any("Git ref mutations cleanup must name" in r for r in result["reasons"]) + ) + + +class TestFinalReportValidatorIntegration(unittest.TestCase): + def test_validator_blocks_raw_branch_delete_in_review_report(self): + report = "\n".join([ + "## Controller Handoff", + "- Cleanup mutations: git branch -d feat/issue-517-test", + ]) + result = assess_final_report_validator(report, "review_pr") + rules = {f["rule_id"] for f in result["findings"]} + self.assertIn("shared.mcp_native_cleanup_proof", rules) + + def test_validator_passes_authorized_cleanup_report(self): + result = assess_final_report_validator(_authorized_cleanup_report(), "review_pr") + blocked = [f for f in result["findings"] if f.get("severity") == "block"] + mcp_blocks = [ + f for f in blocked if f.get("rule_id") == "shared.mcp_native_cleanup_proof" + ] + self.assertEqual(mcp_blocks, []) + + def test_validator_passes_narrative_cleanup_mentions_with_none_ledgers(self): + report = "\n".join([ + "## Review summary", + "Workflow §28B documents raw git push --delete and delete_issue_comment blocks.", + "", + "## Controller Handoff", + "- Cleanup mutations: none", + "- Git ref mutations: none", + "- Worktree mutations: none", + ]) + result = assess_final_report_validator(report, "review_pr") + mcp_blocks = [ + f for f in result["findings"] + if f.get("rule_id") == "shared.mcp_native_cleanup_proof" + and f.get("severity") == "block" + ] + self.assertEqual(mcp_blocks, []) + + def test_validator_blocks_git_ref_cleanup_bypass(self): + report = "\n".join([ + "## Controller Handoff", + "- Cleanup mutations: none", + "- Git ref mutations: git branch -D feat/issue-517-test", + "- Reconciler capability: prgs-reconciler", + ]) + result = assess_final_report_validator(report, "reconcile_already_landed") + rules = {f["rule_id"] for f in result["findings"] if f.get("severity") == "block"} + self.assertIn("shared.mcp_native_cleanup_proof", rules) + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_mcp_server.py b/tests/test_mcp_server.py index 2b8efb4..503e879 100644 --- a/tests/test_mcp_server.py +++ b/tests/test_mcp_server.py @@ -58,6 +58,12 @@ _NO_BLOCKER_FEEDBACK = { } +def _init_reviewer_session(remote="prgs"): + """Seed review decision lock and required workflow-load proof (#389).""" + init_review_decision_lock(remote, "review_pr") + mcp_server.gitea_load_review_workflow() + + def _mark_request_changes_ready(pr_number=8, **kwargs): """Mark a request_changes decision ready with the #332 duplicate- suppression feedback fetch stubbed to 'no existing blocker'.""" @@ -121,6 +127,7 @@ def _install_owned_reviewer_lease( session_id=_DEFAULT_LEASE_SESSION, head_sha="abc123", ): + import merger_lease_adoption as mla import reviewer_pr_lease reviewer_pr_lease.clear_session_lease() @@ -129,7 +136,11 @@ def _install_owned_reviewer_lease( "session_id": session_id, "candidate_head": head_sha, "target_branch": "master", - }) + "comment_id": 9001, + }, lease_provenance=mla.build_lease_provenance( + source=mla.SOURCE_ACQUIRE, + comment_id=9001, + )) return patch( "mcp_server._fetch_pr_comments", return_value=[ @@ -179,6 +190,7 @@ def _seed_ready_review_decision( "correction_authorized": False, "correction_reason": None, }) + _m.gitea_load_review_workflow() # Issue-write tools are profile-gated (#69). @@ -671,6 +683,7 @@ class TestMergePR(unittest.TestCase): def setUp(self): import reviewer_pr_lease + mcp_server.gitea_load_review_workflow() self._lease_patch = _install_owned_reviewer_lease(8) self._lease_patch.start() self._auth_identity_patch = patch( @@ -1049,7 +1062,8 @@ class TestMergePR(unittest.TestCase): "GITEA_ALLOWED_OPERATIONS": "read,merge"} with patch.dict(os.environ, env, clear=True): r = gitea_merge_pr( - pr_number=8, confirmation=self._confirm(8), remote="prgs") + pr_number=8, confirmation=self._confirm(8), remote="prgs", + expected_head_sha=new_sha) self.assertFalse(r["performed"]) self.assertTrue(r.get("approval_visible")) self.assertFalse(r.get("approval_at_current_head")) @@ -1876,7 +1890,7 @@ class TestReviewDecisionValidationGate(unittest.TestCase): def setUp(self): import reviewer_pr_lease - init_review_decision_lock("prgs", "review_pr") + _init_reviewer_session("prgs") self._lease_patch = _install_owned_reviewer_lease( self.PR, head_sha=self.SHA, ) @@ -1996,7 +2010,7 @@ class TestSubmitPrReview(unittest.TestCase): def setUp(self): import reviewer_pr_lease - init_review_decision_lock("prgs", "review_pr") + _init_reviewer_session("prgs") self._lease_patch = _install_owned_reviewer_lease(8) self._lease_patch.start() self._auth_identity_patch = patch( @@ -2416,7 +2430,7 @@ class TestSubmitPrReview(unittest.TestCase): os.remove(spoof_path) def test_mark_final_decision_rejects_remote_mismatch(self): - init_review_decision_lock("prgs", "review_pr") + _init_reviewer_session("prgs") r = gitea_mark_final_review_decision(8, "approve", remote="dadeschools", expected_head_sha="abc123") self.assertFalse(r["marked_ready"]) self.assertTrue(any("does not match locked remote" in x for x in r["reasons"])) @@ -2500,6 +2514,7 @@ if __name__ == "__main__": class TestTrackerHygieneCleanup(unittest.TestCase): def setUp(self): + mcp_server.gitea_load_review_workflow() self.mock_api = patch("mcp_server.api_request").start() self.mock_auth = patch("mcp_server.get_auth_header", return_value=FAKE_AUTH).start() patch("gitea_audit.audit_enabled", return_value=True).start() @@ -3460,6 +3475,45 @@ class TestIssueLocking(unittest.TestCase): self.assertIn("adoption", res) self.assertEqual(res["adoption"]["branch_head_commit"], "abc123") + @patch( + "mcp_server.issue_lock_worktree.read_worktree_git_state", + return_value=_clean_master_git_state_for_lock(), + ) + @patch("mcp_server.api_get_all") + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_lock_issue_adoption_response_has_explicit_proof(self, _auth, mock_api, _git_state): + # #477 AC1: the live lock response must carry citable adoption proof. + branch = "feat/issue-196-mutations" + self.mock_dup_fetcher.return_value = ([], [branch], {"status": "not_claimed"}) + mock_api.return_value = [{"name": branch, "commit": {"id": "abc123"}}] + res = gitea_lock_issue(issue_number=196, branch_name=branch, remote="prgs") + proof = res["adoption"] + self.assertEqual(proof["adoption_decision"], "ADOPT") + self.assertTrue(proof["adopted"]) + self.assertEqual(proof["adopted_branch"], branch) + self.assertEqual(proof["adopted_branch_head"], "abc123") + self.assertEqual(proof["competing_branch_check"]["result"], "clear") + self.assertIn("gitea_create_pr", proof["safe_next_action"]) + self.assertIn("196", proof["matcher_summary"]) + + @patch( + "mcp_server.issue_lock_worktree.read_worktree_git_state", + return_value=_clean_master_git_state_for_lock(), + ) + @patch("mcp_server.api_get_all", return_value=[]) + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_lock_issue_no_match_response_does_not_claim_adoption(self, _auth, _api, _git_state): + # #477 AC2/AC3: a normal (NO_MATCH) lock must carry adoption-free proof + # and must NOT expose an ``adoption`` block. + res = gitea_lock_issue(issue_number=196, branch_name="feat/issue-196-mutations", remote="prgs") + self.assertTrue(res["success"]) + self.assertNotIn("adoption", res) + check = res["adoption_check"] + self.assertEqual(check["adoption_decision"], "NO_MATCH") + self.assertFalse(check["adopted"]) + self.assertIsNone(check["adopted_branch"]) + self.assertEqual(check["competing_branch_check"]["result"], "clear") + @patch( "mcp_server.issue_lock_worktree.read_worktree_git_state", return_value=_clean_master_git_state_for_lock(), @@ -3840,7 +3894,7 @@ class TestPreflightVerification(unittest.TestCase): os.environ["GITEA_TEST_PORCELAIN"] = " M reviewer_edit.py\n" with self.assertRaises(RuntimeError) as ctx: mcp_server.verify_preflight_purity() - self.assertIn("Reviewer profile is forbidden from modifying tracked workspace files", str(ctx.exception)) + self.assertIn("forbidden from modifying tracked workspace files", str(ctx.exception)) self.assertIn("reviewer_edit.py", str(ctx.exception)) # Foreign pre-existing dirty state does not block when unchanged. @@ -3906,7 +3960,172 @@ class TestPreflightVerification(unittest.TestCase): with self.assertRaises(RuntimeError) as ctx: mcp_server.verify_preflight_purity(worktree_path=worktree) msg = str(ctx.exception) - self.assertIn("active task workspace root", msg) - self.assertIn("inspected git root", msg) - self.assertIn("dirty files: task_file.py", msg) - self.assertIn("dirty scope:", msg) + self.assertIn("resolved workspace", msg) + self.assertIn(worktree, msg) + self.assertIn("worktree_path argument", msg) + self.assertIn("task_file.py", msg) + self.assertIn("author namespace", msg) + + +class TestIssue546Deadlock(unittest.TestCase): + def setUp(self): + import reviewer_pr_lease + import review_workflow_load + import mcp_server + reviewer_pr_lease.clear_session_lease() + review_workflow_load.clear_review_workflow_load() + mcp_server._preflight_capability_called = False + mcp_server._preflight_resolved_role = None + mcp_server._REVIEW_DECISION_LOCK = None + + self.auth_user_patch = patch("mcp_server._authenticated_username", return_value="reviewer-bot") + self.auth_user_patch.start() + self.addCleanup(self.auth_user_patch.stop) + + self.verify_purity_patch = patch("mcp_server.verify_preflight_purity", return_value=None) + self.verify_purity_patch.start() + self.addCleanup(self.verify_purity_patch.stop) + + self.verify_workspace_patch = patch("mcp_server._verify_role_mutation_workspace", return_value="/workspace") + self.verify_workspace_patch.start() + self.addCleanup(self.verify_workspace_patch.stop) + + self.get_profile_patch = patch("mcp_server.get_profile", return_value={ + "profile_name": "prgs-reviewer", + "allowed_operations": ["gitea.read", "gitea.pr.comment", "gitea.pr.review", "gitea.pr.approve"], + "forbidden_operations": [], + }) + self.get_profile_patch.start() + self.addCleanup(self.get_profile_patch.stop) + + self.pr_work_lease_patch = patch( + "mcp_server._pr_work_lease_reviewer_block", + return_value=dict(_NO_PR_WORK_LEASE_BLOCK), + ) + self.pr_work_lease_patch.start() + self.addCleanup(self.pr_work_lease_patch.stop) + + self.auth_header_patch = patch("mcp_server.get_auth_header", return_value="token test") + self.auth_header_patch.start() + self.addCleanup(self.auth_header_patch.stop) + + def test_workflow_no_deadlock(self): + import mcp_server + import reviewer_pr_lease + + # 1. Resolve capability + res = mcp_server.gitea_resolve_task_capability(task="review_pr", remote="prgs") + self.assertTrue(res["allowed_in_current_session"]) + + # 2. Load review workflow + res = mcp_server.gitea_load_review_workflow() + self.assertTrue(res["success"]) + + # Mock API requests for lease comments and PR retrieval + comments = [] + def mock_api_side(method, url, auth, data=None): + if "/api/v1/user" in url: + return {"login": "reviewer-bot"} + if "/pulls/" in url: + if "/reviews" in url: + if method == "POST": + return {"id": 2001, "state": "APPROVED"} + return [{"id": 2001, "user": {"login": "reviewer-bot"}, "state": "APPROVED", "commit_id": "abc123"}] + return {"user": {"login": "author-user"}, "state": "open", "head": {"sha": "abc123"}, "mergeable": True} + if "/comments" in url: + if method == "POST": + comments.append({"id": 1001, "body": data["body"], "user": {"login": "reviewer-bot"}}) + return {"id": 1001} + return comments + return {} + + with patch("mcp_server.api_request", side_effect=mock_api_side): + # 3. Acquire reviewer lease + res = mcp_server.gitea_acquire_reviewer_pr_lease( + pr_number=550, + worktree="/workspace", + candidate_head="abc123", + remote="prgs", + ) + self.assertTrue(res["success"]) + self.assertEqual(res["comment_id"], 1001) + + # verify lease is recorded in session + session_lease = reviewer_pr_lease.get_session_lease() + self.assertIsNotNone(session_lease) + self.assertEqual(session_lease["pr_number"], 550) + + # 4. Resolve capability again (simulating subsequent tool call preflight check/token refresh) + # This should NOT clear the session lease or decision lock! + mcp_server._preflight_capability_called = False # simulate clearance from prior mutation + res = mcp_server.gitea_resolve_task_capability(task="review_pr", remote="prgs") + self.assertTrue(res["allowed_in_current_session"]) + self.assertIsNotNone(reviewer_pr_lease.get_session_lease()) # Preserved! + + # 5. Mark final review decision APPROVED + res = mcp_server.gitea_mark_final_review_decision( + pr_number=550, + action="approve", + expected_head_sha="abc123", + remote="prgs", + ) + self.assertTrue(res["marked_ready"]) + + # 6. Submit review (should succeed without deadlock) + res = mcp_server.gitea_submit_pr_review( + pr_number=550, + action="approve", + body="APPROVED", + expected_head_sha="abc123", + final_review_decision_ready=True, + remote="prgs", + worktree_path="/workspace", + ) + self.assertTrue(res["performed"]) + + def test_release_reviewer_pr_lease(self): + import mcp_server + import reviewer_pr_lease + + # Resolve capability and load workflow + mcp_server.gitea_resolve_task_capability(task="review_pr", remote="prgs") + mcp_server.gitea_load_review_workflow() + + comments = [_reviewer_lease_comment(550, session_id=_DEFAULT_LEASE_SESSION, head_sha="abc123")] + posted_comments = [] + + def mock_api_side(method, url, auth, data=None): + if "/api/v1/user" in url: + return {"login": "reviewer-bot"} + if "/pulls/" in url: + return {"user": {"login": "author-user"}, "state": "open", "head": {"sha": "abc123"}, "mergeable": True} + if "/comments" in url: + if method == "POST": + new_c = {"id": 1002, "body": data["body"], "user": {"login": "reviewer-bot"}} + comments.append(new_c) + posted_comments.append(new_c) + return {"id": 1002} + return comments + return {} + + # Set session lease in memory + import merger_lease_adoption as mla + reviewer_pr_lease.record_session_lease({ + "pr_number": 550, + "session_id": _DEFAULT_LEASE_SESSION, + "candidate_head": "abc123", + "target_branch": "master", + "comment_id": 9001, + }, lease_provenance=mla.build_lease_provenance(source=mla.SOURCE_ACQUIRE, comment_id=9001)) + + with patch("mcp_server.api_request", side_effect=mock_api_side): + # Release lease + res = mcp_server.gitea_release_reviewer_pr_lease(pr_number=550, remote="prgs") + self.assertTrue(res["success"]) + self.assertTrue(res["released"]) + self.assertEqual(res["comment_id"], 1002) + + # verify lease is cleared in session + self.assertIsNone(reviewer_pr_lease.get_session_lease()) + # verify comment phase is released + self.assertIn("phase: released", posted_comments[0]["body"]) diff --git a/tests/test_mcp_session_state.py b/tests/test_mcp_session_state.py new file mode 100644 index 0000000..ebfc0ce --- /dev/null +++ b/tests/test_mcp_session_state.py @@ -0,0 +1,207 @@ +"""Tests for durable MCP session state shared across daemon processes (#559).""" + +from __future__ import annotations + +import os +import tempfile +import unittest +from pathlib import Path +from unittest.mock import patch + +sys_path_root = str(Path(__file__).resolve().parent.parent) +import sys + +if sys_path_root not in sys.path: + sys.path.insert(0, sys_path_root) + +import mcp_session_state +import review_workflow_load +import mcp_server + + +class TestMcpSessionStateStore(unittest.TestCase): + def setUp(self): + self._tmpdir = tempfile.TemporaryDirectory() + self.state_dir = self._tmpdir.name + self._env = patch.dict( + os.environ, + { + mcp_session_state.STATE_DIR_ENV: self.state_dir, + mcp_session_state.SESSION_PROFILE_LOCK_ENV: "prgs-reviewer", + "GITEA_MCP_PROFILE": "prgs-reviewer", + }, + clear=False, + ) + self._env.start() + + def tearDown(self): + self._env.stop() + self._tmpdir.cleanup() + + def test_round_trip_same_identity(self): + saved = mcp_session_state.save_state( + kind=mcp_session_state.KIND_WORKFLOW_LOAD, + payload={"loaded": True, "workflow_hash": "abc123def456"}, + remote="prgs", + org="Scaled-Tech-Consulting", + repo="Gitea-Tools", + ) + self.assertIsNotNone(saved) + self.assertEqual(saved["workflow_hash"], "abc123def456") + self.assertIn("recorded_at", saved) + + loaded = mcp_session_state.load_state( + kind=mcp_session_state.KIND_WORKFLOW_LOAD, + remote="prgs", + org="Scaled-Tech-Consulting", + repo="Gitea-Tools", + ) + self.assertIsNotNone(loaded) + self.assertEqual(loaded["workflow_hash"], "abc123def456") + self.assertEqual(loaded["profile_identity"], "prgs-reviewer") + + def test_profile_mismatch_returns_none(self): + mcp_session_state.save_state( + kind=mcp_session_state.KIND_DECISION_LOCK, + payload={"final_review_decision_ready": True, "remote": "prgs"}, + remote="prgs", + ) + with patch.dict( + os.environ, + {mcp_session_state.SESSION_PROFILE_LOCK_ENV: "prgs-author"}, + clear=False, + ): + loaded = mcp_session_state.load_state( + kind=mcp_session_state.KIND_DECISION_LOCK, + remote="prgs", + ) + self.assertIsNone(loaded) + + def test_clear_removes_file(self): + mcp_session_state.save_state( + kind=mcp_session_state.KIND_WORKFLOW_LOAD, + payload={"loaded": True}, + remote="prgs", + ) + path = mcp_session_state.state_file_path( + kind=mcp_session_state.KIND_WORKFLOW_LOAD, + remote="prgs", + profile_identity="prgs-reviewer", + state_dir=self.state_dir, + ) + self.assertTrue(os.path.exists(path)) + mcp_session_state.clear_state( + kind=mcp_session_state.KIND_WORKFLOW_LOAD, + remote="prgs", + profile_identity="prgs-reviewer", + ) + self.assertFalse(os.path.exists(path)) + + def test_files_are_private_mode(self): + mcp_session_state.save_state( + kind=mcp_session_state.KIND_WORKFLOW_LOAD, + payload={"loaded": True}, + remote="prgs", + ) + path = mcp_session_state.state_file_path( + kind=mcp_session_state.KIND_WORKFLOW_LOAD, + remote="prgs", + profile_identity="prgs-reviewer", + state_dir=self.state_dir, + ) + mode = os.stat(path).st_mode & 0o777 + self.assertEqual(mode, 0o600) + + +class TestWorkflowLoadCrossProcess(unittest.TestCase): + def setUp(self): + self._tmpdir = tempfile.TemporaryDirectory() + self._env = patch.dict( + os.environ, + { + mcp_session_state.STATE_DIR_ENV: self._tmpdir.name, + mcp_session_state.SESSION_PROFILE_LOCK_ENV: "prgs-reviewer", + "GITEA_MCP_PROFILE": "prgs-reviewer", + "GITEA_MCP_REMOTE": "prgs", + }, + clear=False, + ) + self._env.start() + review_workflow_load.clear_review_workflow_load() + + def tearDown(self): + review_workflow_load.clear_review_workflow_load() + self._env.stop() + self._tmpdir.cleanup() + + def test_different_pid_same_profile_accepted(self): + root = str(Path(__file__).resolve().parent.parent) + recorded = review_workflow_load.record_review_workflow_load(root) + self.assertTrue(recorded["loaded"]) + + # Simulate another daemon process: clear memory, keep durable file, + # and change apparent PID identity only (profile remains the same). + review_workflow_load._REVIEW_WORKFLOW_LOAD = None + status = review_workflow_load.workflow_load_status(root) + self.assertTrue(status["workflow_load_proof_present"]) + self.assertTrue( + status["workflow_load_valid"], + msg=status.get("reasons"), + ) + + def test_profile_mismatch_blocks(self): + root = str(Path(__file__).resolve().parent.parent) + review_workflow_load.record_review_workflow_load(root) + review_workflow_load._REVIEW_WORKFLOW_LOAD = None + with patch.dict( + os.environ, + {mcp_session_state.SESSION_PROFILE_LOCK_ENV: "prgs-author"}, + clear=False, + ): + # Durable load uses active identity; mismatched profile key misses. + status = review_workflow_load.workflow_load_status(root) + self.assertFalse(status["workflow_load_proof_present"]) + + +class TestDecisionLockCrossProcess(unittest.TestCase): + def setUp(self): + self._tmpdir = tempfile.TemporaryDirectory() + self._env = patch.dict( + os.environ, + { + mcp_session_state.STATE_DIR_ENV: self._tmpdir.name, + mcp_session_state.SESSION_PROFILE_LOCK_ENV: "prgs-reviewer", + }, + clear=False, + ) + self._env.start() + mcp_server._save_review_decision_lock(None) + review_workflow_load.clear_review_workflow_load() + + def tearDown(self): + mcp_server._save_review_decision_lock(None) + review_workflow_load.clear_review_workflow_load() + self._env.stop() + self._tmpdir.cleanup() + + def test_decision_lock_survives_memory_clear(self): + with patch.object(mcp_server, "get_profile", return_value={ + "profile_name": "prgs-reviewer", + }): + mcp_server.init_review_decision_lock("prgs", "review_pr", force=True) + lock = mcp_server._load_review_decision_lock() + self.assertIsNotNone(lock) + self.assertEqual(lock["remote"], "prgs") + self.assertFalse(lock["final_review_decision_ready"]) + + # New process: memory empty, durable state remains. + mcp_server._REVIEW_DECISION_LOCK = None + restored = mcp_server._load_review_decision_lock() + self.assertIsNotNone(restored) + self.assertEqual(restored["remote"], "prgs") + reasons = mcp_server._review_decision_session_reasons(restored) + self.assertEqual(reasons, []) + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/test_mcp_stale_runtime.py b/tests/test_mcp_stale_runtime.py new file mode 100644 index 0000000..19310c2 --- /dev/null +++ b/tests/test_mcp_stale_runtime.py @@ -0,0 +1,71 @@ +import os +import unittest +from unittest.mock import patch, MagicMock +from datetime import datetime + +import gitea_mcp_server + +class TestMcpStaleRuntime(unittest.TestCase): + @patch("subprocess.run") + @patch("os.path.getmtime") + @patch("os.path.exists") + @patch("os.getpid") + @patch.dict("os.environ", {"GITEA_FORCE_MCP_RUNTIME_CHECK": "1"}) + def test_stale_and_missing_runtimes(self, mock_getpid, mock_exists, mock_getmtime, mock_run): + # Setup mocks + mock_getpid.return_value = 12345 + mock_exists.return_value = True + + # Code modification time: Jul 8 2026, 14:00:00 + code_time = datetime(2026, 7, 8, 14, 0, 0) + mock_getmtime.return_value = code_time.timestamp() + + # Mock ps -ax output + # PID 12345 is self (started at 13:00:00 - stale) + # PID 54321 is prgs-author (started at 15:00:00 - fresh) + # prgs-reviewer is missing + ps_output = ( + " PID LSTART COMMAND\n" + "12345 Wed Jul 8 13:00:00 2026 /path/to/python mcp_server.py\n" + "54321 Wed Jul 8 15:00:00 2026 /path/to/python mcp_server.py\n" + ) + + mock_run_ps = MagicMock() + mock_run_ps.stdout = ps_output + + # Mock env output for ps eww + mock_run_env12345 = MagicMock() + mock_run_env12345.stdout = "GITEA_MCP_PROFILE=prgs-reconciler" + + mock_run_env54321 = MagicMock() + mock_run_env54321.stdout = "GITEA_MCP_PROFILE=prgs-author" + + def side_effect(args, **kwargs): + if args[0] == "ps" and "eww" in args: + pid = args[2] + if pid == "12345": + return mock_run_env12345 + elif pid == "54321": + return mock_run_env54321 + elif args[0] == "ps": + return mock_run_ps + raise ValueError(f"Unexpected subprocess run args: {args}") + + mock_run.side_effect = side_effect + + # Test 1: required reviewer role matching prgs-reviewer (which is missing) + reasons = gitea_mcp_server._check_mcp_runtimes_diagnostics("review_pr", ["prgs-reviewer"]) + + self.assertTrue(any("stale-runtime: The active Gitea MCP server process is stale" in r for r in reasons)) + self.assertTrue(any("stale-runtime: None of the matching profiles for task" in r for r in reasons)) + + # Test 2: required author role matching prgs-author (which is fresh) + reasons_author = gitea_mcp_server._check_mcp_runtimes_diagnostics("create_issue", ["prgs-author"]) + + # Still contains self stale error + self.assertTrue(any("stale-runtime: The active Gitea MCP server process is stale" in r for r in reasons_author)) + # But does NOT contain missing author profile error + self.assertFalse(any("None of the matching profiles for task" in r for r in reasons_author)) + +if __name__ == "__main__": + unittest.main() diff --git a/tests/test_merger_lease_adoption.py b/tests/test_merger_lease_adoption.py new file mode 100644 index 0000000..8cd681d --- /dev/null +++ b/tests/test_merger_lease_adoption.py @@ -0,0 +1,431 @@ +"""Merger lease adoption / recovery (#536).""" + +import os +import sys +import unittest +from datetime import datetime, timezone +from unittest.mock import patch + +sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent)) + +import merger_lease_adoption as mla +import reviewer_pr_lease as leases + + +def _lease_comment( + pr_number: int, + session_id: str, + *, + phase: str = "claimed", + profile: str = "prgs-reviewer", + identity: str = "sysadmin", + candidate_head: str = "a" * 40, + comment_id: int = 100, +) -> dict: + body = leases.format_lease_body( + repo="Scaled-Tech-Consulting/Gitea-Tools", + pr_number=pr_number, + issue_number=536, + reviewer_identity=identity, + profile=profile, + session_id=session_id, + worktree="branches/review-pr536", + phase=phase, + candidate_head=candidate_head, + target_branch="master", + target_branch_sha="b" * 40, + last_activity=datetime.now(timezone.utc), + ) + return {"id": comment_id, "body": body, "user": {"login": identity}} + + +HEAD = "f" * 40 + + +class TestSanctionedProvenance(unittest.TestCase): + def setUp(self): + leases.clear_session_lease() + + def test_manual_session_seed_rejected_by_mutation_gate(self): + comments = [_lease_comment(536, "reviewer-session", candidate_head=HEAD)] + leases.record_session_lease({ + "pr_number": 536, + "session_id": "merger-session", + "candidate_head": HEAD, + "comment_id": 999, + }) + result = leases.assess_mutation_lease_gate( + pr_number=536, + comments=comments, + reviewer_identity="sysadmin", + session_id="merger-session", + mutation="merge", + live_head_sha=HEAD, + pinned_head_sha=HEAD, + ) + self.assertTrue(result["block"]) + self.assertTrue( + any("sanctioned provenance" in r for r in result["reasons"]) + ) + + def test_sanctioned_acquire_provenance_allows_mutation(self): + comments = [_lease_comment(536, "my-session", candidate_head=HEAD)] + leases.record_session_lease( + { + "pr_number": 536, + "session_id": "my-session", + "candidate_head": HEAD, + "comment_id": 101, + }, + lease_provenance=mla.build_lease_provenance( + source=mla.SOURCE_ACQUIRE, + comment_id=101, + ), + ) + result = leases.assess_mutation_lease_gate( + pr_number=536, + comments=comments, + reviewer_identity="sysadmin", + session_id="my-session", + mutation="merge", + live_head_sha=HEAD, + pinned_head_sha=HEAD, + ) + self.assertFalse(result["block"]) + + +class TestAdoptAssessment(unittest.TestCase): + def test_adopt_allowed_for_merger_with_reviewer_lease_and_approval(self): + comments = [_lease_comment(536, "reviewer-sess", candidate_head=HEAD)] + result = mla.assess_adopt_merger_lease( + comments, + pr_number=536, + adopter_identity="sysadmin", + adopter_profile="prgs-merger", + adopter_session_id="merger-sess", + repo="Scaled-Tech-Consulting/Gitea-Tools", + issue_number=536, + worktree="branches/merge-pr536", + expected_head_sha=HEAD, + live_head_sha=HEAD, + approval_at_head=True, + pr_open=True, + ) + self.assertTrue(result["adopt_allowed"]) + self.assertIn(mla.ADOPTION_MARKER, result["adoption_body"]) + self.assertIn("adopted_from_session_id: reviewer-sess", result["adoption_body"]) + + def test_adopt_blocked_without_approval(self): + comments = [_lease_comment(536, "reviewer-sess", candidate_head=HEAD)] + result = mla.assess_adopt_merger_lease( + comments, + pr_number=536, + adopter_identity="sysadmin", + adopter_profile="prgs-merger", + adopter_session_id="merger-sess", + repo="Scaled-Tech-Consulting/Gitea-Tools", + issue_number=536, + worktree="branches/merge-pr536", + expected_head_sha=HEAD, + live_head_sha=HEAD, + approval_at_head=False, + pr_open=True, + ) + self.assertFalse(result["adopt_allowed"]) + self.assertTrue(any("APPROVED" in r for r in result["reasons"])) + + def test_adopt_blocked_for_non_merger_profile(self): + comments = [_lease_comment(536, "reviewer-sess", candidate_head=HEAD)] + result = mla.assess_adopt_merger_lease( + comments, + pr_number=536, + adopter_identity="sysadmin", + adopter_profile="prgs-reviewer", + adopter_session_id="merger-sess", + repo="Scaled-Tech-Consulting/Gitea-Tools", + issue_number=536, + worktree="branches/review-pr536", + expected_head_sha=HEAD, + live_head_sha=HEAD, + approval_at_head=True, + pr_open=True, + ) + self.assertFalse(result["adopt_allowed"]) + self.assertTrue(any("merger profile" in r for r in result["reasons"])) + + def test_adopt_blocked_when_pr_not_open(self): + comments = [_lease_comment(536, "reviewer-sess", candidate_head=HEAD)] + result = mla.assess_adopt_merger_lease( + comments, + pr_number=536, + adopter_identity="sysadmin", + adopter_profile="prgs-merger", + adopter_session_id="merger-sess", + repo="Scaled-Tech-Consulting/Gitea-Tools", + issue_number=536, + worktree="branches/merge-pr536", + expected_head_sha=HEAD, + live_head_sha=HEAD, + approval_at_head=True, + pr_open=False, + ) + self.assertFalse(result["adopt_allowed"]) + + +class TestMergerHandoffMutationGate(unittest.TestCase): + def setUp(self): + leases.clear_session_lease() + + def test_cross_session_merge_blocked_without_adoption_comment(self): + reviewer_comment = _lease_comment(536, "reviewer-sess", candidate_head=HEAD) + provenance = mla.build_lease_provenance( + source=mla.SOURCE_ADOPT, + comment_id=7001, + adopted_from_session_id="reviewer-sess", + ) + leases.record_session_lease( + { + "pr_number": 536, + "session_id": "merger-sess", + "candidate_head": HEAD, + "comment_id": 7001, + "phase": "adopted", + }, + lease_provenance=provenance, + ) + result = leases.assess_mutation_lease_gate( + pr_number=536, + comments=[reviewer_comment], + reviewer_identity="sysadmin", + session_id="merger-sess", + mutation="merge", + live_head_sha=HEAD, + pinned_head_sha=HEAD, + ) + self.assertTrue(result["block"]) + self.assertTrue(any("owned by session_id=reviewer-sess" in r for r in result["reasons"])) + + def test_cross_session_merge_allowed_after_adoption_comment(self): + reviewer_comment = _lease_comment(536, "reviewer-sess", candidate_head=HEAD) + adoption_body = mla.format_adoption_body( + repo="Scaled-Tech-Consulting/Gitea-Tools", + pr_number=536, + issue_number=536, + adopter_identity="sysadmin", + adopter_profile="prgs-merger", + adopter_session_id="merger-sess", + worktree="branches/merge-pr536", + candidate_head=HEAD, + target_branch="master", + target_branch_sha="b" * 40, + adopted_from_session_id="reviewer-sess", + adopted_from_profile="prgs-reviewer", + adopted_from_reviewer_identity="sysadmin", + adopted_from_comment_id=100, + ) + adoption_comment = { + "id": 7001, + "body": adoption_body, + "user": {"login": "sysadmin"}, + } + provenance = mla.build_lease_provenance( + source=mla.SOURCE_ADOPT, + comment_id=7001, + adopted_from_session_id="reviewer-sess", + ) + leases.record_session_lease( + { + "pr_number": 536, + "session_id": "merger-sess", + "candidate_head": HEAD, + "comment_id": 7001, + "phase": "adopted", + }, + lease_provenance=provenance, + ) + result = leases.assess_mutation_lease_gate( + pr_number=536, + comments=[reviewer_comment, adoption_comment], + reviewer_identity="sysadmin", + session_id="merger-sess", + mutation="merge", + live_head_sha=HEAD, + pinned_head_sha=HEAD, + ) + self.assertFalse(result["block"]) + + +class TestAdoptTool(unittest.TestCase): + def setUp(self): + leases.clear_session_lease() + # Patch only the role workspace verifier (which internally does the single + # verify_preflight_purity); the direct duplicate call was removed to prevent + # capability state clearing. See fix for #548. + patch("mcp_server._verify_role_mutation_workspace", return_value=None).start() + patch("gitea_audit.audit_enabled", return_value=False).start() + mcp_server = __import__("mcp_server") + mcp_server._IDENTITY_CACHE.clear() + mcp_server.init_review_decision_lock("prgs", "adopt_merger_pr_lease") + mcp_server.record_preflight_check("whoami") + mcp_server.record_preflight_check( + "capability", "reviewer", resolved_task="adopt_merger_pr_lease" + ) + + def tearDown(self): + patch.stopall() + leases.clear_session_lease() + + @patch("mcp_server._authenticated_username", return_value="sysadmin") + @patch("mcp_server.get_auth_header", return_value="Basic dGVzdDp0ZXN0") + @patch("mcp_server.get_profile") + @patch("mcp_server.api_request") + def test_adopt_tool_posts_proof_and_records_sanctioned_session( + self, mock_api, mock_profile, _mock_auth, _mock_user + ): + mock_profile.return_value = { + "profile_name": "prgs-merger", + "allowed_operations": [ + "gitea.read", + "gitea.pr.comment", + "gitea.pr.merge", + ], + "forbidden_operations": [], + } + reviewer_comment = _lease_comment(536, "reviewer-sess", candidate_head=HEAD) + posted_bodies: list[str] = [] + + def _side(method, url, auth=None, payload=None, *a, **k): + m = (method or "").upper() + if m == "POST": + if payload and payload.get("body"): + posted_bodies.append(payload["body"]) + return {"id": 7001} + if "/pulls/" in url and "/files" not in url: + return { + "state": "open", + "number": 536, + "head": {"sha": HEAD}, + "merged": False, + } + if "/reviews" in url: + return [] + if "/comments" in url: + return [reviewer_comment] + return {} + + mock_api.side_effect = _side + from mcp_server import gitea_adopt_merger_pr_lease + + with patch.object( + __import__("mcp_server"), + "gitea_get_pr_review_feedback", + return_value={ + "approval_at_current_head": True, + "latest_approved_head_sha": HEAD, + }, + ): + with patch.dict( + os.environ, + { + "GITEA_PROFILE_NAME": "prgs-merger", + "GITEA_ALLOWED_OPERATIONS": "gitea.read,gitea.pr.comment,gitea.pr.merge", + }, + clear=True, + ): + result = gitea_adopt_merger_pr_lease( + pr_number=536, + worktree="branches/merge-pr536", + expected_head_sha=HEAD, + issue_number=536, + remote="prgs", + ) + self.assertTrue(result["success"]) + self.assertTrue(result["adopted"]) + self.assertEqual(result["adopted_from_session_id"], "reviewer-sess") + self.assertEqual(result["adoption_comment_id"], 7001) + self.assertEqual(len(posted_bodies), 1) + self.assertIn(mla.ADOPTION_MARKER, posted_bodies[0]) + + session = leases.get_session_lease() + self.assertTrue(mla.is_sanctioned_session_lease(session)) + + adoption_comment = { + "id": 7001, + "body": posted_bodies[0], + "user": {"login": "sysadmin"}, + } + gate = leases.assess_mutation_lease_gate( + pr_number=536, + comments=[reviewer_comment, adoption_comment], + reviewer_identity="sysadmin", + session_id=result["session_id"], + mutation="merge", + live_head_sha=HEAD, + pinned_head_sha=HEAD, + ) + self.assertFalse(gate["block"]) + + def test_adopt_preserves_preflight_capability_state_no_duplicate_clear(self): + """Prove fix for #548: adopt calls _verify (single purity) and does not + trigger duplicate verify that would clear capability state, avoiding + need for temp local patches or raw fallbacks. + """ + from unittest.mock import patch, MagicMock + import mcp_server as mcp_server_mod + + # fresh preflight state (as controller/reconciler would have from resolve) + mcp_server_mod.record_preflight_check("whoami") + mcp_server_mod.record_preflight_check( + "capability", "merger", resolved_task="adopt_merger_pr_lease" + ) + initial_cap_called = mcp_server_mod._preflight_capability_called + + verify_patch = patch("mcp_server.verify_preflight_purity", wraps=mcp_server_mod.verify_preflight_purity) + def _role_side(*a, **k): + # simulate the real _verify which calls verify once with path + mcp_server_mod.verify_preflight_purity(k.get("remote") or a[0] if a else None, worktree_path=k.get("worktree") or "branches/work", task=k.get("task")) + return "branches/work" + role_patch = patch("mcp_server._verify_role_mutation_workspace", side_effect=_role_side) + with verify_patch as vmock, role_patch, \ + patch("mcp_server.get_auth_header", return_value="token test"), \ + patch("mcp_server._authenticated_username", return_value="sysadmin"): + # minimal mocks to let adopt reach end without full api + with patch("mcp_server.get_profile") as gp: + gp.return_value = { + "profile_name": "prgs-merger", + "allowed_operations": ["gitea.read", "gitea.pr.comment", "gitea.pr.merge"], + "forbidden_operations": [], + } + with patch("mcp_server.api_request") as api: + def _side(url, *a, **k): + if "/pulls/" in str(url) and "/comments" not in str(url): + return {"state": "open", "head": {"sha": "deadbeef"*5}, "merged": False, "number": 999} + if "/comments" in str(url): + return [] # list of comments for fetch + return {"id": 42} + api.side_effect = _side + with patch("mcp_server.gitea_get_pr_review_feedback", return_value={"approval_at_current_head": True}): + with patch("merger_lease_adoption.assess_adopt_merger_lease", return_value={ + "adopt_allowed": True, + "active_lease": {"session_id": "rev", "profile": "prgs-reviewer", "reviewer_identity": "sysadmin", "issue_number": 548, "target_branch": "master"}, + "adoption_body": " adopted", + }): + res = mcp_server_mod.gitea_adopt_merger_pr_lease( + pr_number=999, + worktree="branches/merge-test", + expected_head_sha="deadbeef"*5, + issue_number=548, + remote="prgs", + ) + self.assertTrue(res.get("success")) + # verify_preflight_purity should have been invoked exactly once (via the _verify_role path) + # not twice (old dup would have cleared state mid-way, requiring temp patches) + self.assertEqual(vmock.call_count, 1) + # capability state management: the single verify consumes (clears) as designed; + # prior state was valid, no unexpected clear between "checks" or hidden reset + # (the removed dup line was the source of the #548 clearing bug) + self.assertTrue(initial_cap_called) + # final may be set by other records in flow, but key is single invocation and success without bypasses + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_namespace_workspace_binding.py b/tests/test_namespace_workspace_binding.py new file mode 100644 index 0000000..b9c270c --- /dev/null +++ b/tests/test_namespace_workspace_binding.py @@ -0,0 +1,240 @@ +"""Tests for namespace-scoped MCP workspace binding (#510).""" + +from __future__ import annotations + +import os +import sys +import unittest +from pathlib import Path +from unittest import mock +from unittest.mock import MagicMock + +sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) + +import gitea_mcp_server as srv # noqa: E402 +import namespace_workspace_binding as nwb # noqa: E402 + +REPO_ROOT = Path(__file__).resolve().parent.parent +if REPO_ROOT.parent.name == "branches": + CONTROL_ROOT = str(REPO_ROOT.parent.parent) +else: + CONTROL_ROOT = str(REPO_ROOT) + +AUTHOR_DIRTY = f"{CONTROL_ROOT}/branches/mcp-author-worktree" +MERGER_CLEAN = f"{CONTROL_ROOT}/branches/merge-pr487-submit" +REVIEWER_CLEAN = f"{CONTROL_ROOT}/branches/review-pr487-submit" +RECONCILER_CLEAN = f"{CONTROL_ROOT}/branches/reconcile-pr487" +MCP_PROCESS_ROOT = CONTROL_ROOT + + +class TestNamespaceWorkspaceModule(unittest.TestCase): + def test_author_env_ignored_for_merger_namespace(self): + workspace, source = nwb.resolve_namespace_workspace( + role_kind="merger", + worktree_path=None, + process_project_root=MCP_PROCESS_ROOT, + env={ + nwb.AUTHOR_WORKTREE_ENV: AUTHOR_DIRTY, + nwb.MERGER_WORKTREE_ENV: MERGER_CLEAN, + }, + profile_name="gitea-merger", + ) + self.assertEqual(workspace, os.path.realpath(MERGER_CLEAN)) + self.assertEqual(source, f"{nwb.MERGER_WORKTREE_ENV} environment variable") + + def test_author_env_ignored_for_reviewer_namespace(self): + workspace, source = nwb.resolve_namespace_workspace( + role_kind="reviewer", + worktree_path=None, + process_project_root=MCP_PROCESS_ROOT, + env={ + nwb.AUTHOR_WORKTREE_ENV: AUTHOR_DIRTY, + nwb.REVIEWER_WORKTREE_ENV: REVIEWER_CLEAN, + }, + ) + self.assertEqual(workspace, os.path.realpath(REVIEWER_CLEAN)) + self.assertEqual(source, f"{nwb.REVIEWER_WORKTREE_ENV} environment variable") + + def test_author_env_ignored_for_reconciler_namespace(self): + workspace, source = nwb.resolve_namespace_workspace( + role_kind="reconciler", + worktree_path=None, + process_project_root=MCP_PROCESS_ROOT, + env={ + nwb.AUTHOR_WORKTREE_ENV: AUTHOR_DIRTY, + nwb.RECONCILER_WORKTREE_ENV: RECONCILER_CLEAN, + }, + ) + self.assertEqual(workspace, os.path.realpath(RECONCILER_CLEAN)) + self.assertEqual(source, f"{nwb.RECONCILER_WORKTREE_ENV} environment variable") + + def test_merger_profile_maps_to_merger_namespace(self): + role = nwb.normalize_role_kind("reviewer", profile_name="gitea-merger") + self.assertEqual(role, "merger") + + def test_error_message_includes_path_and_binding_source(self): + msg = nwb.format_namespace_workspace_binding_error( + role_kind="merger", + workspace_path=AUTHOR_DIRTY, + binding_source=f"{nwb.AUTHOR_WORKTREE_ENV} environment variable", + dirty_files=["gitea_mcp_server.py"], + ignored_bindings=[f"{nwb.AUTHOR_WORKTREE_ENV}={AUTHOR_DIRTY} (ignored for merger namespace)"], + ) + self.assertIn(AUTHOR_DIRTY, msg) + self.assertIn("via", msg.lower()) + self.assertIn(nwb.AUTHOR_WORKTREE_ENV, msg) + self.assertIn("Do not clean or reset foreign role worktrees", msg) + + +class TestNamespaceWorkspaceIntegration(unittest.TestCase): + def setUp(self): + self._saved = { + "whoami_called": srv._preflight_whoami_called, + "capability_called": srv._preflight_capability_called, + "resolved_role": srv._preflight_resolved_role, + "whoami_violation": srv._preflight_whoami_violation, + "capability_violation": srv._preflight_capability_violation, + "in_test": srv._preflight_in_test_mode, + } + srv._preflight_whoami_called = True + srv._preflight_capability_called = True + srv._preflight_whoami_violation = False + srv._preflight_capability_violation = False + srv._preflight_in_test_mode = lambda: False + self._env_patch = mock.patch.dict(os.environ, {"GITEA_TEST_PORCELAIN": ""}, clear=False) + self._env_patch.start() + + def tearDown(self): + srv._preflight_whoami_called = self._saved["whoami_called"] + srv._preflight_capability_called = self._saved["capability_called"] + srv._preflight_resolved_role = self._saved["resolved_role"] + srv._preflight_whoami_violation = self._saved["whoami_violation"] + srv._preflight_capability_violation = self._saved["capability_violation"] + srv._preflight_in_test_mode = self._saved["in_test"] + self._env_patch.stop() + for key in ( + nwb.AUTHOR_WORKTREE_ENV, + nwb.MERGER_WORKTREE_ENV, + nwb.REVIEWER_WORKTREE_ENV, + nwb.RECONCILER_WORKTREE_ENV, + nwb.ACTIVE_WORKTREE_ENV, + ): + os.environ.pop(key, None) + + def _merger_profile(self): + return { + "profile_name": "gitea-merger", + "allowed_operations": ["gitea.pr.merge", "gitea.read"], + "forbidden_operations": ["gitea.pr.create", "gitea.branch.push"], + } + + def _reviewer_profile(self): + return { + "profile_name": "prgs-reviewer", + "allowed_operations": ["gitea.pr.approve", "gitea.pr.review", "gitea.read"], + "forbidden_operations": ["gitea.pr.create", "gitea.branch.push"], + } + + def _reconciler_profile(self): + return { + "profile_name": "prgs-reconciler", + "allowed_operations": ["gitea.pr.close", "gitea.read"], + "forbidden_operations": [ + "gitea.pr.approve", + "gitea.pr.merge", + "gitea.pr.create", + ], + } + + def _author_profile(self): + return { + "profile_name": "prgs-author", + "allowed_operations": ["gitea.pr.create", "gitea.branch.push", "gitea.read"], + "forbidden_operations": ["gitea.pr.merge", "gitea.pr.approve"], + } + + @mock.patch("subprocess.run") + @mock.patch("os.path.isdir", return_value=True) + @mock.patch("os.path.exists", return_value=True) + def test_dirty_author_worktree_does_not_block_merger_with_clean_workspace( + self, _exists, _isdir, mock_run + ): + mock_run.return_value = MagicMock(returncode=0, stdout=f"{CONTROL_ROOT}/.git\n") + os.environ[nwb.AUTHOR_WORKTREE_ENV] = AUTHOR_DIRTY + os.environ[nwb.MERGER_WORKTREE_ENV] = MERGER_CLEAN + srv._preflight_resolved_role = "reviewer" + with mock.patch.object(srv, "PROJECT_ROOT", MCP_PROCESS_ROOT): + with mock.patch("gitea_mcp_server.get_profile", return_value=self._merger_profile()): + srv.verify_preflight_purity("prgs", worktree_path=MERGER_CLEAN) + + @mock.patch("subprocess.run") + @mock.patch("os.path.isdir", return_value=True) + @mock.patch("os.path.exists", return_value=True) + def test_dirty_author_worktree_does_not_block_reviewer_with_clean_workspace( + self, _exists, _isdir, mock_run + ): + mock_run.return_value = MagicMock(returncode=0, stdout=f"{CONTROL_ROOT}/.git\n") + os.environ[nwb.AUTHOR_WORKTREE_ENV] = AUTHOR_DIRTY + os.environ[nwb.REVIEWER_WORKTREE_ENV] = REVIEWER_CLEAN + srv._preflight_resolved_role = "reviewer" + with mock.patch.object(srv, "PROJECT_ROOT", MCP_PROCESS_ROOT): + with mock.patch("gitea_mcp_server.get_profile", return_value=self._reviewer_profile()): + srv.verify_preflight_purity("prgs", worktree_path=REVIEWER_CLEAN) + + @mock.patch("subprocess.run") + @mock.patch("os.path.isdir", return_value=True) + @mock.patch("os.path.exists", return_value=True) + def test_dirty_author_worktree_does_not_block_reconciler_with_clean_workspace( + self, _exists, _isdir, mock_run + ): + mock_run.return_value = MagicMock(returncode=0, stdout=f"{CONTROL_ROOT}/.git\n") + os.environ[nwb.AUTHOR_WORKTREE_ENV] = AUTHOR_DIRTY + os.environ[nwb.RECONCILER_WORKTREE_ENV] = RECONCILER_CLEAN + srv._preflight_resolved_role = "reconciler" + with mock.patch.object(srv, "PROJECT_ROOT", MCP_PROCESS_ROOT): + with mock.patch("gitea_mcp_server.get_profile", return_value=self._reconciler_profile()): + srv.verify_preflight_purity("prgs", worktree_path=RECONCILER_CLEAN) + + @mock.patch("subprocess.run") + @mock.patch("os.path.isdir", return_value=True) + @mock.patch("os.path.exists", return_value=True) + def test_dirty_active_task_workspace_still_blocks_mutations( + self, _exists, _isdir, mock_run + ): + mock_run.return_value = MagicMock(returncode=0, stdout=f"{CONTROL_ROOT}/.git\n") + os.environ[nwb.MERGER_WORKTREE_ENV] = MERGER_CLEAN + srv._preflight_resolved_role = "reviewer" + dirty = " M namespace_workspace_binding.py\n" + with mock.patch.object(srv, "PROJECT_ROOT", MCP_PROCESS_ROOT): + with mock.patch("gitea_mcp_server.get_profile", return_value=self._merger_profile()): + with mock.patch("gitea_mcp_server._get_workspace_porcelain", return_value=dirty): + with self.assertRaises(RuntimeError) as ctx: + srv.verify_preflight_purity("prgs", worktree_path=MERGER_CLEAN) + self.assertIn(MERGER_CLEAN, str(ctx.exception)) + self.assertIn("binding", str(ctx.exception).lower()) + + def test_root_workspace_mutation_still_blocked_for_author(self): + srv._preflight_resolved_role = "author" + with mock.patch.object(srv, "PROJECT_ROOT", CONTROL_ROOT): + with mock.patch("gitea_mcp_server.get_profile", return_value=self._author_profile()): + with mock.patch( + "gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", + return_value={"current_branch": "master"}, + ): + with self.assertRaises(RuntimeError) as ctx: + srv.verify_preflight_purity("prgs") + self.assertIn("stable control checkout", str(ctx.exception)) + + @mock.patch("subprocess.run") + @mock.patch("os.path.isdir", return_value=True) + @mock.patch("os.path.exists", return_value=True) + def test_pr487_style_merge_binds_clean_merger_workspace( + self, _exists, _isdir, mock_run + ): + mock_run.return_value = MagicMock(returncode=0, stdout=f"{CONTROL_ROOT}/.git\n") + os.environ[nwb.AUTHOR_WORKTREE_ENV] = AUTHOR_DIRTY + srv._preflight_resolved_role = "reviewer" + with mock.patch.object(srv, "PROJECT_ROOT", MCP_PROCESS_ROOT): + with mock.patch("gitea_mcp_server.get_profile", return_value=self._merger_profile()): + resolved = srv._verify_role_mutation_workspace("prgs") + self.assertEqual(resolved, os.path.realpath(MCP_PROCESS_ROOT)) \ No newline at end of file diff --git a/tests/test_permission_reports.py b/tests/test_permission_reports.py index 279dc2b..b06e416 100644 --- a/tests/test_permission_reports.py +++ b/tests/test_permission_reports.py @@ -95,10 +95,15 @@ class PermissionReportBase(unittest.TestCase): self._dir = tempfile.TemporaryDirectory() self.config_path = os.path.join(self._dir.name, "profiles.json") self._write_config(CONFIG) + import review_workflow_load + review_workflow_load.record_review_workflow_load(mcp_server.PROJECT_ROOT) def tearDown(self): + import review_workflow_load + review_workflow_load.clear_review_workflow_load() self._remotes.stop() mcp_server._IDENTITY_CACHE.clear() + mcp_server.review_workflow_load.clear_review_workflow_load() gitea_config._active_profile_override = None self._dir.cleanup() @@ -250,6 +255,8 @@ class TestEligibilityDenialReport(PermissionReportBase): return {"login": "author-user"} return PR_PAYLOAD mock_api.side_effect = fake_api + mcp_server.init_review_decision_lock("prgs", "review_pr") + mcp_server.gitea_load_review_workflow() with patch.dict(os.environ, self._env("author-profile")): res = mcp_server.gitea_merge_pr( pr_number=42, confirmation="MERGE PR 42", diff --git a/tests/test_post_merge_cleanup_proof.py b/tests/test_post_merge_cleanup_proof.py new file mode 100644 index 0000000..10e5bf6 --- /dev/null +++ b/tests/test_post_merge_cleanup_proof.py @@ -0,0 +1,169 @@ +"""Tests for post-merge cleanup proof enforcement (#402).""" +import sys +import unittest +from pathlib import Path + +sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) + +from final_report_validator import assess_final_report_validator # noqa: E402 +from post_merge_cleanup_proof import ( # noqa: E402 + CLEANUP_SKIPPED, + assess_post_merge_cleanup_proof, +) + +MERGE_SHA = "a" * 40 +HEAD_BRANCH = "feat/issue-274-branches-only-worktrees" +WORKTREE = "branches/review-pr374-conflicts" + + +def _full_remote_cleanup_report(**overrides): + fields = { + "Task": "review PR #374", + "Merge result": "merged", + "Merge commit SHA": MERGE_SHA, + "Cleanup status": "remote branch deleted", + "Delete-branch capability resolved": "gitea.branch.delete allowed via delete_branch task", + "Merged PR head branch": HEAD_BRANCH, + "Deleted branch": HEAD_BRANCH, + "Branch protection": "none", + "Open PR inventory proof": "no other open PR references branch (inventory complete)", + "Active heartbeat/claim/lease": "none", + "Cleanup mutations": "gitea_delete_branch on remote head branch", + } + fields.update(overrides) + lines = ["## Controller Handoff", ""] + lines.extend(f"- {key}: {value}" for key, value in fields.items()) + return "\n".join(lines) + + +def _full_worktree_cleanup_report(**overrides): + fields = { + "Task": "review PR #374", + "Merge result": "merged", + "Cleanup status": "local worktree removed", + "Removed worktree path": WORKTREE, + "Pre-removal tracked state": "clean", + "Pre-removal untracked state": "clean", + "Git worktree list after removal": "only main checkout listed", + "Cleanup mutations": "git worktree remove on session-owned review worktree", + } + fields.update(overrides) + lines = ["## Controller Handoff", ""] + lines.extend(f"- {key}: {value}" for key, value in fields.items()) + return "\n".join(lines) + + +class TestPostMergeCleanupProof(unittest.TestCase): + def test_no_cleanup_claim_passes(self): + report = "## Controller Handoff\n- Task: review PR #1\n- Merge result: merged" + result = assess_post_merge_cleanup_proof(report) + self.assertFalse(result["block"]) + + def test_missing_capability_proof_blocked(self): + report = _full_remote_cleanup_report( + **{"Delete-branch capability resolved": "delete succeeded"} + ) + result = assess_post_merge_cleanup_proof(report) + self.assertTrue(result["block"]) + self.assertTrue(any("capability" in r.lower() for r in result["reasons"])) + + def test_unmerged_pr_blocked(self): + report = _full_remote_cleanup_report(**{"Merge result": "not merged"}) + result = assess_post_merge_cleanup_proof(report) + self.assertTrue(result["block"]) + self.assertTrue(any("merge result" in r.lower() for r in result["reasons"])) + + def test_wrong_branch_blocked(self): + report = _full_remote_cleanup_report( + **{"Deleted branch": "feat/other-branch"} + ) + report += "\nDeleted branch does not match merged PR head branch" + result = assess_post_merge_cleanup_proof(report) + self.assertTrue(result["block"]) + + def test_protected_branch_blocked(self): + report = _full_remote_cleanup_report(**{"Branch protection": "enabled"}) + result = assess_post_merge_cleanup_proof(report) + self.assertTrue(result["block"]) + + def test_open_pr_reference_blocked(self): + report = _full_remote_cleanup_report( + **{"Open PR inventory proof": "PR #999 still references branch"} + ) + result = assess_post_merge_cleanup_proof(report) + self.assertTrue(result["block"]) + + def test_active_claim_blocked(self): + report = _full_remote_cleanup_report( + **{"Active heartbeat/claim/lease": "status:in-progress on linked issue"} + ) + result = assess_post_merge_cleanup_proof(report) + self.assertTrue(result["block"]) + + def test_dirty_worktree_blocked(self): + report = _full_worktree_cleanup_report( + **{"Pre-removal tracked state": "dirty"} + ) + result = assess_post_merge_cleanup_proof(report) + self.assertTrue(result["block"]) + self.assertTrue(any("tracked" in r.lower() for r in result["reasons"])) + + def test_foreign_worktree_blocked(self): + report = _full_worktree_cleanup_report( + **{"Removed worktree path": "/tmp/foreign-worktree"} + ) + result = assess_post_merge_cleanup_proof(report) + self.assertTrue(result["block"]) + self.assertTrue(any("branches/" in r.lower() for r in result["reasons"])) + + def test_cleanup_skipped_with_blocker_passes(self): + report = "\n".join([ + "## Controller Handoff", + "- Cleanup outcome: CLEANUP_SKIPPED", + "- Cleanup blocker: gitea.branch.delete capability not resolved in active profile", + ]) + result = assess_post_merge_cleanup_proof(report) + self.assertFalse(result["block"]) + self.assertEqual(result["outcome"], CLEANUP_SKIPPED) + + def test_cleanup_skipped_without_blocker_blocked(self): + report = "## Controller Handoff\n- Cleanup outcome: CLEANUP_SKIPPED" + result = assess_post_merge_cleanup_proof(report) + self.assertTrue(result["block"]) + + def test_full_remote_cleanup_passes(self): + result = assess_post_merge_cleanup_proof(_full_remote_cleanup_report()) + self.assertFalse(result["block"]) + self.assertTrue(result["remote_delete_claimed"]) + + def test_full_worktree_cleanup_passes(self): + result = assess_post_merge_cleanup_proof(_full_worktree_cleanup_report()) + self.assertFalse(result["block"]) + self.assertTrue(result["worktree_remove_claimed"]) + + def test_validator_integration_blocks_unproven_delete(self): + report = ( + "## Controller Handoff\n" + "- Cleanup mutations: gitea_delete_branch deleted remote branch\n" + ) + result = assess_final_report_validator(report, task_kind="review_pr") + self.assertTrue(result["blocked"]) + rule_ids = [f["rule_id"] for f in result["findings"]] + self.assertIn("reviewer.post_merge_cleanup_proof", rule_ids) + + def test_validator_integration_allows_skipped(self): + report = "\n".join([ + "## Controller Handoff", + "- Cleanup outcome: CLEANUP_SKIPPED", + "- Cleanup blocker: branch still referenced by open PR #414", + ]) + result = assess_final_report_validator(report, task_kind="review_pr") + cleanup_findings = [ + f for f in result["findings"] + if f["rule_id"] == "reviewer.post_merge_cleanup_proof" + ] + self.assertEqual(cleanup_findings, []) + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_post_merge_moot_lease.py b/tests/test_post_merge_moot_lease.py new file mode 100644 index 0000000..a71cb72 --- /dev/null +++ b/tests/test_post_merge_moot_lease.py @@ -0,0 +1,257 @@ +"""Post-merge moot reviewer-lease handling (#515). + +Covers: + a. Reviewer-lease acquisition/adoption is refused on an already-merged/closed + PR (fail closed, no mutation). + b. The post-merge moot cleanup path is safe and idempotent. + c. An active foreign lease on an *open* PR is never force-cleaned. +""" + +import os +import sys +import unittest +from datetime import datetime, timezone +from unittest.mock import patch + +sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent)) + +import reviewer_pr_lease as leases # noqa: E402 +from mcp_server import ( # noqa: E402 + gitea_acquire_reviewer_pr_lease, + gitea_cleanup_post_merge_moot_lease, +) + +FAKE_AUTH = "Basic dGVzdDp0ZXN0" +MERGER_ENV = { + "GITEA_PROFILE_NAME": "prgs-merger", + "GITEA_ALLOWED_OPERATIONS": "gitea.read,gitea.pr.comment", +} +PR = 487 +ISSUE = 485 +SESSION = "97274-676d20a825c4" + + +def _lease_comment(pr_number=PR, session_id=SESSION, *, phase="claimed", + candidate_head="a" * 40): + body = leases.format_lease_body( + repo="Scaled-Tech-Consulting/Gitea-Tools", + pr_number=pr_number, + issue_number=ISSUE, + reviewer_identity="sysadmin", + profile="prgs-reviewer", + session_id=session_id, + worktree="branches/review-pr487", + phase=phase, + candidate_head=candidate_head, + target_branch="master", + target_branch_sha="b" * 40, + last_activity=datetime.now(timezone.utc), + ) + return {"id": 6603, "body": body, "user": {"login": "sysadmin"}} + + +# --------------------------------------------------------------------------- # +# Pure logic +# --------------------------------------------------------------------------- # +class TestAcquireRefusedOnMergedPR(unittest.TestCase): + def test_acquire_refused_when_pr_merged_or_closed(self): + result = leases.assess_acquire_lease( + [_lease_comment()], + pr_number=PR, + reviewer_identity="sysadmin", + profile="prgs-merger", + session_id="new-session", + repo="Scaled-Tech-Consulting/Gitea-Tools", + issue_number=ISSUE, + worktree="branches/merge-pr487", + candidate_head="a" * 40, + target_branch="master", + target_branch_sha="b" * 40, + pr_merged_or_closed=True, + ) + self.assertFalse(result["acquire_allowed"]) + self.assertTrue(result["post_merge_moot"]) + self.assertIsNone(result["lease_body"]) + self.assertTrue(any("post_merge_moot" in r for r in result["reasons"])) + + def test_acquire_still_allowed_on_open_pr_without_flag(self): + result = leases.assess_acquire_lease( + [], + pr_number=PR, + reviewer_identity="sysadmin", + profile="prgs-reviewer", + session_id="s1", + repo="Scaled-Tech-Consulting/Gitea-Tools", + issue_number=ISSUE, + worktree="branches/review-pr487", + candidate_head="a" * 40, + target_branch="master", + target_branch_sha="b" * 40, + ) + self.assertTrue(result["acquire_allowed"]) + self.assertFalse(result["post_merge_moot"]) + + +class TestPostMergeMootAssessment(unittest.TestCase): + def test_merged_pr_with_active_lease_is_moot_and_cleanable(self): + a = leases.assess_post_merge_moot_lease( + [_lease_comment()], + pr_number=PR, + pr_merged=True, + pr_state="closed", + merge_commit_sha="c" * 40, + ) + self.assertTrue(a["pr_merged_or_closed"]) + self.assertTrue(a["is_moot"]) + self.assertTrue(a["cleanup_allowed"]) + self.assertIsNotNone(a["release_body"]) + self.assertIn("phase: released", a["release_body"]) + self.assertIn("blocker: post-merge-moot", a["release_body"]) + + def test_open_pr_active_lease_never_cleaned(self): + a = leases.assess_post_merge_moot_lease( + [_lease_comment()], + pr_number=PR, + pr_merged=False, + pr_state="open", + ) + self.assertFalse(a["pr_merged_or_closed"]) + self.assertFalse(a["is_moot"]) + self.assertFalse(a["cleanup_allowed"]) + self.assertIsNone(a["release_body"]) + self.assertTrue(any("still open" in r for r in a["reasons"])) + + def test_merged_pr_without_lease_nothing_to_clean(self): + a = leases.assess_post_merge_moot_lease( + [], pr_number=PR, pr_merged=True, pr_state="closed") + self.assertTrue(a["pr_merged_or_closed"]) + self.assertFalse(a["is_moot"]) + self.assertFalse(a["cleanup_allowed"]) + self.assertTrue(any("nothing to clean" in r for r in a["reasons"])) + + def test_cleanup_is_idempotent(self): + """After the released marker is posted, a re-assess finds nothing to clean.""" + first = leases.assess_post_merge_moot_lease( + [_lease_comment()], pr_number=PR, pr_merged=True, pr_state="closed") + self.assertTrue(first["cleanup_allowed"]) + released_comment = { + "id": 7000, "body": first["release_body"], "user": {"login": "sysadmin"}} + second = leases.assess_post_merge_moot_lease( + [_lease_comment(), released_comment], + pr_number=PR, pr_merged=True, pr_state="closed") + self.assertFalse(second["is_moot"]) + self.assertFalse(second["cleanup_allowed"]) + + +# --------------------------------------------------------------------------- # +# Server tools +# --------------------------------------------------------------------------- # +def _api_side_effect(*, pr_state, pr_merged, comments, posted_id=9999): + """Build an api_request side effect keyed on method + url.""" + calls = {"post": []} + + def _side(method, url, auth=None, payload=None, *a, **k): + m = (method or "").upper() + if m == "POST": + calls["post"].append({"url": url, "payload": payload}) + return {"id": posted_id} + if "/comments" in url: + return list(comments) + if "/pulls/" in url: + pr = {"state": pr_state, "number": PR, "merge_commit_sha": "c" * 40} + if pr_merged: + pr["merged"] = True + pr["merged_at"] = "2026-07-08T07:46:04Z" + return pr + if "/issues/" in url: + return {"state": "closed" if pr_merged else "open", "number": ISSUE} + return {} + + return _side, calls + + +class TestAcquireToolRefusesMergedPR(unittest.TestCase): + def setUp(self): + leases.clear_session_lease() + + @patch("mcp_server.verify_preflight_purity", return_value=None) + @patch("mcp_server._authenticated_username", return_value="sysadmin") + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + @patch("mcp_server.api_request") + def test_acquire_tool_fails_closed_on_merged_pr_without_posting( + self, mock_api, _auth, _user, _purity): + side, calls = _api_side_effect( + pr_state="closed", pr_merged=True, comments=[]) + mock_api.side_effect = side + with patch.dict(os.environ, MERGER_ENV, clear=True): + result = gitea_acquire_reviewer_pr_lease( + pr_number=PR, + worktree="branches/merge-pr487", + candidate_head="a" * 40, + issue_number=ISSUE, + remote="prgs", + ) + self.assertFalse(result["success"]) + self.assertFalse(result["acquired"]) + self.assertTrue(result.get("post_merge_moot")) + self.assertEqual(calls["post"], [], "must not post a lease comment") + + +class TestCleanupTool(unittest.TestCase): + def setUp(self): + leases.clear_session_lease() + + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + @patch("mcp_server.api_request") + def test_read_only_reports_moot_without_mutating(self, mock_api, _auth): + side, calls = _api_side_effect( + pr_state="closed", pr_merged=True, comments=[_lease_comment()]) + mock_api.side_effect = side + with patch.dict(os.environ, MERGER_ENV, clear=True): + result = gitea_cleanup_post_merge_moot_lease( + pr_number=PR, apply=False, remote="prgs") + self.assertTrue(result["success"]) + self.assertTrue(result["pr_merged_or_closed"]) + self.assertTrue(result["lease_moot"]) + self.assertFalse(result["cleanup_performed"]) + self.assertTrue(result["no_merge_or_adoption"]) + self.assertEqual(result["mode"], "read_only") + self.assertEqual(calls["post"], []) + + @patch("mcp_server.verify_preflight_purity", return_value=None) + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + @patch("mcp_server.api_request") + def test_apply_posts_released_marker_on_merged_pr( + self, mock_api, _auth, _purity): + side, calls = _api_side_effect( + pr_state="closed", pr_merged=True, comments=[_lease_comment()]) + mock_api.side_effect = side + with patch.dict(os.environ, MERGER_ENV, clear=True): + result = gitea_cleanup_post_merge_moot_lease( + pr_number=PR, apply=True, remote="prgs") + self.assertTrue(result["success"]) + self.assertTrue(result["cleanup_performed"]) + self.assertEqual(result["released_comment_id"], 9999) + self.assertEqual(len(calls["post"]), 1) + self.assertIn("phase: released", calls["post"][0]["payload"]["body"]) + self.assertIn("post-merge-moot", calls["post"][0]["payload"]["body"]) + + @patch("mcp_server.verify_preflight_purity", return_value=None) + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + @patch("mcp_server.api_request") + def test_apply_refuses_to_clean_open_pr(self, mock_api, _auth, _purity): + side, calls = _api_side_effect( + pr_state="open", pr_merged=False, comments=[_lease_comment()]) + mock_api.side_effect = side + with patch.dict(os.environ, MERGER_ENV, clear=True): + result = gitea_cleanup_post_merge_moot_lease( + pr_number=PR, apply=True, remote="prgs") + self.assertFalse(result["cleanup_performed"]) + self.assertFalse(result["pr_merged_or_closed"]) + self.assertEqual(calls["post"], [], "never force-clean an open PR lease") + self.assertTrue( + any("still open" in r for r in result.get("cleanup_skipped_reason", []))) + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/test_pr_lease_comments_non_list_guard.py b/tests/test_pr_lease_comments_non_list_guard.py new file mode 100644 index 0000000..9b21a24 --- /dev/null +++ b/tests/test_pr_lease_comments_non_list_guard.py @@ -0,0 +1,81 @@ +"""Regression tests for non-list API payloads on PR/issue comment listing (#485).""" +import os +import sys +import unittest +from unittest.mock import patch + +sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent)) + +from mcp_server import ( # noqa: E402 + _list_pr_lease_comments, + gitea_list_issue_comments, +) + +FAKE_AUTH = "Basic dGVzdDp0ZXN0" +AUTHOR_ENV = { + "GITEA_PROFILE_NAME": "gitea-author", + "GITEA_ALLOWED_OPERATIONS": "gitea.read,gitea.issue.comment", +} + + +class TestPrLeaseCommentsNonListGuard(unittest.TestCase): + @patch("mcp_server.api_request") + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_list_pr_lease_comments_non_list_payload_returns_empty(self, _auth, mock_api): + mock_api.return_value = {"message": "Unauthorized"} + result = _list_pr_lease_comments( + 12, remote="prgs", host=None, org=None, repo=None) + self.assertEqual(result, []) + # Single comments GET only — no pre-flight PR lookup (#519 isolation). + mock_api.assert_called_once() + _method, url, _auth_arg = mock_api.call_args[0][:3] + self.assertIn("/issues/12/comments", url) + self.assertNotIn("/pulls/", url) + + @patch("mcp_server.api_request") + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_list_pr_lease_comments_none_returns_empty(self, _auth, mock_api): + mock_api.return_value = None + result = _list_pr_lease_comments( + 12, remote="prgs", host=None, org=None, repo=None) + self.assertEqual(result, []) + + @patch("mcp_server.api_request") + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_list_pr_lease_comments_list_payload_unchanged(self, _auth, mock_api): + comment = {"id": 7, "body": ""} + mock_api.return_value = [comment] + result = _list_pr_lease_comments( + 12, remote="prgs", host=None, org=None, repo=None, limit=5) + self.assertEqual(result, [comment]) + + @patch("mcp_server.api_request") + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_list_issue_comments_non_list_payload_returns_empty(self, _auth, mock_api): + mock_api.return_value = {"message": "Unauthorized"} + with patch.dict(os.environ, AUTHOR_ENV, clear=True): + result = gitea_list_issue_comments(issue_number=9, remote="prgs") + self.assertTrue(result["success"]) + self.assertEqual(result["comments"], []) + + @patch("mcp_server.api_request") + @patch("mcp_server.get_auth_header", return_value=FAKE_AUTH) + def test_list_issue_comments_list_payload_unchanged(self, _auth, mock_api): + mock_api.return_value = [ + { + "id": 101, + "user": {"login": "alice"}, + "body": "hello", + "created_at": "2026-07-03T00:00:00Z", + "updated_at": "2026-07-03T01:00:00Z", + } + ] + with patch.dict(os.environ, AUTHOR_ENV, clear=True): + result = gitea_list_issue_comments(issue_number=9, remote="prgs") + self.assertTrue(result["success"]) + self.assertEqual(len(result["comments"]), 1) + self.assertEqual(result["comments"][0]["author"], "alice") + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/test_pr_queue_inventory.py b/tests/test_pr_queue_inventory.py index a61cc3f..ddd9902 100644 --- a/tests/test_pr_queue_inventory.py +++ b/tests/test_pr_queue_inventory.py @@ -11,6 +11,7 @@ from mcp_server import ( gitea_view_pr, gitea_review_pr, gitea_check_pr_eligibility, + gitea_load_review_workflow, ) import gitea_config diff --git a/tests/test_preflight_read_survival.py b/tests/test_preflight_read_survival.py new file mode 100644 index 0000000..6da8af6 --- /dev/null +++ b/tests/test_preflight_read_survival.py @@ -0,0 +1,106 @@ +"""#469: capability preflight survives interleaved read-only whoami calls.""" + +import os +import unittest + +import gitea_mcp_server as mcp_server + + +class TestPreflightReadSurvival(unittest.TestCase): + def setUp(self): + self.orig_whoami = mcp_server._preflight_whoami_called + self.orig_capability = mcp_server._preflight_capability_called + self.orig_whoami_violation = mcp_server._preflight_whoami_violation + self.orig_capability_violation = mcp_server._preflight_capability_violation + self.orig_resolved_role = mcp_server._preflight_resolved_role + self.orig_resolved_task = mcp_server._preflight_resolved_task + self.orig_process_start = mcp_server._process_start_porcelain + self.orig_whoami_baseline = mcp_server._preflight_whoami_baseline_porcelain + self.orig_capability_baseline = mcp_server._preflight_capability_baseline_porcelain + for key in ("GITEA_TEST_FORCE_DIRTY", "GITEA_TEST_PORCELAIN"): + if key in os.environ: + del os.environ[key] + os.environ["GITEA_TEST_PORCELAIN"] = "" + mcp_server._preflight_whoami_called = False + mcp_server._preflight_capability_called = False + mcp_server._preflight_whoami_violation = False + mcp_server._preflight_capability_violation = False + mcp_server._preflight_resolved_role = None + mcp_server._preflight_resolved_task = None + mcp_server._process_start_porcelain = "" + mcp_server._preflight_whoami_baseline_porcelain = None + mcp_server._preflight_capability_baseline_porcelain = None + + def tearDown(self): + mcp_server._preflight_whoami_called = self.orig_whoami + mcp_server._preflight_capability_called = self.orig_capability + mcp_server._preflight_whoami_violation = self.orig_whoami_violation + mcp_server._preflight_capability_violation = self.orig_capability_violation + mcp_server._preflight_resolved_role = self.orig_resolved_role + mcp_server._preflight_resolved_task = self.orig_resolved_task + mcp_server._process_start_porcelain = self.orig_process_start + mcp_server._preflight_whoami_baseline_porcelain = self.orig_whoami_baseline + mcp_server._preflight_capability_baseline_porcelain = self.orig_capability_baseline + for key in ("GITEA_TEST_FORCE_DIRTY", "GITEA_TEST_PORCELAIN"): + if key in os.environ: + del os.environ[key] + + def test_interleaved_whoami_preserves_capability(self): + mcp_server.record_preflight_check("whoami") + mcp_server.record_preflight_check( + "capability", resolved_role="reconciler", resolved_task="close_pr" + ) + self.assertTrue(mcp_server._preflight_capability_called) + self.assertEqual(mcp_server._preflight_resolved_task, "close_pr") + + mcp_server.record_preflight_check("whoami") + self.assertTrue(mcp_server._preflight_capability_called) + self.assertEqual(mcp_server._preflight_resolved_task, "close_pr") + + mcp_server.verify_preflight_purity(task="close_pr") + self.assertFalse(mcp_server._preflight_capability_called) + + def test_missing_capability_still_fails_closed(self): + mcp_server.record_preflight_check("whoami") + with self.assertRaises(RuntimeError) as ctx: + mcp_server.verify_preflight_purity(task="close_pr") + self.assertIn("has not been resolved", str(ctx.exception)) + + def test_task_mismatch_fails_closed(self): + mcp_server.record_preflight_check("whoami") + mcp_server.record_preflight_check( + "capability", resolved_role="author", resolved_task="create_issue" + ) + with self.assertRaises(RuntimeError) as ctx: + mcp_server.verify_preflight_purity(task="close_pr") + self.assertIn("task mismatch", str(ctx.exception)) + + def test_capability_consumed_after_mutation_gate(self): + mcp_server.record_preflight_check("whoami") + mcp_server.record_preflight_check( + "capability", resolved_role="author", resolved_task="create_issue" + ) + mcp_server.verify_preflight_purity(task="create_issue") + with self.assertRaises(RuntimeError) as ctx: + mcp_server.verify_preflight_purity(task="create_issue") + self.assertIn("has not been resolved", str(ctx.exception)) + + def test_whoami_recovery_after_violation_clears_capability(self): + os.environ["GITEA_TEST_FORCE_DIRTY"] = "1" + mcp_server.record_preflight_check("whoami") + self.assertTrue(mcp_server._preflight_whoami_violation) + + del os.environ["GITEA_TEST_FORCE_DIRTY"] + os.environ["GITEA_TEST_PORCELAIN"] = "" + mcp_server.record_preflight_check("whoami") + self.assertFalse(mcp_server._preflight_whoami_violation) + self.assertFalse(mcp_server._preflight_capability_called) + + mcp_server.record_preflight_check( + "capability", resolved_role="reviewer", resolved_task="review_pr" + ) + mcp_server.verify_preflight_purity(task="review_pr") + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_reconciler_close_workspace_guard.py b/tests/test_reconciler_close_workspace_guard.py new file mode 100644 index 0000000..54a24e4 --- /dev/null +++ b/tests/test_reconciler_close_workspace_guard.py @@ -0,0 +1,89 @@ +"""Reconciler close_pr must not require author branches/ worktree (#468).""" +import os +import sys +import unittest +from pathlib import Path +from unittest.mock import patch + +sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) + +import gitea_mcp_server as srv + +FAKE_AUTH = "token test" +CONTROL_CHECKOUT_ROOT = str(Path(__file__).resolve().parents[3]) + +RECONCILER_PROFILE = { + "profile_name": "prgs-reconciler", + "allowed_operations": ["gitea.read", "gitea.pr.close", "gitea.pr.comment"], + "forbidden_operations": [ + "gitea.pr.approve", + "gitea.pr.merge", + "gitea.pr.review", + "gitea.pr.create", + "gitea.branch.push", + "gitea.repo.commit", + ], + "audit_label": "prgs-reconciler", +} + + +class TestReconcilerCloseWorkspaceGuard(unittest.TestCase): + def setUp(self): + srv._preflight_whoami_called = True + srv._preflight_capability_called = True + srv._preflight_whoami_violation = False + srv._preflight_capability_violation = False + self._orig_in_test = srv._preflight_in_test_mode + srv._preflight_in_test_mode = lambda: False + + def tearDown(self): + srv._preflight_in_test_mode = self._orig_in_test + + @patch("gitea_mcp_server._auth", return_value=FAKE_AUTH) + @patch("gitea_mcp_server._namespace_mutation_block", return_value=None) + @patch("gitea_mcp_server.get_profile", return_value=RECONCILER_PROFILE) + @patch("gitea_mcp_server.api_request") + def test_reconciler_close_pr_from_control_checkout_succeeds( + self, mock_api, _profile, _ns, _auth + ): + srv._preflight_resolved_role = "reconciler" + mock_api.return_value = { + "number": 414, + "title": "old", + "body": "", + "state": "closed", + "html_url": "https://gitea.example.com/pulls/414", + } + with patch.object(srv, "PROJECT_ROOT", CONTROL_CHECKOUT_ROOT): + with patch.dict(os.environ, {}, clear=False): + os.environ.pop("GITEA_AUTHOR_WORKTREE", None) + os.environ.pop("GITEA_ACTIVE_WORKTREE", None) + result = srv.gitea_edit_pr(414, state="closed", remote="prgs") + self.assertTrue(result["success"]) + self.assertEqual(result["state"], "closed") + mock_api.assert_called_once() + + @patch("gitea_mcp_server._auth", return_value=FAKE_AUTH) + @patch("gitea_mcp_server._profile_permission_block", return_value=None) + @patch("gitea_mcp_server._namespace_mutation_block", return_value=None) + @patch( + "gitea_mcp_server.role_session_router.check_author_mutation_after_reviewer_stop", + return_value=(True, []), + ) + @patch("gitea_mcp_server.api_get_all", return_value=[]) + @patch( + "gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", + return_value={"current_branch": "master"}, + ) + def test_author_create_issue_still_blocked_on_control_checkout( + self, _git, _get_all, _role, _ns, _prof, _auth + ): + srv._preflight_resolved_role = "author" + with patch.object(srv, "PROJECT_ROOT", CONTROL_CHECKOUT_ROOT): + with self.assertRaises(RuntimeError) as ctx: + srv.gitea_create_issue(title="Test", body="body") + self.assertIn("stable control checkout", str(ctx.exception)) + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_remote_repo_guard.py b/tests/test_remote_repo_guard.py new file mode 100644 index 0000000..9fbfa47 --- /dev/null +++ b/tests/test_remote_repo_guard.py @@ -0,0 +1,178 @@ +"""Regression coverage for the remote/repo mismatch guard (#530). + +Bare ``remote=prgs`` historically resolved to ``Scaled-Tech-Consulting/Timesheet`` +(the hardcoded ``REMOTES`` default) instead of ``Scaled-Tech-Consulting/Gitea-Tools``, +which is what the local git remote actually points at. That silent mismatch caused +false 404s and risked mutating the wrong repository. The guard fails closed when the +MCP-resolved org/repo disagrees with the local git remote URL and the caller did not +pass explicit ``org``/``repo``. +""" + +import os +import unittest +from unittest import mock + +import remote_repo_guard + + +LOCAL_GITEA_TOOLS_URL = "https://gitea.prgs.cc/Scaled-Tech-Consulting/Gitea-Tools.git" + + +class TestAssessRemoteRepoMatch(unittest.TestCase): + def test_explicit_org_and_repo_skips_guard(self): + # Caller supplied both org and repo explicitly: never block, even on mismatch. + assessment = remote_repo_guard.assess_remote_repo_match( + remote="prgs", + resolved_org="Scaled-Tech-Consulting", + resolved_repo="Timesheet", + local_remote_url=LOCAL_GITEA_TOOLS_URL, + org_explicit=True, + repo_explicit=True, + ) + self.assertTrue(assessment["proven"]) + self.assertFalse(assessment["block"]) + self.assertEqual(assessment["reasons"], []) + + def test_missing_local_remote_url_skips_guard(self): + # Best-effort lookup: no local remote URL (non-checkout usage) => do not block. + assessment = remote_repo_guard.assess_remote_repo_match( + remote="prgs", + resolved_org="Scaled-Tech-Consulting", + resolved_repo="Timesheet", + local_remote_url=None, + org_explicit=False, + repo_explicit=False, + ) + self.assertTrue(assessment["proven"]) + self.assertFalse(assessment["block"]) + + def test_matching_repo_is_proven(self): + assessment = remote_repo_guard.assess_remote_repo_match( + remote="prgs", + resolved_org="Scaled-Tech-Consulting", + resolved_repo="Gitea-Tools", + local_remote_url=LOCAL_GITEA_TOOLS_URL, + org_explicit=False, + repo_explicit=False, + ) + self.assertTrue(assessment["proven"]) + self.assertFalse(assessment["block"]) + + def test_regression_prgs_default_timesheet_vs_local_gitea_tools(self): + # The exact #530 scenario: default repo Timesheet, local remote Gitea-Tools. + assessment = remote_repo_guard.assess_remote_repo_match( + remote="prgs", + resolved_org="Scaled-Tech-Consulting", + resolved_repo="Timesheet", + local_remote_url=LOCAL_GITEA_TOOLS_URL, + org_explicit=False, + repo_explicit=False, + ) + self.assertFalse(assessment["proven"]) + self.assertTrue(assessment["block"]) + self.assertTrue(assessment["reasons"]) + self.assertEqual(assessment["resolved_repo"], "Timesheet") + self.assertEqual(assessment["resolved_org"], "Scaled-Tech-Consulting") + + def test_only_org_explicit_still_checks_repo(self): + # Repo left as default => still guarded even if org was explicit. + assessment = remote_repo_guard.assess_remote_repo_match( + remote="prgs", + resolved_org="Scaled-Tech-Consulting", + resolved_repo="Timesheet", + local_remote_url=LOCAL_GITEA_TOOLS_URL, + org_explicit=True, + repo_explicit=False, + ) + self.assertTrue(assessment["block"]) + + def test_case_insensitive_match(self): + assessment = remote_repo_guard.assess_remote_repo_match( + remote="prgs", + resolved_org="scaled-tech-consulting", + resolved_repo="gitea-tools", + local_remote_url=LOCAL_GITEA_TOOLS_URL, + org_explicit=False, + repo_explicit=False, + ) + self.assertTrue(assessment["proven"]) + + def test_format_error_mentions_resolved_and_local(self): + assessment = remote_repo_guard.assess_remote_repo_match( + remote="prgs", + resolved_org="Scaled-Tech-Consulting", + resolved_repo="Timesheet", + local_remote_url=LOCAL_GITEA_TOOLS_URL, + org_explicit=False, + repo_explicit=False, + ) + message = remote_repo_guard.format_remote_repo_guard_error(assessment) + self.assertIn("Timesheet", message) + self.assertIn("Gitea-Tools", message) + self.assertIn("org=", message) + self.assertIn("repo=", message) + + +class TestResolveServerWiring(unittest.TestCase): + """The guard is wired into gitea_mcp_server._resolve, so every read/lookup/ + mutation tool that resolves a target fails closed on a repo mismatch. + + Under pytest the guard is bypassed unless GITEA_FORCE_REMOTE_REPO_CHECK is set, + so these tests set the force flag and patch the local remote lookup + REMOTES. + """ + + def setUp(self): + import gitea_mcp_server + + self.server = gitea_mcp_server + + force = mock.patch.dict( + os.environ, {"GITEA_FORCE_REMOTE_REPO_CHECK": "1"}, clear=False + ) + force.start() + self.addCleanup(force.stop) + + url_patch = mock.patch.object( + gitea_mcp_server, + "_local_git_remote_url", + return_value=LOCAL_GITEA_TOOLS_URL, + ) + url_patch.start() + self.addCleanup(url_patch.stop) + + def _set_prgs_default_repo(self, repo): + original = dict(self.server.REMOTES["prgs"]) + self.addCleanup(self.server.REMOTES.__setitem__, "prgs", original) + self.server.REMOTES["prgs"] = {**original, "repo": repo} + + def test_resolve_blocks_on_default_repo_mismatch(self): + self._set_prgs_default_repo("Timesheet") + with self.assertRaises(RuntimeError) as ctx: + self.server._resolve("prgs", None, None, None) + self.assertIn("Timesheet", str(ctx.exception)) + self.assertIn("Gitea-Tools", str(ctx.exception)) + + def test_resolve_allows_explicit_org_and_repo(self): + self._set_prgs_default_repo("Timesheet") + # Explicit org/repo is authoritative even if it does not match local remote. + host, org, repo = self.server._resolve( + "prgs", None, "Scaled-Tech-Consulting", "Timesheet" + ) + self.assertEqual((org, repo), ("Scaled-Tech-Consulting", "Timesheet")) + + def test_resolve_allows_matching_default(self): + self._set_prgs_default_repo("Gitea-Tools") + host, org, repo = self.server._resolve("prgs", None, None, None) + self.assertEqual((org, repo), ("Scaled-Tech-Consulting", "Gitea-Tools")) + + def test_lookup_tool_cannot_silently_query_different_repo(self): + # gitea_view_issue resolves the target via _resolve first, so a bare + # remote=prgs pointing at the wrong default repo fails closed before any + # API call is made. + self._set_prgs_default_repo("Timesheet") + with self.assertRaises(RuntimeError): + self.server.gitea_view_issue(issue_number=1, remote="prgs") + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/test_review_workflow_boundary.py b/tests/test_review_workflow_boundary.py new file mode 100644 index 0000000..2d1fc1a --- /dev/null +++ b/tests/test_review_workflow_boundary.py @@ -0,0 +1,138 @@ +"""Tests for workflow-load session boundary tracking (#403).""" + +import os +import sys +import unittest +from unittest.mock import patch + +sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent)) + +import final_report_validator +import review_workflow_boundary +import review_workflow_load +import mcp_server + + +class TestPreReviewClassification(unittest.TestCase): + def setUp(self): + review_workflow_boundary.clear_pre_review_commands() + review_workflow_load.clear_review_workflow_load() + + def test_inventory_command_allowed(self): + result = review_workflow_boundary.classify_pre_review_command( + "gitea_list_prs remote=prgs", + cwd="/tmp", + project_root="/repo/Gitea-Tools", + ) + self.assertEqual( + result["classification"], + review_workflow_boundary.CLASSIFICATION_READ_ONLY_INVENTORY, + ) + + def test_main_checkout_pytest_is_boundary_violation(self): + root = "/repo/Gitea-Tools" + result = review_workflow_boundary.classify_pre_review_command( + "python -m pytest tests/", + cwd=root, + project_root=root, + ) + self.assertEqual( + result["classification"], + review_workflow_boundary.CLASSIFICATION_BOUNDARY_VIOLATION, + ) + + def test_profiles_json_inspection_is_boundary_violation(self): + result = review_workflow_boundary.classify_pre_review_command( + "cat profiles.json", + cwd="/repo/Gitea-Tools", + project_root="/repo/Gitea-Tools", + ) + self.assertEqual( + result["classification"], + review_workflow_boundary.CLASSIFICATION_BOUNDARY_VIOLATION, + ) + + +class TestWorkflowLoadBoundaryGate(unittest.TestCase): + def setUp(self): + review_workflow_boundary.clear_pre_review_commands() + review_workflow_load.clear_review_workflow_load() + mcp_server.init_review_decision_lock("prgs", "review_pr") + mcp_server.record_preflight_check("whoami") + mcp_server.record_preflight_check("capability", "reviewer") + + def _root(self) -> str: + return str(__import__("pathlib").Path(__file__).resolve().parent.parent) + + def test_boundary_violation_blocks_mutation_after_load(self): + root = self._root() + review_workflow_boundary.record_pre_review_command( + "python -m pytest tests/", + cwd=root, + project_root=root, + ) + res = mcp_server.gitea_load_review_workflow() + self.assertFalse(res["success"]) + self.assertEqual(res["boundary_status"], "violation") + blocked = mcp_server.gitea_mark_final_review_decision( + 42, "approve", remote="prgs") + self.assertFalse(blocked["marked_ready"]) + joined = " ".join(blocked["reasons"]).lower() + self.assertTrue( + "validation" in joined or "boundary" in joined or "workflow" in joined + ) + + def test_clean_inventory_then_load_passes(self): + root = self._root() + review_workflow_boundary.record_pre_review_command( + "gitea_list_prs remote=prgs", + cwd=root, + project_root=root, + ) + res = mcp_server.gitea_load_review_workflow() + self.assertTrue(res["success"]) + self.assertEqual(res["boundary_status"], "clean") + blockers = review_workflow_load.review_workflow_load_blockers(root) + self.assertEqual(blockers, []) + + def test_file_view_narrative_fails_validator_without_helper(self): + report = ( + "## Controller Handoff\n" + "- Task: review-merge-pr\n" + "- I read the canonical workflow review-merge-pr.md before review.\n" + ) + findings = final_report_validator.assess_final_report_validator( + report, + task_kind="review_pr", + ) + self.assertTrue(any( + f["rule_id"] == "reviewer.workflow_load_boundary" + for f in findings.get("findings") or [] + )) + + def test_helper_result_passes_validator(self): + root = self._root() + review_workflow_load.record_review_workflow_load(root) + helper = review_workflow_boundary.workflow_load_helper_result( + review_workflow_load._REVIEW_WORKFLOW_LOAD, + root, + ) + report = ( + "## Controller Handoff\n" + "- Task: review-merge-pr\n" + f"- Workflow-load helper result: workflow_hash: {helper['workflow_hash']}; " + f"boundary_status: {helper['boundary_status']}\n" + ) + findings = final_report_validator.assess_final_report_validator( + report, + task_kind="review_pr", + ) + boundary_findings = [ + f for f in (findings.get("findings") or []) + if f.get("rule_id") == "reviewer.workflow_load_boundary" + ] + self.assertEqual(boundary_findings, []) + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_review_workflow_load.py b/tests/test_review_workflow_load.py new file mode 100644 index 0000000..8d1e42a --- /dev/null +++ b/tests/test_review_workflow_load.py @@ -0,0 +1,179 @@ +"""Tests for canonical review workflow load proof (#389).""" + +import os +import sys +import unittest +from unittest.mock import patch + +sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent)) + +import review_workflow_load +import mcp_server + + +class TestReviewWorkflowLoadModule(unittest.TestCase): + def setUp(self): + review_workflow_load.clear_review_workflow_load() + mcp_server._save_review_decision_lock(None) + + def test_load_records_hash_and_schema(self): + root = str(__import__("pathlib").Path(__file__).resolve().parent.parent) + recorded = review_workflow_load.record_review_workflow_load(root) + self.assertEqual( + recorded["workflow_source"], + review_workflow_load.WORKFLOW_REL_PATH, + ) + self.assertEqual(recorded["task_mode"], "review-merge-pr") + self.assertRegex(recorded["workflow_hash"], r"^[0-9a-f]{12}$") + self.assertEqual( + recorded["final_report_schema_path"], + review_workflow_load.SCHEMA_REL_PATH, + ) + status = review_workflow_load.workflow_load_status(root) + self.assertTrue(status["workflow_load_proof_present"]) + self.assertTrue(status["workflow_load_valid"]) + + def test_profile_identity_mismatch_blocks(self): + """#559: different daemon PIDs are OK; profile identity mismatch is not.""" + import tempfile + from unittest.mock import patch + + import mcp_session_state + + root = str(__import__("pathlib").Path(__file__).resolve().parent.parent) + with tempfile.TemporaryDirectory() as tmp: + with patch.dict( + os.environ, + { + mcp_session_state.STATE_DIR_ENV: tmp, + mcp_session_state.SESSION_PROFILE_LOCK_ENV: "prgs-reviewer", + }, + clear=False, + ): + review_workflow_load.clear_review_workflow_load() + review_workflow_load.record_review_workflow_load(root) + # Corrupt the in-memory profile identity while keeping PID. + review_workflow_load._REVIEW_WORKFLOW_LOAD[ + "session_profile_lock" + ] = "other-profile" + review_workflow_load._REVIEW_WORKFLOW_LOAD[ + "profile_identity" + ] = "other-profile" + blockers = review_workflow_load.review_workflow_load_blockers(root) + self.assertTrue( + any("profile identity mismatch" in b for b in blockers), + msg=blockers, + ) + review_workflow_load.clear_review_workflow_load() + + def test_prompt_conflict_detected(self): + conflict, reasons = review_workflow_load.assess_prompt_conflict( + "Run work-issue author implementation only") + self.assertTrue(conflict) + self.assertTrue(reasons) + + +class TestReviewWorkflowLoadGates(unittest.TestCase): + def setUp(self): + review_workflow_load.clear_review_workflow_load() + mcp_server.init_review_decision_lock("prgs", "review_pr") + mcp_server.record_preflight_check("whoami") + mcp_server.record_preflight_check("capability", "reviewer") + + def _load_workflow(self): + return mcp_server.gitea_load_review_workflow() + + def test_mcp_helper_returns_required_fields(self): + res = self._load_workflow() + self.assertTrue(res["success"]) + self.assertTrue(res["loaded"]) + self.assertIn("workflow_source", res) + self.assertIn("workflow_hash", res) + self.assertIn("final_report_schema_path", res) + self.assertIn("final_report_schema_hash", res) + + def test_mark_final_blocked_without_load(self): + res = mcp_server.gitea_mark_final_review_decision( + 42, "approve", remote="prgs") + self.assertFalse(res["marked_ready"]) + self.assertTrue(any( + "gitea_load_review_workflow" in r for r in res["reasons"])) + self.assertTrue(any( + "approve/merge replay" in r.lower() or "Do not call" in r + for r in res["reasons"])) + + def test_submit_review_blocked_without_load(self): + with patch("mcp_server.gitea_check_pr_eligibility") as elig: + elig.return_value = { + "eligible": True, + "authenticated_user": "rev", + "profile_name": "prgs-reviewer", + "pr_author": "author", + "head_sha": "abc123", + "reasons": [], + } + res = mcp_server.gitea_submit_pr_review( + 42, + "approve", + remote="prgs", + final_review_decision_ready=True, + ) + self.assertFalse(res["performed"]) + self.assertTrue(any( + "gitea_load_review_workflow" in r for r in res["reasons"])) + + def test_merge_blocked_without_load(self): + res = mcp_server.gitea_merge_pr( + 42, + confirmation="MERGE PR 42", + remote="prgs", + ) + self.assertFalse(res["performed"]) + self.assertTrue(any( + "gitea_load_review_workflow" in r for r in res["reasons"])) + + def test_resolve_capability_reports_missing_load(self): + with patch.object(mcp_server, "_ensure_matching_profile"): + with patch.object( + mcp_server.gitea_config, "is_runtime_switching_enabled", + return_value=False): + with patch.object( + mcp_server, "_authenticated_username", + return_value="rev"): + res = mcp_server.gitea_resolve_task_capability( + "review_pr", remote="prgs") + proof = res.get("workflow_load_proof") or {} + self.assertFalse(proof.get("workflow_load_valid")) + self.assertTrue(any( + "gitea_load_review_workflow" in g + for g in res.get("task_role_guidance") or [])) + + def test_init_review_lock_clears_prior_load(self): + self._load_workflow() + mcp_server.init_review_decision_lock("prgs", "review_pr") + blockers = review_workflow_load.review_workflow_load_blockers( + str(__import__("pathlib").Path(__file__).resolve().parent.parent)) + self.assertTrue(blockers) + + def test_dry_run_allowed_without_load(self): + with patch("mcp_server.get_auth_header", return_value="Basic dGVzdA=="), \ + patch("mcp_server._list_pr_lease_comments", return_value=[]), \ + patch("mcp_server.gitea_check_pr_eligibility") as elig: + elig.return_value = { + "eligible": True, + "authenticated_user": "rev", + "profile_name": "prgs-reviewer", + "pr_author": "author", + "head_sha": "abc123", + "reasons": [], + } + res = mcp_server.gitea_dry_run_pr_review( + 42, "approve", remote="prgs") + self.assertNotIn( + "gitea_load_review_workflow", + " ".join(res.get("reasons") or []), + ) + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_reviewer_mutation_capability_proof.py b/tests/test_reviewer_mutation_capability_proof.py new file mode 100644 index 0000000..2f91dd0 --- /dev/null +++ b/tests/test_reviewer_mutation_capability_proof.py @@ -0,0 +1,141 @@ +"""Tests for exact per-mutation capability proof in reviewer reports (#405).""" + +import sys +import unittest +from pathlib import Path + +sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) + +from reviewer_mutation_capability_proof import assess_mutation_capability_proof +from review_proofs import assess_mutation_capability_proof as proofs_assess +from final_report_validator import assess_final_report_validator + + +REVIEW_ONLY = """ +Review decision: approved +Mutation capability table: +- gitea_submit_pr_review | review_pr (gitea.pr.review) | submitted | resolved before review +""" + +MERGE_PROVEN = """ +Review decision: approved +Merge result: merged 0123456789ab +Mutation capability table: +- gitea_submit_pr_review | review_pr (gitea.pr.review) | submitted | before review +- gitea_merge_pr | merge_pr (gitea.pr.merge) | merged | resolved before merge +""" + +MERGE_AND_DELETE_PROVEN = """ +Review decision: approved +Merge result: merged 0123456789ab +Remote branch deleted: feat/issue-x +Mutation capability table: +- gitea_submit_pr_review | review_pr (gitea.pr.review) | submitted | before review +- gitea_merge_pr | merge_pr (gitea.pr.merge) | merged | before merge +- gitea_delete_branch | delete_branch (gitea.branch.delete) | deleted | before delete +""" + + +class TestModule(unittest.TestCase): + def test_review_only_with_capability_passes(self): + r = assess_mutation_capability_proof(REVIEW_ONLY) + self.assertTrue(r["proven"], r["reasons"]) + + def test_merge_with_exact_capability_passes(self): + r = assess_mutation_capability_proof(MERGE_PROVEN) + self.assertTrue(r["proven"], r["reasons"]) + + def test_merge_and_delete_fully_proven_passes(self): + r = assess_mutation_capability_proof(MERGE_AND_DELETE_PROVEN) + self.assertTrue(r["proven"], r["reasons"]) + + def test_review_pr_does_not_authorize_merge(self): + report = """ +Review decision: approved +Merge result: merged 0123456789ab +Mutation capability table: +- gitea_submit_pr_review | review_pr (gitea.pr.review) | submitted | before review +""" + r = assess_mutation_capability_proof(report) + self.assertFalse(r["proven"]) + self.assertTrue(any("merge" in x.lower() for x in r["reasons"]), r["reasons"]) + + def test_merge_pr_does_not_authorize_branch_deletion(self): + report = """ +Review decision: approved +Merge result: merged 0123456789ab +Remote branch deleted: feat/issue-x +Mutation capability table: +- gitea_submit_pr_review | review_pr (gitea.pr.review) | submitted | before review +- gitea_merge_pr | merge_pr (gitea.pr.merge) | merged | before merge +""" + r = assess_mutation_capability_proof(report) + self.assertFalse(r["proven"]) + self.assertTrue(any("delet" in x.lower() for x in r["reasons"]), r["reasons"]) + + def test_delete_skipped_when_capability_missing_passes(self): + report = """ +Review decision: approved +Merge result: merged 0123456789ab +Branch deletion: skipped — delete_branch capability not available +Mutation capability table: +- gitea_submit_pr_review | review_pr (gitea.pr.review) | submitted | before review +- gitea_merge_pr | merge_pr (gitea.pr.merge) | merged | before merge +""" + r = assess_mutation_capability_proof(report) + self.assertTrue(r["proven"], r["reasons"]) + + def test_missing_table_when_merging_blocks(self): + report = """ +Review decision: approved +Merge result: merged 0123456789ab +merge_pr gitea.pr.merge resolved +""" + r = assess_mutation_capability_proof(report) + self.assertFalse(r["proven"]) + self.assertTrue(any("table" in x.lower() for x in r["reasons"]), r["reasons"]) + + def test_post_hoc_proof_blocks(self): + report = """ +Review decision: approved +Merge result: merged 0123456789ab +Mutation capability table: +- gitea_merge_pr | merge_pr (gitea.pr.merge) | merged | capability resolved after merge +""" + r = assess_mutation_capability_proof(report) + self.assertFalse(r["proven"]) + self.assertTrue(any("after" in x.lower() for x in r["reasons"]), r["reasons"]) + + def test_review_without_capability_blocks(self): + report = "Review decision: approved\nreview submitted\n" + r = assess_mutation_capability_proof(report) + self.assertFalse(r["proven"]) + + def test_no_mutation_no_requirement(self): + r = assess_mutation_capability_proof("Selected PR: #1\nSkipped, no action.") + self.assertTrue(r["proven"], r["reasons"]) + + +class TestWiring(unittest.TestCase): + def test_review_proofs_wrapper_matches_module(self): + self.assertEqual( + proofs_assess(MERGE_PROVEN)["proven"], + assess_mutation_capability_proof(MERGE_PROVEN)["proven"], + ) + + def test_final_report_validator_flags_nearby_capability_merge(self): + report = """ +## Controller Handoff +- Task: review_pr +Review decision: approved +Merge result: merged 0123456789ab +Mutation capability table: +- gitea_submit_pr_review | review_pr (gitea.pr.review) | submitted | before review +""" + result = assess_final_report_validator(report, "review_pr") + rule_ids = [f["rule_id"] for f in result.get("findings", [])] + self.assertIn("reviewer.mutation_capability_proof", rule_ids) + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/test_reviewer_pr_lease.py b/tests/test_reviewer_pr_lease.py index 69ee6d5..97e5a2e 100644 --- a/tests/test_reviewer_pr_lease.py +++ b/tests/test_reviewer_pr_lease.py @@ -7,6 +7,7 @@ from unittest.mock import patch sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent)) +import merger_lease_adoption as mla import reviewer_pr_lease as leases @@ -123,7 +124,11 @@ class TestReviewerLeaseMutationGate(unittest.TestCase): "session_id": "my-session", "candidate_head": head, "target_branch": "master", - }) + "comment_id": 101, + }, lease_provenance=mla.build_lease_provenance( + source=mla.SOURCE_ACQUIRE, + comment_id=101, + )) result = leases.assess_mutation_lease_gate( pr_number=382, comments=comments, @@ -143,7 +148,11 @@ class TestReviewerLeaseMutationGate(unittest.TestCase): "pr_number": 382, "session_id": "my-session", "candidate_head": reviewed, - }) + "comment_id": 102, + }, lease_provenance=mla.build_lease_provenance( + source=mla.SOURCE_ACQUIRE, + comment_id=102, + )) result = leases.assess_mutation_lease_gate( pr_number=382, comments=comments, diff --git a/tests/test_root_checkout_guard.py b/tests/test_root_checkout_guard.py new file mode 100644 index 0000000..86dc064 --- /dev/null +++ b/tests/test_root_checkout_guard.py @@ -0,0 +1,161 @@ +"""Tests for root checkout guard (#475).""" + +from __future__ import annotations + +import os +import sys +import unittest +from pathlib import Path +from unittest.mock import patch + +sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) + +import gitea_mcp_server as srv # noqa: E402 +import root_checkout_guard as rcg # noqa: E402 + +CONTROL_ROOT = str(Path(__file__).resolve().parents[3]) +BRANCHES_WORKTREE = str(Path(__file__).resolve().parents[1]) +MASTER_SHA = "a" * 40 +OTHER_SHA = "b" * 40 + + +class TestAssessRootCheckoutGuard(unittest.TestCase): + def _assess(self, **kwargs): + defaults = { + "workspace_path": CONTROL_ROOT, + "canonical_repo_root": CONTROL_ROOT, + "current_branch": "master", + "head_sha": MASTER_SHA, + "porcelain_status": "", + "remote_master_sha": MASTER_SHA, + "resolved_role": "author", + } + defaults.update(kwargs) + return rcg.assess_root_checkout_guard(**defaults) + + def test_clean_master_control_checkout_allowed(self): + result = self._assess() + self.assertTrue(result["proven"]) + self.assertFalse(result["block"]) + + def test_branches_worktree_allowed_for_author(self): + result = self._assess( + workspace_path=BRANCHES_WORKTREE, + current_branch="feat/issue-475-root-checkout-guard", + head_sha=OTHER_SHA, + resolved_role="author", + ) + self.assertTrue(result["proven"]) + + def test_branches_worktree_allowed_for_reviewer(self): + result = self._assess( + workspace_path=f"{CONTROL_ROOT}/branches/review-pr-1", + current_branch="review-pr-1", + resolved_role="reviewer", + ) + self.assertTrue(result["proven"]) + + def test_reconciler_always_allowed(self): + result = self._assess( + current_branch="feat/some-branch", + head_sha=OTHER_SHA, + porcelain_status=" M gitea_mcp_server.py\n", + resolved_role="reconciler", + ) + self.assertTrue(result["proven"]) + + def test_feature_branch_on_control_checkout_blocked(self): + result = self._assess( + current_branch="feat/issue-99-example", + head_sha=OTHER_SHA, + ) + self.assertTrue(result["block"]) + self.assertIn("not a stable base branch", result["reasons"][0]) + + def test_detached_head_blocked(self): + result = self._assess(current_branch=None) + self.assertTrue(result["block"]) + self.assertIn("detached HEAD", result["reasons"][0]) + + def test_dirty_control_checkout_blocked(self): + result = self._assess(porcelain_status=" M gitea_mcp_server.py\n") + self.assertTrue(result["block"]) + self.assertIn("tracked local edits", result["reasons"][0]) + + def test_head_behind_prgs_master_blocked(self): + result = self._assess( + head_sha=OTHER_SHA, + remote_master_sha=MASTER_SHA, + ) + self.assertTrue(result["block"]) + self.assertIn("does not match prgs/master", result["reasons"][0]) + + def test_merger_requires_clean_control_checkout(self): + result = self._assess( + workspace_path=BRANCHES_WORKTREE, + current_branch="feat/issue-475-root-checkout-guard", + head_sha=OTHER_SHA, + resolved_role="merger", + ) + self.assertTrue(result["block"]) + + +class TestVerifyPreflightRootGuardIntegration(unittest.TestCase): + def setUp(self): + srv._preflight_whoami_called = True + srv._preflight_capability_called = True + srv._preflight_resolved_role = "reviewer" + srv._preflight_whoami_violation = False + srv._preflight_capability_violation = False + self._orig_in_test = srv._preflight_in_test_mode + srv._preflight_in_test_mode = lambda: False + + def tearDown(self): + srv._preflight_in_test_mode = self._orig_in_test + + @patch("gitea_mcp_server._get_workspace_porcelain", return_value="") + @patch("gitea_mcp_server.root_checkout_guard.resolve_remote_master_sha", return_value=MASTER_SHA) + @patch("gitea_mcp_server.issue_lock_worktree.read_worktree_git_state") + @patch("gitea_mcp_server._resolve_author_mutation_context") + def test_reviewer_from_contaminated_root_blocked( + self, mock_ctx, mock_git, _remote_sha, _porcelain, + ): + srv._preflight_capability_baseline_porcelain = "" + mock_ctx.return_value = { + "workspace_path": CONTROL_ROOT, + "canonical_repo_root": CONTROL_ROOT, + "process_project_root": CONTROL_ROOT, + } + mock_git.return_value = { + "current_branch": "feat/hijacked-root", + "head_sha": OTHER_SHA, + "porcelain_status": "", + } + with self.assertRaises(RuntimeError) as ctx: + srv.verify_preflight_purity("prgs", worktree_path=CONTROL_ROOT) + self.assertIn("Root checkout guard (#475)", str(ctx.exception)) + self.assertIn(rcg.REMEDIATION, str(ctx.exception)) + + @patch("gitea_mcp_server._get_workspace_porcelain", return_value="") + @patch("gitea_mcp_server.root_checkout_guard.resolve_remote_master_sha", return_value=MASTER_SHA) + @patch("gitea_mcp_server.issue_lock_worktree.read_worktree_git_state") + @patch("gitea_mcp_server._resolve_author_mutation_context") + def test_reviewer_from_branches_worktree_allowed( + self, mock_ctx, mock_git, _remote_sha, _porcelain, + ): + srv._preflight_capability_baseline_porcelain = "" + mock_ctx.return_value = { + "workspace_path": BRANCHES_WORKTREE, + "canonical_repo_root": CONTROL_ROOT, + "process_project_root": BRANCHES_WORKTREE, + } + mock_git.return_value = { + "current_branch": "feat/issue-475-root-checkout-guard", + "head_sha": OTHER_SHA, + "porcelain_status": "", + } + srv.verify_preflight_purity("prgs", worktree_path=BRANCHES_WORKTREE) + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_run_tests_script.py b/tests/test_run_tests_script.py new file mode 100644 index 0000000..1ba5512 --- /dev/null +++ b/tests/test_run_tests_script.py @@ -0,0 +1,65 @@ +"""Contract checks for the root-level `run-tests.sh` convenience runner (#473). + +`run-tests.sh` is the canonical full-validation entry point. It must invoke the +project virtualenv interpreter, forward extra args to pytest, fail closed when +the venv Python is missing, and stay repo-local (no network, no lock files). +These checks pin that behavior so it cannot silently regress, and confirm the +developer testing guide names the runner. +""" +import stat +from pathlib import Path + +REPO_ROOT = Path(__file__).resolve().parent.parent +SCRIPT = REPO_ROOT / "run-tests.sh" +GUIDE = REPO_ROOT / "docs" / "developer-testing-guidelines.md" + + +def _script_text() -> str: + return SCRIPT.read_text(encoding="utf-8") + + +def test_run_tests_script_exists(): + assert SCRIPT.is_file(), "run-tests.sh must exist at the repository root" + + +def test_run_tests_script_is_executable(): + mode = SCRIPT.stat().st_mode + assert mode & stat.S_IXUSR, "run-tests.sh must be executable (chmod +x)" + + +def test_run_tests_script_has_strict_bash_flags(): + text = _script_text() + assert text.startswith("#!/usr/bin/env bash"), "must use the bash shebang" + assert "set -euo pipefail" in text, "must set -euo pipefail" + + +def test_run_tests_script_uses_venv_pytest_and_forwards_args(): + text = _script_text() + # Resolves the venv interpreter relative to the script's own directory. + assert "venv/bin/python" in text, "must use the project virtualenv Python" + assert "-m pytest" in text, "must run pytest via the module" + assert '"$@"' in text, "must forward extra CLI args to pytest" + + +def test_run_tests_script_fails_closed_without_venv(): + text = _script_text() + # Missing venv must be an explicit, non-zero-exit error, not a silent + # fallback to the wrong interpreter. + assert "if [[ ! -x" in text, "must guard on an executable venv Python" + assert "exit 1" in text, "must exit non-zero when the venv is missing" + assert "ERROR" in text, "must print a clear error message" + + +def test_run_tests_script_stays_repo_local(): + text = _script_text() + # No network calls, no lock-file writes from the runner itself. + for forbidden in ("curl", "wget", "gitea_issue_lock.json"): + assert forbidden not in text, f"runner must not reference {forbidden!r}" + + +def test_guide_names_canonical_runner(): + text = " ".join(GUIDE.read_text(encoding="utf-8").split()) + assert "./run-tests.sh" in text, "testing guide must name ./run-tests.sh" + assert "./run-tests.sh tests/test_mcp_server.py -q" in text, ( + "testing guide must show the focused-validation example" + ) diff --git a/tests/test_terminal_review_hard_stop.py b/tests/test_terminal_review_hard_stop.py index 4c65130..07edd98 100644 --- a/tests/test_terminal_review_hard_stop.py +++ b/tests/test_terminal_review_hard_stop.py @@ -36,7 +36,11 @@ def _lock(mutations=None, correction=False): def _seed(mutations=None, correction=False): + import review_workflow_load + review_workflow_load.record_review_workflow_load(mcp_server.PROJECT_ROOT) mcp_server._save_review_decision_lock(_lock(mutations, correction)) + mcp_server.gitea_load_review_workflow() + APPROVED_A = {"pr_number": 5, "action": "approve", "review_id": 1, @@ -48,6 +52,7 @@ RC_A = {"pr_number": 5, "action": "request_changes", "review_id": 2, class TestTerminalHardStopReasons(unittest.TestCase): def tearDown(self): mcp_server._save_review_decision_lock(None) + mcp_server.review_workflow_load.clear_review_workflow_load() def test_no_lock_no_reasons(self): mcp_server._save_review_decision_lock(None) @@ -92,7 +97,10 @@ class TestTerminalHardStopReasons(unittest.TestCase): class TestMergeHardStopWiring(unittest.TestCase): def tearDown(self): + import review_workflow_load + review_workflow_load.clear_review_workflow_load() mcp_server._save_review_decision_lock(None) + mcp_server.review_workflow_load.clear_review_workflow_load() def test_merge_blocked_after_request_changes(self): _seed([RC_A]) @@ -113,7 +121,10 @@ class TestMergeHardStopWiring(unittest.TestCase): class TestMarkFinalHardStopWiring(unittest.TestCase): def tearDown(self): + import review_workflow_load + review_workflow_load.clear_review_workflow_load() mcp_server._save_review_decision_lock(None) + mcp_server.review_workflow_load.clear_review_workflow_load() def test_mark_ready_blocked_after_terminal_mutation(self): _seed([RC_A]) @@ -145,7 +156,10 @@ def _mark(action, pr_number=6, **kwargs): class TestDuplicateRequestChangesSuppression(unittest.TestCase): def tearDown(self): + import review_workflow_load + review_workflow_load.clear_review_workflow_load() mcp_server._save_review_decision_lock(None) + mcp_server.review_workflow_load.clear_review_workflow_load() def test_duplicate_request_changes_blocked_at_same_head(self): _seed() diff --git a/tests/test_webui_lease_visibility.py b/tests/test_webui_lease_visibility.py new file mode 100644 index 0000000..a515311 --- /dev/null +++ b/tests/test_webui_lease_visibility.py @@ -0,0 +1,94 @@ +"""Tests for web UI lease visibility (#433).""" +import sys +import unittest +from pathlib import Path + +sys.path.insert(0, str(Path(__file__).resolve().parent.parent)) + +from starlette.testclient import TestClient + +from webui.app import create_app +from webui.lease_loader import ( + _duplicate_branch_warnings, + _duplicate_pr_warnings, + load_lease_snapshot, + parse_reviewer_lease_comment, + snapshot_to_dict, +) + + +class TestLeaseLoader(unittest.TestCase): + def test_parse_reviewer_lease_comment(self): + body = ( + "\n" + "pr: #42\n" + "reviewer_identity: sysadmin\n" + "profile: prgs-reviewer\n" + "phase: validating\n" + "expires_at: 2026-07-07T20:00:00Z\n" + ) + parsed = parse_reviewer_lease_comment(body) + self.assertIsNotNone(parsed) + assert parsed is not None + self.assertEqual(parsed["pr_number"], 42) + self.assertEqual(parsed["phase"], "validating") + + def test_duplicate_pr_warnings(self): + raw_prs = [ + {"number": 1, "title": "Closes #99", "body": ""}, + {"number": 2, "title": "fixes #99", "body": ""}, + ] + warnings = _duplicate_pr_warnings(raw_prs) + self.assertEqual(len(warnings), 1) + self.assertEqual(warnings[0].issue_number, 99) + self.assertEqual(warnings[0].pr_numbers, (1, 2)) + + def test_duplicate_branch_warnings(self): + warnings = _duplicate_branch_warnings([ + "feat/issue-12-a", + "feat/issue-12-b", + "master", + ]) + self.assertEqual(len(warnings), 1) + self.assertEqual(warnings[0].issue_number, 12) + + def test_load_snapshot_with_injected_fetch(self): + def fetch_prs(_h, _o, _r, _a): + return ([], None) + + def fetch_issues(_h, _o, _r, _a): + return ([], None) + + snapshot = load_lease_snapshot( + fetch_prs=fetch_prs, + fetch_issues=fetch_issues, + fetch_comments=lambda *_a, **_k: [], + issue_lock_path=str(Path("/nonexistent/lock.json")), + ) + data = snapshot_to_dict(snapshot) + self.assertIn("claim_inventory", data) + self.assertIn("collision_history", data) + + +class TestLeaseRoutes(unittest.TestCase): + def setUp(self): + self.client = TestClient(create_app()) + + def test_leases_page_renders(self): + response = self.client.get("/leases") + self.assertEqual(response.status_code, 200) + self.assertIn("Leases", response.text) + self.assertIn("Collision warnings", response.text) + self.assertNotIn("child issue", response.text.lower()) + + def test_api_leases_json(self): + response = self.client.get("/api/leases") + self.assertEqual(response.status_code, 200) + data = response.json() + self.assertIn("claim_inventory", data) + self.assertIn("duplicate_prs", data) + self.assertIn("reviewer_leases", data) + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_webui_skeleton.py b/tests/test_webui_skeleton.py index 5dd23a1..953b368 100644 --- a/tests/test_webui_skeleton.py +++ b/tests/test_webui_skeleton.py @@ -46,10 +46,14 @@ class TestWebuiSkeleton(unittest.TestCase): self.assertEqual(response.status_code, 200) self.assertIn("Gitea-Tools", response.text) + def test_leases_is_implemented(self): + response = self.client.get("/leases") + self.assertEqual(response.status_code, 200) + self.assertIn("Leases", response.text) + self.assertIn("Collision warnings", response.text) + def test_extra_stub_routes(self): - for path in ("/worktrees", "/leases"): - with self.subTest(path=path): - self.assertEqual(self.client.get(path).status_code, 200) + self.assertEqual(self.client.get("/worktrees").status_code, 200) def test_post_is_rejected(self): response = self.client.post("/health") @@ -62,10 +66,10 @@ class TestWebuiSkeleton(unittest.TestCase): self.assertIn("Live queue", response.text) def test_nav_links_on_all_pages(self): - for path in ("/", "/queue", "/projects", "/prompts", "/runtime", "/audit"): + for path in ("/", "/queue", "/projects", "/prompts", "/runtime", "/audit", "/leases"): with self.subTest(path=path): text = self.client.get(path).text - for href in ("/queue", "/projects", "/prompts", "/runtime", "/audit"): + for href in ("/queue", "/projects", "/prompts", "/runtime", "/audit", "/leases"): self.assertIn(f'href="{href}"', text) diff --git a/tests/test_workspace_guard_alignment.py b/tests/test_workspace_guard_alignment.py index ddcb67f..4454351 100644 --- a/tests/test_workspace_guard_alignment.py +++ b/tests/test_workspace_guard_alignment.py @@ -126,17 +126,37 @@ class TestRuntimeContextGuardAlignment(unittest.TestCase): with mock.patch.dict("os.environ", {"GITEA_TEST_PORCELAIN": ""}, clear=False): srv.verify_preflight_purity(worktree_path=BRANCHES_WORKTREE) - def test_stable_checkout_still_rejected(self): + @mock.patch("gitea_mcp_server.root_checkout_guard.resolve_remote_master_sha", return_value="a" * 40) + @mock.patch( + "gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", + return_value={ + "current_branch": "master", + "head_sha": "a" * 40, + "porcelain_status": "", + }, + ) + def test_stable_checkout_still_rejected(self, _git, _remote_sha): with mock.patch.object(srv, "PROJECT_ROOT", CONTROL_ROOT): with mock.patch.dict("os.environ", {"GITEA_TEST_PORCELAIN": ""}, clear=False): with self.assertRaises(RuntimeError) as ctx: srv.verify_preflight_purity() self.assertIn("stable control checkout", str(ctx.exception)) + @mock.patch("gitea_mcp_server.root_checkout_guard.resolve_remote_master_sha", return_value="a" * 40) + @mock.patch( + "gitea_mcp_server.issue_lock_worktree.read_worktree_git_state", + return_value={ + "current_branch": "master", + "head_sha": "a" * 40, + "porcelain_status": "", + }, + ) @mock.patch("os.path.isdir", return_value=True) @mock.patch("os.path.exists", return_value=True) @mock.patch("subprocess.run") - def test_non_branches_worktree_rejected(self, mock_run, *_exists): + def test_non_branches_worktree_rejected( + self, mock_run, mock_exists, mock_isdir, _git, _remote_sha, + ): outside = "/tmp/outside-repo-checkout" mock_run.return_value = MagicMock( returncode=0, diff --git a/tests/test_worktree_cleanup_audit.py b/tests/test_worktree_cleanup_audit.py new file mode 100644 index 0000000..f55b652 --- /dev/null +++ b/tests/test_worktree_cleanup_audit.py @@ -0,0 +1,532 @@ +"""Tests for session-owned worktree cleanup audit, TTL, and integrity (#401, #404).""" + +import sys +import unittest +from datetime import datetime, timedelta, timezone +from unittest.mock import patch + +sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent)) + +import worktree_cleanup_audit as wca # noqa: E402 + + +NOW = datetime(2026, 7, 7, 12, 0, 0, tzinfo=timezone.utc) + + +def _iso(dt): + return dt.isoformat().replace("+00:00", "Z") + + +class TestWorkflowTypeInference(unittest.TestCase): + def test_review_pr_path(self): + self.assertEqual( + wca.infer_workflow_type("branches/review-pr376", "review-pr376"), + wca.WORKFLOW_REVIEW, + ) + + def test_baseline_path(self): + self.assertEqual( + wca.infer_workflow_type("branches/baseline-master-issue-401"), + wca.WORKFLOW_BASELINE, + ) + + def test_merge_simulation_path(self): + self.assertEqual( + wca.infer_workflow_type("branches/merge-sim-pr380"), + wca.WORKFLOW_MERGE_SIMULATION, + ) + + def test_issue_work_branch(self): + self.assertEqual( + wca.infer_workflow_type( + "branches/issue-401-worktree", "feat/issue-401-worktree" + ), + wca.WORKFLOW_ISSUE_WORK, + ) + + def test_conflict_fix_path(self): + self.assertEqual( + wca.infer_workflow_type("branches/conflict-fix-pr376"), + wca.WORKFLOW_CONFLICT_FIX, + ) + + def test_unknown_path(self): + self.assertEqual( + wca.infer_workflow_type("branches/scratchpad"), wca.WORKFLOW_UNKNOWN + ) + + +class TestMetadata(unittest.TestCase): + def test_metadata_fields_present(self): + meta = wca.build_worktree_metadata( + path="branches/issue-401-worktree", + branch="feat/issue-401-worktree", + head_sha="abc123", + creator="jcwalker3", + profile="prgs-author", + created_at=_iso(NOW), + last_used_at=_iso(NOW), + ) + for field in ( + "path", + "workflow_type", + "issue_number", + "pr_number", + "branch", + "head_sha", + "creator", + "profile", + "created_at", + "last_used_at", + "cleanup_eligibility", + ): + self.assertIn(field, meta) + self.assertEqual(meta["issue_number"], 401) + self.assertEqual(meta["workflow_type"], wca.WORKFLOW_ISSUE_WORK) + + def test_review_metadata_auto_removable_flag(self): + meta = wca.build_worktree_metadata(path="branches/review-pr42") + self.assertTrue(meta["auto_remove_on_success"]) + + +class TestTTL(unittest.TestCase): + def test_expired(self): + old = _iso(NOW - timedelta(hours=48)) + self.assertTrue(wca.is_ttl_expired(last_used_at=old, now=NOW, ttl_hours=24)) + + def test_not_expired(self): + recent = _iso(NOW - timedelta(hours=1)) + self.assertFalse( + wca.is_ttl_expired(last_used_at=recent, now=NOW, ttl_hours=24) + ) + + def test_unknown_timestamp_fails_safe(self): + self.assertFalse(wca.is_ttl_expired(last_used_at=None, now=NOW)) + self.assertFalse( + wca.is_ttl_expired(last_used_at="not-a-date", now=NOW) + ) + + +class TestClassification(unittest.TestCase): + def test_successful_review_cleanup(self): + # Scenario 1: clean review worktree -> removable. + cls = wca.classify_worktree( + workflow_type=wca.WORKFLOW_REVIEW, is_dirty=False + ) + self.assertEqual(cls, wca.CLASS_CLEAN_STALE_REMOVABLE) + self.assertTrue(wca.is_removable(cls)) + + def test_dirty_worktree_preserved(self): + # Scenario 3: dirty worktree is never removable. + cls = wca.classify_worktree( + workflow_type=wca.WORKFLOW_REVIEW, is_dirty=True + ) + self.assertEqual(cls, wca.CLASS_DIRTY_LOCAL) + self.assertFalse(wca.is_removable(cls)) + + def test_active_pr_worktree_preserved(self): + # Scenario 4: open PR wins over an otherwise-removable review worktree. + cls = wca.classify_worktree( + workflow_type=wca.WORKFLOW_REVIEW, + is_dirty=False, + has_open_pr=True, + ) + self.assertEqual(cls, wca.CLASS_ACTIVE_OPEN_PR) + self.assertFalse(wca.is_removable(cls)) + + def test_stale_clean_issue_worktree_removable(self): + # Scenario 5: clean issue worktree, TTL expired, no lock -> removable. + cls = wca.classify_worktree( + workflow_type=wca.WORKFLOW_ISSUE_WORK, + is_dirty=False, + ttl_expired=True, + ) + self.assertEqual(cls, wca.CLASS_CLEAN_STALE_REMOVABLE) + self.assertTrue(wca.is_removable(cls)) + + def test_fresh_clean_issue_worktree_preserved(self): + # Clean issue worktree not yet TTL-expired stays active. + cls = wca.classify_worktree( + workflow_type=wca.WORKFLOW_ISSUE_WORK, + is_dirty=False, + ttl_expired=False, + ) + self.assertEqual(cls, wca.CLASS_ACTIVE_ISSUE_WORK) + self.assertFalse(wca.is_removable(cls)) + + def test_detached_review_worktree_classified(self): + # Scenario 6: detached review worktree -> detached_review_leftover. + cls = wca.classify_worktree( + workflow_type=wca.WORKFLOW_REVIEW, + is_dirty=False, + is_detached=True, + ) + self.assertEqual(cls, wca.CLASS_DETACHED_REVIEW_LEFTOVER) + self.assertTrue(wca.is_removable(cls)) + + def test_ttl_expired_but_dirty_preserved(self): + # Scenario 7: dirty wins over TTL expiry. + cls = wca.classify_worktree( + workflow_type=wca.WORKFLOW_ISSUE_WORK, + is_dirty=True, + ttl_expired=True, + ) + self.assertEqual(cls, wca.CLASS_DIRTY_LOCAL) + self.assertFalse(wca.is_removable(cls)) + + def test_lease_protected_worktree_preserved(self): + # Scenario 8: active lease is never removable. + cls = wca.classify_worktree( + workflow_type=wca.WORKFLOW_REVIEW, + is_dirty=False, + has_active_lease=True, + ) + self.assertEqual(cls, wca.CLASS_ACTIVE_ISSUE_WORK) + self.assertFalse(wca.is_removable(cls)) + + def test_active_issue_lock_preserved(self): + cls = wca.classify_worktree( + workflow_type=wca.WORKFLOW_ISSUE_WORK, + is_dirty=False, + has_active_issue_lock=True, + ttl_expired=True, + ) + self.assertEqual(cls, wca.CLASS_ACTIVE_ISSUE_WORK) + self.assertFalse(wca.is_removable(cls)) + + def test_protected_base_worktree_never_removable(self): + cls = wca.classify_worktree( + workflow_type=wca.WORKFLOW_ISSUE_WORK, + is_dirty=False, + is_protected=True, + ttl_expired=True, + ) + self.assertEqual(cls, wca.CLASS_UNSAFE_UNKNOWN) + self.assertFalse(wca.is_removable(cls)) + + def test_unknown_workflow_type_unsafe(self): + cls = wca.classify_worktree( + workflow_type=wca.WORKFLOW_UNKNOWN, is_dirty=False, ttl_expired=True + ) + self.assertEqual(cls, wca.CLASS_UNSAFE_UNKNOWN) + self.assertFalse(wca.is_removable(cls)) + + +class TestRemovalDecision(unittest.TestCase): + def test_safe_removal_proof(self): + decision = wca.assess_worktree_removal( + path="branches/review-pr42", + branch="review-pr42", + head_sha="abc123", + is_dirty=False, + has_open_pr=False, + has_active_lease=False, + classification=wca.CLASS_CLEAN_STALE_REMOVABLE, + ) + self.assertTrue(decision["safe_to_remove"]) + self.assertEqual(decision["block_reasons"], []) + self.assertTrue(decision["clean"]) + self.assertTrue(decision["no_active_pr"]) + self.assertTrue(decision["no_active_lease"]) + + def test_dirty_blocks_removal(self): + decision = wca.assess_worktree_removal( + path="branches/review-pr42", + branch="review-pr42", + head_sha="abc123", + is_dirty=True, + has_open_pr=False, + has_active_lease=False, + classification=wca.CLASS_DIRTY_LOCAL, + ) + self.assertFalse(decision["safe_to_remove"]) + self.assertIn("worktree has uncommitted changes", decision["block_reasons"]) + + def test_open_pr_and_lease_block_removal(self): + decision = wca.assess_worktree_removal( + path="branches/review-pr42", + branch="review-pr42", + head_sha="abc123", + is_dirty=False, + has_open_pr=True, + has_active_lease=True, + classification=wca.CLASS_ACTIVE_OPEN_PR, + ) + self.assertFalse(decision["safe_to_remove"]) + self.assertIn("worktree branch has an open PR", decision["block_reasons"]) + self.assertIn("worktree has an active lease", decision["block_reasons"]) + + +class TestSuccessCleanupPlan(unittest.TestCase): + def test_review_worktree_removed_at_success(self): + # Scenario 1 end-to-end: clean review worktree removed at success. + meta = wca.build_worktree_metadata(path="branches/review-pr42") + plan = wca.plan_success_cleanup( + metadata=meta, is_dirty=False, has_open_pr=False, has_active_lease=False + ) + self.assertTrue(plan["remove"]) + + def test_failed_review_leaves_worktree_reported(self): + # Scenario 2: dirty review worktree preserved and reported at failure. + meta = wca.build_worktree_metadata(path="branches/review-pr42") + plan = wca.plan_success_cleanup( + metadata=meta, is_dirty=True, has_open_pr=False, has_active_lease=False + ) + self.assertFalse(plan["remove"]) + self.assertIn("uncommitted", plan["reason"]) + report = wca.cleanup_failure_report(meta["path"], plan["reason"]) + self.assertFalse(report["removed"]) + self.assertEqual(report["path"], "branches/review-pr42") + + def test_issue_worktree_preserved_by_policy(self): + meta = wca.build_worktree_metadata( + path="branches/issue-401-worktree", branch="feat/issue-401-worktree" + ) + plan = wca.plan_success_cleanup( + metadata=meta, is_dirty=False, has_open_pr=False, has_active_lease=False + ) + self.assertFalse(plan["remove"]) + self.assertIn("preserved by policy", plan["reason"]) + + +class TestPorcelainParser(unittest.TestCase): + def test_parse_branch_and_detached(self): + text = ( + "worktree /repo\n" + "HEAD 1111111111111111111111111111111111111111\n" + "branch refs/heads/master\n" + "\n" + "worktree /repo/branches/review-pr42\n" + "HEAD 2222222222222222222222222222222222222222\n" + "detached\n" + ) + entries = wca.parse_worktree_porcelain(text) + self.assertEqual(len(entries), 2) + self.assertEqual(entries[0]["branch"], "master") + self.assertFalse(entries[0]["detached"]) + self.assertIsNone(entries[1]["branch"]) + self.assertTrue(entries[1]["detached"]) + + +class TestAuditReportAccuracy(unittest.TestCase): + """Scenario 9: cleanup report accuracy over a mixed branches/ directory.""" + + PORCELAIN = ( + "worktree /repo\n" + "HEAD 1111111111111111111111111111111111111111\n" + "branch refs/heads/master\n" + "\n" + "worktree /repo/branches/review-pr42\n" + "HEAD 2222222222222222222222222222222222222222\n" + "branch refs/heads/review-pr42\n" + "\n" + "worktree /repo/branches/issue-400-open-pr\n" + "HEAD 3333333333333333333333333333333333333333\n" + "branch refs/heads/feat/issue-400-open-pr\n" + "\n" + "worktree /repo/branches/issue-401-dirty\n" + "HEAD 4444444444444444444444444444444444444444\n" + "branch refs/heads/feat/issue-401-dirty\n" + ) + + def _fake_dirty(self, path): + if path.endswith("issue-401-dirty"): + return {"exists": True, "dirty": True, "dirty_files": [" M x.py"]} + return {"exists": True, "dirty": False, "dirty_files": []} + + def test_mixed_directory_classified(self): + with patch.object( + wca, + "list_worktrees", + return_value=wca.parse_worktree_porcelain(self.PORCELAIN), + ), patch.object( + wca, "read_worktree_dirty", side_effect=self._fake_dirty + ), patch.object( + wca, "git_worktree_list", return_value="(mocked)" + ): + report = wca.audit_branches_directory( + "/repo", + open_pr_branches={"feat/issue-400-open-pr"}, + ) + + by_path = {wt["path"]: wt for wt in report["worktrees"]} + # main checkout on master -> protected -> unsafe/unknown, not removable + self.assertEqual( + by_path["/repo"]["classification"], wca.CLASS_UNSAFE_UNKNOWN + ) + # clean review worktree -> removable + self.assertEqual( + by_path["/repo/branches/review-pr42"]["classification"], + wca.CLASS_CLEAN_STALE_REMOVABLE, + ) + # open PR branch -> preserved + self.assertEqual( + by_path["/repo/branches/issue-400-open-pr"]["classification"], + wca.CLASS_ACTIVE_OPEN_PR, + ) + # dirty worktree -> preserved + self.assertEqual( + by_path["/repo/branches/issue-401-dirty"]["classification"], + wca.CLASS_DIRTY_LOCAL, + ) + # exactly one removable candidate (the clean review worktree) + self.assertEqual(report["removable_count"], 1) + self.assertEqual( + report["removable_candidates"][0]["path"], + "/repo/branches/review-pr42", + ) + self.assertEqual(report["total"], 4) + self.assertEqual(report["git_worktree_list"], "(mocked)") + + +def _integrity_entry( + path: str, + *, + classification: str, + registered: bool = True, + preserve: bool | None = None, +) -> dict: + preserve_flag = preserve if preserve is not None else classification in { + "active_open_pr", + "active_issue_work", + "dirty_local_worktree", + "unsafe_unknown", + } + return { + "path": path, + "classification": classification, + "preserve": preserve_flag, + "registered_worktree": registered, + "worktree_state": {"exists": True, "clean": classification == "clean_stale_removable"}, + "worktree_record": {"branch": "feat/x"} if registered else None, + } + + +def _integrity_snapshot(entries: list[dict]) -> dict: + return {"entries": entries} + + +class TestParseWorktreePorcelain(unittest.TestCase): + def test_parses_multiple_worktrees(self): + text = "\n".join([ + "worktree /proj/branches/foo", + "HEAD abcdef0123456789abcdef0123456789abcdef0", + "branch refs/heads/feat/foo", + "", + "worktree /proj", + "HEAD 1111111111111111111111111111111111111111", + "branch refs/heads/master", + ]) + parsed = wca.parse_worktree_list_porcelain(text) + self.assertEqual(len(parsed), 2) + self.assertEqual(parsed[0]["branch"], "feat/foo") + + +class TestClassifyEntry(unittest.TestCase): + def test_active_pr_classification(self): + result = wca.classify_branches_entry( + rel_path="branches/feat-issue-1-x", + worktree_record={"branch": "feat/issue-1-x"}, + worktree_state={"exists": True, "clean": True, "dirty_files": []}, + open_pr_branches={"feat/issue-1-x"}, + ) + self.assertEqual(result, "active_open_pr") + + def test_dirty_classification(self): + result = wca.classify_branches_entry( + rel_path="branches/dirty-one", + worktree_record={"branch": "feat/dirty-one"}, + worktree_state={"exists": True, "dirty_files": ["a.py"]}, + open_pr_branches=set(), + ) + self.assertEqual(result, "dirty_local_worktree") + + +class TestCleanupIntegrity(unittest.TestCase): + def test_preserved_worktree_remains(self): + path = "branches/keep-me" + before = _integrity_snapshot([ + _integrity_entry(path, classification="clean_stale_removable", preserve=False), + ]) + after = _integrity_snapshot([ + _integrity_entry(path, classification="clean_stale_removable", preserve=False), + ]) + result = wca.assess_worktree_cleanup_integrity(before=before, after=after) + self.assertTrue(result["integrity_passed"]) + + def test_intentional_removal_passes(self): + path = "branches/remove-me" + before = _integrity_snapshot([ + _integrity_entry(path, classification="clean_stale_removable", preserve=False), + ]) + after = _integrity_snapshot([]) + result = wca.assess_worktree_cleanup_integrity( + before=before, + after=after, + removals=[{ + "path": path, + "method": "git worktree remove", + "pre_removal_proof": "clean status", + }], + ) + self.assertTrue(result["integrity_passed"]) + + def test_dirty_worktree_disappears_fails(self): + path = "branches/dirty-wt" + before = _integrity_snapshot([_integrity_entry(path, classification="dirty_local_worktree")]) + after = _integrity_snapshot([]) + recon = wca.reconcile_cleanup_audit(before, after) + result = wca.assess_cleanup_audit_integrity(recon) + self.assertFalse(result["proven"]) + + def test_active_pr_worktree_disappears_fails(self): + path = "branches/review-pr99" + before = _integrity_snapshot([_integrity_entry(path, classification="active_open_pr")]) + after = _integrity_snapshot([]) + result = wca.assess_worktree_cleanup_integrity(before=before, after=after) + self.assertFalse(result["integrity_passed"]) + + def test_explained_missing_allowed(self): + path = "branches/review-pr382" + before = _integrity_snapshot([ + _integrity_entry(path, classification="detached_review_leftover", preserve=False), + ]) + after = _integrity_snapshot([]) + result = wca.assess_worktree_cleanup_integrity( + before=before, + after=after, + explained_missing={path: "removed concurrently by sibling session"}, + ) + self.assertTrue(result["integrity_passed"]) + + def test_removal_log_omits_clean_stale_fails(self): + path = "branches/a" + before = _integrity_snapshot([ + _integrity_entry(path, classification="clean_stale_removable", preserve=False), + ]) + after = _integrity_snapshot([]) + recon = wca.reconcile_cleanup_audit(before, after, removal_log=[]) + self.assertFalse(recon["removal_log_complete"]) + + +class TestCleanupReportProof(unittest.TestCase): + def test_complete_report_passes(self): + report = "\n".join([ + "Cleanup audit reconciliation table:", + "Initial count: 10", + "Removed count: 3", + "Preserved count: 7", + "Missing-unexplained count: 0", + "Final count: 7", + "Final verification: git worktree list proof attached", + ]) + result = wca.assess_cleanup_audit_final_report(report) + self.assertTrue(result["proven"]) + + def test_incomplete_report_fails(self): + result = wca.assess_cleanup_audit_final_report("removed some worktrees") + self.assertFalse(result["proven"]) + + +if __name__ == "__main__": + unittest.main() diff --git a/webui/app.py b/webui/app.py index a97d1ad..4f4cb0f 100644 --- a/webui/app.py +++ b/webui/app.py @@ -14,6 +14,8 @@ from webui.project_registry import find_project, load_registry, registry_to_dict from webui.project_views import render_project_detail, render_projects_list from webui.prompt_library import find_prompt, library_to_dict from webui.prompt_views import render_prompt_detail, render_prompts_page +from webui.lease_loader import load_lease_snapshot, snapshot_to_dict as lease_snapshot_to_dict +from webui.lease_views import render_leases_page from webui.queue_loader import load_queue_snapshot, snapshot_to_dict from webui.queue_views import render_queue_page @@ -141,10 +143,12 @@ async def worktrees(_request: Request) -> HTMLResponse: async def leases(_request: Request) -> HTMLResponse: - return _stub_page( - "Leases", - "Lease visibility will show active issue and reviewer PR leases.", - ) + snapshot = load_lease_snapshot() + return HTMLResponse(render_leases_page(snapshot)) + + +async def api_leases(_request: Request) -> JSONResponse: + return JSONResponse(lease_snapshot_to_dict(load_lease_snapshot())) async def method_not_allowed(request: Request, _exc: Exception) -> Response: @@ -175,6 +179,7 @@ def create_app() -> Starlette: Route("/audit", audit, methods=["GET"]), Route("/worktrees", worktrees, methods=["GET"]), Route("/leases", leases, methods=["GET"]), + Route("/api/leases", api_leases, methods=["GET"]), ], exception_handlers={405: method_not_allowed}, ) \ No newline at end of file diff --git a/webui/lease_loader.py b/webui/lease_loader.py new file mode 100644 index 0000000..717feef --- /dev/null +++ b/webui/lease_loader.py @@ -0,0 +1,330 @@ +"""Lease and collision visibility for the web UI (#433).""" + +from __future__ import annotations + +import os +import re +import subprocess +from dataclasses import dataclass +from typing import Any, Callable +from urllib.parse import urlparse + +from gitea_auth import api_fetch_page, get_auth_header, repo_api_url +from issue_claim_heartbeat import build_claim_inventory +from merged_cleanup_reconcile import ISSUE_LOCK_FILE, read_issue_lock + +from webui.project_registry import ProjectRecord, load_registry +from webui.queue_loader import _extract_linked_issue, _fetch_issues, _fetch_prs + +_REVIEWER_LEASE_MARKER = "" +_REVIEWER_FIELD_RE = re.compile( + r"^\s*([a-z_]+)\s*:\s*(.+?)\s*$", + re.IGNORECASE | re.MULTILINE, +) +_ISSUE_BRANCH_RE = re.compile(r"issue-(\d+)", re.IGNORECASE) + +_COLLISION_HISTORY = ( + {"number": 267, "title": "Author work leases"}, + {"number": 268, "title": "Issue claim heartbeat leases"}, + {"number": 400, "title": "Early duplicate-work detection"}, + {"number": 407, "title": "Per-PR reviewer leases"}, +) + + +@dataclass(frozen=True) +class CollisionWarning: + kind: str + message: str + issue_number: int | None = None + pr_numbers: tuple[int, ...] = () + + +@dataclass(frozen=True) +class LeaseSnapshot: + project_id: str + repo_label: str + issue_lock: dict[str, Any] | None + claim_inventory: dict[str, Any] + reviewer_leases: tuple[dict[str, Any], ...] + duplicate_prs: tuple[CollisionWarning, ...] + duplicate_branches: tuple[CollisionWarning, ...] + collision_history: tuple[dict[str, Any], ...] + fetch_error: str | None = None + + +def _repo_root() -> str: + override = (os.environ.get("WEBUI_REPO_ROOT") or "").strip() + if override: + return os.path.realpath(override) + return os.path.realpath(os.path.join(os.path.dirname(__file__), "..")) + + +def _host_from_url(remote_host: str) -> str: + parsed = urlparse(remote_host.strip()) + return parsed.netloc or remote_host.strip().rstrip("/") + + +def _parse_pr_ref(value: str | None) -> int | None: + digits = re.sub(r"[^\d]", "", value or "") + return int(digits) if digits.isdigit() else None + + +def parse_reviewer_lease_comment(body: str) -> dict[str, Any] | None: + text = body or "" + if _REVIEWER_LEASE_MARKER not in text: + return None + fields: dict[str, str] = {} + for match in _REVIEWER_FIELD_RE.finditer(text): + fields[match.group(1).strip().lower()] = match.group(2).strip() + if not fields: + return None + return { + "pr_number": _parse_pr_ref(fields.get("pr")), + "issue_number": _parse_pr_ref(fields.get("issue")), + "reviewer_identity": fields.get("reviewer_identity"), + "profile": fields.get("profile"), + "phase": (fields.get("phase") or "").strip().lower() or None, + "expires_at": fields.get("expires_at"), + "blocker": fields.get("blocker"), + } + + +def _fetch_comments( + host: str, + org: str, + repo: str, + auth: str, + *, + issue_number: int, +) -> list[dict]: + url = f"{repo_api_url(host, org, repo)}/issues/{issue_number}/comments" + comments: list[dict] = [] + page = 1 + while page <= 10: + raw_page, meta = api_fetch_page(url, auth, page=page, limit=50) + comments.extend(raw_page) + if meta.get("is_final_page"): + break + page += 1 + return comments + + +def _list_local_branch_names(project_root: str) -> list[str]: + result = subprocess.run( + ["git", "-C", project_root, "branch", "--list", "--format=%(refname:short)"], + capture_output=True, + text=True, + check=False, + ) + if result.returncode != 0: + return [] + return [line.strip() for line in (result.stdout or "").splitlines() if line.strip()] + + +def _duplicate_pr_warnings(raw_prs: list[dict]) -> list[CollisionWarning]: + issue_to_prs: dict[int, list[int]] = {} + for pr in raw_prs: + linked = _extract_linked_issue(pr.get("title"), pr.get("body")) + if linked is None: + continue + issue_to_prs.setdefault(linked, []).append(int(pr["number"])) + warnings: list[CollisionWarning] = [] + for issue_number, pr_numbers in sorted(issue_to_prs.items()): + if len(pr_numbers) > 1: + warnings.append( + CollisionWarning( + kind="duplicate-pr", + issue_number=issue_number, + pr_numbers=tuple(sorted(pr_numbers)), + message=( + f"Issue #{issue_number} has {len(pr_numbers)} open PRs: " + f"{', '.join(f'#{n}' for n in sorted(pr_numbers))} (#400)" + ), + ) + ) + return warnings + + +def _duplicate_branch_warnings(branch_names: list[str]) -> list[CollisionWarning]: + issue_to_branches: dict[int, list[str]] = {} + for name in branch_names: + match = _ISSUE_BRANCH_RE.search(name) + if not match: + continue + issue_num = int(match.group(1)) + issue_to_branches.setdefault(issue_num, []).append(name) + warnings: list[CollisionWarning] = [] + for issue_number, branches in sorted(issue_to_branches.items()): + if len(branches) > 1: + warnings.append( + CollisionWarning( + kind="duplicate-branch", + issue_number=issue_number, + message=( + f"Issue #{issue_number} has {len(branches)} local branches: " + f"{', '.join(branches)}" + ), + ) + ) + return warnings + + +def _extract_reviewer_leases( + raw_prs: list[dict], + *, + fetch_comments: Callable[[int], list[dict]], +) -> list[dict[str, Any]]: + leases: list[dict[str, Any]] = [] + for pr in raw_prs: + pr_number = int(pr["number"]) + for comment in fetch_comments(pr_number): + parsed = parse_reviewer_lease_comment(comment.get("body") or "") + if not parsed: + continue + leases.append( + { + **parsed, + "pr_number": parsed.get("pr_number") or pr_number, + "comment_id": comment.get("id"), + "author": (comment.get("user") or {}).get("login"), + "created_at": comment.get("created_at"), + } + ) + active_phases = {"claimed", "validating", "approved", "request-changes", "merging"} + return [lease for lease in leases if (lease.get("phase") or "") in active_phases] + + +def load_lease_snapshot( + *, + project_id: str | None = None, + project_root: str | None = None, + issue_lock_path: str | None = None, + fetch_prs: Callable | None = None, + fetch_issues: Callable | None = None, + fetch_comments: Callable[[str, str, str, str, int], list[dict]] | None = None, +) -> LeaseSnapshot: + registry = load_registry() + project: ProjectRecord | None = None + if project_id: + project = next((p for p in registry.projects if p.id == project_id), None) + else: + project = registry.projects[0] if registry.projects else None + + root = project_root or _repo_root() + lock = read_issue_lock(issue_lock_path if issue_lock_path is not None else ISSUE_LOCK_FILE) + + if project is None: + return LeaseSnapshot( + project_id=project_id or "", + repo_label="", + issue_lock=lock, + claim_inventory={"entries": [], "counts": {}}, + reviewer_leases=(), + duplicate_prs=(), + duplicate_branches=(), + collision_history=_COLLISION_HISTORY, + fetch_error="project not found in registry", + ) + + host = _host_from_url(project.remote_host) + pr_fetch = fetch_prs or _fetch_prs + issue_fetch = fetch_issues or _fetch_issues + comment_fetch = fetch_comments or _fetch_comments + using_live = fetch_prs is None or fetch_issues is None + auth = get_auth_header(host) if using_live else "test-auth" + + if using_live and not auth: + return LeaseSnapshot( + project_id=project.id, + repo_label=f"{project.gitea_owner}/{project.repo_name}", + issue_lock=lock, + claim_inventory={"entries": [], "counts": {}}, + reviewer_leases=(), + duplicate_prs=(), + duplicate_branches=_duplicate_branch_warnings(_list_local_branch_names(root)), + collision_history=_COLLISION_HISTORY, + fetch_error=( + f"Gitea credentials unavailable for {host}; " + "remote lease artifacts cannot be loaded" + ), + ) + + try: + raw_prs, _ = pr_fetch(host, project.gitea_owner, project.repo_name, auth) + raw_issues, _ = issue_fetch(host, project.gitea_owner, project.repo_name, auth) + except Exception as exc: # noqa: BLE001 + return LeaseSnapshot( + project_id=project.id, + repo_label=f"{project.gitea_owner}/{project.repo_name}", + issue_lock=lock, + claim_inventory={"entries": [], "counts": {}}, + reviewer_leases=(), + duplicate_prs=(), + duplicate_branches=_duplicate_branch_warnings(_list_local_branch_names(root)), + collision_history=_COLLISION_HISTORY, + fetch_error=f"Gitea fetch failed: {exc}", + ) + + comments_by_issue: dict[int, list[dict]] = {} + for issue in raw_issues: + if not any(lb.get("name") == "status:in-progress" for lb in issue.get("labels", [])): + continue + number = int(issue["number"]) + comments_by_issue[number] = comment_fetch( + host, project.gitea_owner, project.repo_name, auth, issue_number=number + ) + + branch_names = _list_local_branch_names(root) + inventory = build_claim_inventory( + issues=raw_issues, + comments_by_issue=comments_by_issue, + open_prs=raw_prs, + branch_names=branch_names, + ) + + def _pr_comments(pr_number: int) -> list[dict]: + return comment_fetch( + host, project.gitea_owner, project.repo_name, auth, issue_number=pr_number + ) + + reviewer_leases = tuple(_extract_reviewer_leases(raw_prs, fetch_comments=_pr_comments)) + + return LeaseSnapshot( + project_id=project.id, + repo_label=f"{project.gitea_owner}/{project.repo_name}", + issue_lock=lock, + claim_inventory=inventory, + reviewer_leases=reviewer_leases, + duplicate_prs=tuple(_duplicate_pr_warnings(raw_prs)), + duplicate_branches=tuple(_duplicate_branch_warnings(branch_names)), + collision_history=_COLLISION_HISTORY, + ) + + +def snapshot_to_dict(snapshot: LeaseSnapshot) -> dict[str, Any]: + return { + "project_id": snapshot.project_id, + "repo_label": snapshot.repo_label, + "issue_lock": snapshot.issue_lock, + "claim_inventory": snapshot.claim_inventory, + "reviewer_leases": list(snapshot.reviewer_leases), + "duplicate_prs": [ + { + "kind": w.kind, + "message": w.message, + "issue_number": w.issue_number, + "pr_numbers": list(w.pr_numbers), + } + for w in snapshot.duplicate_prs + ], + "duplicate_branches": [ + { + "kind": w.kind, + "message": w.message, + "issue_number": w.issue_number, + } + for w in snapshot.duplicate_branches + ], + "collision_history": list(snapshot.collision_history), + "fetch_error": snapshot.fetch_error, + } \ No newline at end of file diff --git a/webui/lease_views.py b/webui/lease_views.py new file mode 100644 index 0000000..a998dea --- /dev/null +++ b/webui/lease_views.py @@ -0,0 +1,130 @@ +"""HTML views for lease and collision visibility (#433).""" + +from __future__ import annotations + +import html +import json + +from webui.layout import render_page +from webui.lease_loader import LeaseSnapshot + + +def _escape(text: str) -> str: + return html.escape(text, quote=True) + + +def _warnings_block(snapshot: LeaseSnapshot) -> str: + warnings = list(snapshot.duplicate_prs) + list(snapshot.duplicate_branches) + if not warnings: + return "

No duplicate PR/branch collisions detected in current inventory.

" + items = "".join(f"
  • {_escape(w.message)}
    • " for w in warnings) + return f"
      {items}
    " + + +def _lock_block(snapshot: LeaseSnapshot) -> str: + lock = snapshot.issue_lock + if not lock: + return "

    No active local issue lock file.

    " + return ( + "
    "
+        f"{_escape(json.dumps(lock, indent=2, sort_keys=True))}"
+        "
    " + ) + + +def _claims_table(snapshot: LeaseSnapshot) -> str: + entries = snapshot.claim_inventory.get("entries") or [] + if not entries: + return "

    No in-progress issue claims in fetched inventory.

    " + rows = [] + for entry in entries: + rows.append( + "" + f"#{entry.get('issue_number')}" + f"" + f"{_escape(str(entry.get('status')))}" + f"{_escape(str(entry.get('linked_open_pr') or '—'))}" + f"{entry.get('heartbeat_count', 0)}" + f"{_escape(', '.join(entry.get('matching_branches') or []) or '—')}" + f"{_escape('; '.join(entry.get('reasons') or []))}" + "" + ) + return ( + "" + "" + "" + f"{''.join(rows)}
    IssueClaim statusOpen PRHeartbeatsBranchesNotes
    " + ) + + +def _reviewer_leases_table(snapshot: LeaseSnapshot) -> str: + if not snapshot.reviewer_leases: + return ( + "

    No active reviewer PR lease comments found " + "(marker <!-- mcp-review-lease:v1 -->; see #407).

    " + ) + rows = [] + for lease in snapshot.reviewer_leases: + rows.append( + "" + f"#{lease.get('pr_number')}" + f"{_escape(str(lease.get('reviewer_identity') or '—'))}" + f"{_escape(str(lease.get('profile') or '—'))}" + f"{_escape(str(lease.get('phase') or '—'))}" + f"{_escape(str(lease.get('expires_at') or '—'))}" + "" + ) + return ( + "" + "" + "" + f"{''.join(rows)}
    PRReviewerProfilePhaseExpires
    " + ) + + +def _history_links(snapshot: LeaseSnapshot) -> str: + items = "".join( + f"
  • #{item['number']} — {_escape(item['title'])}
    • " + for item in snapshot.collision_history + ) + return f"
      {items}
    " + + +LEASE_PAGE_STYLES = """ + +""" + + +def render_leases_page(snapshot: LeaseSnapshot) -> str: + error_block = "" + if snapshot.fetch_error: + error_block = ( + '

    Partial load: ' + f"{_escape(snapshot.fetch_error)}

    " + ) + body = ( + "

    Leases & collisions

    " + "

    Read-only visibility for issue claims, reviewer PR leases, and " + "duplicate-work risks. Does not acquire or release leases.

    " + f"{error_block}" + f"

    Project: {_escape(snapshot.repo_label)}

    " + "

    Collision warnings

    " + f"{_warnings_block(snapshot)}" + "

    Local issue lock

    " + f"{_lock_block(snapshot)}" + "

    In-progress issue claims (#268)

    " + f"{_claims_table(snapshot)}" + "

    Reviewer PR leases (#407)

    " + f"{_reviewer_leases_table(snapshot)}" + "

    Collision / lease history issues

    " + f"{_history_links(snapshot)}" + "

    JSON API

    " + f"{LEASE_PAGE_STYLES}" + ) + return render_page(title="Leases", body_html=body) \ No newline at end of file diff --git a/worktree_cleanup_audit.py b/worktree_cleanup_audit.py new file mode 100644 index 0000000..6ede83b --- /dev/null +++ b/worktree_cleanup_audit.py @@ -0,0 +1,1023 @@ +"""Session-owned worktree cleanup audit, TTL enforcement, and integrity (#401, #404). + +LLM workflows create many session-owned worktrees under ``branches/`` +(review, baseline, merge-simulation, issue, and conflict-fix worktrees). +When a run stops early, races a sibling session, hits a validation failure, +or loses shell/cwd state, those worktrees are left behind and later workflow +decisions get harder and riskier. + +This module provides: + +* ``build_worktree_metadata`` — ownership/purpose metadata for a worktree. +* ``classify_worktree`` — safety-first classification into the audit + vocabulary (active open PR, active issue work, dirty, clean stale + removable, detached review leftover, unsafe/unknown). +* ``assess_worktree_removal`` — a per-worktree removal decision with an + explicit proof and block reasons. +* git-shelling helpers (``list_worktrees``, ``read_worktree_dirty``, + ``git_worktree_list``, ``remove_worktree``) and ``audit_branches_directory`` + that classify every entry under ``branches/``. +* ``capture_cleanup_snapshot`` / ``reconcile_cleanup_audit`` — before/after + reconciliation for bulk cleanup audits (#404). + +Pure assessment functions take explicit state so they are unit-testable +without a filesystem or network. Only the thin git helpers shell out, and +removal is only ever executed after ``assess_worktree_removal`` proves the +worktree is safe to delete. +""" + +from __future__ import annotations + +import os +import re +import subprocess +from datetime import datetime, timezone +from typing import Any + +from merged_cleanup_reconcile import branch_worktree_folder, read_local_worktree_state +from reviewer_worktree import parse_dirty_tracked_files + +PROTECTED_BRANCHES = frozenset({"master", "main", "dev"}) +DEFAULT_TTL_HOURS = float(os.environ.get("GITEA_WORKTREE_TTL_HOURS", "24") or 24) + +# Workflow types that can create session-owned worktrees. +WORKFLOW_REVIEW = "review" +WORKFLOW_BASELINE = "baseline" +WORKFLOW_MERGE_SIMULATION = "merge_simulation" +WORKFLOW_ISSUE_WORK = "issue_work" +WORKFLOW_CONFLICT_FIX = "conflict_fix" +WORKFLOW_UNKNOWN = "unknown" + +# Workflow types whose worktrees are transient and removed automatically at +# successful workflow completion (acceptance criterion 2). +AUTO_REMOVE_ON_SUCCESS = frozenset( + {WORKFLOW_REVIEW, WORKFLOW_BASELINE, WORKFLOW_MERGE_SIMULATION} +) + +# Cleanup-audit classification vocabulary (acceptance criterion 4). +CLASS_ACTIVE_OPEN_PR = "active_open_pr" +CLASS_ACTIVE_ISSUE_WORK = "active_issue_work" +CLASS_DIRTY_LOCAL = "dirty_local_worktree" +CLASS_CLEAN_STALE_REMOVABLE = "clean_stale_removable" +CLASS_DETACHED_REVIEW_LEFTOVER = "detached_review_leftover" +CLASS_UNSAFE_UNKNOWN = "unsafe_unknown" + +# Only these two classifications may ever be removed automatically. +REMOVABLE_CLASSES = frozenset( + {CLASS_CLEAN_STALE_REMOVABLE, CLASS_DETACHED_REVIEW_LEFTOVER} +) + +_ISSUE_REF_RE = re.compile(r"issue-(\d+)", re.IGNORECASE) +_ISSUE_BRANCH_PREFIXES = ("feat/", "fix/", "docs/", "chore/") + + +def infer_workflow_type(path: str | None, branch: str | None = None) -> str: + """Infer the creating workflow type from a worktree path or branch name.""" + text = f"{path or ''} {branch or ''}".lower() + if "baseline" in text: + return WORKFLOW_BASELINE + if "merge-sim" in text or "merge_sim" in text or "mergesim" in text: + return WORKFLOW_MERGE_SIMULATION + if "review" in text or "review-pr" in text: + return WORKFLOW_REVIEW + if "conflict" in text: + return WORKFLOW_CONFLICT_FIX + if "issue-" in text or (branch or "").startswith(_ISSUE_BRANCH_PREFIXES): + return WORKFLOW_ISSUE_WORK + return WORKFLOW_UNKNOWN + + +def _extract_issue_number(path: str | None, branch: str | None) -> int | None: + match = _ISSUE_REF_RE.search(f"{path or ''} {branch or ''}") + return int(match.group(1)) if match else None + + +def build_worktree_metadata( + *, + path: str, + branch: str | None = None, + head_sha: str | None = None, + workflow_type: str | None = None, + issue_number: int | None = None, + pr_number: int | None = None, + creator: str | None = None, + profile: str | None = None, + created_at: str | None = None, + last_used_at: str | None = None, + cleanup_eligibility: str | None = None, +) -> dict[str, Any]: + """Return ownership/purpose metadata for a session-owned worktree. + + Covers acceptance criterion 1: path, workflow type, issue/PR number, + branch/head SHA, creator identity/profile, created and last-used + timestamps, and cleanup eligibility. + """ + wt = workflow_type or infer_workflow_type(path, branch) + issue = issue_number + if issue is None: + issue = _extract_issue_number(path, branch) + return { + "path": path, + "workflow_type": wt, + "issue_number": issue, + "pr_number": pr_number, + "branch": branch, + "head_sha": head_sha, + "creator": creator, + "profile": profile, + "created_at": created_at, + "last_used_at": last_used_at, + "auto_remove_on_success": wt in AUTO_REMOVE_ON_SUCCESS, + "cleanup_eligibility": cleanup_eligibility, + } + + +def _parse_timestamp(value: str | None) -> datetime | None: + if not value: + return None + text = str(value).strip() + if not text: + return None + if text.endswith("Z"): + text = text[:-1] + "+00:00" + try: + parsed = datetime.fromisoformat(text) + except ValueError: + return None + if parsed.tzinfo is None: + return parsed.replace(tzinfo=timezone.utc) + return parsed + + +def is_ttl_expired( + *, + last_used_at: str | None, + now: datetime | str | None, + ttl_hours: float = DEFAULT_TTL_HOURS, +) -> bool: + """Return True only when the age is known and exceeds ``ttl_hours``. + + Unknown or unparseable timestamps fail safe (not expired) so a worktree + is never treated as removable merely because its age is unknown. + """ + last = _parse_timestamp(last_used_at) + now_dt = now if isinstance(now, datetime) else _parse_timestamp(now) + if last is None or now_dt is None: + return False + if now_dt.tzinfo is None: + now_dt = now_dt.replace(tzinfo=timezone.utc) + return (now_dt - last).total_seconds() > ttl_hours * 3600.0 + + +def classify_worktree( + *, + workflow_type: str, + is_dirty: bool, + has_open_pr: bool = False, + has_active_lease: bool = False, + has_active_issue_lock: bool = False, + is_detached: bool = False, + branch_gone: bool = False, + ttl_expired: bool = False, + is_protected: bool = False, + metadata_known: bool = True, +) -> str: + """Classify a worktree, safety-first: any preservation signal wins. + + Dirty, open-PR, leased, active-lock, protected, and unknown states are + all non-removable and are checked before any removable classification, + so nothing removable can shadow a preservation signal (criteria 6-8). + """ + if is_protected: + # The main checkout / a protected base branch is never removable. + return CLASS_UNSAFE_UNKNOWN + if is_dirty: + return CLASS_DIRTY_LOCAL # never auto-deleted (criterion 6) + if has_open_pr: + return CLASS_ACTIVE_OPEN_PR # never auto-deleted (criterion 7) + if has_active_lease: + return CLASS_ACTIVE_ISSUE_WORK # never auto-deleted (criterion 8) + if has_active_issue_lock: + return CLASS_ACTIVE_ISSUE_WORK + if not metadata_known or workflow_type == WORKFLOW_UNKNOWN: + return CLASS_UNSAFE_UNKNOWN # never auto-deleted without proof + + # Clean, no PR, no lease, no lock, known workflow type. + if workflow_type in AUTO_REMOVE_ON_SUCCESS: + if is_detached or branch_gone: + return CLASS_DETACHED_REVIEW_LEFTOVER + return CLASS_CLEAN_STALE_REMOVABLE + # issue_work / conflict_fix: only removable once the TTL has expired. + if ttl_expired: + return CLASS_CLEAN_STALE_REMOVABLE + return CLASS_ACTIVE_ISSUE_WORK + + +def is_removable(classification: str) -> bool: + """Return True only for the two auto-removable classifications.""" + return classification in REMOVABLE_CLASSES + + +def assess_worktree_removal( + *, + path: str, + branch: str | None, + head_sha: str | None, + is_dirty: bool, + has_open_pr: bool, + has_active_lease: bool, + classification: str, +) -> dict[str, Any]: + """Return a per-worktree removal decision with proof (criterion 9). + + A worktree is only safe to remove when it is clean, has no active PR, + has no active lease, and its classification is auto-removable. + """ + block_reasons: list[str] = [] + if is_dirty: + block_reasons.append("worktree has uncommitted changes") + if has_open_pr: + block_reasons.append("worktree branch has an open PR") + if has_active_lease: + block_reasons.append("worktree has an active lease") + if not is_removable(classification): + block_reasons.append( + f"classification '{classification}' is not auto-removable" + ) + return { + "path": path, + "branch": branch, + "head_sha": head_sha, + "classification": classification, + "clean": not is_dirty, + "no_active_pr": not has_open_pr, + "no_active_lease": not has_active_lease, + "safe_to_remove": not block_reasons, + "block_reasons": block_reasons, + } + + +def plan_success_cleanup( + *, + metadata: dict[str, Any], + is_dirty: bool, + has_open_pr: bool, + has_active_lease: bool, +) -> dict[str, Any]: + """Decide whether a just-completed worktree is removed at success. + + Review/baseline/merge-simulation worktrees are removed automatically at + successful completion (criterion 2); everything else is preserved and + reported. Dirty/PR/leased worktrees are always preserved (criteria 6-8). + """ + workflow_type = metadata.get("workflow_type", WORKFLOW_UNKNOWN) + if not metadata.get("auto_remove_on_success"): + return { + "remove": False, + "reason": f"workflow type '{workflow_type}' is preserved by policy", + } + classification = classify_worktree( + workflow_type=workflow_type, + is_dirty=is_dirty, + has_open_pr=has_open_pr, + has_active_lease=has_active_lease, + ) + decision = assess_worktree_removal( + path=metadata.get("path", ""), + branch=metadata.get("branch"), + head_sha=metadata.get("head_sha"), + is_dirty=is_dirty, + has_open_pr=has_open_pr, + has_active_lease=has_active_lease, + classification=classification, + ) + return { + "remove": decision["safe_to_remove"], + "reason": "clean transient worktree removable at success completion" + if decision["safe_to_remove"] + else "; ".join(decision["block_reasons"]), + "classification": classification, + "decision": decision, + } + + +def cleanup_failure_report(path: str, reason: str) -> dict[str, Any]: + """Structured leftover-worktree record for the final report (criterion 3).""" + return {"path": path, "removed": False, "reason": reason} + + +# -------------------------------------------------------------------------- +# git-shelling helpers (only these touch the filesystem) +# -------------------------------------------------------------------------- + + +def parse_worktree_porcelain(text: str) -> list[dict[str, Any]]: + """Parse ``git worktree list --porcelain`` output into entries.""" + entries: list[dict[str, Any]] = [] + current: dict[str, Any] = {} + for raw in (text or "").splitlines(): + line = raw.rstrip("\n") + if not line: + if current: + entries.append(current) + current = {} + continue + if line.startswith("worktree "): + if current: + entries.append(current) + current = { + "path": line[len("worktree ") :].strip(), + "head": None, + "branch": None, + "detached": False, + "bare": False, + } + elif line.startswith("HEAD "): + current["head"] = line[len("HEAD ") :].strip() + elif line.startswith("branch "): + ref = line[len("branch ") :].strip() + current["branch"] = ref.replace("refs/heads/", "", 1) + elif line == "detached": + current["detached"] = True + elif line == "bare": + current["bare"] = True + if current: + entries.append(current) + return entries + + +def list_worktrees(project_root: str) -> list[dict[str, Any]]: + """Return parsed ``git worktree list`` entries for ``project_root``.""" + result = subprocess.run( + ["git", "-C", project_root, "worktree", "list", "--porcelain"], + capture_output=True, + text=True, + check=False, + ) + if result.returncode != 0: + return [] + return parse_worktree_porcelain(result.stdout) + + +def git_worktree_list(project_root: str) -> str: + """Return plain ``git worktree list`` output for final verification (criterion 10).""" + result = subprocess.run( + ["git", "-C", project_root, "worktree", "list"], + capture_output=True, + text=True, + check=False, + ) + return (result.stdout or "").strip() + + +def read_worktree_dirty(path: str) -> dict[str, Any]: + """Return dirty state for a worktree path via ``git status --porcelain``.""" + if not path or not os.path.isdir(path): + return {"exists": False, "dirty": None, "dirty_files": []} + result = subprocess.run( + ["git", "-C", path, "status", "--porcelain"], + capture_output=True, + text=True, + check=False, + ) + files = [ln for ln in (result.stdout or "").splitlines() if ln.strip()] + return {"exists": True, "dirty": bool(files), "dirty_files": files} + + +def remove_worktree(project_root: str, path: str) -> dict[str, Any]: + """Remove a single worktree via ``git worktree remove`` (no ``--force``).""" + result = subprocess.run( + ["git", "-C", project_root, "worktree", "remove", path], + capture_output=True, + text=True, + check=False, + ) + ok = result.returncode == 0 + return { + "path": path, + "removed": ok, + "reason": None if ok else (result.stderr or "git worktree remove failed").strip(), + } + + +def _is_under_branches(project_root: str, path: str) -> bool: + branches_root = os.path.join(os.path.abspath(project_root), "branches") + return os.path.abspath(path or "").startswith(branches_root + os.sep) + + +def audit_branches_directory( + project_root: str, + *, + open_pr_branches: set[str] | None = None, + leased_branches: set[str] | None = None, + active_issue_branches: set[str] | None = None, + now: datetime | str | None = None, + ttl_hours: float = DEFAULT_TTL_HOURS, +) -> dict[str, Any]: + """Classify every session-owned worktree under ``branches/``. + + Read-only: shells out to git for discovery and dirty state, then applies + the pure classifier. Returns per-worktree classifications, counts, the + list of removable candidates, and the ``git worktree list`` proof. + """ + open_pr_branches = open_pr_branches or set() + leased_branches = leased_branches or set() + active_issue_branches = active_issue_branches or set() + + worktrees: list[dict[str, Any]] = [] + for entry in list_worktrees(project_root): + path = entry.get("path") or "" + branch = entry.get("branch") + is_protected = (branch in PROTECTED_BRANCHES) or not _is_under_branches( + project_root, path + ) + dirty_state = read_worktree_dirty(path) + is_dirty = bool(dirty_state.get("dirty")) + metadata = build_worktree_metadata( + path=path, branch=branch, head_sha=entry.get("head") + ) + has_open_pr = bool(branch) and branch in open_pr_branches + has_active_lease = bool(branch) and branch in leased_branches + has_active_lock = bool(branch) and branch in active_issue_branches + ttl_expired = is_ttl_expired( + last_used_at=metadata.get("last_used_at"), now=now, ttl_hours=ttl_hours + ) + classification = classify_worktree( + workflow_type=metadata["workflow_type"], + is_dirty=is_dirty, + has_open_pr=has_open_pr, + has_active_lease=has_active_lease, + has_active_issue_lock=has_active_lock, + is_detached=bool(entry.get("detached")), + branch_gone=branch is None and not entry.get("detached"), + ttl_expired=ttl_expired, + is_protected=is_protected, + ) + metadata["cleanup_eligibility"] = classification + worktrees.append( + { + **metadata, + "detached": bool(entry.get("detached")), + "dirty": is_dirty, + "dirty_files": dirty_state.get("dirty_files", []), + "has_open_pr": has_open_pr, + "has_active_lease": has_active_lease, + "has_active_issue_lock": has_active_lock, + "is_protected": is_protected, + "classification": classification, + "removable": is_removable(classification), + } + ) + + counts: dict[str, int] = {} + for wt in worktrees: + counts[wt["classification"]] = counts.get(wt["classification"], 0) + 1 + removable = [wt for wt in worktrees if wt["removable"]] + + return { + "project_root": project_root, + "worktrees": worktrees, + "counts": counts, + "removable_candidates": removable, + "removable_count": len(removable), + "total": len(worktrees), + "git_worktree_list": git_worktree_list(project_root), + } +CLASSIFICATIONS = frozenset({ + "active_open_pr", + "active_issue_work", + "dirty_local_worktree", + "clean_stale_removable", + "detached_review_leftover", + "orphan_directory", + "unsafe_unknown", +}) + +PRESERVE_CLASSIFICATIONS = frozenset({ + "active_open_pr", + "active_issue_work", + "dirty_local_worktree", + "unsafe_unknown", +}) + +DISPOSITIONS = frozenset({ + "removed_intentionally", + "preserved_exists", + "preserved_missing_explained", + "not_registered_worktree", + "unsafe_unknown", +}) + +REVIEW_WORKTREE_RE = re.compile( + r"branches/(?:review-pr\d+|merge-simulation-pr\d+|review-[\w-]+)", + re.IGNORECASE, +) + + +def normalize_path(path: str) -> str: + return os.path.normpath((path or "").strip()) + + +def relative_branches_path(project_root: str, path: str) -> str: + root = normalize_path(project_root) + normalized = normalize_path(path) + if normalized.startswith(root + os.sep): + return normalized[len(root) + 1 :] + return normalized.replace("\\", "/") + + +def parse_worktree_list_porcelain(porcelain: str) -> list[dict[str, Any]]: + """Parse ``git worktree list --porcelain`` into worktree records.""" + entries: list[dict[str, Any]] = [] + current: dict[str, Any] = {} + for raw in (porcelain or "").splitlines(): + line = raw.strip() + if not line: + if current: + entries.append(current) + current = {} + continue + if line.startswith("worktree "): + if current: + entries.append(current) + current = {"path": line.split(" ", 1)[1].strip()} + elif line.startswith("HEAD "): + current["head_sha"] = line.split(" ", 1)[1].strip() + elif line.startswith("branch "): + current["branch"] = line.split(" ", 1)[1].strip().removeprefix("refs/heads/") + elif line == "detached": + current["detached"] = True + elif line == "bare": + current["bare"] = True + if current: + entries.append(current) + return entries + + +def list_branches_directories(project_root: str, dir_names: list[str] | None = None) -> list[str]: + """Return relative ``branches/`` paths for first-level directories.""" + branches_root = os.path.join(project_root, "branches") + if dir_names is not None: + return sorted( + f"branches/{name}" + for name in dir_names + if name and not name.startswith(".") + ) + if not os.path.isdir(branches_root): + return [] + names: list[str] = [] + for entry in sorted(os.listdir(branches_root)): + full = os.path.join(branches_root, entry) + if entry.startswith(".") or not os.path.isdir(full): + continue + names.append(f"branches/{entry}") + return names + + +def _untracked_dirty(porcelain: str) -> bool: + return any(line.startswith("??") for line in (porcelain or "").splitlines()) + + +def classify_branches_entry( + *, + rel_path: str, + worktree_record: dict[str, Any] | None, + worktree_state: dict[str, Any] | None, + open_pr_branches: set[str] | None = None, + active_lock_branches: set[str] | None = None, + active_issue_branches: set[str] | None = None, +) -> str: + """Classify a ``branches/`` directory for cleanup policy.""" + open_pr_branches = open_pr_branches or set() + active_lock_branches = active_lock_branches or set() + active_issue_branches = active_issue_branches or set() + state = worktree_state or {} + record = worktree_record or {} + + branch_name = (record.get("branch") or "").strip() + folder_name = rel_path.split("/", 1)[-1] if "/" in rel_path else rel_path + inferred_branch = folder_name.replace("-", "/") if "/" not in folder_name else folder_name + + candidate_branches = {b for b in (branch_name, inferred_branch) if b} + open_folder_names = {branch_worktree_folder(b) for b in open_pr_branches} + if folder_name in open_folder_names or any( + b in open_pr_branches for b in candidate_branches + ): + return "active_open_pr" + if any(b in active_lock_branches or b in active_issue_branches for b in candidate_branches): + return "active_issue_work" + + dirty_tracked = bool(state.get("dirty_files")) + dirty_untracked = bool(state.get("dirty_untracked")) + if dirty_tracked or dirty_untracked: + return "dirty_local_worktree" + + if not record: + return "orphan_directory" + + if record.get("detached") and REVIEW_WORKTREE_RE.search(rel_path): + return "detached_review_leftover" + + if state.get("exists") and state.get("clean"): + return "clean_stale_removable" + + return "unsafe_unknown" + + +def capture_cleanup_snapshot( + project_root: str, + *, + branch_dirs: list[str] | None = None, + worktree_porcelain: str | None = None, + open_pr_branches: set[str] | None = None, + active_lock_branches: set[str] | None = None, + active_issue_branches: set[str] | None = None, + issue_lock_path: str | None = None, +) -> dict[str, Any]: + """Capture audit snapshot for ``branches/`` dirs and registered worktrees.""" + root = normalize_path(project_root) + rel_dirs = list_branches_directories(root, branch_dirs) + worktrees = parse_worktree_list_porcelain(worktree_porcelain or "") + worktree_by_rel: dict[str, dict[str, Any]] = {} + for wt in worktrees: + rel = relative_branches_path(root, wt.get("path") or "") + if rel.startswith("branches/"): + worktree_by_rel[rel] = wt + + lock_branches = set(active_lock_branches or []) + lock = None + if issue_lock_path: + from merged_cleanup_reconcile import read_issue_lock + + lock = read_issue_lock(issue_lock_path) + if lock and lock.get("branch_name"): + lock_branches.add(str(lock["branch_name"])) + + entries: list[dict[str, Any]] = [] + for rel_path in rel_dirs: + abs_path = os.path.join(root, rel_path) + wt_record = worktree_by_rel.get(rel_path) + state = read_local_worktree_state(abs_path) if os.path.isdir(abs_path) else { + "exists": False, + "clean": None, + "dirty_files": [], + } + if state.get("exists"): + status_res = state.get("porcelain_status") + if status_res is None and os.path.isdir(abs_path): + import subprocess + + proc = subprocess.run( + ["git", "-C", abs_path, "status", "--porcelain"], + capture_output=True, + text=True, + check=False, + ) + status_res = proc.stdout or "" + state["dirty_untracked"] = _untracked_dirty(status_res or "") + state["dirty_files"] = state.get("dirty_files") or parse_dirty_tracked_files( + status_res or "" + ) + state["clean"] = not state["dirty_files"] and not state["dirty_untracked"] + + classification = classify_branches_entry( + rel_path=rel_path, + worktree_record=wt_record, + worktree_state=state, + open_pr_branches=open_pr_branches, + active_lock_branches=lock_branches, + active_issue_branches=active_issue_branches, + ) + entries.append( + { + "path": rel_path, + "absolute_path": abs_path, + "classification": classification, + "registered_worktree": bool(wt_record), + "worktree_record": wt_record or None, + "worktree_state": state, + "preserve": classification in PRESERVE_CLASSIFICATIONS, + } + ) + + return { + "project_root": root, + "branch_directory_count": len(rel_dirs), + "registered_branches_worktree_count": len(worktree_by_rel), + "entries": entries, + "worktrees": worktrees, + } + + +def _index_snapshot_entries(snapshot: dict[str, Any]) -> dict[str, dict[str, Any]]: + return {entry["path"]: entry for entry in snapshot.get("entries") or []} + + +def _removal_paths(removal_log: list[dict[str, Any]] | None) -> dict[str, dict[str, Any]]: + indexed: dict[str, dict[str, Any]] = {} + for item in removal_log or []: + rel = (item.get("path") or "").strip().replace("\\", "/") + if rel: + indexed[rel] = item + return indexed + + +def reconcile_cleanup_audit( + before: dict[str, Any], + after: dict[str, Any], + removal_log: list[dict[str, Any]] | None = None, + *, + explained_missing: dict[str, str] | None = None, + concurrent_mutations: list[str] | None = None, +) -> dict[str, Any]: + """Reconcile before/after snapshots to exactly one disposition per path.""" + before_index = _index_snapshot_entries(before) + after_index = _index_snapshot_entries(after) + removals = _removal_paths(removal_log) + explained = { + (k or "").strip().replace("\\", "/"): (v or "").strip() + for k, v in (explained_missing or {}).items() + } + concurrent = { + (p or "").strip().replace("\\", "/") + for p in (concurrent_mutations or []) + } + + rows: list[dict[str, Any]] = [] + for path, before_entry in sorted(before_index.items()): + after_entry = after_index.get(path) + after_exists = bool(after_entry and after_entry.get("worktree_state", {}).get("exists")) + classification = before_entry.get("classification") or "unsafe_unknown" + preserve = bool(before_entry.get("preserve")) or classification in PRESERVE_CLASSIFICATIONS + removal = removals.get(path) + explanation = explained.get(path, "") + + if removal: + disposition = "removed_intentionally" + reasons = [] + elif after_exists: + disposition = "preserved_exists" + reasons = [] + elif explanation: + disposition = "preserved_missing_explained" + reasons = [explanation] + elif not before_entry.get("registered_worktree"): + disposition = "not_registered_worktree" + reasons = ["directory was not a registered git worktree at audit start"] + elif path in concurrent: + disposition = "preserved_missing_explained" + reasons = ["removed or mutated by another session during cleanup"] + elif preserve: + disposition = "unsafe_unknown" + reasons = [ + f"preserved classification '{classification}' disappeared without " + "removal log or explanation" + ] + else: + disposition = "unsafe_unknown" + reasons = [ + "clean/removable path disappeared without removal log entry" + ] + + rows.append( + { + "path": path, + "classification": classification, + "preserve": preserve, + "disposition": disposition, + "removed_intentionally": disposition == "removed_intentionally", + "removal_record": removal, + "after_exists": after_exists, + "reasons": reasons, + } + ) + + for path, removal in removals.items(): + if path not in before_index: + rows.append( + { + "path": path, + "classification": "unsafe_unknown", + "preserve": False, + "disposition": "unsafe_unknown", + "removed_intentionally": True, + "removal_record": removal, + "after_exists": path in after_index, + "reasons": ["removal log references path absent from before snapshot"], + } + ) + + counts = { + "initial_count": len(before_index), + "removed_count": sum(1 for r in rows if r["disposition"] == "removed_intentionally"), + "preserved_count": sum(1 for r in rows if r["disposition"] == "preserved_exists"), + "missing_unexplained_count": sum( + 1 + for r in rows + if r["disposition"] == "unsafe_unknown" + and r.get("preserve") + and not r.get("after_exists") + ), + "missing_explained_count": sum( + 1 for r in rows if r["disposition"] == "preserved_missing_explained" + ), + "final_count": len(after_index), + "orphan_directory_count": sum( + 1 for r in rows if r["disposition"] == "not_registered_worktree" + ), + } + expected_final = ( + counts["initial_count"] + - counts["removed_count"] + - counts["missing_explained_count"] + ) + counts["count_reconciles"] = counts["final_count"] == expected_final + + return { + "rows": rows, + "counts": counts, + "removal_log_complete": _removal_log_complete(before_index, after_index, removals), + } + + +def _removal_log_complete( + before_index: dict[str, dict[str, Any]], + after_index: dict[str, dict[str, Any]], + removals: dict[str, dict[str, Any]], +) -> bool: + for path, before_entry in before_index.items(): + if path in after_index: + continue + classification = before_entry.get("classification") or "" + if classification == "clean_stale_removable" and path not in removals: + return False + return True + + +def assess_cleanup_audit_integrity(reconciliation: dict[str, Any]) -> dict[str, Any]: + """Fail closed when preserved worktrees vanish or counts do not reconcile.""" + reasons: list[str] = [] + counts = reconciliation.get("counts") or {} + + if counts.get("missing_unexplained_count"): + reasons.append( + f"{counts['missing_unexplained_count']} preserved worktree(s) missing " + "without explanation" + ) + + for row in reconciliation.get("rows") or []: + if not row.get("preserve") or row.get("after_exists"): + continue + if row.get("disposition") == "removed_intentionally": + continue + message = ( + f"preserved worktree {row.get('path')} missing " + f"({row.get('disposition')})" + ) + if message not in reasons: + reasons.append(message) + for item in row.get("reasons") or []: + if item not in reasons: + reasons.append(item) + + if not counts.get("count_reconciles"): + reasons.append( + "final directory count does not reconcile with initial minus removed " + f"(initial={counts.get('initial_count')}, removed={counts.get('removed_count')}, " + f"final={counts.get('final_count')})" + ) + + if reconciliation.get("removal_log_complete") is False: + reasons.append("removal log omits one or more removed clean-stale worktrees") + + for row in reconciliation.get("rows") or []: + removal = row.get("removal_record") or {} + if row.get("disposition") == "removed_intentionally": + if not removal.get("method"): + reasons.append(f"removal log for {row.get('path')} missing method") + if not removal.get("pre_removal_proof"): + reasons.append(f"removal log for {row.get('path')} missing pre-removal proof") + + block = bool(reasons) + return { + "block": block, + "proven": not block, + "reasons": reasons, + "counts": counts, + "safe_next_action": ( + "capture before/after snapshots, record every removal with proof, and " + "explain any preserved path that disappears" + if block + else "proceed" + ), + } + + +def build_cleanup_reconciliation_table(reconciliation: dict[str, Any]) -> dict[str, Any]: + """Return the operator-facing reconciliation summary table.""" + counts = dict(reconciliation.get("counts") or {}) + return { + "initial_count": counts.get("initial_count", 0), + "removed_count": counts.get("removed_count", 0), + "preserved_count": counts.get("preserved_count", 0), + "missing_unexplained_count": counts.get("missing_unexplained_count", 0), + "missing_explained_count": counts.get("missing_explained_count", 0), + "final_count": counts.get("final_count", 0), + "count_reconciles": counts.get("count_reconciles", False), + } + + +def _read_worktree_porcelain(project_root: str) -> str: + import subprocess + + try: + res = subprocess.run( + ["git", "-C", project_root, "worktree", "list", "--porcelain"], + capture_output=True, + text=True, + check=False, + ) + except OSError: + return "" + return res.stdout if res.returncode == 0 else "" + + +def capture_branches_worktree_snapshot( + project_root: str, + *, + open_pr_branches: list[str] | None = None, + active_lock_branch: str | None = None, + leased_paths: list[str] | None = None, + issue_lock_path: str | None = None, +) -> dict[str, Any]: + """MCP-facing snapshot capture for live ``branches/`` cleanup audits.""" + root = normalize_path(project_root) + lock_branches = set() + if active_lock_branch: + lock_branches.add(active_lock_branch) + issue_branches = set() + for path in leased_paths or []: + rel = relative_branches_path(root, path) + if rel.startswith("branches/"): + issue_branches.add(rel.split("/", 1)[-1].replace("-", "/")) + return capture_cleanup_snapshot( + root, + worktree_porcelain=_read_worktree_porcelain(root), + open_pr_branches=set(open_pr_branches or []), + active_lock_branches=lock_branches, + active_issue_branches=issue_branches, + issue_lock_path=issue_lock_path, + ) + + +def assess_worktree_cleanup_integrity( + *, + before: dict[str, Any], + after: dict[str, Any], + removals: list[dict[str, Any]] | None = None, + explained_missing: dict[str, str] | None = None, +) -> dict[str, Any]: + """MCP-facing integrity assessment over before/after cleanup snapshots.""" + reconciliation = reconcile_cleanup_audit( + before, + after, + removals, + explained_missing=explained_missing, + ) + integrity = assess_cleanup_audit_integrity(reconciliation) + return { + **integrity, + "integrity_passed": integrity.get("proven", False), + "reconciliation": reconciliation, + "reconciliation_table": build_cleanup_reconciliation_table(reconciliation), + "rows": reconciliation.get("rows") or [], + } + + +_RECON_INITIAL_RE = re.compile(r"initial count\s*:\s*(\d+)", re.I) +_RECON_REMOVED_RE = re.compile(r"removed count\s*:\s*(\d+)", re.I) +_RECON_PRESERVED_RE = re.compile(r"preserved count\s*:\s*(\d+)", re.I) +_RECON_MISSING_RE = re.compile(r"missing-unexplained count\s*:\s*(\d+)", re.I) +_RECON_FINAL_RE = re.compile(r"final count\s*:\s*(\d+)", re.I) +_WORKTREE_LIST_RE = re.compile(r"git worktree list|worktree list proof", re.I) + + +def assess_cleanup_audit_final_report(report_text: str) -> dict[str, Any]: + """Validate cleanup final report includes reconciliation proof (#404).""" + text = report_text or "" + reasons: list[str] = [] + for pattern in ( + _RECON_INITIAL_RE, + _RECON_REMOVED_RE, + _RECON_PRESERVED_RE, + _RECON_MISSING_RE, + _RECON_FINAL_RE, + ): + if not pattern.search(text): + reasons.append( + f"cleanup report missing field matching /{pattern.pattern}/" + ) + if not _WORKTREE_LIST_RE.search(text): + reasons.append("final verification missing git worktree list proof") + proven = not reasons + return {"proven": proven, "block": not proven, "reasons": reasons} \ No newline at end of file