Add standalone wsdebug tool for inspecting the raw Sneedchat WebSocket stream

Loads the same .env credentials as the bridge, connects to the Sneedchat WebSocket independent of Discord, and prints every raw frame to stdout while also appending it to a log file. Adds an OnRaw hook to sneed.Client so the full, untruncated frame is available to callers.
2026-08-02 00:04:53 +00:00
parent adba77b3e5
commit 98eeb2e2ba
4 changed files with 128 additions and 0 deletions
@@ -0,0 +1,93 @@
+// Command wsdebug is a standalone debugging tool that logs into Kiwi Farms
+// using the same .env credentials as the bridge, opens the Sneedchat
+// WebSocket connection, and prints/logs every raw frame received. It does
+// not touch Discord at all — it exists purely to inspect what the
+// WebSocket is sending.
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"io"
+	"log"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"local/sneedchatbridge/config"
+	"local/sneedchatbridge/cookie"
+	"local/sneedchatbridge/sneed"
+)
+
+func main() {
+	envFile := flag.String("env", ".env", "path to .env file")
+	logFile := flag.String("log", "wsdebug.log", "path to log file")
+	room := flag.Int("room", 0, "override SNEEDCHAT_ROOM_ID from .env")
+	flag.Parse()
+
+	cfg, err := config.Load(*envFile)
+	if err != nil {
+		log.Fatalf("Failed to load config: %v", err)
+	}
+	if *room != 0 {
+		cfg.SneedchatRoomID = *room
+	}
+
+	f, err := os.OpenFile(*logFile, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
+	if err != nil {
+		log.Fatalf("Failed to open log file %s: %v", *logFile, err)
+	}
+	defer f.Close()
+
+	logger := log.New(io.MultiWriter(os.Stdout, f), "", log.LstdFlags|log.Lmicroseconds)
+	log.SetOutput(io.MultiWriter(os.Stdout, f))
+	log.SetFlags(log.LstdFlags | log.Lmicroseconds)
+
+	logger.Printf("wsdebug starting — env=%s log=%s room=%d", *envFile, *logFile, cfg.SneedchatRoomID)
+
+	ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer cancel()
+
+	session, err := cookie.NewSessionService(ctx, "kiwifarms.st", cfg.BridgeUsername, cfg.BridgePassword)
+	if err != nil {
+		log.Fatalf("Failed to establish session: %v", err)
+	}
+	defer session.Close()
+
+	client := sneed.NewClient(cfg.SneedchatRoomID, session, true)
+	client.SetBridgeIdentity(cfg.BridgeUserID, cfg.BridgeUsername)
+
+	client.OnRaw = func(raw string) {
+		logger.Printf("RECV: %s", raw)
+	}
+	client.OnConnect = func() {
+		logger.Printf("=== CONNECTED (room %d) ===", cfg.SneedchatRoomID)
+	}
+	client.OnDisconnect = func() {
+		logger.Printf("=== DISCONNECTED ===")
+	}
+	client.OnMessage = func(m map[string]interface{}) {
+		logger.Printf("MESSAGE: user=%v content=%q uuid=%v", m["username"], m["content"], m["message_uuid"])
+	}
+	client.OnEdit = func(uuid, content string) {
+		logger.Printf("EDIT: uuid=%s content=%q", uuid, content)
+	}
+	client.OnDelete = func(uuid string) {
+		logger.Printf("DELETE: uuid=%s", uuid)
+	}
+
+	if err := client.Connect(ctx); err != nil {
+		log.Fatalf("Initial connect failed: %v", err)
+	}
+
+	logger.Println("wsdebug running — press Ctrl+C to stop")
+
+	<-ctx.Done()
+	stopTime := time.Now().Format(time.RFC3339)
+	fmt.Fprintf(f, "--- shutdown requested at %s ---\n", stopTime)
+	logger.Println("Shutdown signal received, disconnecting...")
+	client.Disconnect()
+	logger.Println("wsdebug stopped")
+}