diff --git a/src/board/chess_board/mod.rs b/src/board/chess_board/mod.rs index 33a5580..c594adf 100644 --- a/src/board/chess_board/mod.rs +++ b/src/board/chess_board/mod.rs @@ -170,21 +170,24 @@ impl ChessBoard { } else { self.en_passant = None; } - if move_piece == Piece::King { - *self.castle_rights_mut(self.current_player()) = CastleRights::NoSide; - } - if move_piece == Piece::Rook { - let castle_rights = self.castle_rights_mut(self.current_player()); - *castle_rights = match chess_move.start().file() { - File::A => castle_rights.without_queen_side(), - File::H => castle_rights.without_king_side(), - _ => *castle_rights, - } - } + let castle_rights = self.castle_rights_mut(self.current_player()); + *castle_rights = match (move_piece, chess_move.start().file()) { + (Piece::Rook, File::A) => castle_rights.without_queen_side(), + (Piece::Rook, File::H) => castle_rights.without_king_side(), + (Piece::King, _) => CastleRights::NoSide, + _ => *castle_rights, + }; if let Some(piece) = captured_piece { *self.piece_occupancy_mut(piece) ^= chess_move.destination(); *self.color_occupancy_mut(opponent) ^= chess_move.destination(); self.combined_occupancy ^= chess_move.destination(); + // If a rook is captured, it loses its castling rights + let castle_rights = self.castle_rights_mut(opponent); + *castle_rights = match (piece, chess_move.destination().file()) { + (Piece::Rook, File::A) => castle_rights.without_queen_side(), + (Piece::Rook, File::H) => castle_rights.without_king_side(), + _ => *castle_rights, + }; } // Revertible state modification @@ -768,6 +771,17 @@ mod test { ); } + #[test] + fn do_move_capture_changes_castling() { + let mut position = ChessBoard::from_fen("r3k2r/8/8/8/8/8/8/R3K2R w KQkq - 0 1").unwrap(); + let expected = ChessBoard::from_fen("r3k2R/8/8/8/8/8/8/R3K3 b Qq - 0 1").unwrap(); + + let capture = Move::new(Square::H1, Square::H8, None); + + position.do_move(capture); + assert_eq!(position, expected); + } + #[test] fn do_move_and_undo() { // Start from default position