#!/usr/bin/env python3
"""Run the composed L140 owner gates for all requested seeds."""
from __future__ import annotations

import json
import re
import subprocess
import time
from pathlib import Path


ROOT = Path(__file__).resolve().parents[4]
OUT = ROOT / "Saved/Linux/l140/seeds"
BIN = ROOT / "Saved/Linux/l140/bin/VoxelForgeCoreHeadless-l140-clang"
DATA = ROOT / "Standalone/data/owner_world.json"
STAMP = "6205e6c180ef32c41792dc3e0d8bf443a96df41974f40334af4bf3a157ab88da"
STAMP_RE = re.compile(r"^\[VoxelForgeSourceStamp\] sha256=([0-9a-f]+)$", re.M)
CHECKS = {
    "native_dispatch": re.compile(
        r"^\[DensityDispatch\] strate=1 generator=FloatingIslands "
        r"recipe=FloatingIslands\.ClusterField version=10 "
        r"bound_path=FloatingIslands-native adapter=FFloatingIslandFieldAdapter$", re.M),
    "island_route": re.compile(r"^\[FloatingIslandRouteAudit\].*status=PASS$", re.M),
    "passage_clearance": re.compile(r"^\[FloatingIslandPassageClearance\].*status=PASS$", re.M),
    "strate1_route": re.compile(
        r"^\[ResolvedStrateRouteAudit\] strate=1 .*blocked=0 .*unsupported=0 "
        r"grade_failures=0 mouth_binding=PASS status=PASS$", re.M),
    "route_summary": re.compile(
        r"^\[ResolvedPassageAuditSummary\].*route_walkable=4 .*route_bindings=4 "
        r"route_binding_failures=0 .*blocked_controls=3/3", re.M),
    "passages": re.compile(r"^\[ResolvedPassageAuditSummary\].*status=PASS$", re.M),
    "shell": re.compile(r"^\[PassageShellAuditSummary\].*status=PASS$", re.M),
    "naturalness": re.compile(r"^\[PassageNaturalnessSummary\].*status=PASS$", re.M),
    "mouth_profile": re.compile(r"^\[PassageMouthProfileSummary\].*status=PASS$", re.M),
    "ripple": re.compile(r"^\[PassageRippleAuditSummary\].*status=PASS$", re.M),
}


def run(seed: int, label: str, seams_only: bool) -> tuple[int, str, float]:
    log = OUT / f"seed{seed}-{label}.log"
    layout = OUT / f"seed{seed}-{label}-layout.json"
    command = ["taskset", "-c", "0-11", "prlimit", "--as=6442450944", "--",
        "timeout", "1740s", str(BIN), str(DATA), "--world-assembler-prototype",
        "--stream-benchmark", "--stream-pipeline", "--workers=8", "--clip-radius=3",
        "--max-clip-level=8", "--max-level=0", "--force-generate", f"--seed={seed}",
        "--recipe=surface-world", "--native-owner-fi", "--density-dispatch-report"]
    command.append("--audit-passage-tile-seams" if seams_only else "--audit-resolved-layout-passages")
    if not seams_only:
        command.append("--audit-passage-tile-seams")
    command.append(f"--resolved-layout-output={layout}")
    started = time.monotonic()
    with log.open("w", encoding="utf-8") as output:
        proc = subprocess.run(command, cwd=ROOT, stdout=output,
                              stderr=subprocess.STDOUT, check=False, timeout=1740)
    return proc.returncode, str(log.relative_to(ROOT)), round(time.monotonic() - started, 3)


def main() -> int:
    if not BIN.is_file() or not DATA.is_file():
        raise SystemExit("missing L140 Clang binary or composed owner fixture")
    OUT.mkdir(parents=True, exist_ok=True)
    results = []
    for seed in range(16):
        exit_code, log_name, elapsed = run(seed, "composed", False)
        content = (ROOT / log_name).read_text(encoding="utf-8", errors="replace")
        stamp = STAMP_RE.search(content)
        checks = {name: bool(pattern.search(content)) for name, pattern in CHECKS.items()}
        checks["source_stamp"] = stamp is not None and stamp.group(1) == STAMP
        checks["blocked_controls"] = len(re.findall(
            r"^\[ResolvedPassageBlockedControl\].*detected=true$", content, re.M)) == 3
        checks["negative_controls"] = (
            "[ResolvedLayoutValidatorControls] rejected=10/10 status=PASS" in content
            and len(re.findall(r"^\[PassageNaturalnessNegativeControl\].*detected=true$",
                content, re.M)) == 1
            and len(re.findall(r"^\[PassageMouthProfileNegativeControl\].*detected=true$",
                content, re.M)) == 1
            and len(re.findall(r"^\[PassageFloorStepNegativeControl\].*detected=true$",
                content, re.M)) == 3
            and "synthetic_spheres_negative_control=DETECTED" in content
            and "floor_beaded_negative_control=DETECTED" in content)
        seam_match = re.search(r"^\[PassageTileSeamAudit\].*status=(PASS|FAIL)$", content, re.M)
        seam_exit = exit_code
        seam_log = log_name
        seam_elapsed = 0.0
        if seam_match is None:
            seam_exit, seam_log, seam_elapsed = run(seed, "seams", True)
            seam_content = (ROOT / seam_log).read_text(encoding="utf-8", errors="replace")
            stamp = STAMP_RE.search(seam_content)
            checks["source_stamp"] &= stamp is not None and stamp.group(1) == STAMP
            seam_match = re.search(
                r"^\[PassageTileSeamAudit\].*status=(PASS|FAIL)$", seam_content, re.M)
            content += "\n" + seam_content
        checks["seams"] = seam_match is not None and seam_match.group(1) == "PASS"
        all_gates = all(checks.values()) and exit_code == 0 and seam_exit == 0
        row = {"seed": seed, "exit": exit_code, "seam_exit": seam_exit,
               "passed": all_gates, "checks": checks,
               "elapsed_s": elapsed + seam_elapsed, "log": log_name, "seam_log": seam_log}
        results.append(row)
        print("[L140SeedSweep] " + json.dumps(row, sort_keys=True), flush=True)
        (OUT / "runs-partial.json").write_text(json.dumps(results, indent=2) + "\n")
    passed = sum(row["passed"] for row in results)
    summary = {"source_stamp": STAMP, "seeds": list(range(16)),
               "passed": passed, "failed": 16 - passed, "results": results}
    (OUT / "summary.json").write_text(json.dumps(summary, indent=2) + "\n")
    print(f"[L140SeedSweepSummary] passed={passed}/16 failed={16-passed} status={'PASS' if passed==16 else 'FAIL'}",
          flush=True)
    return 0 if passed == 16 else 1


if __name__ == "__main__":
    raise SystemExit(main())