openra_env/game_data.py

"""Static Red Alert mod data for game knowledge tools.

Provides unit stats, building stats, tech tree, and faction information
extracted from OpenRA Red Alert mod rules. This gives an LLM agent the same
reference knowledge a human player would have from experience.
"""

from typing import Optional


# ─── Unit Data ────────────────────────────────────────────────────────────────

RA_UNITS: dict[str, dict] = {
    # Infantry
    "e1": {
        "name": "Rifle Infantry",
        "category": "infantry",
        "cost": 100,
        "hp": 5000,
        "speed": 56,
        "armor": "none",
        "side": "both",
        "prerequisites": ["barr|tent"],
        "description": "Basic infantry unit. Cheap and fast to produce.",
    },
    "e2": {
        "name": "Grenadier",
        "category": "infantry",
        "cost": 150,
        "hp": 5000,
        "speed": 56,
        "armor": "none",
        "side": "both",
        "prerequisites": ["barr|tent"],
        "description": "Anti-structure infantry. Grenades deal area damage.",
    },
    "e3": {
        "name": "Rocket Soldier",
        "category": "infantry",
        "cost": 300,
        "hp": 4500,
        "speed": 56,
        "armor": "none",
        "side": "both",
        "prerequisites": ["barr|tent"],
        "description": "Anti-armor and anti-air infantry.",
    },
    "e4": {
        "name": "Flamethrower",
        "category": "infantry",
        "cost": 300,
        "hp": 4000,
        "speed": 56,
        "armor": "none",
        "side": "soviet",
        "prerequisites": ["barr", "ftur"],
        "description": "Short-range anti-infantry/structure. Soviet only.",
    },
    "e6": {
        "name": "Engineer",
        "category": "infantry",
        "cost": 400,
        "hp": 4000,
        "speed": 56,
        "armor": "none",
        "side": "both",
        "prerequisites": ["barr|tent"],
        "description": "Captures enemy buildings. Cannot attack.",
    },
    "e7": {
        "name": "Tanya",
        "category": "infantry",
        "cost": 1800,
        "hp": 10000,
        "speed": 68,
        "armor": "none",
        "side": "allied",
        "prerequisites": ["tent", "atek"],
        "build_limit": 1,
        "description": "Elite commando. Destroys buildings with C4, kills infantry instantly. Allied only.",
    },
    "medi": {
        "name": "Medic",
        "category": "infantry",
        "cost": 200,
        "hp": 6000,
        "speed": 49,
        "armor": "none",
        "side": "allied",
        "prerequisites": ["tent"],
        "description": "Heals nearby infantry. Cannot attack.",
    },
    "mech": {
        "name": "Mechanic",
        "category": "infantry",
        "cost": 500,
        "hp": 8000,
        "speed": 49,
        "armor": "none",
        "side": "allied",
        "prerequisites": ["tent", "fix"],
        "description": "Repairs nearby vehicles. Cannot attack.",
    },
    "spy": {
        "name": "Spy",
        "category": "infantry",
        "cost": 500,
        "hp": 2500,
        "speed": 56,
        "armor": "none",
        "side": "allied",
        "prerequisites": ["tent", "dome"],
        "description": "Disguises as enemy infantry. Infiltrates buildings for bonuses.",
    },
    "thf": {
        "name": "Thief",
        "category": "infantry",
        "cost": 500,
        "hp": 5000,
        "speed": 68,
        "armor": "none",
        "side": "allied",
        "prerequisites": ["tent", "dome"],
        "description": "Steals credits from enemy refineries.",
    },
    "shok": {
        "name": "Shock Trooper",
        "category": "infantry",
        "cost": 350,
        "hp": 5000,
        "speed": 49,
        "armor": "none",
        "side": "soviet",
        "prerequisites": ["barr", "stek", "tsla"],
        "description": "Tesla infantry. High damage vs all targets. Soviet only.",
    },
    "dog": {
        "name": "Attack Dog",
        "category": "infantry",
        "cost": 200,
        "hp": 2000,
        "speed": 99,
        "armor": "none",
        "side": "soviet",
        "prerequisites": ["kenn"],
        "description": "Fast anti-infantry unit. Kills spies. Soviet only.",
    },

    # Vehicles
    "1tnk": {
        "name": "Light Tank",
        "category": "vehicle",
        "cost": 700,
        "hp": 23000,
        "speed": 113,
        "armor": "heavy",
        "side": "allied",
        "prerequisites": ["weap"],
        "description": "Fast medium tank. Good all-around. Allied only.",
    },
    "2tnk": {
        "name": "Medium Tank",
        "category": "vehicle",
        "cost": 850,
        "hp": 30000,
        "speed": 72,
        "armor": "heavy",
        "side": "allied",
        "prerequisites": ["weap", "fix"],
        "description": "Main battle tank. Balanced stats. Allied only. Requires Repair Facility.",
    },
    "3tnk": {
        "name": "Heavy Tank",
        "category": "vehicle",
        "cost": 1150,
        "hp": 46000,
        "speed": 64,
        "armor": "heavy",
        "side": "soviet",
        "prerequisites": ["weap", "fix"],
        "description": "Powerful main battle tank. Dual cannons. Soviet only. Requires Repair Facility.",
    },
    "4tnk": {
        "name": "Mammoth Tank",
        "category": "vehicle",
        "cost": 2000,
        "hp": 60000,
        "speed": 43,
        "armor": "heavy",
        "side": "soviet",
        "prerequisites": ["weap", "fix", "stek"],
        "description": "Heaviest tank. Dual cannons + missiles. Self-healing. Soviet only.",
    },
    "v2rl": {
        "name": "V2 Rocket Launcher",
        "category": "vehicle",
        "cost": 900,
        "hp": 15000,
        "speed": 72,
        "armor": "light",
        "side": "soviet",
        "prerequisites": ["weap", "dome"],
        "description": "Long-range artillery. High damage, inaccurate. Soviet only.",
    },
    "jeep": {
        "name": "Ranger",
        "category": "vehicle",
        "cost": 500,
        "hp": 15000,
        "speed": 164,
        "armor": "light",
        "side": "allied",
        "prerequisites": ["weap"],
        "description": "Fast scout vehicle with machine gun. Allied only.",
    },
    "apc": {
        "name": "APC",
        "category": "vehicle",
        "cost": 850,
        "hp": 20000,
        "speed": 128,
        "armor": "heavy",
        "side": "soviet",
        "prerequisites": ["weap"],
        "description": "Armored troop transport. Carries 5 infantry. Soviet only.",
    },
    "arty": {
        "name": "Artillery",
        "category": "vehicle",
        "cost": 850,
        "hp": 7500,
        "speed": 54,
        "armor": "light",
        "side": "allied",
        "prerequisites": ["weap", "dome"],
        "description": "Long-range siege weapon. Allied only.",
    },
    "harv": {
        "name": "Ore Truck",
        "category": "vehicle",
        "cost": 1100,
        "hp": 60000,
        "speed": 72,
        "armor": "heavy",
        "side": "both",
        "prerequisites": ["proc"],
        "description": "Harvests ore and delivers to refinery. Free with refinery.",
    },
    "mcv": {
        "name": "MCV",
        "category": "vehicle",
        "cost": 2000,
        "hp": 60000,
        "speed": 60,
        "armor": "light",
        "side": "both",
        "prerequisites": ["weap", "fix"],
        "description": "Deploys into Construction Yard. Mobile base.",
    },
    "ftrk": {
        "name": "Flak Truck",
        "category": "vehicle",
        "cost": 600,
        "hp": 15000,
        "speed": 113,
        "armor": "light",
        "side": "soviet",
        "prerequisites": ["weap"],
        "description": "Mobile anti-air unit. Soviet only.",
    },
    "mnly": {
        "name": "Minelayer",
        "category": "vehicle",
        "cost": 800,
        "hp": 30000,
        "speed": 113,
        "armor": "heavy",
        "side": "both",
        "prerequisites": ["weap", "fix"],
        "description": "Lays anti-tank mines.",
    },
    "ttnk": {
        "name": "Tesla Tank",
        "category": "vehicle",
        "cost": 1350,
        "hp": 30000,
        "speed": 92,
        "armor": "light",
        "side": "soviet",
        "prerequisites": ["weap", "stek", "tsla"],
        "description": "Tesla weapon on tracks. Effective vs all targets. Soviet only.",
    },
    "ctnk": {
        "name": "Chrono Tank",
        "category": "vehicle",
        "cost": 1350,
        "hp": 20000,
        "speed": 86,
        "armor": "light",
        "side": "allied",
        "prerequisites": ["weap", "atek"],
        "description": "Teleporting tank. Hit and run tactics. Allied only.",
    },
    "stnk": {
        "name": "Phase Transport",
        "category": "vehicle",
        "cost": 1000,
        "hp": 11000,
        "speed": 128,
        "armor": "light",
        "side": "allied",
        "prerequisites": ["weap", "atek"],
        "description": "Cloaked APC. Invisible when not firing. Allied only.",
    },
    "qtnk": {
        "name": "MAD Tank",
        "category": "vehicle",
        "cost": 2000,
        "hp": 22000,
        "speed": 46,
        "armor": "heavy",
        "side": "soviet",
        "prerequisites": ["weap", "stek"],
        "description": "Deploys seismic charge, destroying self and nearby vehicles. Soviet only.",
    },
    "dtrk": {
        "name": "Demolition Truck",
        "category": "vehicle",
        "cost": 2500,
        "hp": 11000,
        "speed": 113,
        "armor": "light",
        "side": "soviet",
        "prerequisites": ["weap", "stek"],
        "description": "Suicide vehicle. Massive area nuclear explosion on death. Soviet only.",
    },
    "mgg": {
        "name": "Mobile Gap Generator",
        "category": "vehicle",
        "cost": 1000,
        "hp": 11000,
        "speed": 72,
        "armor": "heavy",
        "side": "allied",
        "prerequisites": ["weap", "atek"],
        "description": "Creates mobile shroud area. Allied only.",
    },
    "mrj": {
        "name": "Mobile Radar Jammer",
        "category": "vehicle",
        "cost": 1000,
        "hp": 11000,
        "speed": 68,
        "armor": "heavy",
        "side": "allied",
        "prerequisites": ["weap", "atek"],
        "description": "Jams enemy radar in area. Allied only.",
    },
    "truk": {
        "name": "Supply Truck",
        "category": "vehicle",
        "cost": 500,
        "hp": 11000,
        "speed": 113,
        "armor": "light",
        "side": "both",
        "prerequisites": ["weap"],
        "description": "Delivers cash when reaching allied structures.",
    },

    # Aircraft
    "heli": {
        "name": "Longbow",
        "category": "aircraft",
        "cost": 2000,
        "hp": 12000,
        "speed": 149,
        "armor": "light",
        "side": "allied",
        "prerequisites": ["hpad"],
        "description": "Anti-armor helicopter with missiles. Allied only.",
    },
    "hind": {
        "name": "Hind",
        "category": "aircraft",
        "cost": 1500,
        "hp": 12000,
        "speed": 112,
        "armor": "light",
        "side": "soviet",
        "prerequisites": ["afld"],
        "description": "Anti-ground attack helicopter. Soviet only.",
    },
    "mh60": {
        "name": "Black Hawk",
        "category": "aircraft",
        "cost": 1500,
        "hp": 12000,
        "speed": 112,
        "armor": "light",
        "side": "allied",
        "prerequisites": ["hpad"],
        "description": "Transport/attack helicopter. Allied only.",
    },
    "tran": {
        "name": "Chinook",
        "category": "aircraft",
        "cost": 900,
        "hp": 14000,
        "speed": 128,
        "armor": "light",
        "side": "both",
        "prerequisites": ["hpad|afld"],
        "description": "Transport helicopter. Carries 5 infantry.",
    },
    "yak": {
        "name": "Yak",
        "category": "aircraft",
        "cost": 1350,
        "hp": 6000,
        "speed": 178,
        "armor": "light",
        "side": "soviet",
        "prerequisites": ["afld"],
        "description": "Fast anti-infantry attack plane. Soviet only.",
    },
    "mig": {
        "name": "MiG",
        "category": "aircraft",
        "cost": 2000,
        "hp": 8000,
        "speed": 223,
        "armor": "light",
        "side": "soviet",
        "prerequisites": ["afld", "stek"],
        "description": "Anti-structure/armor attack plane with missiles. Soviet only.",
    },

    # Ships
    "ss": {
        "name": "Submarine",
        "category": "ship",
        "cost": 950,
        "hp": 25000,
        "speed": 78,
        "armor": "light",
        "side": "soviet",
        "prerequisites": ["spen"],
        "description": "Invisible anti-ship unit. Soviet only.",
    },
    "dd": {
        "name": "Destroyer",
        "category": "ship",
        "cost": 1000,
        "hp": 40000,
        "speed": 92,
        "armor": "heavy",
        "side": "allied",
        "prerequisites": ["syrd", "dome"],
        "description": "Multi-role warship. Anti-sub, anti-air, anti-surface. Allied only.",
    },
    "ca": {
        "name": "Cruiser",
        "category": "ship",
        "cost": 2400,
        "hp": 80000,
        "speed": 44,
        "armor": "heavy",
        "side": "allied",
        "prerequisites": ["syrd", "atek"],
        "description": "Heavy bombardment ship. Long range. Allied only.",
    },
    "pt": {
        "name": "Gunboat",
        "category": "ship",
        "cost": 500,
        "hp": 20000,
        "speed": 142,
        "armor": "heavy",
        "side": "both",
        "prerequisites": ["syrd|spen"],
        "description": "Fast patrol boat.",
    },
    "lst": {
        "name": "Transport",
        "category": "ship",
        "cost": 500,
        "hp": 40000,
        "speed": 115,
        "armor": "heavy",
        "side": "both",
        "prerequisites": ["syrd|spen"],
        "description": "Naval transport. Carries vehicles and infantry.",
    },
    "msub": {
        "name": "Missile Submarine",
        "category": "ship",
        "cost": 2000,
        "hp": 40000,
        "speed": 44,
        "armor": "light",
        "side": "soviet",
        "prerequisites": ["spen", "stek"],
        "description": "Long-range missile submarine. Soviet only.",
    },
}


