| const http = require("http"); |
| const fs = require("fs"); |
| const path = require("path"); |
| const { WebSocketServer } = require("ws"); |
|
|
| const PORT = process.env.PORT || 7860; |
|
|
| const httpServer = http.createServer((req, res) => { |
| let file = "index.html"; |
| if (req.url === "/admin.html") file = "admin.html"; |
| const filePath = path.join(__dirname, file); |
| fs.readFile(filePath, (err, data) => { |
| if (err) { res.writeHead(404); return res.end("Not found"); } |
| res.writeHead(200, { "Content-Type": "text/html" }); |
| res.end(data); |
| }); |
| }); |
|
|
| const wss = new WebSocketServer({ server: httpServer }); |
|
|
| let streamerWs = null; |
| let isLive = false; |
| const viewers = new Set(); |
|
|
| function sendTo(ws, obj) { |
| if (ws && ws.readyState === 1) ws.send(JSON.stringify(obj)); |
| } |
|
|
| wss.on("connection", (ws) => { |
| ws.role = null; |
| ws._vid = null; |
|
|
| ws.on("message", (raw) => { |
| let msg; try { msg = JSON.parse(raw); } catch { return; } |
|
|
| switch (msg.type) { |
| case "streamer_join": |
| ws.role = "streamer"; streamerWs = ws; |
| sendTo(ws, { type: "streamer_welcome", viewerCount: viewers.size }); |
| break; |
|
|
| case "viewer_join": |
| ws.role = "viewer"; ws._vid = Math.random().toString(36).slice(2); |
| viewers.add(ws); |
| sendTo(ws, { type: "viewer_welcome", isLive, vid: ws._vid }); |
| sendTo(streamerWs, { type: "viewer_count", count: viewers.size }); |
| if (isLive) sendTo(streamerWs, { type: "new_viewer", vid: ws._vid }); |
| break; |
|
|
| case "go_live": |
| if (ws.role !== "streamer") return; |
| isLive = true; |
| viewers.forEach(v => { sendTo(v, { type: "stream_started" }); }); |
| viewers.forEach(v => sendTo(streamerWs, { type: "new_viewer", vid: v._vid })); |
| break; |
|
|
| case "end_live": |
| if (ws.role !== "streamer") return; |
| isLive = false; |
| viewers.forEach(v => sendTo(v, { type: "stream_ended" })); |
| break; |
|
|
| case "offer": |
| if (ws.role !== "streamer") return; |
| viewers.forEach(v => { if (v._vid === msg.vid) sendTo(v, { type: "offer", sdp: msg.sdp }); }); |
| break; |
|
|
| case "answer": |
| sendTo(streamerWs, { type: "answer", sdp: msg.sdp, vid: ws._vid }); |
| break; |
|
|
| case "candidate": |
| if (ws.role === "streamer") { |
| viewers.forEach(v => { if (v._vid === msg.vid) sendTo(v, { type: "candidate", candidate: msg.candidate }); }); |
| } else { |
| sendTo(streamerWs, { type: "candidate", candidate: msg.candidate, vid: ws._vid }); |
| } |
| break; |
|
|
| case "viewer_ready": |
| ws._vid = msg.vid || ws._vid; |
| sendTo(streamerWs, { type: "new_viewer", vid: ws._vid }); |
| break; |
| } |
| }); |
|
|
| ws.on("close", () => { |
| if (ws.role === "streamer") { |
| streamerWs = null; isLive = false; |
| viewers.forEach(v => sendTo(v, { type: "stream_ended" })); |
| } else if (ws.role === "viewer") { |
| viewers.delete(ws); |
| sendTo(streamerWs, { type: "viewer_count", count: viewers.size }); |
| } |
| }); |
| }); |
|
|
| httpServer.listen(PORT, () => console.log(`Running on port ${PORT}`)); |
|
|
