mchess-server/chess/player.go

package chess

import (
	"context"
	"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) WriteMessageToPlayer(msg []byte) error {
	log.Printf("Writing message: %s to player %d", string(msg), p.Uuid)

	return p.Conn.Write(p.context, websocket.MessageText, msg)
}

func (p *Player) ReadMessageFromPlayer() (websocket.MessageType, []byte, error) {
	msgType, msg, err := p.Conn.Read(p.context)
	log.Printf("Reading message: %s (with messagetype %d) from player %d", string(msg), msgType, p.Uuid)

	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
	}
}