import random

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


class KS19Generator(AirDefenseGroupGenerator):
    """
    This generate a KS 19 flak artillery group (KS-19 from the High Digit SAM mod)
    """

    name = "KS-19 AAA Site"
    price = 98

    def generate(self):

        spacing = random.randint(10, 40)

        self.add_unit(
            highdigitsams.AAA_SON_9_Fire_Can,
            "TR",
            self.position.x - 20,
            self.position.y - 20,
            self.heading,
        )

        index = 0
        for i in range(3):
            for j in range(3):
                index = index + 1
                self.add_unit(
                    highdigitsams.AAA_100mm_KS_19,
                    "AAA#" + str(index),
                    self.position.x + spacing * i,
                    self.position.y + spacing * j,
                    self.heading,
                )

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