dcs-retribution/gen/naming.py

from __future__ import annotations

import random
import time
from typing import List, Any, TYPE_CHECKING

from dcs.country import Country

from game.dcs.aircrafttype import AircraftType
from game.dcs.unittype import UnitType

if TYPE_CHECKING:
    from gen.flights.flight import Flight

ALPHA_MILITARY = [
    "Alpha",
    "Bravo",
    "Charlie",
    "Delta",
    "Echo",
    "Foxtrot",
    "Golf",
    "Hotel",
    "India",
    "Juliet",
    "Kilo",
    "Lima",
    "Mike",
    "November",
    "Oscar",
    "Papa",
    "Quebec",
    "Romeo",
    "Sierra",
    "Tango",
    "Uniform",
    "Victor",
    "Whisky",
    "XRay",
    "Yankee",
    "Zulu",
    "Zero",
]

ANIMALS: tuple[str, ...] = (
    "SHARK",
    "TORTOISE",
    "BAT",
    "PANGOLIN",
    "AARDWOLF",
    "MONKEY",
    "BUFFALO",
    "DOG",
    "BOBCAT",
    "LYNX",
    "PANTHER",
    "TIGER",
    "LION",
    "OWL",
    "BUTTERFLY",
    "BISON",
    "DUCK",
    "COBRA",
    "MAMBA",
    "DOLPHIN",
    "PHEASANT",
    "ARMADILLO",
    "RACOON",
    "ZEBRA",
    "COW",
    "COYOTE",
    "FOX",
    "LIGHTFOOT",
    "COTTONMOUTH",
    "TAURUS",
    "VIPER",
    "CASTOR",
    "GIRAFFE",
    "SNAKE",
    "MONSTER",
    "ALBATROSS",
    "HAWK",
    "DOVE",
    "MOCKINGBIRD",
    "GECKO",
    "ORYX",
    "GORILLA",
    "HARAMBE",
    "GOOSE",
    "MAVERICK",
    "HARE",
    "JACKAL",
    "LEOPARD",
    "CAT",
    "MUSK",
    "ORCA",
    "OCELOT",
    "BEAR",
    "PANDA",
    "GULL",
    "PENGUIN",
    "PYTHON",
    "RAVEN",
    "DEER",
    "MOOSE",
    "REINDEER",
    "SHEEP",
    "GAZELLE",
    "INSECT",
    "VULTURE",
    "WALLABY",
    "KANGAROO",
    "KOALA",
    "KIWI",
    "WHALE",
    "FISH",
    "RHINO",
    "HIPPO",
    "RAT",
    "WOODPECKER",
    "WORM",
    "BABOON",
    "YAK",
    "SCORPIO",
    "HORSE",
    "POODLE",
    "CENTIPEDE",
    "CHICKEN",
    "CHEETAH",
    "CHAMELEON",
    "CATFISH",
    "CATERPILLAR",
    "CARACAL",
    "CAMEL",
    "CAIMAN",
    "BARRACUDA",
    "BANDICOOT",
    "ALLIGATOR",
    "BONGO",
    "CORAL",
    "ELEPHANT",
    "ANTELOPE",
    "CRAB",
    "DACHSHUND",
    "DODO",
    "FLAMINGO",
    "FERRET",
    "FALCON",
    "BULLDOG",
    "DONKEY",
    "IGUANA",
    "TAMARIN",
    "HARRIER",
    "GRIZZLY",
    "GREYHOUND",
    "GRASSHOPPER",
    "JAGUAR",
    "LADYBUG",
    "KOMODO",
    "DRAGON",
    "LIZARD",
    "LLAMA",
    "LOBSTER",
    "OCTOPUS",
    "MANATEE",
    "MAGPIE",
    "MACAW",
    "OSTRICH",
    "OYSTER",
    "MOLE",
    "MULE",
    "MOTH",
    "MONGOOSE",
    "MOLLY",
    "MEERKAT",
    "MOUSE",
    "PEACOCK",
    "PIKE",
    "ROBIN",
    "RAGDOLL",
    "PLATYPUS",
    "PELICAN",
    "PARROT",
    "PORCUPINE",
    "PIRANHA",
    "PUMA",
    "PUG",
    "TAPIR",
    "TERMITE",
    "URCHIN",
    "SHRIMP",
    "TURKEY",
    "TOUCAN",
    "TETRA",
    "HUSKY",
    "STARFISH",
    "SWAN",
    "FROG",
    "SQUIRREL",
    "WALRUS",
    "WARTHOG",
    "CORGI",
    "WEASEL",
    "WOMBAT",
    "WOLVERINE",
    "MAMMOTH",
    "TOAD",
    "WOLF",
    "ZEBU",
    "SEAL",
    "SKATE",
    "JELLYFISH",
    "MOSQUITO",
    "LOCUST",
    "SLUG",
    "SNAIL",
    "HEDGEHOG",
    "PIGLET",
    "FENNEC",
    "BADGER",
    "ALPACA",
    "DINGO",
    "COLT",
    "SKUNK",
    "BUNNY",
    "IMPALA",
    "GUANACO",
    "CAPYBARA",
    "ELK",
    "MINK",
    "PRONGHORN",
    "CROW",
    "BUMBLEBEE",
    "FAWN",
    "OTTER",
    "WATERBUCK",
    "JERBOA",
    "KITTEN",
    "ARGALI",
    "OX",
    "MARE",
    "FINCH",
    "BASILISK",
    "GOPHER",
    "HAMSTER",
    "CANARY",
    "WOODCHUCK",
    "ANACONDA",
)


