import random
from typing import List, Optional, Type

from dcs.unitgroup import VehicleGroup

from game import Game
from game.factions.faction import Faction
from game.theater.theatergroundobject import EwrGroundObject
from gen.sam.ewrs import (
    BigBirdGenerator,
    BoxSpringGenerator,
    DogEarGenerator,
    FlatFaceGenerator,
    HawkEwrGenerator,
    PatriotEwrGenerator,
    RolandEwrGenerator,
    SnowDriftGenerator,
    StraightFlushGenerator,
    TallRackGenerator,
    EwrGenerator,
    TinShieldGenerator,
)

EWR_MAP = {
    "BoxSpringGenerator": BoxSpringGenerator,
    "TallRackGenerator": TallRackGenerator,
    "DogEarGenerator": DogEarGenerator,
    "RolandEwrGenerator": RolandEwrGenerator,
    "FlatFaceGenerator": FlatFaceGenerator,
    "PatriotEwrGenerator": PatriotEwrGenerator,
    "BigBirdGenerator": BigBirdGenerator,
    "SnowDriftGenerator": SnowDriftGenerator,
    "StraightFlushGenerator": StraightFlushGenerator,
    "HawkEwrGenerator": HawkEwrGenerator,
    "TinShieldGenerator": TinShieldGenerator,
}


def get_faction_possible_ewrs_generator(
    faction: Faction,
) -> List[Type[EwrGenerator]]:
    """
    Return the list of possible EWR generators for the given faction
    :param faction: Faction name to search units for
    """
    return [EWR_MAP[s] for s in faction.ewrs]


def generate_ewr_group(
    game: Game, ground_object: EwrGroundObject, faction: Faction
) -> Optional[VehicleGroup]:
    """Generates an early warning radar group.

    :param game: The Game.
    :param ground_object: The ground object which will own the EWR group.
    :param faction: Owner faction.
    :return: The generated group, or None if one could not be generated.
    """
    generators = get_faction_possible_ewrs_generator(faction)
    if len(generators) > 0:
        generator_class = random.choice(generators)
        generator = generator_class(game, ground_object)
        generator.generate()
        return generator.get_generated_group()
    return None