"""
Pydantic models for in-memory game state management.

These models represent the active game state cached in memory for fast gameplay.
They are separate from the SQLAlchemy database models (db_models.py) to optimize
for different use cases:
- game_models.py: Fast in-memory operations, type-safe validation, WebSocket serialization
- db_models.py: Database persistence, relationships, audit trail

Author: Claude
Date: 2025-10-22
"""

import logging
from typing import Optional, Dict, List, Any
from uuid import UUID
from pydantic import BaseModel, Field, field_validator, ConfigDict

logger = logging.getLogger(f'{__name__}')


# ============================================================================
# RUNNER & BASE STATE
# ============================================================================

class RunnerState(BaseModel):
    """
    Represents a runner on base.

    Attributes:
        lineup_id: Unique lineup entry ID for this player in this game
        card_id: Player card ID (for fetching player data if needed)
        on_base: Base number (1=first, 2=second, 3=third)
    """
    lineup_id: int
    card_id: int
    on_base: int = Field(..., ge=1, le=3)

    # Future expansion for advanced gameplay
    # lead_off: bool = False
    # steal_attempt: bool = False

    @field_validator('on_base')
    @classmethod
    def validate_base(cls, v: int) -> int:
        """Ensure base is 1, 2, or 3"""
        if v not in [1, 2, 3]:
            raise ValueError("on_base must be 1, 2, or 3")
        return v


# ============================================================================
# LINEUP STATE
# ============================================================================

class LineupPlayerState(BaseModel):
    """
    Represents a player in the game lineup.

    This is a lightweight reference to a player - the full player data
    (ratings, attributes, etc.) will be cached separately in Week 6.
    """
    lineup_id: int
    card_id: int
    position: str
    batting_order: Optional[int] = None
    is_active: bool = True

    @field_validator('position')
    @classmethod
    def validate_position(cls, v: str) -> str:
        """Ensure position is valid"""
        valid_positions = ['P', 'C', '1B', '2B', '3B', 'SS', 'LF', 'CF', 'RF', 'DH']
        if v not in valid_positions:
            raise ValueError(f"Position must be one of {valid_positions}")
        return v

    @field_validator('batting_order')
    @classmethod
    def validate_batting_order(cls, v: Optional[int]) -> Optional[int]:
        """Ensure batting order is 1-9 if provided"""
        if v is not None and (v < 1 or v > 9):
            raise ValueError("batting_order must be between 1 and 9")
        return v


class TeamLineupState(BaseModel):
    """
    Represents a team's active lineup in the game.

    Provides helper methods for common lineup queries.
    """
    team_id: int
    players: List[LineupPlayerState] = Field(default_factory=list)

    def get_batting_order(self) -> List[LineupPlayerState]:
        """
        Get players in batting order (1-9).

        Returns:
            List of players sorted by batting_order
        """
        return sorted(
            [p for p in self.players if p.batting_order is not None],
            key=lambda x: x.batting_order
        )

    def get_pitcher(self) -> Optional[LineupPlayerState]:
        """
        Get the active pitcher for this team.

        Returns:
            Active pitcher or None if not found
        """
        pitchers = [p for p in self.players if p.position == 'P' and p.is_active]
        return pitchers[0] if pitchers else None

    def get_player_by_lineup_id(self, lineup_id: int) -> Optional[LineupPlayerState]:
        """
        Get player by lineup ID.

        Args:
            lineup_id: The lineup entry ID

        Returns:
            Player or None if not found
        """
        for player in self.players:
            if player.lineup_id == lineup_id:
                return player
        return None

    def get_batter(self, batting_order_idx: int) -> Optional[LineupPlayerState]:
        """
        Get batter by batting order index (0-8).

        Args:
            batting_order_idx: Index in batting order (0 = leadoff, 8 = 9th batter)

        Returns:
            Player at that position in the order, or None
        """
        order = self.get_batting_order()
        if 0 <= batting_order_idx < len(order):
            return order[batting_order_idx]
        return None


# ============================================================================
# DECISION STATE
# ============================================================================

class DefensiveDecision(BaseModel):
    """
    Defensive team strategic decisions for a play.

    These decisions affect play outcomes (e.g., infield depth affects double play chances).
    """
    alignment: str = "normal"  # normal, shifted_left, shifted_right, extreme_shift
    infield_depth: str = "normal"  # in, normal, back, double_play
    outfield_depth: str = "normal"  # in, normal, back
    hold_runners: List[int] = Field(default_factory=list)  # [1, 3] = hold 1st and 3rd

    @field_validator('alignment')
    @classmethod
    def validate_alignment(cls, v: str) -> str:
        """Validate alignment"""
        valid = ['normal', 'shifted_left', 'shifted_right', 'extreme_shift']
        if v not in valid:
            raise ValueError(f"alignment must be one of {valid}")
        return v

    @field_validator('infield_depth')
    @classmethod
    def validate_infield_depth(cls, v: str) -> str:
        """Validate infield depth"""
        valid = ['in', 'normal', 'back', 'double_play']
        if v not in valid:
            raise ValueError(f"infield_depth must be one of {valid}")
        return v

    @field_validator('outfield_depth')
    @classmethod
    def validate_outfield_depth(cls, v: str) -> str:
        """Validate outfield depth"""
        valid = ['in', 'normal', 'back']
        if v not in valid:
            raise ValueError(f"outfield_depth must be one of {valid}")
        return v


