"""按固定的种子重新生成开发集的马虎投标文件，沿用复核过的结论（同一招标文件里复核过的「单元 + 问题类型」），
列出没复核过的新目标（evals/out/check/review/pending-units.json）。

    cd server && ../.venv/bin/python ../evals/check/regen.py
"""

from __future__ import annotations

import json
import pathlib
import subprocess
import sys

EVALS = pathlib.Path(__file__).resolve().parents[1]
OUT = EVALS / "out" / "check"
# （规矩的投标文件，马虎的投标文件，种子，密度）：开发集的固定组合（GUIDELINES §4）
SLOPPY = [
    ("S02-B", "S02-B-sloppy", 11, "high"),
    ("S04-A", "S04-A-sloppy", 7, "high"),
    ("S10-B", "S10-B-sloppy", 13, "high"),
    ("S01-A", "S01-A-sloppy", 17, "high"),
    ("S05-B", "S05-B-sloppy", 23, "high"),
    ("S07-A", "S07-A-sloppy", 19, "high"),
    ("S08-B", "S08-B-sloppy", 29, "high"),
    ("S04-B", "S04-B-low", 31, "low"),
    ("S08-A", "S08-A-low", 37, "low"),
    ("S10-A", "S10-A-low", 41, "low"),
]


def reviews(tender: str) -> dict[tuple[str, str], dict]:
    out = {}
    for f in OUT.glob(f"{tender}-*/manifest.json"):
        for uid, e in json.loads(f.read_text())["units"].items():
            if e.get("review") and e.get("expect") == "defect":
                out[(uid, e["defect"])] = {k: v for k, v in e["review"].items() if k != "user_spot_check"}
    return out


def main() -> None:
    prior = {t: reviews(t) for t in {c[:3] for c, *_ in SLOPPY}}
    pending = []
    for clean, sloppy, seed, density in SLOPPY:
        tender = clean[:3]
        cmd = [
            sys.executable,
            str(EVALS / "check" / "mutate.py"),
            clean,
            sloppy,
            "--split",
            "dev",
            "--seed",
            str(seed),
            "--density",
            density,
        ]
        print(subprocess.run(cmd, capture_output=True, text=True, check=True).stdout.strip())
        p = OUT / sloppy / "manifest.json"
        m = json.loads(p.read_text())
        gold = {x["id"]: x for x in json.loads((EVALS / "gold" / f"{tender}.json").read_text())["items"]}
        for uid, e in m["units"].items():
            if e.get("expect") != "defect":
                continue
            if e.get("defect") == "D02":
                e["mode"] = "evidence"
            r = prior[tender].get((uid, e["defect"]))
            if r:
                e["reviewed"], e["review"] = r["ok"], r
            else:
                x = gold[uid]
                pending.append(
                    {
                        "bid": sloppy,
                        "tender": tender,
                        "unit": uid,
                        "cat": e["cat"],
                        "basis": x.get("basis"),
                        "anchors": x.get("anchors"),
                        "note": x.get("note"),
                        "mode": e["mode"],
                        "defect": e["defect"],
                        "variant": e.get("variant"),
                    }
                )
        p.write_text(json.dumps(m, ensure_ascii=False, indent=1))
    (OUT / "review" / "pending-units.json").write_text(json.dumps(pending, ensure_ascii=False, indent=1))
    print("待复核的新目标：", len(pending))


if __name__ == "__main__":
    main()