websocket?

backend/services/websocket.go

@@ -1,6 +1,11 @@
 package services
 
 import (
+	"sync"
+	"time"
+
+	"adam-french.co.uk/backend/models"
+
 	"github.com/gorilla/websocket"
 )
 
@@ -8,3 +13,52 @@ var Upgrader = websocket.Upgrader{
 	ReadBufferSize:  1024,
 	WriteBufferSize: 1024,
 }
+
+var (
+	clients  = make(map[*websocket.Conn]bool)
+	messages = make([]models.Message, 0)
+	mu       sync.Mutex
+)
+
+func HandleWebSocket(conn *websocket.Conn) {
+	defer conn.Close()
+
+	mu.Lock()
+	clients[conn] = true
+
+	// Send existing message history to new client
+	for _, msg := range messages {
+		if err := conn.WriteJSON(msg); err != nil {
+			mu.Unlock()
+			return
+		}
+	}
+	mu.Unlock()
+
+	// Read loop
+	for {
+		var incoming models.Message
+		if err := conn.ReadJSON(&incoming); err != nil {
+			break // client disconnected
+		}
+
+		incoming.CreatedAt = time.Now()
+
+		// Store and broadcast
+		mu.Lock()
+		messages = append(messages, incoming)
+
+		for client := range clients {
+			if err := client.WriteJSON(incoming); err != nil {
+				client.Close()
+				delete(clients, client)
+			}
+		}
+		mu.Unlock()
+	}
+
+	// Cleanup on disconnect
+	mu.Lock()
+	delete(clients, conn)
+	mu.Unlock()
+}