import random

from dcs.vehicles import AirDefence

from gen.sam.airdefensegroupgenerator import (
    AirDefenseRange,
    AirDefenseGroupGenerator,
)


class ZU23Generator(AirDefenseGroupGenerator):
    """
    This generate a ZU23 flak artillery group
    """

    name = "ZU-23 Group"
    price = 54

    def generate(self):
        grid_x = random.randint(2, 3)
        grid_y = random.randint(2, 3)

        spacing = random.randint(10, 40)

        index = 0
        for i in range(grid_x):
            for j in range(grid_y):
                index = index + 1
                self.add_unit(
                    AirDefence.AAA_ZU_23_Closed,
                    "AAA#" + str(index),
                    self.position.x + spacing * i,
                    self.position.y + spacing * j,
                    self.heading,
                )

    @classmethod
    def range(cls) -> AirDefenseRange:
        return AirDefenseRange.Short