import random

from dcs.vehicles import Unarmed, MissilesSS, AirDefence

from gen.sam.group_generator import GroupGenerator


class V1GroupGenerator(GroupGenerator):
    def __init__(self, game, ground_object, faction):
        super(V1GroupGenerator, self).__init__(game, ground_object)
        self.faction = faction

    def generate(self):

        # Ramps
        self.add_unit(
            MissilesSS.SSM_V_1_Launcher,
            "V1#0",
            self.position.x,
            self.position.y + random.randint(1, 8),
            self.heading,
        )
        self.add_unit(
            MissilesSS.SSM_V_1_Launcher,
            "V1#1",
            self.position.x + 50,
            self.position.y + random.randint(1, 8),
            self.heading,
        )
        self.add_unit(
            MissilesSS.SSM_V_1_Launcher,
            "V1#2",
            self.position.x + 100,
            self.position.y + random.randint(1, 8),
            self.heading,
        )

        # Commander
        self.add_unit(
            Unarmed.LUV_Kubelwagen_82,
            "Kubel#0",
            self.position.x - 35,
            self.position.y - 20,
            self.heading,
        )

        # Self defense flak
        flak_unit = random.choice(
            [AirDefence.AAA_Flak_Vierling_38_Quad_20mm, AirDefence.AAA_Flak_38_20mm]
        )

        self.add_unit(
            flak_unit,
            "FLAK#0",
            self.position.x - 55,
            self.position.y - 38,
            self.heading,
        )

        self.add_unit(
            Unarmed.Truck_Opel_Blitz,
            "Blitz#0",
            self.position.x + 200,
            self.position.y + 15,
            90,
        )