manabrew-agent-interface 0.3.3

use forge_foundation::{ManaAtom, ZoneType};
use manabrew_engine::agent::notification::GameNotification;
use manabrew_engine::agent::{
    BinaryChoiceKind, CombatCostAction, GameEntity, ManaCostAction, PlayerAgent,
    PriorityActionSpace, RollSwapChoice, TargetChoice,
};
use manabrew_engine::card::CounterType;
use manabrew_engine::combat::DefenderId;
use manabrew_engine::game::GameState;
use manabrew_engine::ids::{CardId, PlayerId};
use manabrew_engine::mana::ManaPool;
use manabrew_engine::player::actions::player_action::AbilityRef;
use manabrew_engine::player::actions::PlayerAction as EnginePlayerAction;

use crate::game_log_event::GameLogEntryDto;
use crate::game_snapshot_event::GameSnapshotEventDto;
use crate::game_view_dto::{GameViewDto, GameViewDtoExt, StepKind};
use crate::ids_codec::{card_id_str, parse_card_id, parse_player_id, player_id_str};
use crate::mana_action_id::{mana_ability_actions, parse_tap_action_id};
use crate::prompt::*;

mod choices;
mod combat;
mod costs;
mod library;
mod targeting;

/// Match a mana symbol letter (e.g. "U") or a full color name (e.g. "Blue")
/// against a list of color strings.  Handles the Blue/U mismatch where the
/// mana symbol "U" doesn't match the first character of "Blue".
pub(crate) fn find_matching_color<'a>(
    pending: &str,
    colors: impl Iterator<Item = &'a String>,
) -> Option<String> {
    let mana_to_name: &[(&str, &str)] = &[
        ("W", "White"),
        ("U", "Blue"),
        ("B", "Black"),
        ("R", "Red"),
        ("G", "Green"),
        ("C", "Colorless"),
    ];
    colors
        .into_iter()
        .find(|c| {
            // Direct case-insensitive match (covers both "U"=="U" and "Blue"=="Blue")
            c.eq_ignore_ascii_case(pending)
            // Mana symbol → full name lookup (covers "U" matching "Blue")
            || mana_to_name.iter().any(|(sym, name)| {
                pending.eq_ignore_ascii_case(sym) && c.eq_ignore_ascii_case(name)
            })
            // Full name → mana symbol lookup (covers "Blue" matching "U")
            || mana_to_name.iter().any(|(sym, name)| {
                pending.eq_ignore_ascii_case(name) && c.eq_ignore_ascii_case(sym)
            })
        })
        .cloned()
}

pub(crate) fn parse_express_mana_choice(color: Option<&str>) -> Option<u16> {
    color
        .map(|color| ManaAtom::from_name(&color.to_ascii_lowercase()))
        .filter(|&atom| atom != 0)
}

/// Answers the prompts a `PromptAgent` builds.
///
pub trait Responder {
    fn respond(&mut self, prompt: AgentPrompt) -> ClientToServerMessage;
    fn present(&mut self, _message: &AgentMessage) {}
    fn await_ack(&mut self) -> ClientToServerMessage {
        ClientToServerMessage::Response {
            prompt_id: 0,
            action: PromptOutput::DiceRolled(DiceRolledOutput::DiceRolledAcknowledged),
        }
    }
    fn send_log(&mut self, _entry: GameLogEntryDto) {}
    fn send_snapshot(&mut self, _snapshot: GameSnapshotEventDto) {}
}

pub struct PromptAgent<R: Responder> {
    pub player_id: PlayerId,
    pub game_id: String,
    pub responder: R,
    pending_prompt: Option<AgentPrompt>,
    pub(crate) latest_view: Option<GameViewDto>,
    pub(crate) pending_restore_checkpoint: Option<u64>,
    pub pass_until: Option<manabrew_engine::agent::PassUntilTarget>,
    conceded: bool,
    next_prompt_id: u32,
}

impl<R: Responder> PromptAgent<R> {
    pub fn new(player_id: PlayerId, game_id: String, responder: R) -> Self {
        Self {
            player_id,
            game_id,
            responder,
            pending_prompt: None,
            latest_view: None,
            pending_restore_checkpoint: None,
            pass_until: None,
            conceded: false,
            next_prompt_id: 0,
        }
    }

    fn build_prompt(&mut self, inner: PromptInput, source: Option<CardId>) -> AgentPrompt {
        self.next_prompt_id += 1;
        AgentPrompt {
            prompt_id: self.next_prompt_id,
            deciding_player_id: player_id_str(self.player_id),
            source_card_id: source.map(card_id_str),
            input: inner,
        }
    }

    pub(crate) fn send_prompt(&mut self, inner: PromptInput, source: Option<CardId>) {
        let prompt = self.build_prompt(inner, source);
        self.emit_state();
        self.responder
            .present(&AgentMessage::Prompt(prompt.clone()));
        self.pending_prompt = Some(prompt);
    }

    pub(crate) fn recv_action(&mut self) -> PromptOutput {
        let prompt = self
            .pending_prompt
            .take()
            .expect("recv_action called without a pending prompt");
        if self.conceded {
            return Self::default_pass();
        }
        loop {
            match self.responder.respond(prompt.clone()) {
                ClientToServerMessage::Response { prompt_id: 0, .. } => {
                    return Self::default_pass();
                }
                ClientToServerMessage::Response { prompt_id, action } => {
                    if prompt_id != prompt.prompt_id {
                        self.reject(
                            &prompt,
                            ProtocolErrorCode::StalePrompt,
                            format!(
                                "response for prompt {prompt_id}, open prompt is {}",
                                prompt.prompt_id
                            ),
                        );
                        continue;
                    }
                    match prompt.input.validate_response(&action) {
                        Ok(()) => return action,
                        Err(ResponseViolation::WrongPromptType) => {
                            self.reject(
                                &prompt,
                                ProtocolErrorCode::WrongPromptType,
                                "response output does not match the prompt type".to_string(),
                            );
                        }
                        Err(ResponseViolation::UnknownActionId(id)) => {
                            self.reject(
                                &prompt,
                                ProtocolErrorCode::UnknownActionId,
                                format!("action id {id:?} was not advertised by the prompt"),
                            );
                        }
                    }
                }
                ClientToServerMessage::Directive { directive } => {
                    self.handle_directive(directive);
                    return Self::default_pass();
                }
            }
        }
    }

