import json
from pathlib import Path
from typing import Any

from figment.rules import run_red_flag_checks


CARD_DIR = Path("data/protocol_cards")


def _known_card_ids() -> set[str]:
    return {
        json.loads(path.read_text(encoding="utf-8"))["card_id"]
        for path in CARD_DIR.glob("*.json")
    }


def _confirmed(**overrides: Any) -> dict[str, Any]:
    intake = {
        "confirmed": True,
        "setting": "field clinic",
        "patient_age": "42",
        "pregnancy_status": "not_applicable",
        "chief_concern": "",
        "symptoms": "",
        "vitals": "",
        "allergies": "unknown",
        "medications": "unknown",
        "available_supplies": "radio, basic kit",
        "responder_note": "",
    }
    intake.update(overrides)
    return intake


def test_red_flag_rules_emit_only_existing_protocol_card_ids() -> None:
    cases = [
        _confirmed(chief_concern="new confusion", symptoms="difficult to wake"),
        _confirmed(chief_concern="severe respiratory distress", symptoms="gasping and blue lips"),
        _confirmed(
            pregnancy_status="pregnant",
            chief_concern="vaginal bleeding",
            symptoms="severe headache",
        ),
        _confirmed(chief_concern="chest pain", symptoms="chest pressure"),
        _confirmed(chief_concern="stroke concern", symptoms="facial droop and slurred speech"),
        _confirmed(patient_age="7", chief_concern="dehydration", symptoms="lethargic with no urine"),
        _confirmed(chief_concern="fever", symptoms="stiff neck", vitals="temperature 103 F"),
        _confirmed(chief_concern="wound getting worse", symptoms="spreading redness and foul drainage"),
    ]
    known_cards = _known_card_ids()
    emitted_rules = []

    for intake in cases:
        emitted_rules.extend(run_red_flag_checks(intake))

    assert {
        "AMS-001",
        "RESP-001",
        "PREG-001",
        "red_flag_chest_pain",
        "STROKE-001",
        "PED-DEHYD-001",
        "FEVER-001",
        "WOUND-001",
    } == {rule.rule_id for rule in emitted_rules}
    assert {rule.card_id for rule in emitted_rules} <= known_cards