"""Plate normalisation + attachment to the vehicle it sits inside."""

from __future__ import annotations

from core.graph import attach_plates
from core.plates import ChainPlateRecognizer, _as_conf, normalize_plate
from core.schemas import BBox, EvidenceGraph, Plate, Vehicle


def _car(vid: str, box: tuple[float, float, float, float]) -> Vehicle:
    return Vehicle(
        id=vid,
        type="car",
        bbox=BBox(x1=box[0], y1=box[1], x2=box[2], y2=box[3]),
        confidence=0.9,
    )


def test_normalize_valid_indian_plate() -> None:
    assert normalize_plate("up 32 ab 1234") == ("UP32AB1234", True)


def test_normalize_rejects_garbage() -> None:
    cleaned, ok = normalize_plate("!!??")
    assert ok is False
    assert cleaned == ""


def test_plate_attaches_to_enclosing_vehicle() -> None:
    g = EvidenceGraph(image_id="i", vehicles=[_car("v1", (0, 0, 100, 100))])
    plate = Plate(
        text="UP32AB1234",
        regex_ok=True,
        confidence=0.8,
        bbox=BBox(x1=40, y1=70, x2=70, y2=85),
    )
    attach_plates(g, [plate])
    got = g.plate_of("v1")
    assert got is not None
    assert got.text == "UP32AB1234"


def test_plate_outside_vehicle_is_not_attached() -> None:
    g = EvidenceGraph(image_id="i", vehicles=[_car("v1", (0, 0, 50, 50))])
    plate = Plate(
        text="UP32AB1234",
        regex_ok=True,
        confidence=0.8,
        bbox=BBox(x1=200, y1=200, x2=230, y2=215),
    )
    attach_plates(g, [plate])
    assert g.plate_of("v1") is None


def test_as_conf_collapses_list_to_mean() -> None:
    assert _as_conf([0.8, 0.6]) == 0.7
    assert _as_conf(0.5) == 0.5
    assert _as_conf(None) == 0.0


class _Fixed:
    def __init__(self, plates: list[Plate]) -> None:
        self._plates = plates

    def read_all(self, image_path: str) -> list[Plate]:
        return self._plates


def test_chain_uses_primary_when_it_reads_a_plate() -> None:
    chain = ChainPlateRecognizer(
        _Fixed([Plate(text="UP32AB1234")]), _Fixed([Plate(text="ZZ")])
    )
    assert chain.read_all("x")[0].text == "UP32AB1234"


def test_chain_falls_back_when_primary_is_empty() -> None:
    chain = ChainPlateRecognizer(_Fixed([]), _Fixed([Plate(text="MH01AA1111")]))
    assert chain.read_all("x")[0].text == "MH01AA1111"