58 lines
1.2 KiB
Go
58 lines
1.2 KiB
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
|
|
}
|
|
}
|