use crate::{
    board::{BoardSlot, BoardSlotState},
    game::{MillGame, PlayerColor, Stone, StoneState},
};

use anyhow::Result;
use engine::prelude::*;

pub struct SimpleAI {
    pub player_color: PlayerColor,
}

impl SimpleAI {
    pub fn new(player_color: PlayerColor) -> Self {
        Self { player_color }
    }

    pub fn step(
        &self,
        stones: &mut [Stone; 9],
        board: &mut [[[BoardSlot; 3]; 3]; 3],
        scene: &mut SceneHandle,
    ) -> Result<()> {
        if let Some(placable) = stones
            .iter_mut()
            .find(|stone| stone.state == StoneState::ReadyToBePlaced)
        {
            let mut free_slots: Vec<&mut BoardSlot> = board
                .iter_mut()
                .flatten()
                .flatten()
                .filter(|slot| slot.state == BoardSlotState::Empty)
                .collect();

            let len = free_slots.len();

            let slot = &mut free_slots[Random::range(0, len as u32) as usize];

            scene.on_scene(|scene| {
                MillGame::place_stone(placable, slot, self.player_color, scene)?;

                Ok(())
            })?;
        }

        Ok(())
    }
}