Update app.js

app.js

@@ -1,10 +1,7 @@
-// host.js
-import { serve } from "bun";
-
 const rooms = new Map(); // roomId ⇒ Set<ServerWebSocket>
 
-serve({
-  port: 7860, // ← change if you need a different port
+Bun.serve({
+  port: 7860,
 
   fetch(req, server) {
     const url = new URL(req.url);
@@ -23,38 +20,9 @@ serve({
 
     // WebSocket upgrade
     if (req.headers.get("upgrade")?.toLowerCase() === "websocket") {
-      const ws = server.upgrade(req);
-      ws.onmessage = (raw) => {
-        let msg;
-        try { msg = JSON.parse(raw); } catch { return; }
-        const { action, room, message } = msg;
-        if (action === "join" && room) {
-          // join room
-          rooms.get(room)?.delete(ws);
-          let set = rooms.get(room);
-          if (!set) { set = new Set(); rooms.set(room, set); }
-          set.add(ws);
-          ws.send(JSON.stringify({ system: `Joined room ${room}` }));
-        }
-        if (action === "post" && room && message) {
-          // broadcast
-          const set = rooms.get(room);
-          if (!set) return;
-          const payload = JSON.stringify({
-            room,
-            message,
-            from: "host",
-            timestamp: Date.now(),
-          });
-          for (const client of set) {
-            if (client.readyState === 1) client.send(payload);
-          }
-        }
-      };
-      ws.onclose = () => {
-        for (const set of rooms.values()) set.delete(ws);
-      };
-      return ws;
+      const success = server.upgrade(req);
+      if (success) return; // Bun returns 101 automatically
+      return new Response("Upgrade failed", { status: 500 });
     }
 
     // Serve inline HTML test page
@@ -95,6 +63,37 @@ serve({
 
     return new Response("Not Found", { status: 404, headers: { "Access-Control-Allow-Origin": "*" } });
   },
+
+  websocket: {
+    message(ws, raw) {
+      let msg;
+      try { msg = JSON.parse(raw); } catch { return; }
+      const { action, room, message } = msg;
+      if (action === "join" && room) {
+        for (const set of rooms.values()) set.delete(ws);
+        let set = rooms.get(room);
+        if (!set) { set = new Set(); rooms.set(room, set); }
+        set.add(ws);
+        ws.send(JSON.stringify({ system: `Joined room ${room}` }));
+      }
+      if (action === "post" && room && message) {
+        const set = rooms.get(room);
+        if (!set) return;
+        const payload = JSON.stringify({
+          room,
+          message,
+          from: "host",
+          timestamp: Date.now(),
+        });
+        for (const client of set) {
+          if (client.readyState === 1) client.send(payload);
+        }
+      }
+    },
+    close(ws) {
+      for (const set of rooms.values()) set.delete(ws);
+    },
+  },
 });
 
 console.log("✅ Host server running on ws://localhost:7860");