# ─── Building Data ────────────────────────────────────────────────────────────

RA_BUILDINGS: dict[str, dict] = {
    "fact": {
        "name": "Construction Yard",
        "cost": 2000,
        "hp": 150000,
        "power": 0,
        "side": "both",
        "prerequisites": [],
        "produces": ["Building", "Defense"],
        "description": "Primary base structure. Required to build other structures.",
    },
    "powr": {
        "name": "Power Plant",
        "cost": 300,
        "hp": 40000,
        "power": 100,
        "side": "both",
        "prerequisites": [],
        "produces": [],
        "description": "Basic power supply. Most structures need power to function.",
    },
    "apwr": {
        "name": "Advanced Power Plant",
        "cost": 500,
        "hp": 70000,
        "power": 200,
        "side": "both",
        "prerequisites": ["dome"],
        "produces": [],
        "description": "Double power output. Requires radar dome tech.",
    },
    "barr": {
        "name": "Soviet Barracks",
        "cost": 500,
        "hp": 60000,
        "power": -20,
        "side": "soviet",
        "prerequisites": ["powr"],
        "produces": ["Infantry"],
        "description": "Soviet infantry production. Required for all Soviet infantry.",
    },
    "tent": {
        "name": "Allied Barracks",
        "cost": 500,
        "hp": 60000,
        "power": -20,
        "side": "allied",
        "prerequisites": ["powr"],
        "produces": ["Infantry"],
        "description": "Allied infantry production. Required for all Allied infantry.",
    },
    "proc": {
        "name": "Ore Refinery",
        "cost": 1400,
        "hp": 90000,
        "power": -30,
        "side": "both",
        "prerequisites": ["powr"],
        "produces": [],
        "description": "Processes ore into credits. Comes with a free Ore Truck.",
    },
    "weap": {
        "name": "War Factory",
        "cost": 2000,
        "hp": 150000,
        "power": -30,
        "side": "both",
        "prerequisites": ["proc"],
        "produces": ["Vehicle"],
        "description": "Vehicle production facility. Required for all vehicles.",
    },
    "dome": {
        "name": "Radar Dome",
        "cost": 1500,
        "hp": 100000,
        "power": -40,
        "side": "both",
        "prerequisites": ["proc"],
        "produces": [],
        "description": "Provides minimap radar. Unlocks advanced tech.",
    },
    "fix": {
        "name": "Service Depot",
        "cost": 1200,
        "hp": 80000,
        "power": -30,
        "side": "both",
        "prerequisites": ["weap"],
        "produces": [],
        "description": "Repairs vehicles. Unlocks MCV and Minelayer.",
    },
    "atek": {
        "name": "Allied Tech Center",
        "cost": 1500,
        "hp": 60000,
        "power": -200,
        "side": "allied",
        "prerequisites": ["dome", "weap"],
        "produces": [],
        "description": "Unlocks advanced Allied units. GPS satellite.",
    },
    "stek": {
        "name": "Soviet Tech Center",
        "cost": 1500,
        "hp": 80000,
        "power": -100,
        "side": "soviet",
        "prerequisites": ["dome", "weap"],
        "produces": [],
        "description": "Unlocks advanced Soviet units.",
    },
    "hpad": {
        "name": "Helipad",
        "cost": 500,
        "hp": 80000,
        "power": -10,
        "side": "allied",
        "prerequisites": ["dome"],
        "produces": ["Aircraft"],
        "description": "Allied aircraft production. Rearming pad.",
    },
    "afld": {
        "name": "Airfield",
        "cost": 500,
        "hp": 100000,
        "power": -20,
        "side": "soviet",
        "prerequisites": ["dome"],
        "produces": ["Aircraft"],
        "description": "Soviet aircraft production. Rearming strip.",
    },
    "spen": {
        "name": "Sub Pen",
        "cost": 800,
        "hp": 100000,
        "power": -20,
        "side": "soviet",
        "prerequisites": ["powr"],
        "produces": ["Ship"],
        "terrain": "water",
        "description": "Soviet naval production. Repairs ships. REQUIRES WATER — cannot build on land maps.",
    },
    "syrd": {
        "name": "Naval Yard",
        "cost": 1000,
        "hp": 100000,
        "power": -20,
        "side": "allied",
        "prerequisites": ["powr"],
        "produces": ["Ship"],
        "terrain": "water",
        "description": "Allied naval production. Repairs ships. REQUIRES WATER — cannot build on land maps.",
    },
    "silo": {
        "name": "Ore Silo",
        "cost": 150,
        "hp": 30000,
        "power": -10,
        "side": "both",
        "prerequisites": ["proc"],
        "produces": [],
        "description": "Additional ore storage capacity.",
    },
    "kenn": {
        "name": "Kennel",
        "cost": 200,
        "hp": 30000,
        "power": -10,
        "side": "soviet",
        "prerequisites": ["powr"],
        "produces": ["Infantry"],
        "description": "Produces attack dogs. Soviet only.",
    },

    # Defenses
    "pbox": {
        "name": "Pillbox",
        "cost": 600,
        "hp": 40000,
        "power": 0,
        "side": "allied",
        "prerequisites": ["tent"],
        "produces": [],
        "description": "Anti-infantry defense turret. Allied only.",
    },
    "hbox": {
        "name": "Camo Pillbox",
        "cost": 750,
        "hp": 40000,
        "power": 0,
        "side": "allied",
        "prerequisites": ["tent"],
        "produces": [],
        "description": "Hidden anti-infantry defense. Allied only.",
    },
    "gun": {
        "name": "Turret",
        "cost": 800,
        "hp": 40000,
        "power": -20,
        "side": "allied",
        "prerequisites": ["weap"],
        "produces": [],
        "description": "Anti-armor defense turret. Allied only.",
    },
    "ftur": {
        "name": "Flame Tower",
        "cost": 600,
        "hp": 40000,
        "power": -20,
        "side": "soviet",
        "prerequisites": ["barr"],
        "produces": [],
        "description": "Short-range anti-infantry defense. Soviet only.",
    },
    "tsla": {
        "name": "Tesla Coil",
        "cost": 1200,
        "hp": 40000,
        "power": -75,
        "side": "soviet",
        "prerequisites": ["weap"],
        "produces": [],
        "description": "Powerful anti-ground defense. High power cost. Soviet only.",
    },
    "agun": {
        "name": "AA Gun",
        "cost": 800,
        "hp": 40000,
        "power": -50,
        "side": "allied",
        "prerequisites": ["dome"],
        "produces": [],
        "description": "Anti-air defense turret. Allied only.",
    },
    "sam": {
        "name": "SAM Site",
        "cost": 700,
        "hp": 40000,
        "power": -20,
        "side": "soviet",
        "prerequisites": ["dome"],
        "produces": [],
        "description": "Anti-air missile defense. Soviet only.",
    },
    "gap": {
        "name": "Gap Generator",
        "cost": 800,
        "hp": 50000,
        "power": -60,
        "side": "allied",
        "prerequisites": ["atek"],
        "produces": [],
        "description": "Creates shroud area over your base. Allied only.",
    },

    # Superweapons
    "iron": {
        "name": "Iron Curtain",
        "cost": 2000,
        "hp": 100000,
        "power": -200,
        "side": "soviet",
        "prerequisites": ["stek"],
        "produces": [],
        "build_limit": 1,
        "description": "Superweapon: Makes one unit/building invulnerable temporarily.",
    },
    "pdox": {
        "name": "Chronosphere",
        "cost": 1500,
        "hp": 100000,
        "power": -200,
        "side": "allied",
        "prerequisites": ["atek"],
        "produces": [],
        "build_limit": 1,
        "description": "Superweapon: Teleports units across the map.",
    },
    "mslo": {
        "name": "Missile Silo",
        "cost": 2500,
        "hp": 100000,
        "power": -150,
        "side": "soviet",
        "prerequisites": ["stek"],
        "produces": [],
        "build_limit": 1,
        "description": "Superweapon: Launches nuclear missile at target location.",
    },
}


