warchron_app/src/warchron/model/war.py

from __future__ import annotations
from uuid import uuid4 
from datetime import datetime

from warchron.model.campaign import Campaign, Sector, CampaignParticipant


class War:
    def __init__(self, name: str, year: int):
        self.id: str = str(uuid4())
        self.name: str = name
        self.year: int = year
        self.participants:  dict[str, WarParticipant] = {}
        self.objectives: dict[str, Objective] = {}
        self.campaigns = []
        self.is_over: bool = False

    def set_id(self, new_id: str):
        self.id = new_id

    def set_name(self, new_name: str):
        self.name = new_name

    def set_year(self, new_year: int):
        self.year = new_year
        
    def set_state(self, new_state: bool):
        self.is_over = new_state

    def toDict(self):
        return {
            "id" : self.id,
            "name" : self.name,
            "year" : self.year,
            # "participants" : self.participants,
            "campaigns": [camp.toDict() for camp in self.campaigns],
            "is_over": self.is_over
        }
    
    @staticmethod
    def fromDict(data: dict):
        war = War(name=data["name"], year=data["year"])
        war.set_id(data["id"])
        # war.participants = data.get("participants", {})
        for camp_data in data.get("campaigns", []):
            war.campaigns.append(Campaign.fromDict(camp_data))
        war.set_state(data.get("is_over", False))
        return war

# Objective methods

    def add_objective(self, name: str, description: str) -> Objective:
        obj = Objective(name, description)
        self.objectives[obj.id] = obj
        return obj

    def get_objective(self, id: str) -> Objective:
        return self.objectives[id]

    def get_all_objectives(self) -> list[Objective]:
        return list(self.objectives.values())

    def get_objective_name(self, objective_id: str | None) -> str:
        if objective_id is None:
            return ""
        obj = self.objectives.get(objective_id)
        return obj.name if obj else ""
    
    def update_objective(self, objective_id: str, *, name: str, description: str):
        obj = self.get_objective(objective_id)
        obj.set_name(name)
        obj.set_description(description)
 
    def remove_objective(self, objective_id: str):
        del self.objectives[objective_id]

# War participant methods

    def get_all_war_participants_ids(self) -> set[str]:
        return set(self.participants.keys())

    def has_participant(self, player_id: str) -> bool:
        return player_id in self.participants

    def add_war_participant(self, player_id: str, faction: str) -> WarParticipant:
        if player_id in self.participants:
            raise ValueError("Player already registered in this war")
        participant = WarParticipant(player_id, faction)
        self.participants[participant.id] = participant
        return participant

    def get_war_participant(self, id: str) -> WarParticipant:
        return self.participants[id]

    def get_all_war_participants(self) -> list[WarParticipant]:
        return list(self.participants.values())

    def update_war_participant(self, player_id: str, *, faction: str):
        part = self.get_war_participant(player_id)
        part.set_faction(faction)

    def remove_war_participant(self, player_id: str):
        del self.participants[player_id]

# Campaign methods

    def has_campaign(self, campaign_id: str) -> bool:
        return any(c.id == campaign_id for c in self.campaigns)

    def get_default_campaign_values(self) -> dict:
        return {
            "month": datetime.now().month
        }
    
    def add_campaign(self, name: str, month: int | None = None) -> Campaign:
        if month is None:
            month = self.get_default_campaign_values()["month"]
        campaign = Campaign(name, month)
        self.campaigns.append(campaign)
        return campaign

    def get_campaign(self, campaign_id: str) -> Campaign:
        for camp in self.campaigns:
            if camp.id == campaign_id:
                return camp
        raise KeyError(f"Campaign {campaign_id} not found in War {self.id}")

    def get_campaign_by_round(self, round_id: str) -> Campaign:
        for camp in self.campaigns:
            for rnd in camp.rounds:
                if rnd.id == round_id:
                    return camp
        raise KeyError(f"Round {round_id} not found in any Campaign")

    def get_campaign_by_sector(self, sector_id: str) -> Campaign:
        for camp in self.campaigns:
            for sect in camp.sectors.values():
                if sect.id == sector_id:
                    return camp
        raise KeyError(f"Sector {sector_id} not found in any Campaign")

    def get_campaign_by_campaign_participant(self, participant_id: str) -> Campaign:
        for camp in self.campaigns:
            for part in camp.participants.values():
                if part.id == participant_id:
                    return camp
        raise KeyError(f"Participant {participant_id} not found in any Campaign")

    def update_campaign(self, campaign_id: str, *, name: str, month: int):
        camp = self.get_campaign(campaign_id)
        camp.set_name(name)
        camp.set_month(month)
        
    def get_all_campaigns(self) -> list[Campaign]:
        return list(self.campaigns)
    
    def remove_campaign(self, campaign_id: str):
        camp = self.get_campaign(campaign_id)
        self.campaigns.remove(camp)

# Sector methods

    def add_sector(self, campaign_id, name: str, round_id: str, major_id: str, minor_id: str, influence_id: str) -> Sector:
        camp = self.get_campaign(campaign_id)
        return camp.add_sector(name, round_id, major_id, minor_id, influence_id)

    def get_sector(self, id: str) -> Sector:
        return self.sectors[id]

    def update_sector(self, objective_id: str, *, name: str, round_id: str, major_id: str, minor_id: str, influence_id: str):
        obj = self.get_objective(objective_id)
        obj.set_name(name)
 
    def remove_sector(self, sector_id: str):
        camp = self.get_campaign_by_sector(sector_id)
        camp.remove_sector(sector_id)

# Campaign participant methods

    def get_available_war_participants(self, campaign_id: str) -> list[WarParticipant]:
        camp = self.get_campaign(campaign_id)
        return [
            part
            for part in self.participants.values()
            if not camp.has_participant(part.id)
        ]

    def add_campaign_participant(self, campaign_id: str, player_id: str, leader: str, theme: str) -> CampaignParticipant:
        camp = self.get_campaign(campaign_id)
        return camp.add_campaign_participant(player_id, leader, theme)

    def get_campaign_participant(self, participant_id) -> CampaignParticipant:
        for camp in self.campaigns.values():
            for part in camp.participants.values():
                if part.id == participant_id:
                    return part
        raise KeyError("Participant not found")

    def update_campaign_participant(self, participant_id: str, *, faction: str):
        camp = self.get_campaign_by_campaign_participant(participant_id)
        camp.update_campaign_participant(participant_id, faction=faction)

    def remove_campaign_participant(self, participant_id: str):
        camp = self.get_campaign_by_campaign_participant(participant_id)
        camp.remove_campaign_participant(participant_id)

class Objective:
    def __init__(self, name: str, description: str):
        self.id: str = str(uuid4())
        self.name: str = name
        self.description: str = description

    def set_id(self, new_id: str):
        self.id = new_id

    def set_name(self, new_name: str):
        self.name = new_name

    def set_description(self, new_description: str):
        self.description = new_description
    
class WarParticipant:
    def __init__(self, player_id: str, faction: str):
        self.id: str = player_id # ref to Model.players
        self.faction: str = faction

    def set_id(self, new_id: str):
        self.id = new_id

    def set_faction(self, new_faction: str):
        self.faction = new_faction