dcs_liberation/game/sim/combat/defendingsam.py

from __future__ import annotations

import logging
import random
from collections.abc import Iterator
from datetime import datetime, timedelta
from typing import TYPE_CHECKING

from game.ato.flightstate import InCombat
from game.settings.settings import CombatResolutionMethod
from .frozencombat import FrozenCombat
from .. import GameUpdateEvents

if TYPE_CHECKING:
    from game.ato import Flight
    from game.theater import TheaterGroundObject
    from ..simulationresults import SimulationResults


class DefendingSam(FrozenCombat):
    def __init__(
        self,
        freeze_duration: timedelta,
        flight: Flight,
        air_defenses: list[TheaterGroundObject],
    ) -> None:
        super().__init__(freeze_duration)
        self.flight = flight
        self.air_defenses = air_defenses

    def because(self) -> str:
        sams = ", ".join(str(d) for d in self.air_defenses)
        return f"{self.flight} is engaged by enemy air defenses: {sams}"

    def describe(self) -> str:
        return f"engaged by enemy air defenses"

    def iter_flights(self) -> Iterator[Flight]:
        yield self.flight

    def resolve(
        self,
        results: SimulationResults,
        events: GameUpdateEvents,
        time: datetime,
        elapsed_time: timedelta,
        resolution_method: CombatResolutionMethod,
    ) -> None:
        assert isinstance(self.flight.state, InCombat)

        if resolution_method is CombatResolutionMethod.SKIP:
            self.flight.state.exit_combat(events, time, elapsed_time)
            return

        if random.random() >= 0.5:
            logging.debug(f"Air defense combat auto-resolved with {self.flight} lost")
            self.flight.kill(results, events)
        else:
            logging.debug(
                f"Air defense combat auto-resolved with {self.flight} surviving"
            )
            self.flight.state.exit_combat(events, time, elapsed_time)