# ─── Tech Tree ────────────────────────────────────────────────────────────────

RA_TECH_TREE: dict[str, list[str]] = {
    "soviet": [
        "powr",     # Power Plant (base)
        "barr",     # Barracks → infantry (requires powr)
        "kenn",     # Kennel → dogs (requires powr)
        "proc",     # Ore Refinery (requires powr)
        "weap",     # War Factory (requires proc)
        "spen",     # Sub Pen (requires powr, needs water)
        "dome",     # Radar Dome (requires proc)
        "fix",      # Service Depot (requires weap)
        "afld",     # Airfield (requires dome)
        "stek",     # Tech Center (requires dome + weap)
        "tsla",     # Tesla Coil (requires weap)
        "sam",      # SAM Site (requires dome)
        "ftur",     # Flame Tower (requires barr)
        "iron",     # Iron Curtain (requires stek)
        "mslo",     # Missile Silo (requires stek)
    ],
    "allied": [
        "powr",     # Power Plant (base)
        "tent",     # Barracks → infantry (requires powr)
        "proc",     # Ore Refinery (requires powr)
        "weap",     # War Factory (requires proc)
        "syrd",     # Naval Yard (requires powr, needs water)
        "dome",     # Radar Dome (requires proc)
        "fix",      # Service Depot (requires weap)
        "hpad",     # Helipad (requires dome)
        "atek",     # Tech Center (requires dome + weap)
        "gun",      # Turret (requires weap)
        "pbox",     # Pillbox (requires tent)
        "agun",     # AA Gun (requires dome)
        "gap",      # Gap Generator (requires atek)
        "pdox",     # Chronosphere (requires atek)
    ],
}


