#!/usr/bin/env python3
"""解析 PR 评论中的 `/eval <products> [sha=<full-head-sha>] [cases=<ref>]` 触发命令。

供 .github/workflows/eval-dispatch.yml 调用：评论体经环境变量 COMMENT_BODY
传入（避免 argv 注入），解析结果以 GitHub Actions output
（products / cases_ref / reviewed_sha）写出；格式非法时把单行错误写入
output `error` 并以非零退出。

产品集显式必填；`sha=` 在评测他人 PR 时必填（审核背书凭据，由 guard 校验），
评测自己创建的 PR 时可省略（自助模式，自动钉住派发时刻的当前 head）；
`cases=<ref>` 为用例仓库版本逃生舱（破坏性变更配对验证）。
"""

import os
import re
import sys

PRODUCT_RE = re.compile(r"^[a-z0-9][a-z0-9-]*$")
REF_CHARSET_RE = re.compile(r"^[A-Za-z0-9._/-]+$")
SHA_RE = re.compile(r"^[0-9a-fA-F]{40}$")
USAGE = (
    "用法: /eval <product>[,<product>...] [sha=<40位PR-head-SHA>] [cases=<ref>]；"
    "评测他人 PR 时 sha= 必填，自己的 PR 可省略；"
    "示例: /eval drive,doc sha=0123456789abcdef0123456789abcdef01234567"
)


def _is_valid_cases_ref(ref: str) -> bool:
    """等价于 git check-ref-format --allow-onelevel 的结构校验（不执行 git）。

    字符白名单之上补齐结构规则：禁止首尾斜杠与连续斜杠、尾部点号、
    任意位置的 `..`、以点开头或以 .lock 结尾的路径分量；另禁止 `-`
    开头，避免被下游 `git fetch origin <ref>` 当作选项解析。
    """
    if not ref or not REF_CHARSET_RE.match(ref):
        return False
    if ref.startswith(("-", "/")) or ref.endswith(("/", ".")):
        return False
    if ".." in ref or "//" in ref:
        return False
    for component in ref.split("/"):
        if component.startswith(".") or component.endswith(".lock"):
            return False
    return True


def parse(body: str):
    lines = [line.strip() for line in (body or "").splitlines() if line.strip()]
    if not lines:
        raise ValueError(f"评论为空。{USAGE}")
    tokens = lines[0].split()
    if tokens[0] != "/eval":
        raise ValueError(f"首行必须以 /eval 命令开头。{USAGE}")
    if len(tokens) < 2:
        raise ValueError(f"产品集显式必填。{USAGE}")

    products = [p for p in tokens[1].split(",") if p]
    if not products:
        raise ValueError(f"产品集显式必填。{USAGE}")
    for product in products:
        if not PRODUCT_RE.match(product):
            raise ValueError(f"产品名非法: {product}。{USAGE}")

    cases_ref = ""
    reviewed_sha = ""
    for extra in tokens[2:]:
        if extra.startswith("cases="):
            if cases_ref:
                raise ValueError(f"cases 引用只能指定一次。{USAGE}")
            cases_ref = extra[len("cases="):]
            if not _is_valid_cases_ref(cases_ref):
                raise ValueError(f"cases 引用非法: {extra}。{USAGE}")
        elif extra.startswith("sha="):
            if reviewed_sha:
                raise ValueError(f"审核 SHA 只能指定一次。{USAGE}")
            reviewed_sha = extra[len("sha="):]
            if not SHA_RE.match(reviewed_sha):
                raise ValueError(f"审核 SHA 非法: {extra}。{USAGE}")
        else:
            raise ValueError(f"未知参数: {extra}。{USAGE}")

    if not reviewed_sha:
        # 是否允许省略由 guard 按“评论者是否 PR 作者”裁决，解析层不拦
        return ",".join(products), cases_ref, ""

    return ",".join(products), cases_ref, reviewed_sha.lower()


def _write_outputs(pairs):
    lines = [f"{key}={value}" for key, value in pairs]
    output_path = os.environ.get("GITHUB_OUTPUT", "")
    if output_path:
        with open(output_path, "a", encoding="utf-8") as f:
            f.write("\n".join(lines) + "\n")
    print("\n".join(lines))


def main() -> int:
    try:
        products, cases_ref, reviewed_sha = parse(os.environ.get("COMMENT_BODY", ""))
    except ValueError as exc:
        _write_outputs([("error", str(exc))])
        print(str(exc), file=sys.stderr)
        return 1
    _write_outputs(
        [
            ("products", products),
            ("cases_ref", cases_ref),
            ("reviewed_sha", reviewed_sha),
        ]
    )
    return 0


if __name__ == "__main__":
    sys.exit(main())