plugin_server

Sleeping

App Files Files Community

Pepguy commited on Jun 25, 2025

Commit

67d31ad

verified ·

1 Parent(s): be81bd0

Update app.js

Browse files

Files changed (1) hide show

app.js +50 -57

app.js CHANGED Viewed

@@ -1,57 +1,21 @@
 // server.js (Bun)
-import { serve } from "bun";
-const rooms = new Map(); // roomId ⇒ Set<WebSocket>
-serve({
-  port: Number(process.env.PORT) || 7860,
-  async fetch(req) {
     const url = new URL(req.url);
-    // Handle WebSocket upgrade
     if (req.headers.get("upgrade")?.toLowerCase() === "websocket") {
-      // Bun’s upgrade helper
-      const { socket, response } = Bun.upgradeWebSocket(req);
-      socket.onmessage = (raw) => {
-        let msg;
-        try { msg = JSON.parse(raw.data); } catch { return; }
-        if (msg.action === "join" && msg.roomId) {
-          // leave all rooms
-          for (const set of rooms.values()) set.delete(socket);
-          // join target room
-          const set = rooms.get(msg.roomId) || new Set();
-          set.add(socket);
-          rooms.set(msg.roomId, set);
-        }
-        if (msg.action === "post" && msg.roomId && msg.message) {
-          const set = rooms.get(msg.roomId);
-          if (!set) return;
-          const payload = JSON.stringify({
-            roomId:  msg.roomId,
-            message: msg.message,
-            timestamp: Date.now(),
-          });
-          for (const client of set) {
-            if (client.readyState === WebSocket.OPEN) {
-              client.send(payload);
-            }
-          }
-        }
-      };
-      socket.onclose = () => {
-        // clean up on disconnect
-        for (const set of rooms.values()) set.delete(socket);
-      };
-      return response;
     }
-    // Fallback to serving the HTML for non-WS requests
     if (url.pathname === "/") {
       return new Response(`<!doctype html>
 <html><head>
@@ -69,29 +33,20 @@ serve({
   <div id="log"></div>
   <input id="msg" placeholder="Type a message" style="width:80%;"/>
   <button onclick="sendMsg()">Send</button>
   <script>
     var ws, currentRoom;
     window.addEventListener('load', function() {
       var scheme = location.protocol === 'https:' ? 'wss' : 'ws';
       var socketUrl = scheme + '://' + location.host + '/';
-      console.log('Connecting to ' + socketUrl);
       ws = new WebSocket(socketUrl);
       ws.onopen = function(){ log('🔌 Connected'); };
       ws.onmessage = function(ev){
         var m=JSON.parse(ev.data);
         log('['+m.roomId+'] '+m.message);
       };
-      ws.onerror = function(e){
-        console.error('WS error', e);
-        log('⚠️ WebSocket error');
-      };
-      ws.onclose = function(c){
-        log('❌ Disconnected (code='+c.code+')');
-      };
     });
     function joinRoom(){
       var id = document.getElementById('room').value.trim();
       if(!id) return alert('Enter room ID');
@@ -117,6 +72,44 @@ serve({
     return new Response("Not Found", { status: 404 });
   },
 });
-console.log("✅ Bun realtime server running on port " + (process.env.PORT||7860));

 // server.js (Bun)
+// import { serve } from "bun";
+const rooms = new Map(); // roomId ⇒ Set<ServerWebSocket>
+Bun.serve({
+  port: Number(Bun.env.PORT) || 7860,
+  fetch(req, server) {
     const url = new URL(req.url);
+    // Upgrade to WebSocket if requested
     if (req.headers.get("upgrade")?.toLowerCase() === "websocket") {
+      // Accept the upgrade and attach room data if needed
+      return server.upgrade(req) ? undefined : new Response("Upgrade failed", { status: 500 });
     }
+    // Serve HTML UI for non-WebSocket requests
     if (url.pathname === "/") {
       return new Response(`<!doctype html>
 <html><head>
   <div id="log"></div>
   <input id="msg" placeholder="Type a message" style="width:80%;"/>
   <button onclick="sendMsg()">Send</button>
   <script>
     var ws, currentRoom;
     window.addEventListener('load', function() {
       var scheme = location.protocol === 'https:' ? 'wss' : 'ws';
       var socketUrl = scheme + '://' + location.host + '/';
       ws = new WebSocket(socketUrl);
       ws.onopen = function(){ log('🔌 Connected'); };
       ws.onmessage = function(ev){
         var m=JSON.parse(ev.data);
         log('['+m.roomId+'] '+m.message);
       };
+      ws.onerror = function(e){ log('⚠️ WebSocket error'); };
+      ws.onclose = function(c){ log('❌ Disconnected (code='+c.code+')'); };
     });
     function joinRoom(){
       var id = document.getElementById('room').value.trim();
       if(!id) return alert('Enter room ID');
     return new Response("Not Found", { status: 404 });
   },
+  websocket: {
+    message(ws, raw) {
+      let msg;
+      try { msg = JSON.parse(raw); } catch { return; }
+      if (msg.action === "join" && msg.roomId) {
+        // Leave all rooms
+        for (const set of rooms.values()) set.delete(ws);
+        // Join target room
+        let set = rooms.get(msg.roomId);
+        if (!set) {
+          set = new Set();
+          rooms.set(msg.roomId, set);
+        }
+        set.add(ws);
+      }
+      if (msg.action === "post" && msg.roomId && msg.message) {
+        const set = rooms.get(msg.roomId);
+        if (!set) return;
+        const payload = JSON.stringify({
+          roomId:  msg.roomId,
+          message: msg.message,
+          timestamp: Date.now(),
+        });
+        for (const client of set) {
+          if (client.readyState === 1) { // 1 === OPEN
+            client.send(payload);
+          }
+        }
+      }
+    },
+    close(ws) {
+      // Clean up on disconnect
+      for (const set of rooms.values()) set.delete(ws);
+    }
+  }
 });
+console.log("✅ Bun realtime server running on port " + (Bun.env.PORT || 7860));