    fn default_pass() -> PromptOutput {
        PromptOutput::ChooseAction(ChooseActionOutput::Pass {
            until: None,
            exhaust_stack: false,
        })
    }

    fn reject(&mut self, prompt: &AgentPrompt, code: ProtocolErrorCode, message: String) {
        self.responder.present(&AgentMessage::Error(ProtocolError {
            code,
            message,
            prompt_id: Some(prompt.prompt_id),
        }));
        self.responder
            .present(&AgentMessage::Prompt(prompt.clone()));
    }

    fn handle_directive(&mut self, directive: DirectiveInput) {
        match directive {
            DirectiveInput::Concede => self.conceded = true,
        }
    }

    pub(crate) fn present_prompt(&mut self, inner: PromptInput, source: Option<CardId>) {
        let prompt = self.build_prompt(inner, source);
        self.emit_state();
        self.responder.present(&AgentMessage::Prompt(prompt));
    }

    pub(crate) fn emit_state(&mut self) {
        let game_view = self.view();
        self.responder
            .present(&AgentMessage::State(StateUpdate { game_view }));
    }

    pub(crate) fn emit_display(&mut self, event: DisplayEvent) {
        self.responder.present(&AgentMessage::Display(event));
    }

    pub(crate) fn view(&self) -> GameViewDto {
        self.latest_view.clone().unwrap_or_else(|| GameViewDto {
            game_id: self.game_id.clone(),
            step: StepKind::Main1,
            ..Default::default()
        })
    }

    pub(crate) fn card_ids(cards: &[CardId]) -> Vec<String> {
        cards.iter().map(|&c| card_id_str(c)).collect()
    }

    pub(crate) fn player_ids(players: &[PlayerId]) -> Vec<String> {
        players.iter().map(|&p| player_id_str(p)).collect()
    }

    pub(crate) fn attack_targets_to_dtos(
        defenders: &[DefenderId],
    ) -> Vec<crate::prompt::AttackTargetDto> {
        use crate::prompt::AttackTargetKind;
        defenders
            .iter()
            .map(|d| match d {
                DefenderId::Player(pid) => crate::prompt::AttackTargetDto {
                    id: format!("player-{}", pid.0),
                    label: format!("Player {}", pid.0),
                    kind: AttackTargetKind::Player,
                },
                // The Rust engine can't distinguish a planeswalker from a
                // battle yet; approximate any permanent target as a walker.
                DefenderId::Permanent(cid) => crate::prompt::AttackTargetDto {
                    id: format!("card-{}", cid.0),
                    label: format!("Permanent {}", cid.0),
                    kind: AttackTargetKind::Planeswalker,
                },
            })
            .collect()
    }

    fn alt_cost_kind(alt: manabrew_engine::spellability::AlternativeCost) -> AlternativeCostKind {
        use manabrew_engine::spellability::AlternativeCost as A;
        match alt {
            A::Flashback => AlternativeCostKind::Flashback,
            A::Spectacle => AlternativeCostKind::Spectacle,
            A::Evoke => AlternativeCostKind::Evoke,
            A::Dash => AlternativeCostKind::Dash,
            A::Blitz => AlternativeCostKind::Blitz,
            A::Escape => AlternativeCostKind::Escape,
            A::Overload => AlternativeCostKind::Overload,
            A::Madness => AlternativeCostKind::Madness,
            A::Foretell => AlternativeCostKind::Foretell,
            A::Emerge => AlternativeCostKind::Emerge,
            A::Suspend => AlternativeCostKind::Suspend,
            A::Morph => AlternativeCostKind::Morph,
            A::Megamorph => AlternativeCostKind::Megamorph,
            A::Bestow => AlternativeCostKind::Bestow,
            A::Warp => AlternativeCostKind::Warp,
            A::SacrificeAlt => AlternativeCostKind::SacrificeAlt,
            A::Plot => AlternativeCostKind::Plot,
            A::Awaken => AlternativeCostKind::Awaken,
            A::Disturb => AlternativeCostKind::Disturb,
            A::Harmonize => AlternativeCostKind::Harmonize,
            A::Freerunning => AlternativeCostKind::Freerunning,
            A::Impending => AlternativeCostKind::Impending,
            A::Mayhem => AlternativeCostKind::Mayhem,
            A::MTMtE => AlternativeCostKind::MTMtE,
            A::Mutate => AlternativeCostKind::Mutate,
            A::Prowl => AlternativeCostKind::Prowl,
            A::Sneak => AlternativeCostKind::Sneak,
            A::Surge => AlternativeCostKind::Surge,
            A::WebSlinging => AlternativeCostKind::WebSlinging,
            A::Plotted => AlternativeCostKind::Plotted,
        }
    }

    fn play_mode_dto(mode: &manabrew_engine::agent::PlayCardMode) -> (PlayCardMode, String) {
        use manabrew_engine::agent::PlayCardMode as E;
        match mode {
            E::Normal => (PlayCardMode::Normal, "Cast normally".to_string()),
            E::BackFaceLand => (
                PlayCardMode::BackFaceLand,
                "Play back face as land".to_string(),
            ),
            E::RoomRightSplit => (PlayCardMode::RoomRightSplit, "Cast right room".to_string()),
            E::StaticAlternative => (
                PlayCardMode::StaticAlternative,
                "Cast with alternative cost".to_string(),
            ),
            E::ForetellExile => (
                PlayCardMode::ForetellExile,
                "Foretell (exile face-down)".to_string(),
            ),
            E::UnlockDoor => (PlayCardMode::UnlockDoor, "Unlock door".to_string()),
            E::Alternative(alt) => (
                PlayCardMode::Alternative {
                    cost: Self::alt_cost_kind(*alt),
                },
                format!("Cast with {alt:?}"),
            ),
        }
    }