class NameGenerator:
    number = 0
    infantry_number = 0
    aircraft_number = 0
    convoy_number = 0
    cargo_ship_number = 0

    animals: list[str] = list(ANIMALS)
    existing_alphas: List[str] = []

    @classmethod
    def reset(cls) -> None:
        cls.number = 0
        cls.infantry_number = 0
        cls.convoy_number = 0
        cls.cargo_ship_number = 0
        cls.animals = list(ANIMALS)
        cls.existing_alphas = []

    @classmethod
    def reset_numbers(cls) -> None:
        cls.number = 0
        cls.infantry_number = 0
        cls.aircraft_number = 0
        cls.convoy_number = 0
        cls.cargo_ship_number = 0

    @classmethod
    def next_aircraft_name(
        cls, country: Country, parent_base_id: int, flight: Flight
    ) -> str:
        cls.aircraft_number += 1
        try:
            if flight.custom_name:
                name_str = flight.custom_name
            else:
                name_str = "{} {}".format(
                    flight.package.target.name, flight.flight_type
                )
        except AttributeError:  # Here to maintain save compatibility with 2.3
            name_str = "{} {}".format(flight.package.target.name, flight.flight_type)
        return "{}|{}|{}|{}|{}|".format(
            name_str,
            country.id,
            cls.aircraft_number,
            parent_base_id,
            flight.unit_type.name,
        )

    @classmethod
    def next_unit_name(
        cls, country: Country, parent_base_id: int, unit_type: UnitType[Any]
    ) -> str:
        cls.number += 1
        return "unit|{}|{}|{}|{}|".format(
            country.id, cls.number, parent_base_id, unit_type.name
        )

    @classmethod
    def next_infantry_name(
        cls, country: Country, parent_base_id: int, unit_type: UnitType[Any]
    ) -> str:
        cls.infantry_number += 1
        return "infantry|{}|{}|{}|{}|".format(
            country.id,
            cls.infantry_number,
            parent_base_id,
            unit_type.name,
        )

    @classmethod
    def next_awacs_name(cls, country: Country) -> str:
        cls.number += 1
        return "awacs|{}|{}|0|".format(country.id, cls.number)

    @classmethod
    def next_tanker_name(cls, country: Country, unit_type: AircraftType) -> str:
        cls.number += 1
        return "tanker|{}|{}|0|{}".format(country.id, cls.number, unit_type.name)

    @classmethod
    def next_carrier_name(cls, country: Country) -> str:
        cls.number += 1
        return "carrier|{}|{}|0|".format(country.id, cls.number)

    @classmethod
    def next_convoy_name(cls) -> str:
        cls.convoy_number += 1
        return f"Convoy {cls.convoy_number:03}"

    @classmethod
    def next_cargo_ship_name(cls) -> str:
        cls.cargo_ship_number += 1
        return f"Cargo Ship {cls.cargo_ship_number:03}"

    @classmethod
    def random_objective_name(cls) -> str:
        if cls.animals:
            animal = random.choice(cls.animals)
            cls.animals.remove(animal)
            return animal

        for _ in range(10):
            alpha = random.choice(ALPHA_MILITARY).upper()
            number = random.randint(0, 100)
            alpha_mil_name = f"{alpha} #{number:02}"
            if alpha_mil_name not in cls.existing_alphas:
                cls.existing_alphas.append(alpha_mil_name)
                return alpha_mil_name

        # At this point, give up trying - something has gone wrong and we haven't been
        # able to make a new name in 10 tries. We'll just make a longer name using the
        # current unix epoch in nanoseconds. That should be unique... right?
        last_chance_name = alpha_mil_name + str(time.time_ns())
        cls.existing_alphas.append(last_chance_name)
        return last_chance_name


namegen = NameGenerator