from enum import IntFlag
from typing import NamedTuple
from BaseClasses import ItemClassification

class ItemGroups(IntFlag):
    Parts = 1 << 1
    Equipment = 1 << 2
    Ammo = 1 << 3
    Recipe = 1 << 4
    Building = 1 << 5
    Trap = 1 << 6
    Lights = 1 << 7
    Foundations = 1 << 8
    Transport = 1 << 9
    Trains = 1 << 10
    ConveyorMk1 = 1 << 11
    ConveyorMk2 = 1 << 12
    ConveyorMk3 = 1 << 13
    ConveyorMk4 = 1 << 14
    ConveyorMk5 = 1 << 15
    ConveyorSupports = 1 << 16
    PipesMk1 = 1 << 17
    PipesMk2 = 1 << 18
    PipelineSupports = 1 << 19
    HyperTubes = 1 << 20
    Signs = 1 << 21
    Pilars = 1 << 22
    Beams = 1 << 23
    Walls = 1 << 24
    Upgrades = 1 << 25
    Vehicles = 1 << 26
    Customizer = 1 << 27
    ConveyorMk6 = 1 << 28
    NeverExclude = 1 << 29

# Workaround for Python 3.10 support. Iterating Flag instances was only added in Python 3.11.
if getattr(ItemGroups.Parts, "__iter__", None) is None:
    def __iter__(self: ItemGroups):
        for flag in ItemGroups:
            if flag in self:
                yield flag
    ItemGroups.__iter__ = __iter__  # type: ignore
    del __iter__


class ItemData(NamedTuple):
    """Represents an item in the pool, it could be a resource bundle, production recipe, trap, etc."""
    category: ItemGroups
    code: int
    type: ItemClassification = ItemClassification.filler
    count: int = 1
    """How many of this item exists in the pool. 0 means none, but still defines the item so it can be added in the starting inventory for example"""