feat: add WebSocket handlers for device and dashboard connections

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

server/src/index.ts

@@ -2,6 +2,12 @@ import { Hono } from "hono";
 import { cors } from "hono/cors";
 import { auth } from "./auth.js";
 import { env } from "./env.js";
+import { handleDeviceMessage, handleDeviceClose } from "./ws/device.js";
+import {
+  handleDashboardMessage,
+  handleDashboardClose,
+} from "./ws/dashboard.js";
+import type { WebSocketData } from "./ws/sessions.js";
 
 const app = new Hono();
 
@@ -25,18 +31,54 @@ app.on(["POST", "GET"], "/api/auth/*", (c) => {
 app.get("/health", (c) => c.json({ status: "ok" }));
 
 // Start server with WebSocket support
-const server = Bun.serve({
+const server = Bun.serve<WebSocketData>({
   port: env.PORT,
-  fetch: app.fetch,
+  fetch(req, server) {
+    const url = new URL(req.url);
+
+    // WebSocket upgrade for device connections
+    if (url.pathname === "/ws/device") {
+      const upgraded = server.upgrade(req, {
+        data: { path: "/ws/device" as const, authenticated: false },
+      });
+      if (upgraded) return undefined;
+      return new Response("WebSocket upgrade failed", { status: 400 });
+    }
+
+    // WebSocket upgrade for dashboard connections
+    if (url.pathname === "/ws/dashboard") {
+      const upgraded = server.upgrade(req, {
+        data: { path: "/ws/dashboard" as const, authenticated: false },
+      });
+      if (upgraded) return undefined;
+      return new Response("WebSocket upgrade failed", { status: 400 });
+    }
+
+    // Non-WebSocket requests go to Hono
+    return app.fetch(req);
+  },
   websocket: {
     open(ws) {
-      console.log("WebSocket connected");
+      console.log(`WebSocket opened: ${ws.data.path}`);
     },
     message(ws, message) {
-      // placeholder — Task 4 implements device/dashboard handlers
+      const raw =
+        typeof message === "string"
+          ? message
+          : new TextDecoder().decode(message);
+
+      if (ws.data.path === "/ws/device") {
+        handleDeviceMessage(ws, raw);
+      } else if (ws.data.path === "/ws/dashboard") {
+        handleDashboardMessage(ws, raw);
+      }
     },
     close(ws) {
-      console.log("WebSocket disconnected");
+      if (ws.data.path === "/ws/device") {
+        handleDeviceClose(ws);
+      } else if (ws.data.path === "/ws/dashboard") {
+        handleDashboardClose(ws);
+      }
     },
   },
 });