working socket integration

backend/handler/websocket.go

@@ -0,0 +1,85 @@
+package handler
+
+import (
+	"encoding/json"
+	"log"
+	"net/http"
+
+	"github.com/arnab-afk/monaco/model"
+	"github.com/gorilla/websocket"
+)
+
+var upgrader = websocket.Upgrader{
+	ReadBufferSize:  1024,
+	WriteBufferSize: 1024,
+	// Allow all origins for development
+	CheckOrigin: func(r *http.Request) bool { return true },
+}
+
+// WebSocketHandler handles WebSocket connections for code execution
+func (h *Handler) WebSocketHandler(w http.ResponseWriter, r *http.Request) {
+	// Upgrade the HTTP connection to a WebSocket connection
+	conn, err := upgrader.Upgrade(w, r, nil)
+	if err != nil {
+		log.Printf("Failed to upgrade connection: %v", err)
+		return
+	}
+
+	// Read the initial message containing the code submission
+	_, message, err := conn.ReadMessage()
+	if err != nil {
+		log.Printf("Failed to read message: %v", err)
+		conn.Close()
+		return
+	}
+
+	// Parse the message as a code submission
+	var submission model.CodeSubmission
+	if err := json.Unmarshal(message, &submission); err != nil {
+		log.Printf("Failed to parse message: %v", err)
+		conn.WriteMessage(websocket.TextMessage, []byte("Error: Invalid submission format"))
+		conn.Close()
+		return
+	}
+
+	// Validate the submission
+	if submission.Code == "" {
+		conn.WriteMessage(websocket.TextMessage, []byte("Error: Code is required"))
+		conn.Close()
+		return
+	}
+
+	// Set default language if not provided
+	if submission.Language == "" {
+		submission.Language = "python" // Default to Python
+	}
+
+	// Validate language
+	supportedLanguages := map[string]bool{
+		"python": true,
+		"java":   true,
+		"c":      true,
+		"cpp":    true,
+	}
+
+	if !supportedLanguages[submission.Language] {
+		conn.WriteMessage(websocket.TextMessage, []byte("Error: Unsupported language: "+submission.Language))
+		conn.Close()
+		return
+	}
+
+	// Generate a unique ID for the submission
+	submission.ID = h.generateID()
+	submission.Status = "pending"
+
+	// Store the submission
+	h.mu.Lock()
+	h.submissions[submission.ID] = &submission
+	h.mu.Unlock()
+
+	// Send the submission ID to the client
+	conn.WriteMessage(websocket.TextMessage, []byte("Submission ID: "+submission.ID))
+
+	// Execute the code with WebSocket communication
+	h.executionService.HandleWebSocket(conn, &submission)
+}