from __future__ import annotations

from dataclasses import dataclass
from datetime import datetime, timedelta
from typing import Iterator, Type

from game.ato.flightplans.ibuilder import IBuilder
from game.ato.flightplans.standard import StandardFlightPlan, StandardLayout
from game.ato.flightplans.waypointbuilder import WaypointBuilder
from game.ato.flightwaypoint import FlightWaypoint


@dataclass(frozen=True)
class RecoveryTankerLayout(StandardLayout):
    nav_to: list[FlightWaypoint]
    recovery_ship: FlightWaypoint
    nav_from: list[FlightWaypoint]

    def iter_waypoints(self) -> Iterator[FlightWaypoint]:
        yield self.departure
        yield from self.nav_to
        yield self.recovery_ship
        yield from self.nav_from
        yield self.arrival
        if self.divert is not None:
            yield self.divert
        yield self.bullseye


class RecoveryTankerFlightPlan(StandardFlightPlan[RecoveryTankerLayout]):
    @staticmethod
    def builder_type() -> Type[Builder]:
        return Builder

    @property
    def tot_waypoint(self) -> FlightWaypoint:
        return self.layout.recovery_ship

    @property
    def mission_begin_on_station_time(self) -> datetime:
        return self.package.time_over_target

    @property
    def mission_departure_time(self) -> datetime:
        return self.patrol_end_time

    @property
    def patrol_start_time(self) -> datetime:
        return self.package.time_over_target

    @property
    def patrol_end_time(self) -> datetime:
        return self.tot + timedelta(hours=2)

    def tot_for_waypoint(self, waypoint: FlightWaypoint) -> datetime | None:
        if waypoint == self.tot_waypoint:
            return self.tot
        return None

    def depart_time_for_waypoint(self, waypoint: FlightWaypoint) -> datetime | None:
        if waypoint == self.tot_waypoint:
            return self.mission_departure_time
        return None


class Builder(IBuilder[RecoveryTankerFlightPlan, RecoveryTankerLayout]):
    def layout(self) -> RecoveryTankerLayout:
        builder = WaypointBuilder(self.flight, self.coalition)

        # TODO: Propagate the ship position to the Tanker's TOT,
        # so that we minimize the tanker's need to catch up to the carrier.
        recovery_ship = self.package.target.position
        recovery_tanker = builder.recovery_tanker(recovery_ship)

        # We don't have per aircraft cruise altitudes, so just reuse patrol altitude?
        tanker_type = self.flight.unit_type
        nav_cruise_altitude = tanker_type.preferred_patrol_altitude

        return RecoveryTankerLayout(
            departure=builder.takeoff(self.flight.departure),
            nav_to=builder.nav_path(
                self.flight.departure.position, recovery_ship, nav_cruise_altitude
            ),
            nav_from=builder.nav_path(
                recovery_ship, self.flight.arrival.position, nav_cruise_altitude
            ),
            recovery_ship=recovery_tanker,
            arrival=builder.land(self.flight.arrival),
            divert=builder.divert(self.flight.divert),
            bullseye=builder.bullseye(),
        )

    def build(self, dump_debug_info: bool = False) -> RecoveryTankerFlightPlan:
        return RecoveryTankerFlightPlan(self.flight, self.layout())