mchess-server/chess/player.go

package chess

import (
	"context"
	"encoding/json"
	"errors"
	"log"
	"mchess_server/api"
	"mchess_server/types"
	"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) SendMoveAndPosition(move types.Move, boardPosition string) error {
	messageToSend, err := json.Marshal(api.WebsocketMessage{
		Type:     api.MoveMessage,
		Move:     &move,
		Position: &boardPosition,
	})
	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.IsValid() {
		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(5 * time.Second)

	select {
	case <-p.wsConnEstablished:
		c <- true
	case <-timer.C:
		return
	}
}