#!/usr/bin/env python3
"""Quick demo assertions — run before recording."""
import os
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parent.parent
sys.path.insert(0, str(ROOT))

from greenlight.config import load_env

for k, v in load_env().items():
    os.environ.setdefault(k, v)

from greenlight.engine.events import EventStream
from greenlight.engine.coordinator import Coordinator
from greenlight.human.channel import DemoHuman


def main():
    agent_mode = os.getenv("GREENLIGHT_LIVE_LLM", "0") == "1"
    events = EventStream(echo=False)
    ledger = Coordinator(events, DemoHuman()).run()
    blocked_ids = {c.claim_id for c in ledger.blocked}
    cleared_ids = {c.claim_id for c in ledger.cleared}

    assert "C1" in blocked_ids, f"C1 should be blocked, got {blocked_ids}"
    assert "C2" in blocked_ids, f"C2 should be blocked, got {blocked_ids}"
    c1 = next(c for c in ledger.claims if c.claim_id == "C1")
    assert c1.status == "blocked"
    assert "40" in (c1.remediation or c1.citation or ""), "C1 should cite 40% TC gap"

    if agent_mode:
        assert len(cleared_ids) >= 3, f"agent should clear at least 3, got {cleared_ids}"
        assert ledger.determination, "agent should produce determination"
        print(
            f"demo_test OK (agent) — {len(cleared_ids)}/6 cleared, "
            f"{len(blocked_ids)} blocked, Kimi tool loop"
        )
    else:
        assert "C3" in cleared_ids, "C3 GOTS should be cleared"
        assert len(cleared_ids) == 3, f"expected 3 cleared, got {cleared_ids}"
        c4 = next(c for c in ledger.claims if c.claim_id == "C4")
        assert c4.status == "needs-evidence", "C4 OEKO-TEX should be needs-evidence without cert"
        assert ledger.determination["confidence"] == 0.5
        assert ledger.determination["max_exposure_eur"] == 80_000_000
        print("demo_test OK — 3/6 cleared, 2 blocked, 1 needs-evidence (C4 OEKO-TEX), determination ready")
    return 0


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