Spaces:
Sleeping
Sleeping
File size: 5,372 Bytes
67d31ad 003dc51 7fe989f 67d31ad 7fe989f 67d31ad d3bdf51 b977bfa 003dc51 b977bfa 67d31ad be81bd0 003dc51 b977bfa be81bd0 67d31ad be81bd0 7fe989f be81bd0 d441744 be81bd0 d441744 be81bd0 7fe989f 003dc51 7fe989f be81bd0 7fe989f be81bd0 003dc51 be81bd0 003dc51 be81bd0 b977bfa be81bd0 b977bfa 67d31ad 003dc51 be81bd0 7fe989f be81bd0 7fe989f be81bd0 b977bfa be81bd0 b977bfa be81bd0 904caf7 d3bdf51 b977bfa d3bdf51 67d31ad b977bfa 67d31ad b977bfa 67d31ad 7fe989f 6d4d78a 003dc51 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 | const rooms = new Map(); // roomId ⇒ Set<ServerWebSocket>
const HARDCODED_TOKEN = "test-token-123";
Bun.serve({
port: Number(Bun.env.PORT) || 7860,
fetch(req, server) {
const url = new URL(req.url);
// Handle CORS preflight
if (req.method === "OPTIONS") {
return new Response(null, {
status: 204,
headers: {
"Access-Control-Allow-Origin": "*",
"Access-Control-Allow-Methods": "GET, POST, OPTIONS",
"Access-Control-Allow-Headers": "Content-Type, Upgrade, Authorization",
"Access-Control-Allow-Credentials": "true",
},
});
}
// Upgrade to WebSocket if requested
if (req.headers.get("upgrade")?.toLowerCase() === "websocket") {
// Bearer token check
const auth = req.headers.get("authorization");
if (auth !== `Bearer ${HARDCODED_TOKEN}`) {
return new Response("Unauthorized", { status: 401, headers: { "Access-Control-Allow-Origin": "*" } });
}
// Accept the upgrade with CORS headers on the handshake
const upgradeRes = server.upgrade(req, {
headers: {
"Access-Control-Allow-Origin": "*",
"Access-Control-Allow-Credentials": "true",
},
});
return upgradeRes ?? new Response("Upgrade failed", { status: 500 });
}
// Serve HTML UI for non-WebSocket requests
if (url.pathname === "/") {
return new Response(`<!doctype html>
<html><head>
<title>Dubem Realtime Rooms</title>
<meta charset="utf-8"/>
<style>
body { font-family: sans-serif; padding: 20px; background: #f0f0f0; }
#log { border: 1px solid #ccc; height:200px; overflow-y:auto; padding:10px; background:#fff;}
input, button { margin:5px 0; padding:8px; font-size:16px; }
</style>
</head><body>
<h2>Join a Room & Send Messages</h2>
<input id="room" placeholder="Room ID"/>
<button onclick="joinRoom()">Join Room</button>
<div>
<input type="checkbox" id="sendToken" checked>Send Bearer Token
</div>
<div id="log"></div>
<input id="msg" placeholder="Type a message" style="width:80%;"/>
<button onclick="sendMsg()">Send</button>
<script>
var ws, currentRoom;
function connectWS() {
var scheme = location.protocol === 'https:' ? 'wss' : 'ws';
var socketUrl = scheme + '://' + location.host + '/';
var sendToken = document.getElementById('sendToken').checked;
if (ws && ws.readyState === 1) ws.close();
// Bun supports custom headers in WebSocket constructor, browsers do not!
try {
ws = sendToken
? new WebSocket(socketUrl, { headers: { "Authorization": "Bearer test-token-123" } })
: new WebSocket(socketUrl);
} catch (e) {
log('Custom headers not supported in this environment.');
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(){ log('⚠️ WebSocket error'); };
ws.onclose = function(c){ log('❌ Disconnected (code='+c.code+')'); };
}
window.addEventListener('load', connectWS);
document.getElementById('sendToken').addEventListener('change', connectWS);
function joinRoom(){
var id = document.getElementById('room').value.trim();
if(!id) return alert('Enter room ID');
ws.send(JSON.stringify({ action:'join', roomId:id }));
currentRoom = id; log('➡️ Joined '+id);
}
function sendMsg(){
var t = document.getElementById('msg').value.trim();
if(!t) return;
if(!currentRoom) return alert('Join a room first');
ws.send(JSON.stringify({ action:'post', roomId:currentRoom, message:t }));
document.getElementById('msg').value = '';
}
function log(txt){
var e = document.getElementById('log');
e.innerHTML += '<div>'+txt+'</div>';
e.scrollTop = e.scrollHeight;
}
</script>
</body></html>`, {
headers: {
"Content-Type": "text/html; charset=utf-8",
"Access-Control-Allow-Origin": "*",
}
});
}
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; }
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) { // 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)); |