package chess import ( "context" "encoding/json" "errors" "mchess_server/api" "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.IsValidMoveMessage() { 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 } }