"""Combo box for selecting a departure airfield."""
from typing import Iterable, Optional

from PySide6.QtWidgets import QComboBox
from dcs.unittype import FlyingType

from game.dcs.aircrafttype import AircraftType
from game.theater.controlpoint import ControlPoint


class QArrivalAirfieldSelector(QComboBox):
    """A combo box for selecting a flight's arrival or divert airfield.

    The combo box will automatically be populated with all airfields the given
    aircraft type is able to land at.
    """

    def __init__(
        self,
        destinations: Iterable[ControlPoint],
        aircraft: Optional[AircraftType],
        optional_text: str,
    ) -> None:
        super().__init__()
        self.destinations = list(destinations)
        self.aircraft = aircraft
        self.optional_text = optional_text
        self.rebuild_selector()
        self.setCurrentIndex(0)

    def change_aircraft(self, aircraft: Optional[FlyingType]) -> None:
        if self.aircraft == aircraft:
            return
        self.aircraft = aircraft
        self.rebuild_selector()

    def rebuild_selector(self) -> None:
        self.clear()
        if self.aircraft is None:
            return
        for destination in self.destinations:
            if destination.can_operate(self.aircraft):
                self.addItem(destination.name, destination)
        self.model().sort(0)
        self.insertItem(0, self.optional_text, None)
        self.setCurrentIndex(0)