bootstrap_meta.py

#!/usr/bin/env python3
"""Create workspace/inputs/pr-<id>/meta.json from defaults if missing (no LLM)."""
from __future__ import annotations

import argparse
import json
import os
import sys
from pathlib import Path


def repo_root() -> Path:
    return Path(__file__).resolve().parents[2]


def default_meta(pr_id: int) -> dict:
    host = os.environ.get("CODE_REVIEW_GITCODE_HOST", "gitcode.com")
    owner = os.environ.get("CODE_REVIEW_GITCODE_OWNER", "openeuler")
    repo = os.environ.get("CODE_REVIEW_GITCODE_REPO", "yuanrong-datasystem")
    upstream = os.environ.get(
        "CODE_REVIEW_UPSTREAM_GIT",
        f"https://{host}/{owner}/{repo}.git",
    )
    fork_url = os.environ.get(
        "CODE_REVIEW_FORK_REMOTE_URL",
        f"https://{host}/yaohaolin/{repo}.git",
    )
    fork_name = os.environ.get("CODE_REVIEW_FORK_REMOTE_NAME", "contributor")
    branch_local = f"pr-{pr_id}-head"
    return {
        "pr_number": pr_id,
        "pr_url": f"https://{host}/{owner}/{repo}/pull/{pr_id}",
        "upstream_git": upstream,
        "issues": {},
        "related": {
            "description": "",
            "issues": [],
            "rfcs": [],
            "closes": [],
            "comment": "Link this PR to GitCode issues and RFCs; used in review prompt traceability.",
        },
        "gitcode": {
            "comment": "Optional. api_base_url defaults to https://api.gitcode.com.",
            "api_base_url": "https://api.gitcode.com",
        },
        "notes": "Auto-generated by scripts/lib/bootstrap_meta.py (first run). "
        "Set fork_fallback.branch from PR page if fetch fails; align base_ref/head_ref with the PR.",
        "diff": {
            "base_ref": "origin/master",
            "head_ref": branch_local,
            "exclude_paths": [],
            "fetch": [
                {
                    "comment": "Tries multiple refs; GitCode often needs fork_fallback.branch.",
                    "type": "pr_head",
                    "remote": "origin",
                    "pr": pr_id,
                    "local_branch": branch_local,
                }
            ],
        },
        "fork_fallback": {
            "comment": "If pr_head fails: set branch to the PR source/compare branch on GitCode.",
            "remote_name": fork_name,
            "remote_url": fork_url,
            "branch": "REPLACE_WITH_SOURCE_BRANCH_FROM_PR_PAGE",
        },
    }


def write_meta(pr_id: int, root: Path, *, force: bool = False) -> Path:
    out = root / "workspace" / "inputs" / f"pr-{pr_id}" / "meta.json"
    if out.is_file() and not force:
        return out
    out.parent.mkdir(parents=True, exist_ok=True)
    meta = default_meta(pr_id)
    with open(out, "w", encoding="utf-8") as f:
        json.dump(meta, f, indent=2, ensure_ascii=False)
        f.write("\n")
    print(f"Wrote {out}", file=sys.stderr)
    return out


def main() -> None:
    ap = argparse.ArgumentParser(description="Bootstrap workspace/inputs/pr-<id>/meta.json")
    ap.add_argument("pr_id", type=int, help="PR number (folder id under workspace/inputs/)")
    ap.add_argument(
        "--repo-root",
        type=Path,
        default=None,
        help="Repository root (default: parent of scripts/)",
    )
    ap.add_argument(
        "--force",
        action="store_true",
        help="Overwrite existing meta.json",
    )
    args = ap.parse_args()
    root = args.repo_root or repo_root()
    write_meta(args.pr_id, root, force=args.force)


if __name__ == "__main__":
    main()