#!/usr/bin/env python3
"""Build a final conditional-workflow result and freshness summary."""

from __future__ import annotations

import argparse
import json
import os
from datetime import datetime, timezone
from pathlib import Path


def summarize(policy: dict, results: dict, selected: dict) -> tuple[list[dict], list[str]]:
    """Return per-gate rows and selected-job failures."""
    failures: list[str] = []
    rows: list[dict] = []
    failed_jobs: set[str] = set()
    for gate in policy["gates"]:
        job = gate["job"]
        is_selected = bool(selected.get(job, False))
        result = results.get(job, "unknown")
        status = result if is_selected else "not_selected_no_fresh_evidence"
        if is_selected and result != "success" and job not in failed_jobs:
            failures.append(f"{job}: selected but result was {result}")
            failed_jobs.add(job)
        rows.append({"gate": gate["id"], "job": job, "selected": is_selected, "status": status})
    return rows, failures


def main() -> int:
    parser = argparse.ArgumentParser()
    parser.add_argument("--policy", type=Path, default=Path("ci/conditional-test-policy.json"))
    parser.add_argument("--output", type=Path, default=Path("conditional-evidence/summary.json"))
    parser.add_argument("--results-json", default=os.environ.get("CONDITIONAL_RESULTS", "{}"))
    parser.add_argument("--selected-json", default=os.environ.get("CONDITIONAL_SELECTED", "{}"))
    args = parser.parse_args()
    policy = json.loads(args.policy.read_text())
    results = json.loads(args.results_json)
    selected = json.loads(args.selected_json)
    rows, failures = summarize(policy, results, selected)
    payload = {
        "schema_version": 1,
        "generated_at": datetime.now(timezone.utc).isoformat(),
        "git_sha": os.environ.get("GITHUB_SHA", "local"),
        "maximum_evidence_age_days": policy["maximum_evidence_age_days"],
        "gates": rows,
        "failures": failures,
    }
    args.output.parent.mkdir(parents=True, exist_ok=True)
    args.output.write_text(json.dumps(payload, indent=2, sort_keys=True) + "\n")
    summary_path = os.environ.get("GITHUB_STEP_SUMMARY")
    if summary_path:
        lines = ["## Conditional test evidence", "", "| Gate | Job | Status |", "|---|---|---|"]
        lines.extend(f"| {row['gate']} | {row['job']} | {row['status']} |" for row in rows)
        lines.extend(["", f"Evidence becomes stale after {policy['maximum_evidence_age_days']} days."])
        with Path(summary_path).open("a") as handle:
            handle.write("\n".join(lines) + "\n")
    for failure in failures:
        print(f"ERROR: {failure}")
    return 1 if failures else 0


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