feat(epic-3): implémentation du Dashboard, mode lecture et notifications temps réel

internal/handler/ws.go

@@ -0,0 +1,41 @@
+package handler
+
+import (
+	"log"
+	"net/http"
+
+	"github.com/michael/flowreader/internal/service"
+	"github.com/michael/flowreader/internal/ws"
+)
+
+// WSHandler handles WebSocket connections.
+type WSHandler struct {
+	hub         *ws.Hub
+	authService *service.AuthService
+}
+
+// NewWSHandler creates a new WS handler.
+func NewWSHandler(hub *ws.Hub, authService *service.AuthService) *WSHandler {
+	return &WSHandler{
+		hub:         hub,
+		authService: authService,
+	}
+}
+
+// Connect handles WebSocket initiation.
+func (h *WSHandler) Connect(w http.ResponseWriter, r *http.Request) {
+	cookie, err := r.Cookie("session_id")
+	if err != nil {
+		http.Error(w, "Unauthorized", http.StatusUnauthorized)
+		return
+	}
+
+	user, err := h.authService.GetUserByToken(cookie.Value)
+	if err != nil || user == nil {
+		http.Error(w, "Unauthorized", http.StatusUnauthorized)
+		return
+	}
+
+	log.Printf("Setting up WS for user %s", user.ID)
+	h.hub.ServeWS(user.ID, w, r)
+}