🐛 Fix user info not found panic

2024-12-22 14:43:02 +08:00 · 2024-12-22 14:43:02 +08:00 · 8e1adfae9a
commit 8e1adfae9a
parent cd58e7a429
1 changed files with 6 additions and 1 deletions
--- a/pkg/internal/http/ws/ws.go
+++ b/pkg/internal/http/ws/ws.go
@ -3,6 +3,7 @@ package ws
 import (
 	"context"
 	"fmt"
+
 	"git.solsynth.dev/hypernet/nexus/pkg/internal/directory"
 	"git.solsynth.dev/hypernet/nexus/pkg/nex"
 	"git.solsynth.dev/hypernet/nexus/pkg/nex/sec"
@ -14,7 +15,11 @@ import (
 )

 func Listen(c *websocket.Conn) {
-	user := c.Locals("nex_user").(sec.UserInfo)
+	user, ok := c.Locals("nex_user").(sec.UserInfo)
+	if !ok {
+		c.Close()
+		return
+	}

 	// Push connection
 	clientId := ClientRegister(user, c)