import typing

from dcs.mapping import Point

NAME_BY_CATEGORY = {
    "power": "Power plant",
    "ammo": "Ammo depot",
    "fuel": "Fuel depot",
    "defense": "AA Defense Site",
    "warehouse": "Warehouse",
    "farp": "FARP",
    "comms": "Comms. tower",
    "oil": "Oil platform"
}

ABBREV_NAME = {
    "power": "PLANT",
    "ammo": "AMMO",
    "fuel": "FUEL",
    "defense": "AA",
    "warehouse": "WARE",
    "farp": "FARP",
    "comms": "COMMST",
    "oil": "OILP"
}


class TheaterGroundObject:
    object_id = 0
    cp_id = 0
    group_id = 0
    heading = 0
    position = None  # type: Point
    category = None  # type: str

    def __init__(self, category, cp_id, group_id, object_id, position, heading):
        self.category = category
        self.cp_id = cp_id
        self.group_id = group_id
        self.object_id = object_id
        self.position = position
        self.heading = heading

    @property
    def string_identifier(self):
        return "{}|{}|{}|{}".format(self.category, self.cp_id, self.group_id, self.object_id)

    @property
    def group_identifier(self) -> str:
        return "{}|{}".format(self.category, self.group_id)

    @property
    def name_abbrev(self) -> str:
        return ABBREV_NAME[self.category]

    def __str__(self):
        return NAME_BY_CATEGORY[self.category]

    def matches_string_identifier(self, id):
        return self.string_identifier == id