# Faction class
# You are welcome to add more data here if you want to keep
# persistent info in the faction object rather than the AI object.
# Or if you want to make the game's resources more complex.
# Currently, there's only money.

import unit


class Faction:
    def __init__(self, ID, money, ai, color):
        self.ID = ID
        self.money = money
        self.ai = ai
        self.next_unit_id = 0
        self.color = color

    def get_next_unit_id(self):
        uid = self.next_unit_id
        self.next_unit_id += 1
        return uid

    def can_build_unit(self, cost):
        return cost <= self.money
        
    # ################################################################
    def run_ai(self, factions, cities, units, gmap):
        match self.ID:
            case "Zombies":
                return self.ai.run_ai_zombies(self.ID, factions, cities, units, gmap)
            case "Survivors":
                return self.ai.run_ai_survivors(self.ID, factions, cities, units, gmap)
            case "Cured":
                return self.ai.run_ai_cured(self.ID, factions, cities, units, gmap)