items.py

from typing import NamedTuple
from BaseClasses import Item


class TABSItem(Item):
    game = "Totally Accurate Battle Simulator"


class Unit(NamedTuple):
    name: str
    cost: int | None
    faction_num: int


class Faction(NamedTuple):
    name: str
    units: list[Unit]
    start_id: int


tribal_units = [
    Unit("Clubber", 70, 0),
    Unit("Protector", 80, 0),
    Unit("Spear Thrower", 120, 0),
    Unit("Stoner", 160, 0),
    Unit("Bone Mage", 300, 0),
    Unit("Chieftain", 400, 0),
    Unit("Mammoth", 2200, 0),
]

farmer_units = [
    Unit("Halfling", 50, 1),
    Unit("Farmer", 80, 1),
    Unit("Hay Baler", 140, 1),
    Unit("Potionseller", 340, 1),
    Unit("Harvester", 500, 1),
    Unit("Wheelbarrow", 1000, 1),
    Unit("Scarecrow", 1200, 1),
]

medieval_units = [
    Unit("Bard", 60, 2),
    Unit("Squire", 100, 2),
    Unit("Archer", 140, 2),
    Unit("Healer", 180, 2),
    Unit("Knight", 650, 2),
    Unit("Catapult", 1000, 2),
    Unit("The King", 1500, 2),
]

ancient_units = [
    Unit("Shield Bearer", 100, 3),
    Unit("Sarissa", 120, 3),
    Unit("Hoplite", 180, 3),
    Unit("Snake Archer", 300, 3),
    Unit("Ballista", 900, 3),
    Unit("Minotaur", 1600, 3),
    Unit("Zeus", 2000, 3),
]

viking_units = [
    Unit("Headbutter", 90, 4),
    Unit("Ice Archer", 160, 4),
    Unit("Brawler", 220, 4),
    Unit("Berserker", 250, 4),
    Unit("Valkyrie", 500, 4),
    Unit("Longship", 1000, 4),
    Unit("Jarl", 1500, 4),
]

dynasty_units = [
    Unit("Samurai", 140, 5),
    Unit("Firework Archer", 180, 5),
    Unit("Monk", 250, 5),
    Unit("Ninja", 500, 5),
    Unit("Dragon", 1000, 5),
    Unit("Hwacha", 1500, 5),
    Unit("Monkey King", 2000, 5),
]

renaissance_units = [
    Unit("Painter", 50, 6),
    Unit("Fencer", 150, 6),
    Unit("Balloon Archer", 200, 6),
    Unit("Musketeer", 250, 6),
    Unit("Halberd", 400, 6),
    Unit("Jouster", 1000, 6),
    Unit("Da Vinci Tank", 4000, 6),
]

pirate_units = [
    Unit("Flintlock", 100, 7),
    Unit("Blunderbuss", 160, 7),
    Unit("Bomb Thrower", 250, 7),
    Unit("Harpooner", 300, 7),
    Unit("Cannon", 1000, 7),
    Unit("Captain", 1500, 7),
    Unit("Pirate Queen", 2500, 7),
]

spooky_units = [
    Unit("Skeleton Warrior", 80, 8),
    Unit("Skeleton Archer", 180, 8),
    Unit("Candlehead", 200, 8),
    Unit("Vampire", 200, 8),
    Unit("Pumpkin Catapult", 1000, 8),
    Unit("Swordcaster", 1000, 8),
    Unit("Reaper", 2500, 8),
]

wild_west_units = [
    Unit("Dynamite Thrower", 100, 9),
    Unit("Miner", 200, 9),
    Unit("Cactus", 400, 9),
    Unit("Gunslinger", 650, 9),
    Unit("Lasso", 740, 9),
    Unit("Deadeye", 900, 9),
    Unit("Quick Draw", 1200, 9),
]

legacy_units = [
    Unit("Peasant", 30, 10),
    Unit("Banner Bearer", 100, 10),
    Unit("Poacher", 120, 10),
    Unit("Blowdarter", 220, 10),
    Unit("Pike", 300, 10),
    Unit("Barrel Roller", 350, 10),
    Unit("Boxer", 450, 10),
    Unit("Flag Bearer", 500, 10),
    Unit("Pharaoh", 750, 10),
    Unit("Wizard", 1200, 10),
    Unit("Chariot", 1800, 10),
    Unit("Thor", 2200, 10),
    Unit("Tank", 6000, 10),
    Unit("Super Boxer", 10000, 10),
    Unit("Dark Peasant", 9999999, 10),
    Unit("Super Peasant", 9999999, 10),
]

