Initial commit: EduBox V2 platform

2026-06-06 19:55:41 +00:00
commit 0a73a70820
69 changed files with 5634 additions and 0 deletions
@@ -0,0 +1,105 @@
+import { WebSocketServer, WebSocket } from "ws";
+import { prisma } from "./prisma";
+
+interface NodeMessage {
+  action: string;
+  nodeId?: string;
+  code?: string;
+  instanceId?: string;
+  type?: string;
+  port?: number;
+  composeConfig?: string;
+  studentName?: string;
+  error?: string;
+}
+
+const nodes = new Map<string, WebSocket>();
+
+export function initWebSocketServer(wss: WebSocketServer) {
+  wss.on("connection", (ws: WebSocket) => {
+    let nodeId: string | null = null;
+
+    ws.on("message", async (raw) => {
+      try {
+        const msg: NodeMessage = JSON.parse(raw.toString());
+
+        if (msg.action === "register" && msg.nodeId) {
+          nodeId = msg.nodeId;
+          nodes.set(nodeId, ws);
+          await prisma.node.upsert({
+            where: { id: nodeId },
+            update: { status: "online", lastSeen: new Date() },
+            create: { id: nodeId, status: "online", lastSeen: new Date() },
+          });
+          ws.send(JSON.stringify({ action: "registered" }));
+          return;
+        }
+
+        if (msg.action === "activate" && msg.code && msg.nodeId) {
+          nodeId = msg.nodeId;
+          const student = await prisma.student.findUnique({
+            where: { activationCode: msg.code },
+          });
+          if (!student) {
+            ws.send(JSON.stringify({ action: "activation_failed", error: "Invalid code" }));
+            return;
+          }
+          await prisma.node.upsert({
+            where: { id: nodeId },
+            update: { studentId: student.id, status: "online", lastSeen: new Date() },
+            create: { id: nodeId, studentId: student.id, status: "online", lastSeen: new Date() },
+          });
+          ws.send(JSON.stringify({ action: "activated", studentName: `${student.firstName} ${student.lastName}` }));
+          return;
+        }
+
+        if (msg.action === "heartbeat" && nodeId) {
+          await prisma.node.upsert({
+            where: { id: nodeId },
+            update: { lastSeen: new Date() },
+            create: { id: nodeId, status: "online", lastSeen: new Date() },
+          });
+          return;
+        }
+
+        if (msg.action === "instance_started" && msg.instanceId) {
+          await prisma.instance.update({
+            where: { id: msg.instanceId },
+            data: { status: "running" },
+          });
+          return;
+        }
+
+        if (msg.action === "instance_error" && msg.instanceId) {
+          await prisma.instance.update({
+            where: { id: msg.instanceId },
+            data: { status: "error" },
+          });
+          return;
+        }
+      } catch (err) {
+        console.error("WS error:", err);
+      }
+    });
+
+    ws.on("close", async () => {
+      if (nodeId) {
+        nodes.delete(nodeId);
+        await prisma.node.upsert({
+          where: { id: nodeId },
+          update: { status: "offline" },
+          create: { id: nodeId, status: "offline" },
+        });
+      }
+    });
+  });
+}
+
+export function sendToNode(nodeId: string, message: object) {
+  const ws = nodes.get(nodeId);
+  if (ws && ws.readyState === WebSocket.OPEN) {
+    ws.send(JSON.stringify(message));
+    return true;
+  }
+  return false;
+}