Add test websocket connection

Author: ValorZard

ebiten-game/game/main.go

@@ -19,6 +19,8 @@ import (
 	_ "image/jpeg"
 	_ "image/png"
 
+	"github.com/coder/websocket"
+	"github.com/coder/websocket/wsjson"
 	"github.com/ebitengine/debugui"
 	"github.com/hajimehoshi/ebiten/v2"
 	"github.com/hajimehoshi/ebiten/v2/ebitenutil"
@@ -299,7 +301,26 @@ func (g *game) onHostReceivedDataChannel(d *webrtc.DataChannel) {
 	})
 }
 
+func (g *game) openWebsocket() {
+	ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
+	defer cancel()
+
+	c, _, err := websocket.Dial(ctx, fmt.Sprintf("ws://%s:%d/lobbies", g.signalingIP, g.port), nil)
+	if err != nil {
+		println("Failed to connect to websocket:", err.Error())
+	}
+	defer c.CloseNow()
+
+	err = wsjson.Write(ctx, c, "open websocket")
+	if err != nil {
+		println("Failed to write websocket message:", err.Error())
+	}
+
+	c.Close(websocket.StatusNormalClosure, "")
+}
+
 func (g *game) startHost() {
+	g.openWebsocket()
 	g.writeLog("Hosting a lobby")
 	// Host creates lobby.
 	req, err := http.NewRequestWithContext(context.Background(), "GET", g.getSignalingURL()+"/lobby/host", nil)

ebiten-game/signaling-server/main.go

@@ -4,15 +4,20 @@
 package main
 
 import (
+	"context"
 	"encoding/json"
 	"errors"
 	"fmt"
 	"io"
+	"log"
 	"math/rand"
 	"net/http"
 	"strconv"
 	"sync"
+	"time"
 
+	"github.com/coder/websocket"
+	"github.com/coder/websocket/wsjson"
 	"github.com/pion/webrtc/v4"
 	"github.com/rs/cors"
 )
@@ -112,6 +117,7 @@ func main() {
 			fmt.Printf("Failed to write response: %s", err)
 		}
 	})
+	mux.HandleFunc("/lobbies", db.lobbies)
 
 	fmt.Println("Server started on port 3000")
 	// cors.Default() setup the middleware with default options being
@@ -126,6 +132,29 @@ func main() {
 	}
 }
 
+func (db *lobbyDatabase) lobbies(w http.ResponseWriter, r *http.Request) {
+	c, err := websocket.Accept(w, r, nil)
+	if err != nil {
+		println("Failed to accept websocket:", err.Error())
+	}
+	defer c.CloseNow()
+
+	// Set the context as needed. Use of r.Context() is not recommended
+	// to avoid surprising behavior (see http.Hijacker).
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
+	defer cancel()
+
+	var v any
+	err = wsjson.Read(ctx, c, &v)
+	if err != nil {
+		println("Failed to read websocket message:", err.Error())
+	}
+
+	log.Printf("received: %v", v)
+
+	c.Close(websocket.StatusNormalClosure, "")
+}
+
 func (db *lobbyDatabase) lobbyHost(res http.ResponseWriter, _ *http.Request) {
 	lobbyID := db.makeLobby()
 	lobby := db.lobbyList[lobbyID]

go.mod

@@ -7,6 +7,7 @@ toolchain go1.25.1
 require (
 	github.com/asticode/go-astiav v0.19.0
 	github.com/at-wat/ebml-go v0.17.2
+	github.com/coder/websocket v1.8.14
 	github.com/ebitengine/debugui v0.1.1
 	github.com/emiago/sipgo v0.33.0
 	github.com/go-gst/go-gst v1.3.0
@@ -29,7 +30,6 @@ require (
 
 require (
 	github.com/asticode/go-astikit v0.42.0 // indirect
-	github.com/coder/websocket v1.8.14 // indirect
 	github.com/ebitengine/gomobile v0.0.0-20240911145611-4856209ac325 // indirect
 	github.com/ebitengine/hideconsole v1.0.0 // indirect
 	github.com/ebitengine/purego v0.8.0 // indirect