botWebWars/backend/app/game.py

import asyncio
import random
import uuid
from typing import Dict, List, Optional, Set, Tuple
from app.config import settings
from app.models import (
    AiStepResponse,
    AvailableMovesResponse,
    BattleBout,
    BattleResult,
    BoardConfig,
    BoardState,
    BotMemoryResponse,
    BotRadarResponse,
    DIRECTION_OFFSETS,
    GameConclusion,
    MoveCheckResult,
    MoveResponse,
    Party,
    PartyDefeatResult,
    PartyInvite,
    Player,
    PlayerCreate,
    RadarTarget,
    TurnInfo,
)

STANDARD_DIRECTIONS = [
    ("UP", 0, -1),
    ("UP_RIGHT", 1, -1),
    ("RIGHT", 1, 0),
    ("DOWN_RIGHT", 1, 1),
    ("DOWN", 0, 1),
    ("DOWN_LEFT", -1, 1),
    ("LEFT", -1, 0),
    ("UP_LEFT", -1, -1),
]


class GameEngine:
    def __init__(self):
        self._lock = asyncio.Lock()
        self.players: Dict[str, Player] = {}
        self.parties: Dict[str, Party] = {}
        self.invites: Dict[str, PartyInvite] = {}
        self.turn_order: List[str] = []
        self.current_turn_index: int = 0
        self.round_number: int = 1
        self.turn_number: int = 0

        self.config = BoardConfig(
            min_x=settings.GRID_MIN_X,
            max_x=settings.GRID_MAX_X,
            min_y=settings.GRID_MIN_Y,
            max_y=settings.GRID_MAX_Y,
            grid_cells_x=settings.GRID_MAX_X - settings.GRID_MIN_X,
            grid_cells_y=settings.GRID_MAX_Y - settings.GRID_MIN_Y,
        )

    def _get_occupied_coordinates(self) -> Dict[Tuple[int, int], Player]:
        return {(p.x, p.y): p for p in self.players.values()}

    def _find_random_free_position(self) -> Tuple[int, int]:
        occupied = self._get_occupied_coordinates()
        total_possible = (self.config.max_x - self.config.min_x + 1) * (
            self.config.max_y - self.config.min_y + 1
        )

        if len(occupied) >= total_possible:
            return (
                random.randint(self.config.min_x, self.config.max_x),
                random.randint(self.config.min_y, self.config.max_y),
            )

        for _ in range(100):
            rx = random.randint(self.config.min_x, self.config.max_x)
            ry = random.randint(self.config.min_y, self.config.max_y)
            if (rx, ry) not in occupied:
                return (rx, ry)

        all_coords = [
            (x, y)
            for x in range(self.config.min_x, self.config.max_x + 1)
            for y in range(self.config.min_y, self.config.max_y + 1)
            if (x, y) not in occupied
        ]
        return random.choice(all_coords)

    def _get_active_turn_actors(self) -> List[str]:
        actors = []
        for pid in self.turn_order:
            p = self.players.get(pid)
            if not p:
                continue
            if not p.party_id or p.is_party_leader:
                actors.append(pid)
        return actors

    def _get_current_player(self) -> Optional[Player]:
        actors = self._get_active_turn_actors()
        if not actors:
            return None
        safe_index = self.current_turn_index % len(actors)
        player_id = actors[safe_index]
        return self.players.get(player_id)

    def _get_turn_info(self) -> TurnInfo:
        current = self._get_current_player()
        actors = self._get_active_turn_actors()
        return TurnInfo(
            current_player_id=current.id if current else None,
            current_player_name=current.name if current else None,
            round_number=self.round_number,
            turn_number=self.turn_number,
            turn_order=actors,
        )

    def _advance_turn(self):
        actors = self._get_active_turn_actors()
        if not actors:
            self.current_turn_index = 0
            return
        self.turn_number += 1
        self.current_turn_index = (self.current_turn_index + 1) % len(actors)
        if self.current_turn_index == 0:
            self.round_number += 1

    def _update_party_strength(self, party: Party):
        total = sum(self.players[m].strength for m in party.member_ids if m in self.players)
        party.total_strength = max(1, total)

    def _check_game_concluded(self) -> GameConclusion:
        """The game concludes when there is just 1 party left and all bots are in that party."""
        if len(self.players) >= 2 and len(self.parties) == 1:
            only_party = list(self.parties.values())[0]
            if len(only_party.member_ids) == len(self.players) and all(
                p.party_id == only_party.id for p in self.players.values()
            ):
                rankings = sorted(
                    self.players.values(),
                    key=lambda p: (p.score, p.strength),
                    reverse=True,
                )
                return GameConclusion(
                    concluded=True,
                    winning_party_id=only_party.id,
                    winning_party_name=only_party.name,
                    winning_leader_id=only_party.leader_id,
                    winning_leader_name=only_party.leader_name,
                    total_bots=len(self.players),
                    rankings=rankings,
                )
        return GameConclusion(concluded=False)

    async def get_game_conclusion(self) -> GameConclusion:
        async with self._lock:
            return self._check_game_concluded()

    async def register_player(self, player_in: PlayerCreate) -> Player:
        async with self._lock:
            player_id = f"bot_{uuid.uuid4().hex[:8]}"
            x, y = self._find_random_free_position()

            player = Player(
                id=player_id,
                name=player_in.name,
                color=player_in.color,
                x=x,
                y=y,
                strength=player_in.strength,
                score=0,
                party_id=None,
                is_party_leader=False,
                visited_locations=[{"x": x, "y": y}],
            )
            self.players[player_id] = player
            self.turn_order.append(player_id)
            return player

    async def get_player(self, player_id: str) -> Optional[Player]:
        async with self._lock:
            return self.players.get(player_id)

    async def get_all_players(self) -> List[Player]:
        async with self._lock:
            return list(self.players.values())

    async def remove_player(self, player_id: str) -> bool:
        async with self._lock:
            if player_id not in self.players:
                return False

            player = self.players[player_id]

            if player.party_id and player.party_id in self.parties:
                party = self.parties[player.party_id]
                if player_id in party.member_ids:
                    party.member_ids.remove(player_id)
                if party.leader_id == player_id:
                    if party.member_ids:
                        new_leader_id = random.choice(party.member_ids)
                        party.leader_id = new_leader_id
                        new_lead = self.players.get(new_leader_id)
                        if new_lead:
                            new_lead.is_party_leader = True
                            party.leader_name = new_lead.name
                        self._update_party_strength(party)
                    else:
                        del self.parties[party.id]
                elif party.member_ids:
                    self._update_party_strength(party)
                else:
                    del self.parties[party.id]

            del self.players[player_id]

            if player_id in self.turn_order:
                self.turn_order.remove(player_id)

            actors = self._get_active_turn_actors()
            if actors:
                self.current_turn_index %= len(actors)
            else:
                self.current_turn_index = 0

            return True

    async def reset(self) -> None:
        async with self._lock:
            self.players.clear()
            self.parties.clear()
            self.invites.clear()
            self.turn_order.clear()
            self.current_turn_index = 0
            self.round_number = 1
            self.turn_number = 0

    async def get_board_state(self) -> BoardState:
        async with self._lock:
            players_list = list(self.players.values())
            parties_list = list(self.parties.values())
            return BoardState(
                config=self.config,
                player_count=len(players_list),
                players=players_list,
                parties=parties_list,
                turn=self._get_turn_info(),
                conclusion=self._check_game_concluded(),
            )

    # ==========================================\n    # Bot Memory & Radar Awareness
    # ==========================================

    async def get_bot_memory(
        self, player_id: str, check_x: Optional[int] = None, check_y: Optional[int] = None
    ) -> BotMemoryResponse:
        async with self._lock:
            player = self.players.get(player_id)
            if not player:
                raise KeyError(f"Player '{player_id}' not found")

            has_visited = True
            if check_x is not None and check_y is not None:
                has_visited = any(loc["x"] == check_x and loc["y"] == check_y for loc in player.visited_locations)

            return BotMemoryResponse(
                player_id=player.id,
                player_name=player.name,
                current_x=player.x,
                current_y=player.y,
                visited_count=len(player.visited_locations),
                visited_history=list(player.visited_locations),
                has_visited_current=has_visited,
            )

    async def get_bot_radar(self, player_id: str) -> BotRadarResponse:
        async with self._lock:
            player = self.players.get(player_id)
            if not player:
                raise KeyError(f"Player '{player_id}' not found")

            bot_goal = "find_and_defeat_all_parties" if player.party_id else "form_party"
            targets: List[RadarTarget] = []

            for other in self.players.values():
                if other.id == player.id:
                    continue
                dist = max(abs(player.x - other.x), abs(player.y - other.y))
                is_ally = bool(player.party_id and player.party_id == other.party_id)
                is_enemy = not is_ally
                party_obj = self.parties.get(other.party_id) if other.party_id else None

                # Recruitment vs Battle rules:
                # 1. Unpartied vs unpartied: recruit (form party)
                # 2. Party leader vs solo bot:
                #    - If solo bot strength <= leader strength: recruit (bot desires equal/stronger leader)
                #    - If solo bot strength > leader strength: battle (bot refuses weaker leader, party responds with battle!)
                # 3. Solo bot vs party:
                #    - If solo bot strength <= other party leader strength: recruit (join)
                #    - If solo bot strength > other party leader strength: battle (refuses, party responds with battle)
                # 4. Party vs Party: battle
                if player.party_id and player.is_party_leader and other.party_id is None:
                    can_recruit = other.strength <= player.strength
                    can_battle = other.strength > player.strength
                elif player.party_id is None and other.party_id is None:
                    can_recruit = True
                    can_battle = False
                elif player.party_id is None and other.party_id:
                    other_lead = self.players.get(party_obj.leader_id) if party_obj else None
                    lead_str = other_lead.strength if other_lead else 1
                    can_recruit = player.strength <= lead_str
                    can_battle = player.strength > lead_str
                else:
                    can_recruit = False
                    can_battle = is_enemy and bool(player.party_id and other.party_id)

                targets.append(
                    RadarTarget(
                        id=other.id,
                        name=other.name,
                        color=other.color,
                        x=other.x,
                        y=other.y,
                        strength=other.strength,
                        distance=dist,
                        party_id=other.party_id,
                        party_name=party_obj.name if party_obj else None,
                        is_ally=is_ally,
                        is_enemy=is_enemy,
                        can_recruit=can_recruit,
                        can_battle=can_battle,
                    )
                )

            targets.sort(key=lambda t: t.distance)

            # Choose primary target based on goal:
            # - If unpartied: seek closest bot/party to join/form party with
            # - If partied: seek closest enemy party or refusing solo bot to battle
            primary_targets = []
            if bot_goal == "form_party":
                recruit_targets = [t for t in targets if t.can_recruit or not t.party_id]
                primary_targets = recruit_targets if recruit_targets else targets
            else:
                battle_targets = [t for t in targets if t.is_enemy and (t.party_id or t.can_battle)]
                primary_targets = battle_targets if battle_targets else [t for t in targets if t.is_enemy]

            nearest = primary_targets[0] if primary_targets else (targets[0] if targets else None)

            rec_dir = None
            rec_act = "explore_unvisited"

            if nearest:
                dx = 1 if nearest.x > player.x else (-1 if nearest.x < player.x else 0)
                dy = 1 if nearest.y > player.y else (-1 if nearest.y < player.y else 0)

                for name, (ox, oy) in DIRECTION_OFFSETS.items():
                    if ox == dx and oy == dy and "_" in name:
                        rec_dir = name
                        break
                if not rec_dir:
                    for name, (ox, oy) in DIRECTION_OFFSETS.items():
                        if ox == dx and oy == dy:
                            rec_dir = name
                            break

                if nearest.distance <= 1:
                    if nearest.can_recruit:
                        rec_act = "form_party"
                    else:
                        rec_act = "engage_battle"
                else:
                    if bot_goal == "form_party":
                        rec_act = "seek_partner"
                    else:
                        rec_act = "hunt_party"

            return BotRadarResponse(
                player_id=player.id,
                current_x=player.x,
                current_y=player.y,
                bot_goal=bot_goal,
                targets=targets,
                nearest_target=nearest,
                recommended_direction=rec_dir,
                recommended_action=rec_act,
            )

    # ==========================================\n    # Party Logic & Leadership Negotiation
    # ==========================================

    @staticmethod
    def _are_adjacent(p1: Player, p2: Player) -> bool:
        return max(abs(p1.x - p2.x), abs(p1.y - p2.y)) <= 1

    def _verify_party_connectivity(self, member_players: List[Player]) -> bool:
        if len(member_players) <= 1:
            return True

        visited: Set[str] = set()
        queue = [member_players[0]]
        visited.add(member_players[0].id)

        while queue:
            curr = queue.pop(0)
            for other in member_players:
                if other.id not in visited and self._are_adjacent(curr, other):
                    visited.add(other.id)
                    queue.append(other)

        return len(visited) == len(member_players)

    def _negotiate_party_leader_id(self, bot_a: Player, bot_b: Player) -> str:
        if bot_a.strength > bot_b.strength:
            return bot_a.id
        elif bot_b.strength > bot_a.strength:
            return bot_b.id
        else:
            if bot_a.score > bot_b.score:
                return bot_a.id
            elif bot_b.score > bot_a.score:
                return bot_b.id
            return bot_a.id

    async def form_party(
        self, member_ids: List[str], leader_id: str, name: Optional[str] = None
    ) -> Party:
        async with self._lock:
            if len(member_ids) < 2:
                raise ValueError("A party must have at least 2 members")
            if leader_id not in member_ids:
                raise ValueError("Agreed leader must be one of the party members")

            member_players: List[Player] = []
            for mid in member_ids:
                p = self.players.get(mid)
                if not p:
                    raise KeyError(f"Player '{mid}' not found")
                member_players.append(p)

            if not self._verify_party_connectivity(member_players):
                raise ValueError(
                    "All party members must be linked within 1 distance of each other (directly or via connected teammates)."
                )

            for p in member_players:
                if p.party_id and p.party_id in self.parties:
                    old_party = self.parties[p.party_id]
                    if p.id in old_party.member_ids:
                        old_party.member_ids.remove(p.id)
                    if not old_party.member_ids:
                        del self.parties[old_party.id]

            party_id = f"party_{uuid.uuid4().hex[:8]}"
            leader = self.players[leader_id]
            party_name = name or f"Squad {leader.name}"

            party = Party(
                id=party_id,
                name=party_name,
                leader_id=leader_id,
                leader_name=leader.name,
                member_ids=list(member_ids),
                total_strength=sum(p.strength for p in member_players),
            )

            for p in member_players:
                p.party_id = party_id
                p.is_party_leader = p.id == leader_id

            self.parties[party_id] = party
            return party

    async def invite_to_party(
        self,
        inviter_id: str,
        invitee_id: str,
        proposed_leader_id: str,
        party_name: Optional[str] = None,
    ) -> PartyInvite:
        async with self._lock:
            inviter = self.players.get(inviter_id)
            invitee = self.players.get(invitee_id)
            if not inviter or not invitee:
                raise KeyError("Inviter or invitee not found")

            eligible_hosts = [inviter]
            if inviter.party_id and inviter.party_id in self.parties:
                party = self.parties[inviter.party_id]
                eligible_hosts = [self.players[mid] for mid in party.member_ids if mid in self.players]

            is_adjacent = any(self._are_adjacent(invitee, host) for host in eligible_hosts)
            if not is_adjacent:
                raise ValueError(
                    f"Bot '{invitee.name}' is too far away. Must be within 1 distance of a party member to join."
                )

            invite_id = f"inv_{uuid.uuid4().hex[:8]}"
            invite = PartyInvite(
                id=invite_id,
                inviter_id=inviter_id,
                invitee_id=invitee_id,
                proposed_leader_id=proposed_leader_id,
                party_id=inviter.party_id,
            )
            self.invites[invite_id] = invite
            return invite

    async def respond_to_invite(self, invite_id: str, accept: bool) -> Optional[Party]:
        async with self._lock:
            invite = self.invites.get(invite_id)
            if not invite:
                raise KeyError(f"Invite '{invite_id}' not found")
            if invite.status != "pending":
                raise ValueError(f"Invite already {invite.status}")

            if not accept:
                invite.status = "rejected"
                return None

            invite.status = "accepted"
            inviter = self.players.get(invite.inviter_id)
            invitee = self.players.get(invite.invitee_id)
            if not inviter or not invitee:
                raise KeyError("Inviter or invitee no longer active")

            if inviter.party_id and inviter.party_id in self.parties:
                party = self.parties[inviter.party_id]
                if invitee.id not in party.member_ids:
                    party.member_ids.append(invitee.id)
                invitee.party_id = party.id
                if invite.proposed_leader_id in party.member_ids:
                    party.leader_id = invite.proposed_leader_id
                    for mid in party.member_ids:
                        p = self.players.get(mid)
                        if p:
                            p.is_party_leader = p.id == party.leader_id
                    party.leader_name = self.players[party.leader_id].name
                self._update_party_strength(party)
                return party

            members = [inviter.id, invitee.id]
            leader_id = invite.proposed_leader_id if invite.proposed_leader_id in members else inviter.id
            party_id = f"party_{uuid.uuid4().hex[:8]}"
            leader = self.players[leader_id]
            party = Party(
                id=party_id,
                name=f"Squad {leader.name}",
                leader_id=leader_id,
                leader_name=leader.name,
                member_ids=members,
                total_strength=inviter.strength + invitee.strength,
            )
            inviter.party_id = party_id
            inviter.is_party_leader = inviter.id == leader_id
            invitee.party_id = party_id
            invitee.is_party_leader = invitee.id == leader_id
            self.parties[party_id] = party
            return party

    async def get_all_parties(self) -> List[Party]:
        async with self._lock:
            return list(self.parties.values())

    async def get_party(self, party_id: str) -> Optional[Party]:
        async with self._lock:
            return self.parties.get(party_id)

    # ==========================================\n    # Movement Checking & Execution
    # ==========================================

    def _check_move_internal(
        self,
        player: Player,
        dx: int,
        dy: int,
        direction_name: str,
        occupied_map: Dict[Tuple[int, int], Player],
    ) -> MoveCheckResult:
        if player.party_id and player.party_id in self.parties and player.is_party_leader:
            party = self.parties[player.party_id]
            party_members = [self.players[mid] for mid in party.member_ids if mid in self.players]
            party_member_ids = {m.id for m in party_members}

            for m in party_members:
                tx = m.x + dx
                ty = m.y + dy

                if tx < self.config.min_x or tx > self.config.max_x or ty < self.config.min_y or ty > self.config.max_y:
                    return MoveCheckResult(
                        direction=direction_name,
                        dx=dx,
                        dy=dy,
                        target_x=player.x + dx,
                        target_y=player.y + dy,
                        available=False,
                        reason=f"Party member '{m.name}' would hit boundary wall at ({tx}, {ty}).",
                    )

                occupant = occupied_map.get((tx, ty))
                if occupant is not None and occupant.id not in party_member_ids:
                    return MoveCheckResult(
                        direction=direction_name,
                        dx=dx,
                        dy=dy,
                        target_x=player.x + dx,
                        target_y=player.y + dy,
                        available=False,
                        reason=f"Party member '{m.name}' path blocked by '{occupant.name}' at ({tx}, {ty}).",
                    )

            return MoveCheckResult(
                direction=direction_name,
                dx=dx,
                dy=dy,
                target_x=player.x + dx,
                target_y=player.y + dy,
                available=True,
                reason=None,
            )

        target_x = player.x + dx
        target_y = player.y + dy

        if (
            target_x < self.config.min_x
            or target_x > self.config.max_x
            or target_y < self.config.min_y
            or target_y > self.config.max_y
        ):
            return MoveCheckResult(
                direction=direction_name,
                dx=dx,
                dy=dy,
                target_x=target_x,
                target_y=target_y,
                available=False,
                reason=f"Hit boundary wall at ({target_x}, {target_y}). Coordinates must remain between {self.config.min_x} and {self.config.max_x}.",
            )

        occupant = occupied_map.get((target_x, target_y))
        if occupant is not None and occupant.id != player.id:
            return MoveCheckResult(
                direction=direction_name,
                dx=dx,
                dy=dy,
                target_x=target_x,
                target_y=target_y,
                available=False,
                reason=f"Target square ({target_x}, {target_y}) is occupied by player '{occupant.name}'.",
            )

        return MoveCheckResult(
            direction=direction_name,
            dx=dx,
            dy=dy,
            target_x=target_x,
            target_y=target_y,
            available=True,
            reason=None,
        )

    async def check_single_move(self, player_id: str, direction_input: str) -> MoveCheckResult:
        async with self._lock:
            player = self.players.get(player_id)
            if not player:
                raise KeyError(f"Player '{player_id}' not found")

            if player.party_id and not player.is_party_leader:
                party = self.parties.get(player.party_id)
                leader_name = party.leader_name if party else "Leader"
                return MoveCheckResult(
                    direction=direction_input,
                    dx=0,
                    dy=0,
                    target_x=player.x,
                    target_y=player.y,
                    available=False,
                    reason=f"Only party leader '{leader_name}' controls movement for the party.",
                )

            normalized = direction_input.strip().upper().replace(" ", "_")
            if normalized not in DIRECTION_OFFSETS:
                raise ValueError(f"Unknown direction '{direction_input}'")

            dx, dy = DIRECTION_OFFSETS[normalized]
            occupied = self._get_occupied_coordinates()
            return self._check_move_internal(player, dx, dy, normalized, occupied)

    async def get_available_moves(self, player_id: str) -> AvailableMovesResponse:
        async with self._lock:
            player = self.players.get(player_id)
            if not player:
                raise KeyError(f"Player '{player_id}' not found")

            current_turn_player = self._get_current_player()
            is_turn = current_turn_player is not None and current_turn_player.id == player_id
            occupied = self._get_occupied_coordinates()

            party = self.parties.get(player.party_id) if player.party_id else None
            is_leader = player.is_party_leader or party is None
            member_count = len(party.member_ids) if party else 1

            moves: Dict[str, MoveCheckResult] = {}
            for name, dx, dy in STANDARD_DIRECTIONS:
                if player.party_id and not player.is_party_leader:
                    moves[name] = MoveCheckResult(
                        direction=name,
                        dx=dx,
                        dy=dy,
                        target_x=player.x + dx,
                        target_y=player.y + dy,
                        available=False,
                        reason=f"Only party leader '{party.leader_name if party else 'Leader'}' controls group movement.",
                    )
                else:
                    moves[name] = self._check_move_internal(player, dx, dy, name, occupied)

            return AvailableMovesResponse(
                player_id=player.id,
                player_name=player.name,
                current_x=player.x,
                current_y=player.y,
                is_turn=is_turn and is_leader,
                is_party_leader=player.is_party_leader,
                party_id=player.party_id,
                party_member_count=member_count,
                current_turn_player_id=current_turn_player.id if current_turn_player else None,
                moves=moves,
            )

    def _check_adjacent_encounter(
        self, player: Player
    ) -> Tuple[Optional[Party], Optional[BattleResult]]:
        """Evaluates immediate adjacent encounters for party formation or battle:
        1. Two unpartied bots meet -> negotiate leadership (stronger leads), form party.
        2. Unpartied bot meets a party:
           - If bot strength <= leader strength: bot joins party willingly (equal or stronger leader).
           - If bot strength > leader strength: bot refuses to join weaker leader;
             party responds by doing battle!
        3. Two opposing parties meet -> engage in 3-bout D20 battle!
        """
        # Case A: Player is in a party
        if player.party_id and player.party_id in self.parties:
            if not player.is_party_leader:
                return None, None

            my_party = self.parties[player.party_id]
            my_members = [self.players[mid] for mid in my_party.member_ids if mid in self.players]

            for m in my_members:
                for other in list(self.players.values()):
                    if other.id in my_party.member_ids:
                        continue
                    if not self._are_adjacent(m, other):
                        continue

                    # Opposing party
                    if other.party_id and other.party_id in self.parties:
                        opposing_party = self.parties[other.party_id]
                        battle_res = self._resolve_3bout_battle_internal(my_party, opposing_party)
                        return None, battle_res

                    # Solo bot (other.party_id is None)
                    if other.party_id is None:
                        if other.strength <= player.strength:
                            # Bot willingly joins party under equal/stronger leader
                            other.party_id = my_party.id
                            other.is_party_leader = False
                            if other.id not in my_party.member_ids:
                                my_party.member_ids.append(other.id)
                            self._update_party_strength(my_party)
                            return my_party, None
                        else:
                            # other.strength > player.strength:
                            # Bot refuses to join weaker leader! Party responds by doing battle!
                            solo_party_id = f"party_{uuid.uuid4().hex[:8]}"
                            solo_party = Party(
                                id=solo_party_id,
                                name=f"Squad {other.name}",
                                leader_id=other.id,
                                leader_name=other.name,
                                member_ids=[other.id],
                                total_strength=other.strength,
                            )
                            other.party_id = solo_party_id
                            other.is_party_leader = True
                            self.parties[solo_party_id] = solo_party

                            battle_res = self._resolve_3bout_battle_internal(my_party, solo_party)
                            return None, battle_res

            return None, None

        # Case B: Player is a Solo Bot
        else:
            for other in list(self.players.values()):
                if other.id == player.id:
                    continue
                if not self._are_adjacent(player, other):
                    continue

                # Solo Bot meets another Solo Bot -> form party
                if other.party_id is None:
                    leader_id = self._negotiate_party_leader_id(player, other)
                    leader = self.players[leader_id]
                    party_id = f"party_{uuid.uuid4().hex[:8]}"
                    party = Party(
                        id=party_id,
                        name=f"Squad {leader.name}",
                        leader_id=leader_id,
                        leader_name=leader.name,
                        member_ids=[player.id, other.id],
                        total_strength=player.strength + other.strength,
                    )
                    player.party_id = party_id
                    player.is_party_leader = player.id == leader_id
                    other.party_id = party_id
                    other.is_party_leader = other.id == leader_id
                    self.parties[party_id] = party
                    return party, None

                # Solo Bot meets a Party Member
                if other.party_id and other.party_id in self.parties:
                    other_party = self.parties[other.party_id]
                    leader = self.players.get(other_party.leader_id)
                    leader_strength = leader.strength if leader else 1

                    if player.strength <= leader_strength:
                        # Player willingly joins party under equal/stronger leader
                        player.party_id = other_party.id
                        player.is_party_leader = False
                        if player.id not in other_party.member_ids:
                            other_party.member_ids.append(player.id)
                        self._update_party_strength(other_party)
                        return other_party, None
                    else:
                        # player.strength > leader_strength:
                        # Player refuses to join weaker leader! Party responds by doing battle!
                        solo_party_id = f"party_{uuid.uuid4().hex[:8]}"
                        solo_party = Party(
                            id=solo_party_id,
                            name=f"Squad {player.name}",
                            leader_id=player.id,
                            leader_name=player.name,
                            member_ids=[player.id],
                            total_strength=player.strength,
                        )
                        player.party_id = solo_party_id
                        player.is_party_leader = True
                        self.parties[solo_party_id] = solo_party

                        battle_res = self._resolve_3bout_battle_internal(other_party, solo_party)
                        return None, battle_res

            return None, None

    def _check_and_auto_form_party(self, player: Player) -> Optional[Party]:
        party, _ = self._check_adjacent_encounter(player)
        return party

    async def move_player(
        self, player_id: str, dx: int, dy: int, direction_name: str
    ) -> MoveResponse:
        async with self._lock:
            player = self.players.get(player_id)
            if not player:
                raise KeyError(f"Player '{player_id}' not found")

            if player.party_id and not player.is_party_leader:
                party = self.parties.get(player.party_id)
                leader_name = party.leader_name if party else "Leader"
                raise PermissionError(
                    f"Party member '{player.name}' cannot move individually. Only party leader '{leader_name}' controls movement for the party."
                )

            current_turn_player = self._get_current_player()
            if not current_turn_player or current_turn_player.id != player_id:
                curr_name = current_turn_player.name if current_turn_player else "Nobody"
                curr_id = current_turn_player.id if current_turn_player else "None"
                raise PermissionError(
                    f"It is not your turn. Current turn belongs to '{curr_name}' ({curr_id})."
                )

            occupied = self._get_occupied_coordinates()
            check = self._check_move_internal(player, dx, dy, direction_name, occupied)

            if not check.available:
                raise ValueError(f"Illegal move: {check.reason}")

            prev_pos = {"x": player.x, "y": player.y}
            affected_players: List[Player] = []

            # Move party in unison or single bot
            if player.party_id and player.party_id in self.parties:
                party = self.parties[player.party_id]
                for mid in party.member_ids:
                    m = self.players.get(mid)
                    if m:
                        m.x += dx
                        m.y += dy
                        m.visited_locations.append({"x": m.x, "y": m.y})
                        affected_players.append(m)
            else:
                player.x = check.target_x
                player.y = check.target_y
                player.visited_locations.append({"x": player.x, "y": player.y})
                affected_players.append(player)

            new_pos = {"x": player.x, "y": player.y}

            # Check for adjacent interactions (party formation or battle)
            formed_party, battle_result = self._check_adjacent_encounter(player)

            self._advance_turn()
            turn_info = self._get_turn_info()
            conclusion = self._check_game_concluded()

            return MoveResponse(
                success=True,
                player=player,
                direction=direction_name,
                party_moved=bool(player.party_id),
                affected_players=affected_players,
                previous_position=prev_pos,
                new_position=new_pos,
                party_formed_triggered=bool(formed_party),
                formed_party=formed_party,
                battle_triggered=bool(battle_result),
                battle_result=battle_result,
                game_concluded=conclusion if conclusion.concluded else None,
                turn=turn_info,
            )

    async def pass_turn(self, player_id: str) -> TurnInfo:
        async with self._lock:
            player = self.players.get(player_id)
            if not player:
                raise KeyError(f"Player '{player_id}' not found")

            current_turn_player = self._get_current_player()
            if not current_turn_player or current_turn_player.id != player_id:
                curr_name = current_turn_player.name if current_turn_player else "Nobody"
                curr_id = current_turn_player.id if current_turn_player else "None"
                raise PermissionError(
                    f"Cannot pass: it is not your turn. Current turn belongs to '{curr_name}' ({curr_id})."
                )

            self._advance_turn()
            return self._get_turn_info()

    # ==========================================\n    # Autonomous Goal Step (AI Logic)
    # ==========================================

    async def step_bot_ai(self, player_id: str) -> AiStepResponse:
        async with self._lock:
            player = self.players.get(player_id)
            if not player:
                raise KeyError(f"Player '{player_id}' not found")

            current_turn_player = self._get_current_player()
            if not current_turn_player or current_turn_player.id != player_id:
                raise PermissionError(
                    f"Not your turn. Current turn: '{current_turn_player.name if current_turn_player else 'None'}'"
                )

            bot_goal = "find_and_defeat_all_parties" if player.party_id else "form_party"

            # 1. Check for immediate adjacent encounter before moving
            formed_party, battle_res = self._check_adjacent_encounter(player)
            if battle_res:
                self._advance_turn()
                conclusion = self._check_game_concluded()
                return AiStepResponse(
                    action_taken="battled",
                    player_id=player.id,
                    player_name=player.name,
                    bot_goal=bot_goal,
                    direction=None,
                    move_result=None,
                    formed_party=None,
                    battle_result=battle_res,
                    game_concluded=conclusion if conclusion.concluded else None,
                    turn=self._get_turn_info(),
                )

            if formed_party:
                self._advance_turn()
                conclusion = self._check_game_concluded()
                return AiStepResponse(
                    action_taken="formed_party",
                    player_id=player.id,
                    player_name=player.name,
                    bot_goal=bot_goal,
                    direction=None,
                    move_result=None,
                    formed_party=formed_party,
                    battle_result=None,
                    game_concluded=conclusion if conclusion.concluded else None,
                    turn=self._get_turn_info(),
                )

            # 2. Navigate towards target using radar & memory
            occupied = self._get_occupied_coordinates()
            moves_map: Dict[str, MoveCheckResult] = {}
            for name, dx, dy in STANDARD_DIRECTIONS:
                moves_map[name] = self._check_move_internal(player, dx, dy, name, occupied)

            available_dirs = [name for name, chk in moves_map.items() if chk.available]
            if not available_dirs:
                self._advance_turn()
                conclusion = self._check_game_concluded()
                return AiStepResponse(
                    action_taken="passed",
                    player_id=player.id,
                    player_name=player.name,
                    bot_goal=bot_goal,
                    direction=None,
                    move_result=None,
                    formed_party=None,
                    battle_result=None,
                    game_concluded=conclusion if conclusion.concluded else None,
                    turn=self._get_turn_info(),
                )

            # Find nearest target according to goal
            targets = []
            for other in self.players.values():
                if other.id == player.id:
                    continue
                if player.party_id and player.party_id == other.party_id:
                    continue
                dist = max(abs(player.x - other.x), abs(player.y - other.y))
                targets.append((dist, other))

            targets.sort(key=lambda t: t[0])

            if bot_goal == "form_party":
                preferred = [t for t in targets if not t[1].party_id]
                target_bot = preferred[0][1] if preferred else (targets[0][1] if targets else None)
            else:
                preferred = [t for t in targets if t[1].party_id]
                target_bot = preferred[0][1] if preferred else (targets[0][1] if targets else None)

            visited_set = {(loc["x"], loc["y"]) for loc in player.visited_locations}
            best_dir = available_dirs[0]
            best_score = float("-inf")

            for dir_name in available_dirs:
                dx, dy = DIRECTION_OFFSETS[dir_name]
                tx = player.x + dx
                ty = player.y + dy
                score = 0.0

                if target_bot:
                    old_dist = max(abs(player.x - target_bot.x), abs(player.y - target_bot.y))
                    new_dist = max(abs(tx - target_bot.x), abs(ty - target_bot.y))
                    score += (old_dist - new_dist) * 10.0

                if (tx, ty) not in visited_set:
                    score += 2.0

                if score > best_score:
                    best_score = score
                    best_dir = dir_name

            dx, dy = DIRECTION_OFFSETS[best_dir]
            prev_pos = {"x": player.x, "y": player.y}
            affected_players: List[Player] = []

            if player.party_id and player.party_id in self.parties:
                party = self.parties[player.party_id]
                for mid in party.member_ids:
                    m = self.players.get(mid)
                    if m:
                        m.x += dx
                        m.y += dy
                        m.visited_locations.append({"x": m.x, "y": m.y})
                        affected_players.append(m)
            else:
                player.x += dx
                player.y += dy
                player.visited_locations.append({"x": player.x, "y": player.y})
                affected_players.append(player)

            new_pos = {"x": player.x, "y": player.y}

            # Check for adjacent interactions after movement
            formed_party, battle_result = self._check_adjacent_encounter(player)

            self._advance_turn()
            turn_info = self._get_turn_info()
            conclusion = self._check_game_concluded()

            move_res = MoveResponse(
                success=True,
                player=player,
                direction=best_dir,
                party_moved=bool(player.party_id),
                affected_players=affected_players,
                previous_position=prev_pos,
                new_position=new_pos,
                party_formed_triggered=bool(formed_party),
                formed_party=formed_party,
                battle_triggered=bool(battle_result),
                battle_result=battle_result,
                game_concluded=conclusion if conclusion.concluded else None,
                turn=turn_info,
            )

            action_taken = "battled" if battle_result else ("formed_party" if formed_party else "moved")
            return AiStepResponse(
                action_taken=action_taken,
                player_id=player.id,
                player_name=player.name,
                bot_goal=bot_goal,
                direction=best_dir,
                move_result=move_res,
                formed_party=formed_party,
                battle_result=battle_result,
                game_concluded=conclusion if conclusion.concluded else None,
                turn=turn_info,
            )

    # ==========================================\n    # 3-Bout D20 Battle Mechanics
    # ==========================================

    def _find_adjacent_opposing_party(self, party: Party) -> Optional[Party]:
        my_members = [self.players[mid] for mid in party.member_ids if mid in self.players]
        for other_party in self.parties.values():
            if other_party.id == party.id:
                continue
            other_members = [self.players[mid] for mid in other_party.member_ids if mid in self.players]
            for m1 in my_members:
                for m2 in other_members:
                    if self._are_adjacent(m1, m2):
                        return other_party
        return None

    def _resolve_3bout_battle_internal(self, party1: Party, party2: Party) -> BattleResult:
        str1 = sum(self.players[m].strength for m in party1.member_ids if m in self.players)
        str2 = sum(self.players[m].strength for m in party2.member_ids if m in self.players)

        bouts: List[BattleBout] = []
        p1_bouts_won = 0
        p2_bouts_won = 0
        p1_total_score = 0
        p2_total_score = 0

        for i in range(1, 4):
            roll1 = random.randint(1, 20)
            roll2 = random.randint(1, 20)
            score1 = str1 * roll1
            score2 = str2 * roll2

            p1_total_score += score1
            p2_total_score += score2

            if score1 > score2:
                winner_name = party1.name
                p1_bouts_won += 1
            elif score2 > score1:
                winner_name = party2.name
                p2_bouts_won += 1
            else:
                tie_roll1 = random.randint(1, 20)
                tie_roll2 = random.randint(1, 20)
                if tie_roll1 >= tie_roll2:
                    winner_name = party1.name
                    p1_bouts_won += 1
                else:
                    winner_name = party2.name
                    p2_bouts_won += 1

            bouts.append(
                BattleBout(
                    bout_number=i,
                    party1_roll=roll1,
                    party1_strength=str1,
                    party1_score=score1,
                    party2_roll=roll2,
                    party2_strength=str2,
                    party2_score=score2,
                    winner_name=winner_name,
                )
            )

        # Overall battle winner
        if p1_bouts_won > p2_bouts_won:
            winner_party, defeated_party = party1, party2
        elif p2_bouts_won > p1_bouts_won:
            winner_party, defeated_party = party2, party1
        else:
            if p1_total_score >= p2_total_score:
                winner_party, defeated_party = party1, party2
            else:
                winner_party, defeated_party = party2, party1

        # Winning leader receives +2 points, and the rest of the winning party receives +1 point
        for mid in list(winner_party.member_ids):
            member = self.players.get(mid)
            if member:
                if mid == winner_party.leader_id:
                    member.score += 2
                else:
                    member.score += 1

        killed_leader_id = defeated_party.leader_id
        killed_leader = self.players.get(killed_leader_id)
        absorbed_members = []

        if len(defeated_party.member_ids) == 1:
            # Defeated party was a single refusing/solo bot:
            # The party conquered the solo bot: it loses 1 point (-1 pt) and is absorbed into the winning party!
            if killed_leader:
                killed_leader.score -= 1
                killed_leader.party_id = winner_party.id
                killed_leader.is_party_leader = False
                if killed_leader_id not in winner_party.member_ids:
                    winner_party.member_ids.append(killed_leader_id)
                absorbed_members = [killed_leader_id]
                respawn_pos = {"x": killed_leader.x, "y": killed_leader.y}
            else:
                respawn_pos = {"x": 0, "y": 0}
        else:
            # Multi-bot defeated party:
            # Defeated party loses leader: leader gets -1 point, leaves party, and respawns at random position
            if killed_leader:
                killed_leader.score -= 1
                killed_leader.party_id = None
                killed_leader.is_party_leader = False
                rx, ry = self._find_random_free_position()
                killed_leader.x = rx
                killed_leader.y = ry
                killed_leader.visited_locations.append({"x": rx, "y": ry})
                respawn_pos = {"x": rx, "y": ry}
            else:
                respawn_pos = {"x": 0, "y": 0}

            # Remainder of defeated party joins the winning party (scores remain unchanged)
            for mid in list(defeated_party.member_ids):
                if mid != killed_leader_id:
                    m = self.players.get(mid)
                    if m:
                        m.party_id = winner_party.id
                        m.is_party_leader = False
                        if mid not in winner_party.member_ids:
                            winner_party.member_ids.append(mid)
                        absorbed_members.append(mid)

        # Defeated party is dissolved
        if defeated_party.id in self.parties:
            del self.parties[defeated_party.id]

        # Update winning party strength
        self._update_party_strength(winner_party)

        return BattleResult(
            bouts=bouts,
            party1_name=party1.name,
            party2_name=party2.name,
            party1_bouts_won=p1_bouts_won,
            party2_bouts_won=p2_bouts_won,
            party1_total_score=p1_total_score,
            party2_total_score=p2_total_score,
            winner_party_id=winner_party.id,
            winner_party_name=winner_party.name,
            winner_leader_id=winner_party.leader_id,
            winner_leader_name=winner_party.leader_name,
            defeated_party_id=defeated_party.id,
            defeated_party_name=defeated_party.name,
            killed_leader_id=killed_leader_id,
            killed_leader_name=killed_leader.name if killed_leader else "Unknown",
            killed_leader_new_score=killed_leader.score if killed_leader else -1,
            killed_leader_respawn_position=respawn_pos,
            absorbed_members=absorbed_members,
            new_party_size=len(winner_party.member_ids),
            new_party_strength=winner_party.total_strength,
        )

    async def defeat_party(self, party_id: str) -> PartyDefeatResult:
        async with self._lock:
            party = self.parties.get(party_id)
            if not party:
                raise KeyError(f"Party '{party_id}' not found")

            old_leader_id = party.leader_id
            leader = self.players.get(old_leader_id)
            if not leader:
                raise KeyError(f"Leader '{old_leader_id}' not found")

            leader.score -= 1
            leader.party_id = None
            leader.is_party_leader = False
            if old_leader_id in party.member_ids:
                party.member_ids.remove(old_leader_id)

            rx, ry = self._find_random_free_position()
            leader.x = rx
            leader.y = ry
            leader.visited_locations.append({"x": rx, "y": ry})
            respawn_pos = {"x": rx, "y": ry}

            new_leader_id = None
            new_leader_name = None
            party_dissolved = False

            if len(party.member_ids) > 0:
                new_leader_id = random.choice(party.member_ids)
                party.leader_id = new_leader_id
                new_leader = self.players[new_leader_id]
                new_leader.is_party_leader = True
                new_leader_name = new_leader.name
                party.leader_name = new_leader_name
                self._update_party_strength(party)
            else:
                del self.parties[party_id]
                party_dissolved = True

            return PartyDefeatResult(
                party_id=party_id,
                killed_leader_id=leader.id,
                killed_leader_name=leader.name,
                killed_leader_new_score=leader.score,
                killed_leader_respawn_position=respawn_pos,
                new_leader_id=new_leader_id,
                new_leader_name=new_leader_name,
                remaining_members=list(party.member_ids),
                party_dissolved=party_dissolved,
            )

    async def battle(self, challenger_id: str, defender_id: str) -> BattleResult:
        async with self._lock:
            p1 = self.players.get(challenger_id)
            p2 = self.players.get(defender_id)
            if not p1 or not p2:
                raise KeyError("Both combatants must exist")

            if p1.party_id and p1.party_id == p2.party_id:
                raise ValueError("Cannot battle your own party member")

            if not self._are_adjacent(p1, p2):
                raise ValueError(
                    f"Players are not adjacent. Combat distance must be <= 1 (current distance: ({abs(p1.x - p2.x)}, {abs(p1.y - p2.y)}))"
                )

            # Form temporary 1-bot parties if they don't have one
            party1 = self.parties.get(p1.party_id) if p1.party_id else None
            if not party1:
                party1_id = f"party_{uuid.uuid4().hex[:8]}"
                party1 = Party(
                    id=party1_id,
                    name=f"Squad {p1.name}",
                    leader_id=p1.id,
                    leader_name=p1.name,
                    member_ids=[p1.id],
                    total_strength=p1.strength,
                )
                p1.party_id = party1_id
                p1.is_party_leader = True
                self.parties[party1_id] = party1

            party2 = self.parties.get(p2.party_id) if p2.party_id else None
            if not party2:
                party2_id = f"party_{uuid.uuid4().hex[:8]}"
                party2 = Party(
                    id=party2_id,
                    name=f"Squad {p2.name}",
                    leader_id=p2.id,
                    leader_name=p2.name,
                    member_ids=[p2.id],
                    total_strength=p2.strength,
                )
                p2.party_id = party2_id
                p2.is_party_leader = True
                self.parties[party2_id] = party2

            return self._resolve_3bout_battle_internal(party1, party2)


# Global game engine instance
game_engine = GameEngine()