    fn play_mode_key(mode: &manabrew_engine::agent::PlayCardMode) -> String {
        use manabrew_engine::agent::PlayCardMode as E;
        match mode {
            E::Normal => "normal".to_string(),
            E::BackFaceLand => "backFaceLand".to_string(),
            E::RoomRightSplit => "roomRightSplit".to_string(),
            E::StaticAlternative => "staticAlternative".to_string(),
            E::ForetellExile => "foretellExile".to_string(),
            E::UnlockDoor => "unlockDoor".to_string(),
            E::Alternative(alt) => format!("alternative:{}", format!("{alt:?}").to_lowercase()),
        }
    }

    fn parse_play_mode(mode_str: &str) -> Option<manabrew_engine::agent::PlayCardMode> {
        use manabrew_engine::agent::PlayCardMode;
        use manabrew_engine::spellability::AlternativeCost as A;
        match mode_str {
            "normal" => Some(PlayCardMode::Normal),
            "backFaceLand" => Some(PlayCardMode::BackFaceLand),
            "staticAlternative" => Some(PlayCardMode::StaticAlternative),
            "foretellExile" => Some(PlayCardMode::ForetellExile),
            "unlockDoor" => Some(PlayCardMode::UnlockDoor),
            "roomRightSplit" => Some(PlayCardMode::RoomRightSplit),
            s if s.starts_with("alternative:") => {
                let alt = match &s["alternative:".len()..] {
                    "flashback" => A::Flashback,
                    "spectacle" => A::Spectacle,
                    "evoke" => A::Evoke,
                    "dash" => A::Dash,
                    "blitz" => A::Blitz,
                    "escape" => A::Escape,
                    "overload" => A::Overload,
                    "madness" => A::Madness,
                    "foretell" => A::Foretell,
                    "emerge" => A::Emerge,
                    "suspend" => A::Suspend,
                    "morph" => A::Morph,
                    "megamorph" => A::Megamorph,
                    "bestow" => A::Bestow,
                    "warp" => A::Warp,
                    "sacrificealt" => A::SacrificeAlt,
                    "plot" => A::Plot,
                    "awaken" => A::Awaken,
                    "disturb" => A::Disturb,
                    "harmonize" => A::Harmonize,
                    "freerunning" => A::Freerunning,
                    "impending" => A::Impending,
                    "mayhem" => A::Mayhem,
                    "mtmte" => A::MTMtE,
                    "mutate" => A::Mutate,
                    "prowl" => A::Prowl,
                    "sneak" => A::Sneak,
                    "surge" => A::Surge,
                    "webslinging" => A::WebSlinging,
                    "plotted" => A::Plotted,
                    _ => return None,
                };
                Some(PlayCardMode::Alternative(alt))
            }
            _ => None,
        }
    }

    pub(crate) fn parse_defender_id(id: &str, possible: &[DefenderId]) -> Option<DefenderId> {
        if let Some(rest) = id.strip_prefix("player-") {
            let idx: u32 = rest.parse().ok()?;
            possible
                .iter()
                .find(|d| matches!(d, DefenderId::Player(p) if p.0 == idx))
                .copied()
        } else if let Some(rest) = id.strip_prefix("card-") {
            let idx: u32 = rest.parse().ok()?;
            possible
                .iter()
                .find(|d| matches!(d, DefenderId::Permanent(c) if c.0 == idx))
                .copied()
        } else {
            None
        }
    }

    pub(crate) fn recv_card_choice_or_first(&mut self, valid: &[CardId]) -> Option<CardId> {
        match self.recv_action() {
            PromptOutput::ChooseBoardTargets(ChooseBoardTargetsOutput::BoardTargets { chosen }) => {
                chosen.into_iter().find_map(|r| match r.kind {
                    TargetKind::Card => parse_card_id(&r.id),
                    _ => None,
                })
            }
            _ => valid.first().copied(),
        }
    }

    pub(crate) fn recv_player_choice_or_first(&mut self, valid: &[PlayerId]) -> Option<PlayerId> {
        match self.recv_action() {
            PromptOutput::ChooseBoardTargets(ChooseBoardTargetsOutput::BoardTargets { chosen }) => {
                chosen.into_iter().find_map(|r| match r.kind {
                    TargetKind::Player => parse_player_id(&r.id),
                    _ => None,
                })
            }
            _ => valid.first().copied(),
        }
    }

    pub(crate) fn recv_spell_choice_or_first(&mut self, valid: &[u32]) -> Option<u32> {
        match self.recv_action() {
            PromptOutput::ChooseBoardTargets(ChooseBoardTargetsOutput::BoardTargets { chosen }) => {
                chosen.into_iter().find_map(|r| match r.kind {
                    TargetKind::Spell => crate::ids_codec::parse_stack_id(&r.id),
                    _ => None,
                })
            }
            _ => valid.first().copied(),
        }
    }
}

impl<R: Responder> PlayerAgent for PromptAgent<R> {
    fn choose_targets_for(
        &mut self,
        sa: &mut manabrew_engine::spellability::SpellAbility,
        game: &GameState,
        mana_pools: &[ManaPool],
    ) -> bool {
        manabrew_engine::spellability::choose_targets_by_kind(self, sa, game, mana_pools)
    }

    fn get_pass_until(&self) -> Option<manabrew_engine::agent::PassUntilTarget> {
        self.pass_until
    }

    fn clear_pass_until(&mut self) {
        self.pass_until = None;
    }

    fn snapshot_state(&mut self, game: &GameState, mana_pools: &[ManaPool]) {
        self.latest_view = Some(GameViewDto::from_engine(
            game,
            mana_pools,
            self.player_id,
            &self.game_id,
        ));
    }

    fn mulligan_decision(
        &mut self,
        player: PlayerId,
        hand: &[CardId],
        mulligan_count: u32,
    ) -> bool {
        choices::mulligan_decision(self, player, hand, mulligan_count)
    }

