bot

bot/src/lib.rs

@@ -0,0 +1,108 @@
+mod bot;
+
+use store::{CheckerMove, Color, Dice, GameEvent, GameState, Player, PlayerId, Stage, TurnStage};
+
+#[derive(Debug)]
+pub struct Bot {
+    pub game: GameState,
+    player_id: PlayerId,
+    color: Color,
+}
+
+impl Default for Bot {
+    fn default() -> Bot {
+        Bot {
+            game: GameState::default(),
+            player_id: 1,
+            color: Color::Black,
+        }
+    }
+}
+
+// impl PlayerEngine for Bot {}
+
+impl Bot {
+    /// new initialize a bot
+    /// # Examples
+    /// ```let mut bot = Bot::new(Color::Black);
+    ///    assert_eq!(bot.game.stage, Stage::PreGame);
+    /// ```
+    pub fn new(color: Color) -> Self {
+        let mut game = GameState::default();
+        game.init_player("p1");
+        game.init_player("p2");
+
+        let player_id = match color {
+            Color::White => 1,
+            Color::Black => 2,
+        };
+
+        Self {
+            game,
+            player_id,
+            color,
+        }
+    }
+
+    pub fn consume(&mut self, event: &GameEvent) -> Option<GameEvent> {
+        self.game.consume(event);
+        // println!("{:?}", self.game);
+        if self.game.active_player_id == self.player_id {
+            return match self.game.turn_stage {
+                TurnStage::RollDice => Some(GameEvent::Roll {
+                    player_id: self.player_id,
+                }),
+                TurnStage::MarkPoints => Some(GameEvent::Mark {
+                    player_id: self.player_id,
+                    points: 0,
+                }),
+                TurnStage::Move => Some(GameEvent::Move {
+                    player_id: self.player_id,
+                    moves: self.choose_move(),
+                }),
+            };
+        }
+        None
+    }
+
+    fn choose_move(&self) -> (CheckerMove, CheckerMove) {
+        let (dice1, dice2) = match self.color {
+            Color::White => self.game.dice.values,
+            Color::Black => (0 - self.game.dice.values.0, 0 - self.game.dice.values.1),
+        };
+
+        let fields = self.game.board.get_color_fields(self.color);
+        let first_field = fields.first().unwrap();
+        (
+            CheckerMove::new(first_field.0, first_field.0 + dice1 as usize).unwrap(),
+            CheckerMove::new(first_field.0, first_field.0 + dice2 as usize).unwrap(),
+        )
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_new() {
+        let bot = Bot::new(Color::Black);
+        assert_eq!(bot.game.stage, Stage::PreGame);
+    }
+
+    #[test]
+    fn test_consume() {
+        let mut bot = Bot::new(Color::Black);
+        let mut event = bot.consume(&GameEvent::BeginGame { goes_first: 2 });
+        assert_eq!(event, Some(GameEvent::Roll { player_id: 2 }));
+
+        event = bot.consume(&GameEvent::BeginGame { goes_first: 1 });
+        assert_eq!(event, None);
+
+        event = bot.consume(&GameEvent::RollResult {
+            player_id: 2,
+            dice: Dice { values: (2, 3) },
+        });
+        assert_eq!(bot.game.turn_stage, TurnStage::MarkPoints);
+    }
+}