update

internal/web/handlers.go

@@ -1 +1,60 @@
 package web
+
+import (
+	"chat/internal/cache"
+	"chat/internal/models"
+	"net/http"
+	"time"
+
+	"github.com/coder/websocket"
+	"github.com/coder/websocket/wsjson"
+)
+
+func HandleChat(w http.ResponseWriter, r *http.Request, cache *cache.Cache) {
+	// Get channel id from Cache
+	roomID := r.URL.Query().Get("room")
+	val, found := cache.Get(roomID)
+	if !found {
+		http.Error(w, "Channel not found", 404)
+		return
+	}
+	ch := val.(*models.Channel)
+
+	// Accept websocket
+	conn, err := websocket.Accept(w, r, &websocket.AcceptOptions{
+		InsecureSkipVerify: true,
+	})
+	if err != nil {
+		return
+	}
+	defer conn.Close(websocket.StatusNormalClosure, "")
+
+	ctx := r.Context()
+
+	// Register client to this channel
+	ch.Mu.RLock()
+	for _, msg := range ch.Messages {
+		_ = wsjson.Write(ctx, conn, msg)
+	}
+	ch.Mu.RUnlock()
+
+	// Register for live updates
+	ch.Mu.Lock()
+	if ch.Clients == nil {
+		ch.Clients = make(map[*websocket.Conn]bool)
+	}
+	ch.Clients[conn] = true
+	ch.Mu.Unlock()
+
+	// Simple read loop
+	for {
+		var msg models.Message
+		err := wsjson.Read(ctx, conn, &msg)
+		if err != nil {
+			break
+		}
+
+		msg.Timestamp = time.Now()
+		ch.Broadcast(ctx, msg)
+	}
+}