    fn mulligan_decision_send(&mut self, player: PlayerId, hand: &[CardId], mulligan_count: u32) {
        choices::mulligan_decision_send(self, player, hand, mulligan_count);
    }

    fn mulligan_decision_recv(
        &mut self,
        player: PlayerId,
        hand: &[CardId],
        mulligan_count: u32,
    ) -> bool {
        choices::mulligan_decision_recv(self, player, hand, mulligan_count)
    }

    fn choose_cards_to_bottom(
        &mut self,
        player: PlayerId,
        hand: &[CardId],
        count: usize,
    ) -> Vec<CardId> {
        choices::choose_cards_to_bottom(self, player, hand, count)
    }

    fn choose_cards_to_bottom_send(&mut self, player: PlayerId, hand: &[CardId], count: usize) {
        choices::choose_cards_to_bottom_send(self, player, hand, count);
    }

    fn choose_cards_to_bottom_recv(
        &mut self,
        player: PlayerId,
        hand: &[CardId],
        count: usize,
    ) -> Vec<CardId> {
        choices::choose_cards_to_bottom_recv(self, player, hand, count)
    }

    fn choose_action(
        &mut self,
        _player: PlayerId,
        action_space: Option<&PriorityActionSpace>,
        request_action_space: &mut dyn FnMut() -> PriorityActionSpace,
    ) -> EnginePlayerAction {
        if self.conceded {
            return EnginePlayerAction::Concede;
        }
        let requested_action_space;
        let action_space = match action_space {
            Some(action_space) => action_space,
            None => {
                requested_action_space = request_action_space();
                &requested_action_space
            }
        };
        let playable = &action_space.playable;
        let untappable_lands = &action_space.untappable_lands;
        let _activatable = &action_space.activatable;
        let untappable_land_ids: Vec<String> =
            untappable_lands.iter().map(|&c| card_id_str(c)).collect();

        let mut actions: Vec<AvailableAction> = Vec::new();
        for play in playable.iter() {
            let card_id = card_id_str(play.card_id);
            let (mode, label) = Self::play_mode_dto(&play.mode);
            let key = Self::play_mode_key(&play.mode);
            actions.push(AvailableAction {
                id: format!("cast:{card_id}:{key}"),
                kind: AvailableActionKind::Cast {
                    card_id: card_id.clone(),
                    mode,
                    label,
                },
            });
        }
        for a in action_space
            .activatable
            .iter()
            .chain(action_space.mana_abilities.iter())
        {
            let card_id = card_id_str(a.card_id);
            if a.is_mana_ability {
                actions.extend(
                    mana_ability_actions(
                        &card_id,
                        a.ability_index,
                        &a.description,
                        a.cost.clone(),
                        a.produced_mana.clone(),
                        a.produced_mana_amount,
                    )
                    .into_iter()
                    .map(|(id, info)| AvailableAction {
                        id,
                        kind: AvailableActionKind::ActivateAbility(info),
                    }),
                );
            } else {
                actions.push(AvailableAction {
                    id: format!("ability:{card_id}:{}", a.ability_index),
                    kind: AvailableActionKind::ActivateAbility(ActivatableAbilityInfo {
                        card_id,
                        ability_index: a.ability_index,
                        description: a.description.clone(),
                        cost: a.cost.clone(),
                        is_mana_ability: false,
                        produced_mana: None,
                    }),
                });
            }
        }
        for card_id in &untappable_land_ids {
            actions.push(AvailableAction {
                id: format!("untap:{card_id}"),
                kind: AvailableActionKind::UndoMana {
                    card_id: card_id.clone(),
                },
            });
        }

        self.send_prompt(
            PromptInput::ChooseAction(
                manabrew_protocol::prompts::choose_action::ChooseActionInput { actions },
            ),
            None,
        );
        let prompt = self
            .pending_prompt
            .take()
            .expect("choose_action called without a pending prompt");
        let action = match self.responder.respond(prompt) {
            ClientToServerMessage::Response { action, .. } => action,
            ClientToServerMessage::Directive {
                directive: DirectiveInput::Concede,
            } => return EnginePlayerAction::Concede,
        };
        match action {
            PromptOutput::ChooseAction(ChooseActionOutput::Act { action_id }) => {
                if let Some(rest) = action_id.strip_prefix("cast:") {
                    let (id_part, mode) = rest.split_once(':').unwrap_or((rest, "normal"));
                    let resolved = parse_card_id(id_part).and_then(|cid| {
                        Self::parse_play_mode(mode)
                            .and_then(|m| {
                                playable
                                    .iter()
                                    .copied()
                                    .find(|play| play.card_id == cid && play.mode == m)
                            })
                            .or_else(|| playable.iter().copied().find(|play| play.card_id == cid))
                    });
                    resolved
                        .map(EnginePlayerAction::CastSpell)
                        .unwrap_or(EnginePlayerAction::PassPriority)
                } else if let Some(rest) = action_id.strip_prefix("tap:") {
                    let tap = parse_tap_action_id(rest);
                    match parse_card_id(tap.card_id) {
                        Some(cid) => EnginePlayerAction::ActivateMana(
                            cid,
                            tap.ability_index,
                            parse_express_mana_choice(tap.color),
                        ),
                        None => EnginePlayerAction::PassPriority,
                    }
                } else if let Some(rest) = action_id.strip_prefix("ability:") {
                    let (id_part, idx) = rest.split_once(':').unwrap_or((rest, ""));
                    match (parse_card_id(id_part), idx.parse::<usize>()) {
                        (Some(cid), Ok(ability_index)) => {
                            EnginePlayerAction::ActivateAbility(AbilityRef {
                                card_id: cid,
                                ability_index,
                            })
                        }
                        _ => EnginePlayerAction::PassPriority,
                    }
                } else if let Some(id_part) = action_id.strip_prefix("untap:") {
                    parse_card_id(id_part)
                        .map(EnginePlayerAction::UndoMana)
                        .unwrap_or(EnginePlayerAction::PassPriority)
                } else {
                    EnginePlayerAction::PassPriority
                }
            }
            PromptOutput::ChooseAction(ChooseActionOutput::Pass { until, .. }) => {
                self.pass_until = until.and_then(|u| {
                    Some(manabrew_engine::agent::PassUntilTarget {
                        player: crate::ids_codec::parse_player_id(&u.player_id)?,
                        phase: crate::game_view_dto::step_to_phase(u.phase),
                    })
                });
                EnginePlayerAction::PassPriority
            }
            PromptOutput::ChooseAction(ChooseActionOutput::RestoreSnapshot { checkpoint_id }) => {
                self.pending_restore_checkpoint = Some(checkpoint_id);
                EnginePlayerAction::PassPriority
            }
            _ => EnginePlayerAction::PassPriority,
        }
    }