# ─── Faction Data ─────────────────────────────────────────────────────────────

RA_FACTIONS: dict[str, dict] = {
    "england": {
        "side": "allied",
        "display_name": "England",
        "unique_units": [],
        "description": "Standard Allied faction.",
    },
    "france": {
        "side": "allied",
        "display_name": "France",
        "unique_units": ["stnk"],
        "description": "Allied faction with Phase Transport (cloaked APC).",
    },
    "germany": {
        "side": "allied",
        "display_name": "Germany",
        "unique_units": ["ctnk"],
        "description": "Allied faction with Chrono Tank (teleporting tank).",
    },
    "russia": {
        "side": "soviet",
        "display_name": "Russia",
        "unique_units": ["ttnk"],
        "description": "Soviet faction with Tesla Tank.",
    },
    "ukraine": {
        "side": "soviet",
        "display_name": "Ukraine",
        "unique_units": ["dtrk"],
        "description": "Soviet faction with Demolition Truck (nuclear suicide vehicle).",
    },
}


# ─── Query Functions ──────────────────────────────────────────────────────────


def get_unit_stats(unit_type: str) -> Optional[dict]:
    """Get stats for a unit type. Returns None if not found."""
    return RA_UNITS.get(unit_type.lower())


def get_building_stats(building_type: str) -> Optional[dict]:
    """Get stats for a building type. Returns None if not found."""
    return RA_BUILDINGS.get(building_type.lower())


