init

cmd/ws/ws.go

@@ -0,0 +1,71 @@
+package main
+
+import (
+	"context"
+	"database/sql"
+	"log"
+	"os"
+	"time"
+
+	"github.com/genudine/saerro-go/types"
+	"nhooyr.io/websocket"
+	"nhooyr.io/websocket/wsjson"
+)
+
+func main() {
+	wsAddr := os.Getenv("WS_ADDR")
+	if wsAddr == "" {
+		log.Fatalln("WS_ADDR is not set.")
+	}
+
+	db, err := sql.Open("sqlite", ":memory:")
+	if err != nil {
+		log.Fatalln("database connection failed", err)
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
+	defer cancel()
+
+	wsConn, _, err := websocket.Dial(ctx, wsAddr, nil)
+	if err != nil {
+		log.Fatalln("Connection to ESS failed.", err)
+	}
+	defer wsConn.Close(websocket.StatusInternalError, "internal error. bye")
+
+	err = wsjson.Write(ctx, wsConn, map[string]interface{}{
+		"action":     "subscribe",
+		"worlds":     "all",
+		"eventNames": getEventNames(),
+		"characters": []string{"all"},
+		"service":    "event",
+
+		"logicalAndCharactersWithWorlds": true,
+	})
+	if err != nil {
+		log.Fatalln("subscription write failed", err)
+	}
+
+	log.Println("subscribe done")
+
+	eventHandler := EventHandler{
+		Ingest: &Ingest{
+			DB: db,
+		},
+	}
+
+	for {
+		ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
+
+		var event types.ESSData
+		err := wsjson.Read(ctx, wsConn, &event)
+		if err != nil {
+			log.Println("wsjson read failed", err)
+			cancel()
+			continue
+		}
+
+		go eventHandler.HandleEvent(ctx, event.Payload)
+	}
+
+	wsConn.Close(websocket.StatusNormalClosure, "")
+}