mchess-server/chess/rook.go

package chess

import (
	"mchess_server/types"
)

type Rook struct {
	Color types.ChessColor
}

func (r Rook) GetAllAttackedSquares(board Board, fromSquare types.Coordinate) []types.Coordinate {
	return r.GetAllNonBlockedMoves(board, fromSquare)
}

func (r Rook) GetColor() types.ChessColor {
	return r.Color
}

func (r Rook) GetAllNonBlockedMoves(board Board, fromSquare types.Coordinate) []types.Coordinate {
	return board.GetNonBlockedRowAndColumn(fromSquare)
}

func (r Rook) AfterMoveAction(board *Board, fromSquare types.Coordinate) {
	switch r.Color {
	case types.Black:
        if fromSquare.Col == types.RangeLastValid {
            board.state.BlackHRookMoved = true
        }
        if fromSquare.Col == types.RangeFirstValid {
            board.state.BlackARookMoved = true
        }
	case types.White:
        if fromSquare.Col == types.RangeLastValid {
            board.state.WhiteHRookMoved = true
        }
        if fromSquare.Col == types.RangeFirstValid {
            board.state.WhiteARookMoved = true
        }
	}
}