from pydantic import BaseModel

from idfm_interface.idfm_types import TrainStatus


class NextPassage(BaseModel):
    line: int
    operator: str
    destinations: list[str]
    atStop: bool
    aimedArrivalTs: int | None
    expectedArrivalTs: int | None
    arrivalPlatformName: str | None
    aimedDepartTs: int | None
    expectedDepartTs: int | None
    arrivalStatus: TrainStatus
    departStatus: TrainStatus


class NextPassages(BaseModel):
    ts: int
    passages: dict[int, dict[str, list[NextPassage]]]