from __future__ import annotations

from typing import TYPE_CHECKING

from dcs.forcedoptions import ForcedOptions
from dcs.mission import Mission

if TYPE_CHECKING:
    from game.game import Game


class ForcedOptionsGenerator:
    def __init__(self, mission: Mission, game: Game) -> None:
        self.mission = mission
        self.game = game

    def _set_options_view(self) -> None:
        self.mission.forced_options.options_view = (
            self.game.settings.map_coalition_visibility
        )

    def _set_external_views(self) -> None:
        if not self.game.settings.external_views_allowed:
            self.mission.forced_options.external_views = (
                self.game.settings.external_views_allowed
            )

    def _set_labels(self) -> None:
        # TODO: Fix settings to use the real type.
        # TODO: Allow forcing "full" and have default do nothing.
        if self.game.settings.labels == "Abbreviated":
            self.mission.forced_options.labels = ForcedOptions.Labels.Abbreviate
        elif self.game.settings.labels == "Dot Only":
            self.mission.forced_options.labels = ForcedOptions.Labels.DotOnly
        elif self.game.settings.labels == "Neutral Dot":
            self.mission.forced_options.labels = ForcedOptions.Labels.NeutralDot
        elif self.game.settings.labels == "Off":
            self.mission.forced_options.labels = ForcedOptions.Labels.None_

    def _set_unrestricted_satnav(self) -> None:
        blue = self.game.player_faction
        red = self.game.enemy_faction
        if blue.unrestricted_satnav or red.unrestricted_satnav:
            self.mission.forced_options.unrestricted_satnav = True

    def generate(self):
        self._set_options_view()
        self._set_external_views()
        self._set_labels()
        self._set_unrestricted_satnav()