def get_tech_tree(faction: Optional[str] = None) -> dict:
    """Get the tech tree build order.

    Args:
        faction: Faction name (e.g., 'russia') or side ('allied', 'soviet').
                If None, returns both sides.
    """
    if faction is None:
        return RA_TECH_TREE

    # Map faction to side
    side = faction.lower()
    if side in RA_FACTIONS:
        side = RA_FACTIONS[side]["side"]

    if side in RA_TECH_TREE:
        return {side: RA_TECH_TREE[side]}

    return {}


def get_faction_info(faction: str) -> Optional[dict]:
    """Get faction info including available units and buildings."""
    faction = faction.lower()
    info = RA_FACTIONS.get(faction)
    if info is None:
        return None

    side = info["side"]

    # Collect units available to this faction
    available_units = []
    for unit_type, data in RA_UNITS.items():
        unit_side = data.get("side", "")
        if unit_side == "both" or unit_side == side:
            available_units.append(unit_type)

    # Add faction-unique units
    for u in info.get("unique_units", []):
        if u not in available_units and u in RA_UNITS:
            available_units.append(u)

    # Collect buildings
    available_buildings = []
    for bldg_type, data in RA_BUILDINGS.items():
        bldg_side = data.get("side", "")
        if bldg_side == "both" or bldg_side == side:
            available_buildings.append(bldg_type)

    return {
        **info,
        "faction": faction,
        "available_units": sorted(available_units),
        "available_buildings": sorted(available_buildings),
    }


def get_all_unit_types() -> list[str]:
    """Get all available unit type names."""
    return sorted(RA_UNITS.keys())


def get_all_building_types() -> list[str]:
    """Get all available building type names."""
    return sorted(RA_BUILDINGS.keys())


def get_all_units_for_side(side: str) -> dict[str, dict]:
    """Get all units available to a side ('allied' or 'soviet') with full stats.

    Returns dict keyed by unit type name, each value is the full stats dict.
    Includes units with side='both' plus units specific to the given side.
    """
    side = side.lower()
    return {
        utype: dict(data)
        for utype, data in RA_UNITS.items()
        if data.get("side") in (side, "both")
    }


def get_all_buildings_for_side(side: str) -> dict[str, dict]:
    """Get all buildings available to a side ('allied' or 'soviet') with full stats.

    Returns dict keyed by building type name, each value is the full stats dict.
    Includes buildings with side='both' plus buildings specific to the given side.
    """
    side = side.lower()
    return {
        btype: dict(data)
        for btype, data in RA_BUILDINGS.items()
        if data.get("side") in (side, "both")
    }