    fn choose_attackers(
        &mut self,
        player: PlayerId,
        available: &[CardId],
        possible_defenders: &[DefenderId],
    ) -> Vec<(CardId, DefenderId)> {
        combat::choose_attackers(self, player, available, possible_defenders)
    }

    fn choose_blockers(
        &mut self,
        player: PlayerId,
        attackers: &[CardId],
        available_blockers: &[CardId],
        max_blockers: Option<usize>,
    ) -> Vec<(CardId, CardId)> {
        combat::choose_blockers(self, player, attackers, available_blockers, max_blockers)
    }

    fn choose_damage_assignment_order(
        &mut self,
        player: PlayerId,
        attacker: CardId,
        blockers: &[CardId],
    ) -> Vec<CardId> {
        combat::choose_damage_assignment_order(self, player, attacker, blockers)
    }

    fn assign_combat_damage(
        &mut self,
        game: &GameState,
        player: PlayerId,
        attacker: CardId,
        blockers_in_order: &[CardId],
        defender_id: Option<DefenderId>,
        damage_to_assign: i32,
    ) -> Vec<(Option<CardId>, i32)> {
        let attacker_has_deathtouch = game.card(attacker).has_deathtouch();
        combat::choose_combat_damage_assignment(
            self,
            player,
            attacker,
            blockers_in_order,
            defender_id,
            damage_to_assign,
            attacker_has_deathtouch,
        )
    }

    fn choose_target_player(
        &mut self,
        player: PlayerId,
        valid: &[PlayerId],
        sa: Option<&manabrew_engine::spellability::SpellAbility>,
    ) -> Option<PlayerId> {
        let source = sa.and_then(|s| s.source);
        let intent = sa
            .map(crate::game_view_dto::targeting_intent_of)
            .unwrap_or(crate::game_view_dto::TargetingIntent::Hostile);
        let hostile = crate::game_view_dto::intent_is_hostile(intent);
        targeting::choose_target_player(self, player, valid, source, hostile, intent)
    }

    fn choose_target_card(
        &mut self,
        player: PlayerId,
        valid: &[CardId],
        sa: Option<&manabrew_engine::spellability::SpellAbility>,
    ) -> Option<CardId> {
        let source = sa.and_then(|s| s.source);
        let intent = sa
            .map(crate::game_view_dto::targeting_intent_of)
            .unwrap_or(crate::game_view_dto::TargetingIntent::Hostile);
        let hostile = crate::game_view_dto::intent_is_hostile(intent);
        targeting::choose_target_card(self, player, valid, source, hostile, intent)
    }

    fn choose_target_card_from_zone(
        &mut self,
        player: PlayerId,
        zone: ZoneType,
        valid: &[CardId],
        sa: Option<&manabrew_engine::spellability::SpellAbility>,
    ) -> Option<CardId> {
        let source = sa.and_then(|s| s.source);
        let intent = sa
            .map(crate::game_view_dto::targeting_intent_of)
            .unwrap_or(crate::game_view_dto::TargetingIntent::Hostile);
        let hostile = crate::game_view_dto::intent_is_hostile(intent);
        targeting::choose_target_card_from_zone(self, player, zone, valid, source, hostile, intent)
    }

    fn choose_target_any(
        &mut self,
        player: PlayerId,
        valid_players: &[PlayerId],
        valid_cards: &[CardId],
        sa: Option<&manabrew_engine::spellability::SpellAbility>,
    ) -> TargetChoice {
        let source = sa.and_then(|s| s.source);
        let intent = sa
            .map(crate::game_view_dto::targeting_intent_of)
            .unwrap_or(crate::game_view_dto::TargetingIntent::Hostile);
        let hostile = crate::game_view_dto::intent_is_hostile(intent);
        targeting::choose_target_any(
            self,
            player,
            valid_players,
            valid_cards,
            source,
            hostile,
            intent,
        )
    }

    fn choose_sacrifice(
        &mut self,
        player: PlayerId,
        valid: &[CardId],
        source: Option<CardId>,
    ) -> Option<CardId> {
        targeting::choose_sacrifice(self, player, valid, source)
    }

    fn reveal_cards(
        &mut self,
        game: &GameState,
        _player: PlayerId,
        cards: &[CardId],
        zone: ZoneType,
        owner: PlayerId,
        message_prefix: Option<&str>,
    ) {
        choices::reveal_cards(self, game, cards, zone, owner, message_prefix)
    }

    fn choose_scry(
        &mut self,
        game: &GameState,
        player: PlayerId,
        source: Option<CardId>,
        cards: &[CardId],
    ) -> Vec<Vec<CardId>> {
        library::choose_scry(self, game, player, source, cards)
    }

    fn choose_surveil(
        &mut self,
        game: &GameState,
        player: PlayerId,
        source: Option<CardId>,
        cards: &[CardId],
    ) -> Vec<Vec<CardId>> {
        library::choose_surveil(self, game, player, source, cards)
    }

    fn choose_dig(
        &mut self,
        game: &GameState,
        player: PlayerId,
        valid: &[CardId],
        max: usize,
        optional: bool,
    ) -> Vec<CardId> {
        library::choose_dig(self, game, player, valid, max, optional)
    }

