add user count backend

hub.go

@@ -25,6 +25,22 @@ func IsUsernameTaken(h *server.Hub, username string) bool {
 	return false
 }
 
+func HandleUserCountCheck(hub *server.Hub, w http.ResponseWriter, r *http.Request) {
+	if r.Method != "GET" {
+		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
+		return
+	}
+
+	response := map[string]any{
+		"userCount": int64(len(hub.Clients)),
+	}
+
+	err := json.NewEncoder(w).Encode(response)
+	if err != nil {
+		return
+	}
+}
+
 func HandleUsernameCheck(hub *server.Hub, w http.ResponseWriter, r *http.Request) {
 	if r.Method != "POST" {
 		http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)

main.go

@@ -50,6 +50,10 @@ func main() {
 		HandleFileDownload(filesDir, w, r)
 	})
 
+	mux.HandleFunc("/user-count", func(w http.ResponseWriter, r *http.Request) {
+		HandleUserCountCheck(hub, w, r)
+	})
+
 	mux.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
 		HandleWebSocket(hub, w, r)
 	})