"""WFIGS/FIRMS × firefighting ADS-B correlation within 20 miles."""

from __future__ import annotations

from fire_aircraft import (
    FIREFIGHTER_ICAO,
    RADIUS_MILES,
    correlate_aircraft_to_fires,
    is_firefighter,
)


def _ac(hex_id, lat, lon, icao, **extra):
    return {
        "id": hex_id,
        "lat": lat,
        "lon": lon,
        "label": hex_id,
        "heading": 0,
        "speed": 120,
        "extra": {"type": icao, "hex": hex_id, **extra},
    }


def _fire(name, lat, lon, **extra):
    return {
        "id": name,
        "lat": lat,
        "lon": lon,
        "label": name,
        "extra": extra,
    }


def test_air_tractor_is_firefighter_airliner_is_not():
    assert is_firefighter(_ac("aaa", 0, 0, "AT802")) is True
    assert is_firefighter(_ac("bbb", 0, 0, "C130")) is True
    assert is_firefighter(_ac("ccc", 0, 0, "B738")) is False
    assert "AT802" in FIREFIGHTER_ICAO


def test_flags_tanker_within_20_miles_of_fire():
    # ~10 miles north of a Piedmont fire
    fire = _fire("Jones Gap", 35.00, -82.00)
    tanker = _ac("acf001", 35.145, -82.00, "AT802")
    airliner = _ac("a0b738", 35.145, -82.00, "B738")
    far = _ac("acfar", 35.50, -82.00, "C130")  # ~34 miles
    hits = correlate_aircraft_to_fires([fire], [tanker, airliner, far])
    assert RADIUS_MILES == 20.0
    assert len(hits) == 1
    h = hits[0]
    assert h["aircraft_hex"] == "acf001"
    assert h["fire_id"] == "Jones Gap"
    assert h["aircraft_type"] == "AT802"
    assert 0 < h["distance_mi"] <= 20.0


def test_persist_shape_has_reload_fields():
    fire = _fire("Jones Gap", 35.00, -82.00, src="wfigs")
    tanker = _ac("acf001", 35.10, -82.00, "S64")
    hits = correlate_aircraft_to_fires([fire], [tanker])
    assert set(hits[0]).issuperset({
        "fire_id", "fire_lat", "fire_lon",
        "aircraft_hex", "aircraft_type", "aircraft_lat", "aircraft_lon",
        "distance_mi",
    })