    fn choose_discard(&mut self, player: PlayerId, hand: &[CardId], num: usize) -> Vec<CardId> {
        choices::choose_discard(self, player, hand, num)
    }

    fn choose_discard_any_number(
        &mut self,
        player: PlayerId,
        hand: &[CardId],
        min: usize,
        max: usize,
    ) -> Vec<CardId> {
        choices::choose_discard_any_number(self, player, hand, min, max)
    }

    fn choose_legend_keep(&mut self, player: PlayerId, duplicates: &[CardId]) -> CardId {
        choices::choose_legend_keep(self, player, duplicates)
    }

    fn choose_target_spell(
        &mut self,
        player: PlayerId,
        valid: &[u32],
        source: Option<CardId>,
    ) -> Option<u32> {
        targeting::choose_target_spell(self, player, valid, source)
    }

    fn choose_mode(
        &mut self,
        player: PlayerId,
        descriptions: &[String],
        min: usize,
        max: usize,
        source_card_id: Option<CardId>,
    ) -> Vec<usize> {
        choices::choose_mode(self, player, descriptions, min, max, source_card_id)
    }

    fn choose_spell_abilities_for_effect(
        &mut self,
        player: PlayerId,
        abilities: &[manabrew_engine::spellability::SpellAbility],
        num: usize,
    ) -> Vec<usize> {
        choices::choose_spell_abilities_for_effect(self, player, abilities, num)
    }

    fn get_ability_to_play(
        &mut self,
        player: PlayerId,
        abilities: &[manabrew_engine::spellability::SpellAbility],
    ) -> Option<usize> {
        choices::get_ability_to_play(self, player, abilities)
    }

    fn choose_single_entity_for_effect(
        &mut self,
        player: PlayerId,
        valid: &[GameEntity],
        is_optional: bool,
    ) -> Option<GameEntity> {
        choices::choose_single_entity_for_effect(self, player, valid, is_optional)
    }

    fn choose_entities_for_effect(
        &mut self,
        player: PlayerId,
        candidates: &[GameEntity],
        min: usize,
        max: usize,
    ) -> Vec<GameEntity> {
        choices::choose_entities_for_effect(self, player, candidates, min, max)
    }

    fn choose_single_replacement_effect(
        &mut self,
        player: PlayerId,
        descriptions: &[String],
    ) -> usize {
        choices::choose_single_replacement_effect(self, player, descriptions)
    }

    fn confirm_replacement_effect(
        &mut self,
        player: PlayerId,
        question: &str,
        effect_description: &str,
        source: Option<CardId>,
    ) -> bool {
        choices::confirm_replacement_effect(self, player, question, effect_description, source)
    }

    fn choose_optional_trigger(
        &mut self,
        player: PlayerId,
        description: &str,
        source: Option<CardId>,
        api: Option<manabrew_engine::ability::api_type::ApiType>,
    ) -> bool {
        choices::choose_optional_trigger(self, player, description, source, api)
    }

    fn confirm_action(
        &mut self,
        player: PlayerId,
        mode: Option<&str>,
        message: &str,
        options: &[String],
        source: Option<CardId>,
        api: Option<manabrew_engine::ability::api_type::ApiType>,
    ) -> bool {
        choices::confirm_action(self, player, mode, message, options, source, api)
    }

    fn confirm_payment(
        &mut self,
        player: PlayerId,
        cost_kind: &str,
        message: &str,
        source: Option<CardId>,
        api: Option<manabrew_engine::ability::api_type::ApiType>,
    ) -> bool {
        choices::confirm_payment(self, player, cost_kind, message, source, api)
    }

    fn pay_cost_to_prevent_effect(
        &mut self,
        player: PlayerId,
        cost_kind: &str,
        message: &str,
        source: Option<CardId>,
        api: Option<manabrew_engine::ability::api_type::ApiType>,
        can_pay: bool,
        targets: &[manabrew_engine::agent::GameEntity],
        effect_text: &str,
    ) -> bool {
        choices::pay_cost_to_prevent_effect(
            self,
            player,
            cost_kind,
            message,
            source,
            api,
            can_pay,
            targets,
            effect_text,
        )
    }

    fn choose_binary(
        &mut self,
        player: PlayerId,
        question: &str,
        kind: BinaryChoiceKind,
        default_choice: Option<bool>,
        source: Option<CardId>,
        api: Option<manabrew_engine::ability::api_type::ApiType>,
    ) -> bool {
        choices::choose_binary(self, player, question, kind, default_choice, source, api)
    }

    fn choose_phyrexian_pay_life(
        &mut self,
        player: PlayerId,
        color: &str,
        source: Option<CardId>,
    ) -> bool {
        costs::choose_phyrexian_pay_life(self, player, color, source)
    }

    fn choose_kicker(
        &mut self,
        player: PlayerId,
        kicker_cost: &str,
        source: Option<CardId>,
    ) -> bool {
        costs::choose_kicker(self, player, kicker_cost, source)
    }

    fn choose_buyback(
        &mut self,
        player: PlayerId,
        buyback_cost: &str,
        source: Option<CardId>,
    ) -> bool {
        costs::choose_buyback(self, player, buyback_cost, source)
    }

    fn choose_multikicker(
        &mut self,
        player: PlayerId,
        cost: &str,
        max_kicks: u32,
        source: Option<CardId>,
    ) -> u32 {
        costs::choose_multikicker(self, player, cost, max_kicks, source)
    }

    fn choose_replicate(
        &mut self,
        player: PlayerId,
        cost: &str,
        max_replicates: u32,
        source: Option<CardId>,
    ) -> u32 {
        costs::choose_replicate(self, player, cost, max_replicates, source)
    }

    fn choose_color(&mut self, player: PlayerId, valid_colors: &[String]) -> Option<String> {
        choices::choose_color(self, player, valid_colors)
    }

    fn choose_colors(
        &mut self,
        player: PlayerId,
        valid_colors: &[String],
        min: usize,
        max: usize,
    ) -> Vec<String> {
        choices::choose_colors(self, player, valid_colors, min, max)
    }

