import logging

from game import db
from .conflictgen import *
from .naming import *

from dcs.mission import *
from dcs.statics import *

FARP_FRONTLINE_DISTANCE = 10000


CATEGORY_MAPPING = {
    "power": [Fortification.Workshop_A],
    "warehouse": [Warehouse.Warehouse],
    "fuel": [Warehouse.Tank],
    "ammo": [Warehouse.Ammunition_depot],
    "farp": [Fortification.FARP_Tent],
    "comms": [Fortification.TV_tower],
    "oil": [Fortification.Oil_platform],
}


class GroundObjectsGenerator:
    FARP_CAPACITY = 4

    def __init__(self, mission: Mission, conflict: Conflict, game):
        self.m = mission
        self.conflict = conflict
        self.game = game

    def generate_farps(self, number_of_units=1) -> typing.Collection[StaticGroup]:
        assert self.conflict.is_vector, "FARP could be generated only on frontline conflicts!"

        for i, _ in enumerate(range(0, number_of_units, self.FARP_CAPACITY)):
            heading = self.conflict.heading - 90
            position = self.conflict.find_ground_position(self.conflict.center.point_from_heading(heading, FARP_FRONTLINE_DISTANCE), heading)
            position = position.point_from_heading(0, i * 275)

            yield self.m.farp(
                country=self.m.country(self.game.player),
                name="FARP",
                position=position,
            )

    def generate(self):
        side = self.m.country(self.game.enemy)

        cp = None  # type: ControlPoint
        if self.conflict.attackers_side.name == self.game.player:
            cp = self.conflict.to_cp
        else:
            cp = self.conflict.from_cp

        for ground_object in cp.ground_objects:
            if ground_object.category == "defense":
                unit_type = random.choice(self.game.commision_unit_types(cp, AirDefence))
                assert unit_type is not None, "Cannot find unit type for GroundObject defense ({})!".format(cp)

                group = self.m.vehicle_group(
                    country=side,
                    name=ground_object.string_identifier,
                    _type=unit_type,
                    position=ground_object.position,
                    heading=ground_object.heading
                )

                logging.info("generated defense object identifier {} with mission id {}".format(group.name, group.id))
            else:
                group = self.m.static_group(
                    country=side,
                    name=ground_object.string_identifier,
                    _type=random.choice(CATEGORY_MAPPING[ground_object.category]),
                    position=ground_object.position,
                    heading=ground_object.heading
                )

                logging.info("generated object identifier {} with mission id {}".format(group.name, group.id))