#!/usr/bin/env python3
"""Install the released CLI in a fresh cache, then test its generated consumer.

The fixture tests public templates, numeric auth errors, raw JSON, route override,
full int64 precision and regeneration protection. Only local temporary resources
are used; the framework and CLI are fetched from the specified remote version.
"""
import argparse
import hashlib
import json
import os
from pathlib import Path
import re
import subprocess
import tempfile
import time

ROOT = Path(__file__).resolve().parents[1]
MODULE = "git.yuntuo.hk/k2safe/go-omni"


def run(args, cwd, env, timeout=300):
    result = subprocess.run([str(arg) for arg in args], cwd=cwd, env=env,
                            capture_output=True, text=True, timeout=timeout)
    output = re.sub(r"https://[^/@\s]+:[^/@\s]+@", "https://", result.stdout + result.stderr)
    if result.returncode:
        raise RuntimeError(f"{args[0]} failed ({result.returncode}):\n{output}")
    return output


def main():
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--version", required=True)
    parser.add_argument("--report", type=Path, default=ROOT / "performance/responsePoliciesRemote.json")
    args = parser.parse_args()
    if not re.fullmatch(r"v\d+\.\d+\.\d+(?:-[0-9A-Za-z.-]+)?", args.version):
        parser.error("a fixed release version is required")
    env = {k: v for k, v in os.environ.items() if not k.startswith("GO_OMNI_")}
    env.update(GOWORK="off", GOFLAGS="", GOPRIVATE="git.yuntuo.hk", GONOSUMDB="git.yuntuo.hk")
    start = time.monotonic()
    with tempfile.TemporaryDirectory(prefix="goOmniResponseRelease-") as directory:
        temp = Path(directory)
        (temp / "bin").mkdir()
        env.update(GOBIN=str(temp / "bin"), GOMODCACHE=str(temp / "modcache"))
        print(f"Installing CLI {args.version} in an empty module cache", flush=True)
        run(["go", "install", MODULE + "/cmd/go-omni@" + args.version], temp, env)
        cli = temp / "bin/go-omni"
        run([cli, "api", "validate", "--api", ROOT / "examples/contracts/responses.api"], temp, env)
        env.update(GO_OMNI_SCAFFOLD_INTEGRATION="1", GO_OMNI_RELEASE_CLI=str(cli), GO_OMNI_RELEASE_VERSION=args.version)
        print("Testing remote generation, int64 precision and consumer runtime", flush=True)
        output = run(["go", "test", "-race", "-count=1", "-json", "-timeout=180s",
                      "./cmd/go-omni/apigen", "-run", "^TestResponseTemplateConsumer$"], ROOT, env)
        events = [json.loads(line) for line in output.splitlines() if line.startswith("{")]
        passed = [e["Test"] for e in events if e.get("Action") == "pass" and "Test" in e]
        assert passed == ["TestResponseTemplateConsumer"], output
        evidence = "".join(e.get("Output", "") for e in events)
        checksum = re.search(r"checksum (h1:\S+); no local replace", evidence)
        assert checksum, "consumer did not verify its remote dependency"
        report = {
            "passed": True, "frameworkVersion": args.version, "frameworkSum": checksum.group(1),
            "freshModuleCache": True, "noLocalReplace": True,
            "durationSeconds": round(time.monotonic() - start, 2),
            "checks": ["remoteCLIInstall", "int32ContractValidation", "templateProjectGeneration",
                       "int64MaxValueOnWire", "numericAuthErrors", "rawJSON", "routeOverride",
                       "handwrittenFilesPreserved", "generatedConsumerRaceTests", "fixedRemoteModuleChecksum"],
            "sourceHashes": {name: hashlib.sha256((ROOT / name).read_bytes()).hexdigest() for name in
                             ("scripts/verifyResponseRelease.py", "cmd/go-omni/apigen/response_test.go",
                              "examples/contracts/responses.api")},
        }
        args.report.write_text(json.dumps(report, ensure_ascii=False, indent=2) + "\n")
    print(f"PASS remote response contract {args.version}; temporary files removed", flush=True)


if __name__ == "__main__":
    main()