first commit

backend/internal/handlers/document.go

@@ -0,0 +1,144 @@
+package handlers
+
+import (
+	"net/http"
+
+	"github.com/M1ngdaXie/realtime-collab/internal/models"
+	"github.com/M1ngdaXie/realtime-collab/internal/store"
+	"github.com/gin-gonic/gin"
+	"github.com/google/uuid"
+)
+
+  type DocumentHandler struct {
+  	store *store.Store
+  }
+
+  func NewDocumentHandler(s *store.Store) *DocumentHandler {
+  	return &DocumentHandler{store: s}
+  }
+
+  // CreateDocument creates a new document
+  func (h *DocumentHandler) CreateDocument(c *gin.Context) {
+  	var req models.CreateDocumentRequest
+  	if err := c.ShouldBindJSON(&req); err != nil {
+  		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+  		return
+  	}
+
+  	// Validate document type
+  	if req.Type != models.DocumentTypeEditor && req.Type != models.DocumentTypeKanban {
+  		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid document type"})
+  		return
+  	}
+
+  	doc, err := h.store.CreateDocument(req.Name, req.Type)
+  	if err != nil {
+  		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+  		return
+  	}
+
+  	c.JSON(http.StatusCreated, doc)
+  }
+
+  // ListDocuments returns all documents
+  func (h *DocumentHandler) ListDocuments(c *gin.Context) {
+  	documents, err := h.store.ListDocuments()
+  	if err != nil {
+  		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+  		return
+  	}
+
+  	if documents == nil {
+  		documents = []models.Document{}
+  	}
+
+  	c.JSON(http.StatusOK, models.DocumentListResponse{
+  		Documents: documents,
+  		Total:     len(documents),
+  	})
+  }
+
+  // GetDocument returns a single document
+  func (h *DocumentHandler) GetDocument(c *gin.Context) {
+  	idStr := c.Param("id")
+  	id, err := uuid.Parse(idStr)
+  	if err != nil {
+  		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid document ID"})
+  		return
+  	}
+
+  	doc, err := h.store.GetDocument(id)
+  	if err != nil {
+  		c.JSON(http.StatusNotFound, gin.H{"error": "document not found"})
+  		return
+  	}
+
+  	c.JSON(http.StatusOK, doc)
+  }
+
+  // GetDocumentState returns the Yjs state for a document
+  func (h *DocumentHandler) GetDocumentState(c *gin.Context) {
+  	idStr := c.Param("id")
+  	id, err := uuid.Parse(idStr)
+  	if err != nil {
+  		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid document ID"})
+  		return
+  	}
+
+  	doc, err := h.store.GetDocument(id)
+  	if err != nil {
+  		c.JSON(http.StatusNotFound, gin.H{"error": "document not found"})
+  		return
+  	}
+
+  	// Return binary state
+  	if doc.YjsState == nil {
+  		c.Data(http.StatusOK, "application/octet-stream", []byte{})
+  		return
+  	}
+
+  	c.Data(http.StatusOK, "application/octet-stream", doc.YjsState)
+  }
+
+  // UpdateDocumentState updates the Yjs state for a document
+  func (h *DocumentHandler) UpdateDocumentState(c *gin.Context) {
+  	idStr := c.Param("id")
+  	id, err := uuid.Parse(idStr)
+  	if err != nil {
+  		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid document ID"})
+  		return
+  	}
+
+  	// Read binary body
+  	state, err := c.GetRawData()
+  	if err != nil {
+  		c.JSON(http.StatusBadRequest, gin.H{"error": "failed to read request body"})
+  		return
+  	}
+
+  	err = h.store.UpdateDocumentState(id, state)
+  	if err != nil {
+  		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+  		return
+  	}
+
+  	c.JSON(http.StatusOK, gin.H{"message": "state updated successfully"})
+  }
+
+  // DeleteDocument deletes a document
+  func (h *DocumentHandler) DeleteDocument(c *gin.Context) {
+  	idStr := c.Param("id")
+  	id, err := uuid.Parse(idStr)
+  	if err != nil {
+  		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid document ID"})
+  		return
+  	}
+
+  	err = h.store.DeleteDocument(id)
+  	if err != nil {
+  		c.JSON(http.StatusNotFound, gin.H{"error": "document not found"})
+  		return
+  	}
+
+  	c.JSON(http.StatusOK, gin.H{"message": "document deleted successfully"})
+  }

backend/internal/handlers/websocket.go

@@ -0,0 +1,59 @@
+package handlers
+
+import (
+	"log"
+	"net/http"
+
+	"github.com/M1ngdaXie/realtime-collab/internal/hub"
+	"github.com/gin-gonic/gin"
+	"github.com/google/uuid"
+	"github.com/gorilla/websocket"
+)
+
+
+var upgrader = websocket.Upgrader{
+  	ReadBufferSize:  1024,
+  	WriteBufferSize: 1024,
+  	CheckOrigin: func(r *http.Request) bool {
+  		// Allow all origins for development
+  		// TODO: Restrict in production
+  		return true
+  	},
+}
+
+ type WebSocketHandler struct {
+  	hub *hub.Hub
+  }
+
+  func NewWebSocketHandler(h *hub.Hub) *WebSocketHandler {
+  	return &WebSocketHandler{hub: h}
+  }
+
+  func (wsh *WebSocketHandler) HandleWebSocket(c *gin.Context){
+	roomID := c.Param("roomId")
+
+	if(roomID == ""){
+		c.JSON(http.StatusBadRequest, gin.H{"error": "roomId is required"})
+		return
+	}
+	conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to upgrade to WebSocket"})
+		return
+	}
+    
+		// Create a new client
+  	clientID := uuid.New().String()
+  	client := hub.NewClient(clientID, conn, wsh.hub, roomID)
+
+  	// Register client with hub
+  	wsh.hub.Register <- client
+
+  	// Start read and write pumps in separate goroutines
+  	go client.WritePump()
+  	go client.ReadPump()
+
+  	log.Printf("WebSocket connection established for client %s in room %s", clientID, roomID)
+  }
+
+