67 lines
1.2 KiB
Go
67 lines
1.2 KiB
Go
package chess
|
|
|
|
import (
|
|
"mchess_server/types"
|
|
"strings"
|
|
)
|
|
|
|
type Piece interface {
|
|
GetAllNonBlockedSquares(board Board, fromSquare types.Coordinate) []types.Coordinate
|
|
GetAllAttackedSquares(board Board, fromSquare types.Coordinate) []types.Coordinate
|
|
GetColor() types.ChessColor
|
|
AfterMoveAction(board *Board, fromSquare types.Coordinate)
|
|
}
|
|
|
|
func GetPieceForShortName(name types.PieceShortName) Piece {
|
|
var piece Piece
|
|
nameString := name.String()
|
|
|
|
var color = types.Black
|
|
if nameString < "Z" && nameString > "A" {
|
|
color = types.White
|
|
}
|
|
|
|
nameString = strings.ToLower(nameString)
|
|
|
|
switch nameString {
|
|
case "p":
|
|
piece = Pawn{Color: color}
|
|
case "q":
|
|
piece = Queen{Color: color}
|
|
case "k":
|
|
piece = King{Color: color}
|
|
case "b":
|
|
piece = Bishop{Color: color}
|
|
case "r":
|
|
piece = Rook{Color: color}
|
|
case "n":
|
|
piece = Knight{Color: color}
|
|
}
|
|
|
|
return piece
|
|
}
|
|
|
|
func GetShortNameForPiece(piece Piece) types.PieceShortName {
|
|
var name string
|
|
|
|
switch piece.(type) {
|
|
case Pawn:
|
|
name = "p"
|
|
case Queen:
|
|
name = "q"
|
|
case King:
|
|
name = "k"
|
|
case Bishop:
|
|
name = "b"
|
|
case Rook:
|
|
name = "r"
|
|
case Knight:
|
|
name = "n"
|
|
}
|
|
|
|
if piece.GetColor() == types.White {
|
|
name = strings.ToUpper(name)
|
|
}
|
|
return types.PieceShortName(name)
|
|
}
|