Update app.js

app.js

@@ -1,25 +1,38 @@
 // server.js
+const path = require('path');
+const express = require('express');
 const http = require('http');
 const WebSocket = require('ws');
 const mediasoup = require('mediasoup');
 
+const app = express();
 const PORT = 7860;
+
+// Serve static files from ./public
+app.use(express.static(path.join(__dirname, 'public')));
+
 let worker, router;
 const producerTransports = new Map(); // ws → { transport, producer }
 const consumerTransports = new Map(); // ws → [ { transport, consumer } ]
 
-(async () => {
-  // 1) Start HTTP + WebSocket server
-  const server = http.createServer();
-  const wss = new WebSocket.Server({ server });
-  server.listen(PORT, () => console.log(`Server running on ${PORT}`));
-
-  // 2) Create a single Mediasoup Worker + Router
+;(async () => {
+  // 1) Create the mediasoup Worker + Router
   worker = await mediasoup.createWorker();
-  router = await worker.createRouter({ mediaCodecs: [
-    { kind: 'audio', mimeType: 'audio/opus', clockRate: 48000, channels: 2 },
-    { kind: 'video', mimeType: 'video/VP8', clockRate: 90000 },
-  ]});
+  router = await worker.createRouter({
+    mediaCodecs: [
+      { kind: 'audio', mimeType: 'audio/opus', clockRate: 48000, channels: 2 },
+      { kind: 'video', mimeType: 'video/VP8', clockRate: 90000 }
+    ]
+  });
+
+  // 2) Expose RTP capabilities via HTTP for the clients
+  app.get('/rtpCapabilities', (req, res) => {
+    res.json({ rtpCapabilities: router.rtpCapabilities });
+  });
+
+  // 3) Create HTTP + WebSocket server
+  const server = http.createServer(app);
+  const wss = new WebSocket.Server({ server });
 
   wss.on('connection', ws => {
     ws.on('message', async msg => {
@@ -27,12 +40,20 @@ const consumerTransports = new Map(); // ws → [ { transport, consumer } ]
 
       switch (action) {
         case 'createProducerTransport': {
-          // Create transport for broadcaster
-          const transport = await router.createWebRtcTransport({ listenIps: ['127.0.0.1'], enableUdp: true, enableTcp: true });
+          const transport = await router.createWebRtcTransport({
+            listenIps: ['127.0.0.1'],
+            enableUdp: true,
+            enableTcp: true
+          });
           producerTransports.set(ws, { transport, producer: null });
           ws.send(JSON.stringify({
             action: 'producerTransportCreated',
-            data: { id: transport.id, iceParameters: transport.iceParameters, iceCandidates: transport.iceCandidates, dtlsParameters: transport.dtlsParameters }
+            data: {
+              id: transport.id,
+              iceParameters: transport.iceParameters,
+              iceCandidates: transport.iceCandidates,
+              dtlsParameters: transport.dtlsParameters
+            }
           }));
           break;
         }
@@ -50,26 +71,36 @@ const consumerTransports = new Map(); // ws → [ { transport, consumer } ]
           const { transport } = producerTransports.get(ws);
           const producer = await transport.produce({ kind, rtpParameters });
           producerTransports.get(ws).producer = producer;
-          ws.send(JSON.stringify({ action: 'produced', data: { producerId: producer.id } }));
+          ws.send(JSON.stringify({
+            action: 'produced',
+            data: { producerId: producer.id }
+          }));
           break;
         }
 
         case 'createConsumerTransport': {
-          // Create transport for a viewer
-          const transport = await router.createWebRtcTransport({ listenIps: ['127.0.0.1'], enableUdp: true, enableTcp: true });
+          const transport = await router.createWebRtcTransport({
+            listenIps: ['127.0.0.1'],
+            enableUdp: true,
+            enableTcp: true
+          });
           const arr = consumerTransports.get(ws) || [];
           arr.push({ transport, consumer: null });
           consumerTransports.set(ws, arr);
           ws.send(JSON.stringify({
             action: 'consumerTransportCreated',
-            data: { id: transport.id, iceParameters: transport.iceParameters, iceCandidates: transport.iceCandidates, dtlsParameters: transport.dtlsParameters }
+            data: {
+              id: transport.id,
+              iceParameters: transport.iceParameters,
+              iceCandidates: transport.iceCandidates,
+              dtlsParameters: transport.dtlsParameters
+            }
           }));
           break;
         }
 
         case 'connectConsumerTransport': {
           const { dtlsParameters } = data;
-          // assume latest transport
           const arr = consumerTransports.get(ws);
           await arr[arr.length - 1].transport.connect({ dtlsParameters });
           ws.send(JSON.stringify({ action: 'consumerTransportConnected' }));
@@ -77,7 +108,6 @@ const consumerTransports = new Map(); // ws → [ { transport, consumer } ]
         }
 
         case 'consume': {
-          // find the broadcaster’s producer
           const producerEntry = [...producerTransports.values()].find(e => e.producer);
           if (!producerEntry) {
             ws.send(JSON.stringify({ action: 'consumeError', data: 'No producer available yet' }));
@@ -107,5 +137,24 @@ const consumerTransports = new Map(); // ws → [ { transport, consumer } ]
         }
       }
     });
+
+    ws.on('close', () => {
+      // Clean up transports & producers/consumers for this ws
+      const pEntry = producerTransports.get(ws);
+      if (pEntry) {
+        pEntry.producer?.close();
+        pEntry.transport.close();
+        producerTransports.delete(ws);
+      }
+      const cEntries = consumerTransports.get(ws) || [];
+      cEntries.forEach(({ consumer, transport }) => {
+        consumer?.close();
+        transport.close();
+      });
+      consumerTransports.delete(ws);
+    });
   });
+
+  // 4) Start the server
+  server.listen(PORT, () => console.log(`SFU demo listening on http://localhost:${PORT}`));
 })();