class OffensiveDecision(BaseModel):
    """
    Offensive team strategic decisions for a play.

    These decisions affect batter approach and baserunner actions.
    """
    approach: str = "normal"  # normal, contact, power, patient
    steal_attempts: List[int] = Field(default_factory=list)  # [2] = steal second, [2, 3] = double steal
    hit_and_run: bool = False
    bunt_attempt: bool = False

    @field_validator('approach')
    @classmethod
    def validate_approach(cls, v: str) -> str:
        """Validate batting approach"""
        valid = ['normal', 'contact', 'power', 'patient']
        if v not in valid:
            raise ValueError(f"approach must be one of {valid}")
        return v

    @field_validator('steal_attempts')
    @classmethod
    def validate_steal_attempts(cls, v: List[int]) -> List[int]:
        """Validate steal attempt bases"""
        for base in v:
            if base not in [2, 3, 4]:  # 2nd, 3rd, home
                raise ValueError(f"steal_attempts must contain only bases 2, 3, or 4")
        return v


# ============================================================================
# GAME STATE
# ============================================================================

class GameState(BaseModel):
    """
    Complete in-memory game state.

    This is the core state model representing an active game. It is optimized
    for fast in-memory operations during gameplay.

    Attributes:
        game_id: Unique game identifier
        league_id: League identifier ('sba' or 'pd')
        home_team_id: Home team ID (from league API)
        away_team_id: Away team ID (from league API)
        home_team_is_ai: Whether home team is AI-controlled
        away_team_is_ai: Whether away team is AI-controlled
        status: Game status (pending, active, paused, completed)
        inning: Current inning (1-9+)
        half: Current half ('top' or 'bottom')
        outs: Current outs (0-2)
        home_score: Home team score
        away_score: Away team score
        runners: List of runners currently on base
        current_batter_idx: Index in batting order (0-8)
        current_pitcher_lineup_id: Active pitcher's lineup ID
        pending_decision: Type of decision awaiting ('defensive', 'offensive', 'result_selection')
        decisions_this_play: Accumulated decisions for current play
        play_count: Total plays so far
        last_play_result: Description of last play outcome
    """
    game_id: UUID
    league_id: str

    # Teams
    home_team_id: int
    away_team_id: int
    home_team_is_ai: bool = False
    away_team_is_ai: bool = False

    # Game state
    status: str = "pending"  # pending, active, paused, completed
    inning: int = Field(default=1, ge=1)
    half: str = "top"  # top or bottom
    outs: int = Field(default=0, ge=0, le=2)

    # Score
    home_score: int = Field(default=0, ge=0)
    away_score: int = Field(default=0, ge=0)

    # Runners
    runners: List[RunnerState] = Field(default_factory=list)

    # Current at-bat
    current_batter_idx: int = Field(default=0, ge=0, le=8)  # 0-8 for 9 batters
    current_pitcher_lineup_id: Optional[int] = None

    # Decision tracking
    pending_decision: Optional[str] = None  # 'defensive', 'offensive', 'result_selection'
    decisions_this_play: Dict[str, Any] = Field(default_factory=dict)

    # Play tracking
    play_count: int = Field(default=0, ge=0)
    last_play_result: Optional[str] = None

    @field_validator('league_id')
    @classmethod
    def validate_league_id(cls, v: str) -> str:
        """Ensure league_id is valid"""
        valid_leagues = ['sba', 'pd']
        if v not in valid_leagues:
            raise ValueError(f"league_id must be one of {valid_leagues}")
        return v

    @field_validator('status')
    @classmethod
    def validate_status(cls, v: str) -> str:
        """Ensure status is valid"""
        valid_statuses = ['pending', 'active', 'paused', 'completed']
        if v not in valid_statuses:
            raise ValueError(f"status must be one of {valid_statuses}")
        return v

    @field_validator('half')
    @classmethod
    def validate_half(cls, v: str) -> str:
        """Ensure half is valid"""
        if v not in ['top', 'bottom']:
            raise ValueError("half must be 'top' or 'bottom'")
        return v

    @field_validator('pending_decision')
    @classmethod
    def validate_pending_decision(cls, v: Optional[str]) -> Optional[str]:
        """Ensure pending_decision is valid"""
        if v is not None:
            valid = ['defensive', 'offensive', 'result_selection', 'substitution']
            if v not in valid:
                raise ValueError(f"pending_decision must be one of {valid}")
        return v

    # Helper methods

    def get_batting_team_id(self) -> int:
        """Get the ID of the team currently batting"""
        return self.away_team_id if self.half == "top" else self.home_team_id

    def get_fielding_team_id(self) -> int:
        """Get the ID of the team currently fielding"""
        return self.home_team_id if self.half == "top" else self.away_team_id

    def is_runner_on_first(self) -> bool:
        """Check if there's a runner on first base"""
        return any(r.on_base == 1 for r in self.runners)

    def is_runner_on_second(self) -> bool:
        """Check if there's a runner on second base"""
        return any(r.on_base == 2 for r in self.runners)

    def is_runner_on_third(self) -> bool:
        """Check if there's a runner on third base"""
        return any(r.on_base == 3 for r in self.runners)

    def get_runner_at_base(self, base: int) -> Optional[RunnerState]:
        """Get runner at specified base (1, 2, or 3)"""
        for runner in self.runners:
            if runner.on_base == base:
                return runner
        return None

    def bases_occupied(self) -> List[int]:
        """Get list of occupied bases"""
        return sorted([r.on_base for r in self.runners])

    def clear_bases(self) -> None:
        """Clear all runners from bases"""
        self.runners = []

    def add_runner(self, lineup_id: int, card_id: int, base: int) -> None:
        """Add a runner to a base"""
        if base not in [1, 2, 3]:
            raise ValueError("base must be 1, 2, or 3")

        # Remove any existing runner at that base
        self.runners = [r for r in self.runners if r.on_base != base]

        # Add new runner
        self.runners.append(RunnerState(
            lineup_id=lineup_id,
            card_id=card_id,
            on_base=base
        ))

    def advance_runner(self, from_base: int, to_base: int) -> None:
        """
        Advance a runner from one base to another.

        Args:
            from_base: Starting base (1, 2, or 3)
            to_base: Ending base (2, 3, or 4 for home)
        """
        runner = self.get_runner_at_base(from_base)
        if runner:
            self.runners.remove(runner)
            if to_base == 4:
                # Runner scored
                if self.half == "top":
                    self.away_score += 1
                else:
                    self.home_score += 1
            else:
                # Runner advanced to another base
                runner.on_base = to_base
                self.runners.append(runner)

    def increment_outs(self) -> bool:
        """
        Increment outs by 1.

        Returns:
            True if half-inning is over (3 outs), False otherwise
        """
        self.outs += 1
        if self.outs >= 3:
            return True
        return False

    def end_half_inning(self) -> None:
        """End the current half-inning and prepare for next"""
        self.outs = 0
        self.clear_bases()

        if self.half == "top":
            # Switch to bottom of same inning
            self.half = "bottom"
        else:
            # End inning, go to top of next inning
            self.half = "top"
            self.inning += 1

    def is_game_over(self) -> bool:
        """
        Check if game is over.

        Game ends after 9 innings if home team is ahead or tied,
        or immediately if home team takes lead in bottom of 9th or later.
        """
        if self.inning < 9:
            return False

        if self.inning == 9 and self.half == "bottom":
            # Bottom of 9th - game ends if home team ahead
            if self.home_score > self.away_score:
                return True

        if self.inning > 9 and self.half == "bottom":
            # Extra innings, bottom half - walk-off possible
            if self.home_score > self.away_score:
                return True

        if self.inning >= 9 and self.half == "top" and self.outs >= 3:
            # Top of 9th or later just ended
            if self.home_score != self.away_score:
                return True

        return False

    model_config = ConfigDict(
        json_schema_extra={
            "example": {
                "game_id": "550e8400-e29b-41d4-a716-446655440000",
                "league_id": "sba",
                "home_team_id": 1,
                "away_team_id": 2,
                "home_team_is_ai": False,
                "away_team_is_ai": True,
                "status": "active",
                "inning": 3,
                "half": "top",
                "outs": 1,
                "home_score": 2,
                "away_score": 1,
                "runners": [
                    {"lineup_id": 5, "card_id": 123, "on_base": 2}
                ],
                "current_batter_idx": 3,
                "current_pitcher_lineup_id": 10,
                "pending_decision": None,
                "decisions_this_play": {},
                "play_count": 15,
                "last_play_result": "Single to left field"
            }
        }
    )


# ============================================================================
# EXPORTS
# ============================================================================

__all__ = [
    'RunnerState',
    'LineupPlayerState',
    'TeamLineupState',
    'DefensiveDecision',
    'OffensiveDecision',
    'GameState',
]