fantasy_good_units = [
    Unit("Devout Gauntlet", 200, 11),
    Unit("Celestial Aegis", 300, 11),
    Unit("Radiant Glaive", 500, 11),
    Unit("Rigtheous Paladin", 800, 11),
    Unit("Divine Arbiter", 1000, 11),
    Unit("Sacred Elephant", 2000, 11),
    Unit("Chronomancer", 3000, 11),
]

fantasy_evil_units = [
    Unit("Shadow Walker", 200, 12),
    Unit("Exiled Sentinel", 300, 12),
    Unit("Mad Mechanic", 500, 12),
    Unit("Void Cultist", 800, 12),
    Unit("Tempest Lich", 1000, 12),
    Unit("Death Bringer", 2000, 12),
    Unit("Void Monarch", 3000, 12),
]

secret_units = [
    Unit("Ballooner", 120, 13),
    Unit("Bomb on a Stick", 150, 13),
    Unit("Fan Bearer", 200, 13),
    Unit("Raptor", 200, 13),
    Unit("The Teacher", 200, 13),
    Unit("Jester", 300, 13),
    Unit("Ball 'n' Chain", 350, 13),
    Unit("Chu Ko Nu", 350, 13),
    Unit("Executioner", 350, 13),
    Unit("Shouter", 400, 13),
    Unit("Taekwondo", 400, 13),
    Unit("Raptor Rider", 450, 13),
    Unit("Cheerleader", 500, 13),
    Unit("Cupid", 500, 13),
    Unit("Mace Spinner", 500, 13),
    Unit("CLAMS", 500, 13),
    Unit("Present Elf", 500, 13),
    Unit("Ice Mage", 650, 13),
    Unit("Infernal Whip", 800, 13),
    Unit("Bank Robbers", 850, 13),
    Unit("Witch", 1000, 13),
    Unit("Banshee", 1100, 13),
    Unit("Necromancer", 1200, 13),
    Unit("Solar Architect", 1200, 13),
    Unit("Wheelbarrow Dragon", 1400, 13),
    Unit("Bomb Cannon", 1500, 13),
    Unit("Skeleton Giant", 1700, 13),
    Unit("Cavalry", 1800, 13),
    Unit("Vlad", 1800, 13),
    Unit("Gatling Gun", 2000, 13),
    Unit("Blackbeard", 2600, 13),
    Unit("Samurai Giant", 3000, 13),
    Unit("Ullr", 3000, 13),
    Unit("Lady Red Jade", 3500, 13),
    Unit("Sensei", 3500, 13),
    Unit("Shogun", 3500, 13),
    Unit("Tree Giant", 4000, 13),
    Unit("Artemis", 5500, 13),
    Unit("Ice Giant", 6000, 13),
]

# Too expensive to use in campaign, mark as filler instead
bad_units = {
    "Dark Peasant",
    "Super Peasant",
}

factions = [
    Faction("Tribal", tribal_units, 100_000),
    Faction("Farmer", farmer_units, 101_000),
    Faction("Medieval", medieval_units, 102_000),
    Faction("Ancient", ancient_units, 103_000),
    Faction("Viking", viking_units, 104_000),
    Faction("Dynasty", dynasty_units, 105_000),
    Faction("Renaissance", renaissance_units, 106_000),
    Faction("Pirate", pirate_units, 107_000),
    Faction("Spooky", spooky_units, 108_000),
    Faction("Wild West", wild_west_units, 109_000),
    Faction("Legacy", legacy_units, 110_000),
    Faction("Fantasy Good", fantasy_good_units, 111_000),
    Faction("Fantasy Evil", fantasy_evil_units, 112_000),
    Faction("Secret", secret_units, 113_000),
]

unit_map: dict[str, Unit] = {}
for i, faction in enumerate(factions):
    unit_map |= {unit.name: unit for unit in faction.units}
    progressive_name = f"Progressive {faction.name} Unit"
    unit_map[progressive_name] = Unit(progressive_name, None, i)

item_map: dict[str, int] = {}
for faction in factions:
    item_map |= {unit.name: id for id, unit in enumerate(faction.units, faction.start_id)}
    progressive_name = f"Progressive {faction.name} Unit"
    item_map[progressive_name] = faction.start_id + 999 # Progressive unit offset
item_map["Morale Boost"] = 1