import typing
import math
import random

from dcs.task import *
from dcs.vehicles import *

from game import db
from game.operation.navalintercept import NavalInterceptionOperation
from userdata.debriefing import Debriefing

from .event import Event


class NavalInterceptEvent(Event):
    STRENGTH_INFLUENCE = 0.3
    SUCCESS_RATE = 0.5

    targets = None  # type: db.ShipDict

    def _targets_count(self) -> int:
        from gen.conflictgen import IMPORTANCE_LOW, IMPORTANCE_HIGH
        factor = (self.to_cp.importance - IMPORTANCE_LOW) * 10
        return max(int(factor), 1)

    def __str__(self) -> str:
        return "Naval intercept at {}".format(self.to_cp)

    @property
    def threat_description(self):
        s = "{} ship(s)".format(self._targets_count())
        if not self.from_cp.captured:
            s += ", {} aircraft".format(self.from_cp.base.scramble_count(self.game.settings.multiplier))
        return s

    def is_successfull(self, debriefing: Debriefing):
        total_targets = sum(self.targets.values())
        destroyed_targets = 0
        for unit, count in debriefing.destroyed_units[self.defender_name].items():
            if unit in self.targets:
                destroyed_targets += count

        if self.from_cp.captured:
            return math.ceil(float(destroyed_targets) / total_targets) > self.SUCCESS_RATE
        else:
            return math.ceil(float(destroyed_targets) / total_targets) < self.SUCCESS_RATE

    def commit(self, debriefing: Debriefing):
        super(NavalInterceptEvent, self).commit(debriefing)

        if self.attacker_name == self.game.player:
            if self.is_successfull(debriefing):
                self.to_cp.base.affect_strength(-self.STRENGTH_INFLUENCE)
            else:
                self.from_cp.base.affect_strength(-self.STRENGTH_INFLUENCE)
        else:
            # enemy attacking
            if self.is_successfull(debriefing):
                self.from_cp.base.affect_strength(-self.STRENGTH_INFLUENCE)
            else:
                self.to_cp.base.affect_strength(-self.STRENGTH_INFLUENCE)

    def skip(self):
        if self.to_cp.captured:
            self.to_cp.base.affect_strength(-self.STRENGTH_INFLUENCE)

    def player_attacking(self, strikegroup: db.PlaneDict, clients: db.PlaneDict):
        self.targets = {
            random.choice(db.find_unittype(CargoTransportation, self.defender_name)): self._targets_count(),
        }

        op = NavalInterceptionOperation(
            self.game,
            attacker_name=self.attacker_name,
            defender_name=self.defender_name,
            attacker_clients=clients,
            defender_clients={},
            from_cp=self.from_cp,
            to_cp=self.to_cp
        )

        op.setup(strikegroup=strikegroup,
                 interceptors={},
                 targets=self.targets)

        self.operation = op

    def player_defending(self, interceptors: db.PlaneDict, clients: db.PlaneDict):
        self.targets = {
            random.choice(db.find_unittype(CargoTransportation, self.defender_name)): self._targets_count(),
        }

        op = NavalInterceptionOperation(
            self.game,
            attacker_name=self.attacker_name,
            defender_name=self.defender_name,
            attacker_clients={},
            defender_clients=clients,
            from_cp=self.from_cp,
            to_cp=self.to_cp
        )

        strikegroup = self.from_cp.base.scramble_cas(self.game.settings.multiplier)
        op.setup(strikegroup=strikegroup,
                 interceptors=interceptors,
                 targets=self.targets)

        self.operation = op