    fn choose_cards_for_effect(
        &mut self,
        player: PlayerId,
        valid: &[CardId],
        min: usize,
        max: usize,
    ) -> Vec<CardId> {
        choices::choose_cards_for_effect(self, player, valid, min, max)
    }

    fn choose_single_card_for_zone_change(
        &mut self,
        game: &GameState,
        player: PlayerId,
        valid: &[CardId],
        select_prompt: &str,
        is_optional: bool,
    ) -> Option<CardId> {
        choices::choose_single_card_for_zone_change(
            self,
            game,
            player,
            valid,
            select_prompt,
            is_optional,
        )
    }

    fn choose_cards_for_zone_change(
        &mut self,
        game: &GameState,
        player: PlayerId,
        valid: &[CardId],
        min: usize,
        max: usize,
        select_prompt: &str,
    ) -> Vec<CardId> {
        choices::choose_cards_for_zone_change(self, game, player, valid, min, max, select_prompt)
    }

    fn choose_type(
        &mut self,
        player: PlayerId,
        type_category: &str,
        valid_types: &[String],
    ) -> Option<String> {
        choices::choose_type(self, player, type_category, valid_types)
    }

    fn choose_counter_type(
        &mut self,
        player: PlayerId,
        options: &[CounterType],
        prompt: &str,
    ) -> Option<CounterType> {
        choices::choose_counter_type(self, player, options, prompt)
    }

    fn choose_card_name(&mut self, player: PlayerId, valid_names: &[String]) -> Option<String> {
        choices::choose_card_name(self, player, valid_names)
    }

    fn choose_number(
        &mut self,
        player: PlayerId,
        source: Option<CardId>,
        title: &str,
        description: Option<&str>,
        min: i32,
        max: i32,
    ) -> Option<i32> {
        choices::choose_number(self, player, source, title, description, min, max)
    }

    fn choose_number_from_list(
        &mut self,
        player: PlayerId,
        choices: &[i32],
        message: &str,
        source_card_id: Option<CardId>,
    ) -> Option<i32> {
        choices::choose_number_from_list(self, player, choices, message, source_card_id)
    }

    fn choose_roll_to_ignore(
        &mut self,
        player: PlayerId,
        rolls: &[i32],
        source: Option<CardId>,
    ) -> Option<i32> {
        choices::choose_roll_to_ignore(self, player, rolls, source)
    }

    fn choose_roll_to_swap(
        &mut self,
        player: PlayerId,
        rolls: &[i32],
        source: Option<CardId>,
    ) -> Option<i32> {
        choices::choose_roll_to_swap(self, player, rolls, source)
    }

    fn choose_dice_to_reroll(
        &mut self,
        player: PlayerId,
        rolls: &[i32],
        source: Option<CardId>,
    ) -> Vec<i32> {
        choices::choose_dice_to_reroll(self, player, rolls, source)
    }

    fn choose_roll_to_modify(
        &mut self,
        player: PlayerId,
        rolls: &[i32],
        source: Option<CardId>,
    ) -> Option<i32> {
        choices::choose_roll_to_modify(self, player, rolls, source)
    }

    fn choose_roll_swap_value(
        &mut self,
        player: PlayerId,
        current_result: i32,
        power: i32,
        toughness: i32,
        source: Option<CardId>,
    ) -> Option<RollSwapChoice> {
        choices::choose_roll_swap_value(self, player, current_result, power, toughness, source)
    }

    fn flip_coin_call(&mut self, player: PlayerId) -> bool {
        choices::flip_coin_call(self, player)
    }

    fn pay_combat_cost(
        &mut self,
        player: PlayerId,
        attacker: CardId,
        cost: i32,
        description: &str,
        mana_ability_options: &[manabrew_engine::agent::ManaAbilityOption],
        tappable_lands: &[CardId],
        untappable_lands: &[CardId],
        mana_pool_total: i32,
    ) -> CombatCostAction {
        combat::pay_combat_cost(
            self,
            player,
            attacker,
            cost,
            description,
            mana_ability_options,
            tappable_lands,
            untappable_lands,
            mana_pool_total,
        )
    }

    fn choose_improvise(
        &mut self,
        player: PlayerId,
        untapped_artifacts: &[CardId],
        remaining_cost: &forge_foundation::ManaCost,
        source: Option<CardId>,
    ) -> Vec<CardId> {
        costs::choose_improvise(self, player, untapped_artifacts, remaining_cost, source)
    }

    fn choose_convoke(
        &mut self,
        player: PlayerId,
        untapped_creatures: &[CardId],
        remaining_cost: &forge_foundation::ManaCost,
        source: Option<CardId>,
    ) -> Vec<CardId> {
        costs::choose_convoke(self, player, untapped_creatures, remaining_cost, source)
    }

    fn pay_mana_cost(
        &mut self,
        player: PlayerId,
        card_id: CardId,
        card_name: &str,
        mana_cost: &str,
        mana_cost_display: &str,
        _mana_cost_checkpoint: &str,
        can_confirm_from_pool: bool,
        _allow_reserved_source_reuse: bool,
        _reserved_sacrifices: &[CardId],
        mana_ability_options: &[manabrew_engine::agent::ManaAbilityOption],
        tappable_lands: &[CardId],
        untappable_lands: &[CardId],
        mana_pool: &ManaPool,
    ) -> ManaCostAction {
        costs::pay_mana_cost(
            self,
            player,
            card_id,
            card_name,
            mana_cost,
            mana_cost_display,
            can_confirm_from_pool,
            mana_ability_options,
            tappable_lands,
            untappable_lands,
            mana_pool,
        )
    }

    fn await_display_ack(&mut self) {
        if self.conceded {
            return;
        }
        if let ClientToServerMessage::Directive { directive } = self.responder.await_ack() {
            self.handle_directive(directive);
        }
    }

    fn specify_mana_combo(
        &mut self,
        player: PlayerId,
        available_colors: &[String],
        amount: usize,
        source: Option<CardId>,
        express_choice: Option<u16>,
    ) -> Vec<String> {
        costs::specify_mana_combo(
            self,
            player,
            available_colors,
            amount,
            source,
            express_choice,
        )
    }

