105 lines
3.6 KiB
Go
105 lines
3.6 KiB
Go
package chess
|
|
|
|
import "local/m/mchess_server/types"
|
|
|
|
type Board map[types.Coordinate]types.Piece
|
|
|
|
func (b Board) Init() {
|
|
var coord types.Coordinate
|
|
|
|
for i := 1; i <= 8; i++ {
|
|
coord.Row = 2
|
|
coord.Col = i
|
|
b[coord] = types.Piece{Class: types.Pawn, Color: types.White}
|
|
|
|
coord.Row = 7
|
|
coord.Col = i
|
|
b[coord] = types.Piece{Class: types.Pawn, Color: types.Black}
|
|
}
|
|
|
|
b[types.Coordinate{Row: 1, Col: 1}] = types.Piece{Class: types.Rook, Color: types.White}
|
|
b[types.Coordinate{Row: 1, Col: 2}] = types.Piece{Class: types.Knight, Color: types.White}
|
|
b[types.Coordinate{Row: 1, Col: 3}] = types.Piece{Class: types.Bishop, Color: types.White}
|
|
b[types.Coordinate{Row: 1, Col: 4}] = types.Piece{Class: types.Queen, Color: types.White}
|
|
b[types.Coordinate{Row: 1, Col: 5}] = types.Piece{Class: types.King, Color: types.White}
|
|
b[types.Coordinate{Row: 1, Col: 6}] = types.Piece{Class: types.Bishop, Color: types.White}
|
|
b[types.Coordinate{Row: 1, Col: 7}] = types.Piece{Class: types.Knight, Color: types.White}
|
|
b[types.Coordinate{Row: 1, Col: 8}] = types.Piece{Class: types.Rook, Color: types.White}
|
|
|
|
b[types.Coordinate{Row: 8, Col: 1}] = types.Piece{Class: types.Rook, Color: types.Black}
|
|
b[types.Coordinate{Row: 8, Col: 2}] = types.Piece{Class: types.Knight, Color: types.Black}
|
|
b[types.Coordinate{Row: 8, Col: 3}] = types.Piece{Class: types.Bishop, Color: types.Black}
|
|
b[types.Coordinate{Row: 8, Col: 4}] = types.Piece{Class: types.Queen, Color: types.Black}
|
|
b[types.Coordinate{Row: 8, Col: 5}] = types.Piece{Class: types.King, Color: types.Black}
|
|
b[types.Coordinate{Row: 8, Col: 6}] = types.Piece{Class: types.Bishop, Color: types.Black}
|
|
b[types.Coordinate{Row: 8, Col: 7}] = types.Piece{Class: types.Knight, Color: types.Black}
|
|
b[types.Coordinate{Row: 8, Col: 8}] = types.Piece{Class: types.Rook, Color: types.Black}
|
|
}
|
|
|
|
func (b Board) GetPieceAt(coord types.Coordinate) (types.Piece, bool) {
|
|
piece, found := b[coord]
|
|
|
|
if !found {
|
|
piece = types.Piece{}
|
|
}
|
|
|
|
return piece, found
|
|
}
|
|
|
|
func (b Board) CheckMove(move types.Move) (bool, string) {
|
|
pieceAtStartSquare, found := b.GetPieceAt(move.StartSquare)
|
|
if !found {
|
|
return false, "no piece at start square"
|
|
}
|
|
movingColor := pieceAtStartSquare.Color
|
|
|
|
pieceAtEndSquare, found := b.GetPieceAt(move.EndSquare)
|
|
if found {
|
|
if pieceAtEndSquare.Color == pieceAtStartSquare.Color {
|
|
return false, "same-coloured piece at end square"
|
|
}
|
|
}
|
|
// At the moment, we do not need to check if the correct color is moving,
|
|
//since we are only reading moves from the player whose turn it is.
|
|
|
|
//Check if king of moving color is in check -> move not allowed
|
|
//Do that by checking if the king is in a square attacked by the other color.
|
|
oppKingCoordinate := b.getSquareOfPiece(types.Piece{
|
|
Class: types.King,
|
|
Color: movingColor})
|
|
if oppKingCoordinate == nil {
|
|
return false, string(movingColor) + " king not found"
|
|
}
|
|
|
|
b.isSquareAttacked(*oppKingCoordinate, movingColor.Opposite())
|
|
|
|
//Check for checkmate
|
|
//Is every square that the king can move to attacked? And can no other
|
|
//piece block? -> checkmate
|
|
|
|
//Maybe for checking checkmate, we have to check the 'path' in which the
|
|
//checkmate is given
|
|
|
|
// |K| | | | |Q|
|
|
// in this scenaria the path are all the squares between queen and king.
|
|
// If a piece can be moved into the path, it is no checkmate
|
|
|
|
return true, ""
|
|
}
|
|
|
|
func (b Board) getSquareOfPiece(piece types.Piece) *types.Coordinate {
|
|
for k, v := range b {
|
|
if v == piece {
|
|
return &k
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (b Board) isSquareAttacked(square types.Coordinate, byColor types.ChessColor) bool {
|
|
attacked := false
|
|
|
|
//get every legal move of color to check if this square is attacked
|
|
return attacked
|
|
}
|