99 lines
2.0 KiB
Go
99 lines
2.0 KiB
Go
package chess
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"local/m/mchess_server/api"
|
|
"local/m/mchess_server/types"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"nhooyr.io/websocket"
|
|
)
|
|
|
|
type Player struct {
|
|
Uuid uuid.UUID
|
|
Conn *websocket.Conn
|
|
InGame bool
|
|
wsConnEstablished chan bool
|
|
context context.Context
|
|
}
|
|
|
|
func NewPlayer(uuid uuid.UUID) *Player {
|
|
return &Player{
|
|
Uuid: uuid,
|
|
Conn: nil,
|
|
InGame: false,
|
|
wsConnEstablished: make(chan bool),
|
|
}
|
|
}
|
|
|
|
func (p *Player) SetConnection(ctx context.Context, conn *websocket.Conn) {
|
|
p.Conn = conn
|
|
p.context = ctx
|
|
p.wsConnEstablished <- true
|
|
}
|
|
|
|
func (p *Player) SendMove(move types.Move) error {
|
|
messageToSend, err := json.Marshal(api.WebsocketMessage{
|
|
Type: api.MoveMessage,
|
|
Move: &move,
|
|
})
|
|
if err != nil {
|
|
log.Println("Error while marshalling: ", err)
|
|
return err
|
|
}
|
|
|
|
err = p.writeMessage(messageToSend)
|
|
if err != nil {
|
|
log.Println("Error during message writing:", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *Player) writeMessage(msg []byte) error {
|
|
log.Printf("Writing message: %s to player %s", string(msg), p.Uuid.String())
|
|
|
|
return p.Conn.Write(p.context, websocket.MessageText, msg)
|
|
}
|
|
|
|
func (p *Player) ReadMove() (types.Move, error) {
|
|
_, receivedMessage, err := p.readMessage()
|
|
if err != nil {
|
|
return types.Move{}, err
|
|
}
|
|
|
|
var msg api.WebsocketMessage
|
|
err = json.Unmarshal(receivedMessage, &msg)
|
|
if err != nil {
|
|
return types.Move{}, err
|
|
}
|
|
|
|
if !msg.IsValidMove() {
|
|
return types.Move{}, errors.New("not a valid move")
|
|
}
|
|
|
|
return *msg.Move, nil
|
|
}
|
|
|
|
func (p *Player) readMessage() (websocket.MessageType, []byte, error) {
|
|
msgType, msg, err := p.Conn.Read(p.context)
|
|
log.Printf("Reading message: %s from player %s", string(msg), p.Uuid.String())
|
|
|
|
return msgType, msg, err
|
|
}
|
|
|
|
func (p *Player) WaitForWebsocketConnection(c chan bool) {
|
|
timer := time.NewTimer(500 * time.Second)
|
|
|
|
select {
|
|
case <-p.wsConnEstablished:
|
|
c <- true
|
|
case <-timer.C:
|
|
return
|
|
}
|
|
}
|