from __future__ import annotations
from typing import Literal
from pydantic import BaseModel, Field, model_validator


class Location(BaseModel):
    id: str
    name: str
    type: Literal["public", "commerce", "governance", "residential", "social"]
    description: str


class TimeConfig(BaseModel):
    total_days: int = 365
    rounds_per_day: int = 3
    active_agents_per_round_min: int = 3
    active_agents_per_round_max: int = 10

    @model_validator(mode="after")
    def validate_ranges(self) -> "TimeConfig":
        if self.total_days < 1:
            raise ValueError("total_days must be positive")
        if self.rounds_per_day < 1:
            raise ValueError("rounds_per_day must be positive")
        if self.active_agents_per_round_min < 0 or self.active_agents_per_round_max < 0:
            raise ValueError("active agent counts cannot be negative")
        if self.active_agents_per_round_min > self.active_agents_per_round_max:
            raise ValueError("active_agents_per_round_min cannot exceed max")
        return self


class WorldBlueprint(BaseModel):
    name: str
    description: str
    rules: list[str]
    locations: list[Location]
    resources: list[str]
    initial_tensions: list[str]
    time_config: TimeConfig


class WorldMetrics(BaseModel):
    stability: float = Field(default=0.5, ge=0.0, le=1.0)
    prosperity: float = Field(default=0.5, ge=0.0, le=1.0)
    trust: float = Field(default=0.5, ge=0.0, le=1.0)
    freedom: float = Field(default=0.5, ge=0.0, le=1.0)
    conflict: float = Field(default=0.2, ge=0.0, le=1.0)
    brand_sentiment: float = Field(default=0.5, ge=0.0, le=1.0)
    purchase_intent: float = Field(default=0.5, ge=0.0, le=1.0)
    word_of_mouth: float = Field(default=0.0, ge=0.0, le=1.0)
    churn_risk: float = Field(default=0.2, ge=0.0, le=1.0)
    adoption_rate: float = Field(default=0.0, ge=0.0, le=1.0)
    information_spread: float = Field(default=0.0, ge=0.0, le=1.0)
    echo_chamber_index: float = Field(default=0.0, ge=0.0, le=1.0)
    rumor_distortion: float = Field(default=0.0, ge=0.0, le=1.0)


class Institution(BaseModel):
    name: str
    purpose: str
    founder_id: int
    member_ids: list[int] = Field(default_factory=list)
    rules: list[str] = Field(default_factory=list)
    created_day: int = 0


class Proposal(BaseModel):
    id: str
    proposer_id: int
    content: str
    votes_for: list[int] = Field(default_factory=list)
    votes_against: list[int] = Field(default_factory=list)
    status: Literal["open", "passed", "rejected"] = "open"
    created_round: int = 0


class WorldState(BaseModel):
    blueprint: WorldBlueprint
    metrics: WorldMetrics = Field(default_factory=WorldMetrics)
    institutions: list[Institution] = Field(default_factory=list)
    proposals: list[Proposal] = Field(default_factory=list)
    day: int = 1
    round_in_day: int = 0
    active_disputes: list[str] = Field(default_factory=list)
    community_rules: list[str] = Field(default_factory=list)