73 lines
1.6 KiB
Go
73 lines
1.6 KiB
Go
package websocket
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"mchess_server/api"
|
|
"net/http"
|
|
|
|
lobbies "mchess_server/lobby_registry"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
gorillaws "github.com/gorilla/websocket"
|
|
)
|
|
|
|
var upgrader = gorillaws.Upgrader{
|
|
CheckOrigin: func(r *http.Request) bool {
|
|
return true
|
|
},
|
|
}
|
|
|
|
func RegisterWebSocketConnection(c *gin.Context) {
|
|
log.Println(c.Request)
|
|
conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
|
|
if err != nil {
|
|
log.Println(err)
|
|
return
|
|
}
|
|
go waitForAndHandlePlayerID(c, conn)
|
|
}
|
|
|
|
func waitForAndHandlePlayerID(ctx context.Context, conn *gorillaws.Conn) {
|
|
msgType, msg, err := conn.ReadMessage()
|
|
if err != nil {
|
|
errorMessage := fmt.Sprintf("Reading from websocket connection did not work: %s", err)
|
|
log.Println(errorMessage)
|
|
conn.Close()
|
|
return
|
|
}
|
|
|
|
log.Println("read from websocket endpoint: ", msgType, string(msg), err)
|
|
|
|
var info api.PlayerInfo
|
|
err = json.Unmarshal(msg, &info)
|
|
if err != nil {
|
|
errorMessage := fmt.Sprintf("Unmarshaling message did not work: %s", err)
|
|
log.Println(errorMessage)
|
|
conn.WriteMessage(msgType, []byte(errorMessage))
|
|
conn.Close()
|
|
return
|
|
}
|
|
|
|
lobby := lobbies.GetLobbyRegistry().GetLobbyByUUID(*info.LobbyID)
|
|
if lobby == nil {
|
|
conn.WriteMessage(msgType, []byte("lobby not found"))
|
|
conn.Close()
|
|
return
|
|
}
|
|
|
|
player, found := lobby.GetPlayerByUUID(*info.PlayerID)
|
|
if !found {
|
|
conn.WriteMessage(msgType, []byte("player not found"))
|
|
conn.Close()
|
|
return
|
|
}
|
|
if player.Conn.HasWebsocketConnection() {
|
|
player.Conn.Close("closing existing connection")
|
|
}
|
|
lobby.Game.SetWebsocketConnectionFor(ctx, player, conn)
|
|
log.Println("player after setting connection: ", player)
|
|
}
|