sune-org/chatroom
1
0
Fork 0
mirror of https://github.com/sune-org/chatroom.git synced 2026-01-13 16:17:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

init chatroom

Browse Source
This commit is contained in:
multipleof4
2025-09-07 20:00:49 -07:00
parent 36f3d73b57
commit 856f3ba6b9
1 changed files with 42 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
index.js
View File

@@ -24,20 +24,51 @@ export default {
export class ChatsuneDurableObject {
constructor(state, env) {
this.state = state;
this.env = env;
this.sockets = new Set();
this.state.blockConcurrencyWhile(async () => this.messages = await this.state.storage.get('messages') || []);
}
broadcast(message) { this.sockets.forEach(s => s.readyState === WebSocket.OPEN && s.send(JSON.stringify(message))); }
broadcastConnectionCount() { this.broadcast({ type: 'CONNECTION_COUNT', payload: { count: this.sockets.size } }); }
async fetch(req) {
if (req.method === 'OPTIONS') return new Response(null, { status: 204, headers: CORS_HEADERS });
if (req.headers.get('Upgrade') === 'websocket') {
const [client, server] = Object.values(new WebSocketPair());
server.accept();
this.sockets.add(server);
server.addEventListener('close', () => this.sockets.delete(server));
server.addEventListener('message', () => {});
return new Response(null, { status: 101, webSocket: client });
}
return new Response(JSON.stringify({ ok: true }), { status: 200, headers: { 'Content-Type': 'application/json', ...CORS_HEADERS } });
if (req.headers.get('Upgrade') !== 'websocket') return new Response(JSON.stringify({ ok: true }), { status: 200, headers: { 'Content-Type': 'application/json', ...CORS_HEADERS } });
const [client, server] = Object.values(new WebSocketPair());
server.accept();
this.sockets.add(server);
server.addEventListener('message', async (event) => {
try {
const data = JSON.parse(event.data);
switch (data.type) {
case 'USER_JOINED': {
server.username = data.payload.name || `anon-${crypto.randomUUID().slice(0, 8)}`;
server.send(JSON.stringify({ type: 'HISTORY', payload: this.messages }));
this.broadcast({ type: 'NEW_MESSAGE', payload: { author: { name: 'system' }, text: `${server.username} has joined.`, timestamp: new Date().toISOString() } });
this.broadcastConnectionCount();
break;
}
case 'NEW_MESSAGE': {
if (!server.username) return;
const messagePayload = { author: { name: server.username }, text: data.payload.text, timestamp: new Date().toISOString() };
this.messages.push(messagePayload);
this.messages = this.messages.slice(-4);
this.broadcast({ type: 'NEW_MESSAGE', payload: messagePayload });
await this.state.storage.put('messages', this.messages);
break;
}
}
} catch (e) { /* Ignore invalid JSON */ }
});
server.addEventListener('close', () => {
this.sockets.delete(server);
if (server.username) this.broadcast({ type: 'NEW_MESSAGE', payload: { author: { name: 'system' }, text: `${server.username} has left.`, timestamp: new Date().toISOString() } });
this.broadcastConnectionCount();
});
return new Response(null, { status: 101, webSocket: client });
}
}