import random
from typing import List, Optional, Type

from dcs.vehicles import AirDefence
from dcs.unitgroup import VehicleGroup

from game import Game, db
from gen.sam.aaa_bofors import BoforsGenerator
from gen.sam.aaa_flak import FlakGenerator
from gen.sam.aaa_flak18 import Flak18Generator
from gen.sam.aaa_ww2_ally_flak import AllyWW2FlakGenerator
from gen.sam.aaa_zu23_insurgent import ZU23InsurgentGenerator
from gen.sam.cold_war_flak import EarlyColdWarFlakGenerator, ColdWarFlakGenerator
from gen.sam.group_generator import GroupGenerator
from gen.sam.sam_avenger import AvengerGenerator
from gen.sam.sam_chaparral import ChaparralGenerator
from gen.sam.sam_gepard import GepardGenerator
from gen.sam.sam_hawk import HawkGenerator
from gen.sam.sam_hq7 import HQ7Generator
from gen.sam.sam_linebacker import LinebackerGenerator
from gen.sam.sam_patriot import PatriotGenerator
from gen.sam.sam_rapier import RapierGenerator
from gen.sam.sam_roland import RolandGenerator
from gen.sam.sam_sa10 import SA10Generator
from gen.sam.sam_sa11 import SA11Generator
from gen.sam.sam_sa13 import SA13Generator
from gen.sam.sam_sa15 import SA15Generator
from gen.sam.sam_sa19 import SA19Generator
from gen.sam.sam_sa2 import SA2Generator
from gen.sam.sam_sa3 import SA3Generator
from gen.sam.sam_sa6 import SA6Generator
from gen.sam.sam_sa8 import SA8Generator
from gen.sam.sam_sa9 import SA9Generator
from gen.sam.sam_vulcan import VulcanGenerator
from gen.sam.sam_zsu23 import ZSU23Generator
from gen.sam.sam_zu23 import ZU23Generator
from gen.sam.sam_zu23_ural import ZU23UralGenerator
from gen.sam.sam_zu23_ural_insurgent import ZU23UralInsurgentGenerator
from gen.sam.freya_ewr import FreyaGenerator
from theater import TheaterGroundObject

SAM_MAP = {
    "HawkGenerator": HawkGenerator,
    "ZU23Generator": ZU23Generator,
    "ZU23UralGenerator": ZU23UralGenerator,
    "ZU23UralInsurgentGenerator": ZU23UralInsurgentGenerator,
    "ZU23InsurgentGenerator": ZU23InsurgentGenerator,
    "ZSU23Generator": ZSU23Generator,
    "VulcanGenerator": VulcanGenerator,
    "LinebackerGenerator": LinebackerGenerator,
    "RapierGenerator": RapierGenerator,
    "AvengerGenerator": AvengerGenerator,
    "GepardGenerator": GepardGenerator,
    "RolandGenerator": RolandGenerator,
    "PatriotGenerator": PatriotGenerator,
    "ChaparralGenerator": ChaparralGenerator,
    "BoforsGenerator": BoforsGenerator,
    "FlakGenerator": FlakGenerator,
    "SA2Generator": SA2Generator,
    "SA3Generator": SA3Generator,
    "SA6Generator": SA6Generator,
    "SA8Generator": SA8Generator,
    "SA9Generator": SA9Generator,
    "SA10Generator": SA10Generator,
    "SA11Generator": SA11Generator,
    "SA13Generator": SA13Generator,
    "SA15Generator": SA15Generator,
    "SA19Generator": SA19Generator,
    "HQ7Generator": HQ7Generator,
    "Flak18Generator": Flak18Generator,
    "ColdWarFlakGenerator": ColdWarFlakGenerator,
    "EarlyColdWarFlakGenerator": EarlyColdWarFlakGenerator,
    "FreyaGenerator": FreyaGenerator,
    "AllyWW2FlakGenerator": AllyWW2FlakGenerator
}

SAM_PRICES = {
    AirDefence.SAM_Hawk_PCP: 35,
    AirDefence.AAA_ZU_23_Emplacement: 10,
    AirDefence.AAA_ZU_23_Closed: 10,
    AirDefence.AAA_ZU_23_on_Ural_375: 10,
    AirDefence.AAA_ZU_23_Insurgent_on_Ural_375: 10,
    AirDefence.AAA_ZU_23_Insurgent_Closed: 10,
    AirDefence.AAA_ZU_23_Insurgent: 10,
    AirDefence.SPAAA_ZSU_23_4_Shilka: 10,
    AirDefence.AAA_Vulcan_M163: 15,
    AirDefence.SAM_Linebacker_M6: 20,
    AirDefence.Rapier_FSA_Launcher: 20,
    AirDefence.SAM_Avenger_M1097: 22,
    AirDefence.SPAAA_Gepard: 24,
    AirDefence.SAM_Roland_ADS: 40,
    AirDefence.SAM_Patriot_LN_M901: 85,
    AirDefence.SAM_Patriot_EPP_III: 85,
    AirDefence.SAM_Chaparral_M48: 25,
    AirDefence.AAA_Bofors_40mm: 15,
    AirDefence.AAA_8_8cm_Flak_36: 15,
    AirDefence.SAM_SA_2_LN_SM_90: 30,
    AirDefence.SAM_SA_3_S_125_LN_5P73: 35,
    AirDefence.SAM_SA_6_Kub_LN_2P25: 45,
    AirDefence.SAM_SA_8_Osa_9A33: 30,
    AirDefence.SAM_SA_9_Strela_1_9P31: 25,
    AirDefence.SAM_SA_10_S_300PS_LN_5P85C: 80,
    AirDefence.SAM_SA_10_S_300PS_CP_54K6: 80,
    AirDefence.SAM_SA_11_Buk_LN_9A310M1: 60,
    AirDefence.SAM_SA_13_Strela_10M3_9A35M3: 30,
    AirDefence.SAM_SA_15_Tor_9A331: 40,
    AirDefence.SAM_SA_19_Tunguska_2S6: 35,
    AirDefence.HQ_7_Self_Propelled_LN: 35
}


def get_faction_possible_sams_generator(faction: str) -> List[Type[GroupGenerator]]:
    """
    Return the list of possible SAM generator for the given faction
    :param faction: Faction name to search units for
    """
    return [SAM_MAP[s] for s in db.FACTIONS[faction].sams if s in SAM_MAP.keys()]


def generate_anti_air_group(game: Game, ground_object: TheaterGroundObject,
                            faction: str) -> Optional[VehicleGroup]:
    """
    This generate a SAM group
    :param game: The Game.
    :param ground_object: The ground object which will own the sam group.
    :param faction: Owner faction.
    :return: The generated group, or None if one could not be generated.
    """
    possible_sams_generators = get_faction_possible_sams_generator(faction)
    if len(possible_sams_generators) > 0:
        sam_generator_class = random.choice(possible_sams_generators)
        generator = sam_generator_class(game, ground_object, db.FACTIONS[faction])
        generator.generate()
        return generator.get_generated_group()
    return None


def generate_shorad_group(game: Game, ground_object: TheaterGroundObject,
                          faction_name: str) -> Optional[VehicleGroup]:
    faction = db.FACTIONS[faction_name]

    if len(faction.shorads) > 0:
        sam = random.choice(faction.shorads)
        generator = SAM_MAP[sam](game, ground_object)
        generator.generate()
        return generator.get_generated_group()
    else:
        return generate_anti_air_group(game, ground_object, faction_name)