initial commitHEADmaster

1 files changed, 106 insertions, 0 deletions

diff --git a/ws.go b/ws.go
new file mode 100644
index 0000000..aea8621
--- /dev/null
+++ b/ws.go
@@ -0,0 +1,106 @@
+// ws.go
+package main
+
+import (
+	"encoding/json"
+	"log"
+	"net/http"
+	"time"
+	"unicode/utf8"
+
+	"github.com/gorilla/websocket"
+)
+
+var upgrader = websocket.Upgrader{
+	ReadBufferSize:  1024,
+	WriteBufferSize: 1024,
+	// okay while you're behind nginx/caddy
+	CheckOrigin: func(r *http.Request) bool { return true },
+}
+
+func wsHandler(hub *Hub) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		conn, err := upgrader.Upgrade(w, r, nil)
+		if err != nil {
+			log.Printf("upgrade error: %v", err)
+			return
+		}
+
+		// Expect first message to be auth
+		_, firstMsg, err := conn.ReadMessage()
+		if err != nil {
+			log.Printf("failed to read auth message: %v", err)
+			conn.Close()
+			return
+		}
+
+		var auth IncomingMessage
+		if err := json.Unmarshal(firstMsg, &auth); err != nil {
+			log.Printf("invalid auth json: %v", err)
+			_ = conn.WriteJSON(AuthError{Type: "auth_error", Error: "bad_json"})
+			conn.Close()
+			return
+		}
+
+		if auth.Type != "auth" || auth.Token == "" {
+			_ = conn.WriteJSON(AuthError{Type: "auth_error", Error: "missing_token"})
+			conn.Close()
+			return
+		}
+
+		// Lookup token
+		hub.mu.RLock()
+		tc, ok := hub.tokens[auth.Token]
+		hub.mu.RUnlock()
+
+		if !ok || tc.Revoked {
+			_ = conn.WriteJSON(AuthError{Type: "auth_error", Error: "invalid_token"})
+			conn.Close()
+			return
+		}
+
+		if auth.ServerID == "" {
+			auth.ServerID = "default"
+		}
+		if auth.Nick == "" {
+			auth.Nick = tc.Name
+		}
+
+		// Enforce nickname length limit on initial/auth nick.
+		if utf8.RuneCountInString(auth.Nick) > maxNickLen {
+			log.Printf("auth nick %q too long, falling back to internal name %q", auth.Nick, tc.Name)
+			auth.Nick = tc.Name
+		}
+		if utf8.RuneCountInString(auth.Nick) > maxNickLen {
+			// still too long, truncate internal name
+			log.Printf("internal name %q too long, truncating to %d runes", auth.Nick, maxNickLen)
+			auth.Nick = truncateRunes(auth.Nick, maxNickLen)
+		}
+
+		client := &Client{
+			hub:        hub,
+			conn:       conn,
+			send:       make(chan []byte, 32),
+			token:      tc,
+			nick:       auth.Nick,
+			serverID:   auth.ServerID,
+			playerUUID: auth.PlayerUUID,
+			lastSeen:   time.Now(),
+		}
+
+		hub.register(client)
+
+		if err := conn.WriteJSON(AuthOK{
+			Type:         "auth_ok",
+			InternalName: tc.Name,
+		}); err != nil {
+			log.Printf("failed to send auth_ok: %v", err)
+			hub.unregister(client)
+			conn.Close()
+			return
+		}
+
+		go client.writePump()
+		client.readPump()
+	}
+}
\ No newline at end of file