    fn exert_attackers(&mut self, player: PlayerId, attackers: &[CardId]) -> Vec<CardId> {
        combat::exert_attackers(self, player, attackers)
    }

    fn enlist_attackers(&mut self, player: PlayerId, attackers: &[CardId]) -> Vec<CardId> {
        combat::enlist_attackers(self, player, attackers)
    }

    fn choose_reorder_library(
        &mut self,
        game: &GameState,
        player: PlayerId,
        cards: &[CardId],
    ) -> Vec<CardId> {
        library::choose_reorder_library(self, game, player, cards)
    }

    fn help_pay_assist(&mut self, player: PlayerId, card_name: &str, max_generic: u32) -> u32 {
        choices::help_pay_assist(self, player, card_name, max_generic)
    }

    fn choose_random_discard(
        &mut self,
        player: PlayerId,
        hand: &[CardId],
        num: usize,
    ) -> Vec<CardId> {
        choices::choose_random_discard(self, player, hand, num)
    }

    fn choose_land_or_spell(&mut self, player: PlayerId) -> Option<bool> {
        choices::choose_land_or_spell(self, player)
    }

    fn notify(&mut self, event: GameNotification) {
        match event {
            GameNotification::Event(log_event) => {
                self.responder
                    .send_log(GameLogEntryDto::from_event(log_event));
            }
            GameNotification::CardPlayed {
                player,
                card_id,
                card_name,
                set_code,
            } => {
                self.emit_display(DisplayEvent::CardPlayed {
                    card_id: card_id_str(card_id),
                    card_name,
                    set_code,
                    player_id: player_id_str(player),
                });
                self.emit_state();
            }
            GameNotification::TurnChanged {
                active_player,
                turn_number,
            } => {
                let player_id = player_id_str(active_player);
                let active_player_name = self
                    .latest_view
                    .as_ref()
                    .and_then(|v| v.players.iter().find(|p| p.id == player_id))
                    .map(|p| p.name.clone())
                    .unwrap_or_else(|| format!("Player {}", active_player.0));
                self.responder.send_log(GameLogEntryDto::from_event(
                    manabrew_engine::agent::GameLogEvent::rule(format!(
                        "TURN {} — {}",
                        turn_number, active_player_name
                    ))
                    .with_player(active_player),
                ));
                self.emit_display(DisplayEvent::TurnChanged {
                    active_player_id: player_id,
                    active_player_name,
                    turn_number,
                });
                self.emit_state();
            }
            GameNotification::PhaseChanged { .. } | GameNotification::StateChanged => {
                self.emit_state();
            }
            GameNotification::PriorityChanged { .. } => {
                self.emit_state();
            }
            GameNotification::FirstPlayerRoll {
                sides,
                rolls,
                winner,
            } => {
                let view = self.view();
                let winner_id = player_id_str(winner);
                let entries = rolls
                    .into_iter()
                    .map(|(pid, value)| {
                        let id = player_id_str(pid);
                        let name = view
                            .players
                            .iter()
                            .find(|p| p.id == id)
                            .map(|p| p.name.clone())
                            .unwrap_or_else(|| id.clone());
                        manabrew_protocol::prompts::dice_rolled::DiceRollEntry {
                            label: Some(name),
                            highlighted: id == winner_id,
                            player_id: Some(id),
                            natural_results: vec![value],
                            final_results: vec![value],
                            ignored_rolls: vec![],
                        }
                    })
                    .collect();
                self.present_prompt(
                    PromptInput::DiceRolled(
                        manabrew_protocol::prompts::dice_rolled::DiceRolledInput {
                            presentation: PromptPresentation {
                                title: "Roll for first player".to_string(),
                                description: None,
                                text: None,
                                source_card_id: None,
                                targets: Vec::new(),
                            },
                            sides,
                            rolls: entries,
                            source_card_name: None,
                        },
                    ),
                    None,
                );
                // Caller is responsible for `await_display_ack` after the
                // full broadcast — see `roll_for_first_player`.
            }
            GameNotification::DiceRolled {
                player,
                sides,
                natural_results,
                final_results,
                ignored_rolls,
                source_card_name,
            } => {
                // Send the prompt to every agent's transport. The caller
                // is responsible for issuing a parallel `await_display_ack`
                // pass after broadcasting — that way all clients see the
                // animation start at the same time and we wait once for
                // the slowest player rather than serially per-agent.
                self.present_prompt(
                    PromptInput::DiceRolled(
                        manabrew_protocol::prompts::dice_rolled::DiceRolledInput {
                            presentation: PromptPresentation {
                                title: "Dice roll".to_string(),
                                description: None,
                                text: None,
                                source_card_id: None,
                                targets: Vec::new(),
                            },
                            sides,
                            rolls: vec![manabrew_protocol::prompts::dice_rolled::DiceRollEntry {
                                label: None,
                                player_id: Some(player_id_str(player)),
                                natural_results,
                                final_results,
                                ignored_rolls,
                                highlighted: false,
                            }],
                            source_card_name,
                        },
                    ),
                    None,
                );
            }
            GameNotification::SnapshotCreated {
                checkpoint_id,
                label,
            } => {
                if let Some(view) = self.latest_view.clone() {
                    self.responder.send_snapshot(GameSnapshotEventDto::new(
                        checkpoint_id,
                        label,
                        view,
                    ));
                }
            }
            GameNotification::GameOver => {
                self.emit_state();
                self.present_prompt(
                    PromptInput::GameOver(manabrew_protocol::prompts::game_over::GameOverInput {}),
                    None,
                );
            }
            GameNotification::ManaPaymentResolved { .. } => {}
            GameNotification::ActivatedAbilityPaymentFailed { .. } => {
                self.emit_state();
            }
        }
    }

    fn take_restore_request(&mut self) -> Option<u64> {
        self.pending_restore_checkpoint.take()
    }
}