ws handler added
This commit is contained in:
KeshavAnandCode
289
src/ws-handler.ts
Normal file
289
src/ws-handler.ts
Normal file
@@ -0,0 +1,289 @@
|
|||||||
|
import type { ServerWebSocket } from "bun";
|
||||||
|
import {
|
||||||
|
rooms, wsToPlayer,
|
||||||
|
Room, Player,
|
||||||
|
generateId, defaultSettings, freshBuzzerState, sanitize,
|
||||||
|
getPublicRoom, broadcastToRoom, sendToMod,
|
||||||
|
} from "./rooms";
|
||||||
|
|
||||||
|
type WS = ServerWebSocket<unknown>;
|
||||||
|
|
||||||
|
function send(ws: WS, msg: object) {
|
||||||
|
try { ws.send(JSON.stringify(msg)); } catch {}
|
||||||
|
}
|
||||||
|
|
||||||
|
function err(ws: WS, message: string) {
|
||||||
|
send(ws, { type: "error", message });
|
||||||
|
}
|
||||||
|
|
||||||
|
export function handleMessage(ws: WS, raw: string) {
|
||||||
|
let msg: any;
|
||||||
|
try { msg = JSON.parse(raw); } catch { return; }
|
||||||
|
|
||||||
|
const type: string = msg?.type ?? "";
|
||||||
|
|
||||||
|
// ── CREATE ROOM ──────────────────────────────────────────────────────────
|
||||||
|
if (type === "create_room") {
|
||||||
|
const roomId = generateId(6);
|
||||||
|
const modSecret = generateId(24);
|
||||||
|
const room: Room = {
|
||||||
|
id: roomId,
|
||||||
|
moderatorSecret: modSecret,
|
||||||
|
modWs: ws,
|
||||||
|
settings: defaultSettings(),
|
||||||
|
players: new Map(),
|
||||||
|
buzzerState: freshBuzzerState(),
|
||||||
|
locked: false,
|
||||||
|
createdAt: Date.now(),
|
||||||
|
};
|
||||||
|
rooms.set(roomId, room);
|
||||||
|
send(ws, { type: "room_created", roomId, modSecret, room: getPublicRoom(room) });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ── MOD REJOIN ────────────────────────────────────────────────────────────
|
||||||
|
if (type === "mod_rejoin") {
|
||||||
|
const { roomId, modSecret } = msg;
|
||||||
|
const room = rooms.get(sanitize(roomId, 12));
|
||||||
|
if (!room || room.moderatorSecret !== sanitize(modSecret, 32)) {
|
||||||
|
err(ws, "Invalid moderator credentials"); return;
|
||||||
|
}
|
||||||
|
room.modWs = ws;
|
||||||
|
send(ws, { type: "mod_joined", room: getPublicRoom(room) });
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) }, ws);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ── JOIN ROOM (player) ────────────────────────────────────────────────────
|
||||||
|
if (type === "join_room") {
|
||||||
|
const roomId = sanitize(msg.roomId ?? "", 12);
|
||||||
|
const room = rooms.get(roomId);
|
||||||
|
if (!room) { err(ws, "Room not found"); return; }
|
||||||
|
if (room.locked) { err(ws, "Room is locked"); return; }
|
||||||
|
|
||||||
|
const name = sanitize(msg.playerName ?? "Player", 24) || "Player";
|
||||||
|
const existingId = sanitize(msg.playerId ?? "", 12);
|
||||||
|
|
||||||
|
// Block anyone trying to present the mod secret as a player
|
||||||
|
if (msg.modSecret) { err(ws, "Unauthorized"); return; }
|
||||||
|
|
||||||
|
let player: Player | undefined;
|
||||||
|
if (existingId && room.players.has(existingId)) {
|
||||||
|
player = room.players.get(existingId)!;
|
||||||
|
player.ws = ws;
|
||||||
|
player.isConnected = true;
|
||||||
|
player.name = name;
|
||||||
|
} else {
|
||||||
|
const playerId = generateId();
|
||||||
|
player = {
|
||||||
|
id: playerId, name,
|
||||||
|
teamIndex: null, score: 0,
|
||||||
|
ws, isConnected: true,
|
||||||
|
joinedAt: Date.now(),
|
||||||
|
};
|
||||||
|
room.players.set(playerId, player);
|
||||||
|
}
|
||||||
|
|
||||||
|
wsToPlayer.set(ws, { roomId, playerId: player.id });
|
||||||
|
send(ws, { type: "joined", playerId: player.id, room: getPublicRoom(room) });
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) }, ws);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ── BUZZ ──────────────────────────────────────────────────────────────────
|
||||||
|
if (type === "buzz") {
|
||||||
|
const ctx = wsToPlayer.get(ws);
|
||||||
|
if (!ctx) { err(ws, "Not a player"); return; }
|
||||||
|
const room = rooms.get(ctx.roomId);
|
||||||
|
if (!room) return;
|
||||||
|
const bz = room.buzzerState;
|
||||||
|
|
||||||
|
if (!bz.roundOpen) { send(ws, { type: "buzz_rejected", reason: "Round not open" }); return; }
|
||||||
|
if (bz.buzzOrder.includes(ctx.playerId) && !room.settings.allowRebuzz) {
|
||||||
|
send(ws, { type: "buzz_rejected", reason: "Already buzzed" }); return;
|
||||||
|
}
|
||||||
|
if (room.settings.buzzerLockout && bz.buzzOrder.length > 0) {
|
||||||
|
send(ws, { type: "buzz_rejected", reason: "Buzzer locked out" }); return;
|
||||||
|
}
|
||||||
|
|
||||||
|
bz.buzzOrder.push(ctx.playerId);
|
||||||
|
bz.buzzTimes.set(ctx.playerId, Date.now());
|
||||||
|
bz.active = true;
|
||||||
|
|
||||||
|
const player = room.players.get(ctx.playerId)!;
|
||||||
|
const publicOrder = room.settings.showOrder ? bz.buzzOrder : [bz.buzzOrder[0]];
|
||||||
|
|
||||||
|
broadcastToRoom(room, {
|
||||||
|
type: "buzz_event",
|
||||||
|
playerId: ctx.playerId,
|
||||||
|
playerName: player.name,
|
||||||
|
teamIndex: player.teamIndex,
|
||||||
|
buzzOrder: publicOrder,
|
||||||
|
room: getPublicRoom(room),
|
||||||
|
});
|
||||||
|
// mod gets full order + timestamps always
|
||||||
|
sendToMod(room, {
|
||||||
|
type: "buzz_event",
|
||||||
|
playerId: ctx.playerId,
|
||||||
|
playerName: player.name,
|
||||||
|
teamIndex: player.teamIndex,
|
||||||
|
buzzOrder: bz.buzzOrder,
|
||||||
|
buzzTimes: Object.fromEntries(bz.buzzTimes),
|
||||||
|
room: getPublicRoom(room),
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ── MOD-ONLY FROM HERE ────────────────────────────────────────────────────
|
||||||
|
// find room where this ws is the mod
|
||||||
|
let room: Room | undefined;
|
||||||
|
for (const r of rooms.values()) {
|
||||||
|
if (r.modWs === ws) { room = r; break; }
|
||||||
|
}
|
||||||
|
if (!room) { err(ws, "Not authorized"); return; }
|
||||||
|
|
||||||
|
switch (type) {
|
||||||
|
case "open_round": {
|
||||||
|
room.buzzerState = freshBuzzerState();
|
||||||
|
room.buzzerState.roundOpen = true;
|
||||||
|
broadcastToRoom(room, { type: "round_open", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "round_open", room: getPublicRoom(room) });
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "close_round": {
|
||||||
|
room.buzzerState.roundOpen = false;
|
||||||
|
broadcastToRoom(room, { type: "round_closed", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "round_closed", room: getPublicRoom(room) });
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "reset_buzzer": {
|
||||||
|
room.buzzerState = freshBuzzerState();
|
||||||
|
broadcastToRoom(room, { type: "buzzer_reset", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "buzzer_reset", room: getPublicRoom(room) });
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "update_settings": {
|
||||||
|
const s = msg.settings ?? {};
|
||||||
|
const st = room.settings;
|
||||||
|
if (typeof s.teamMode === "boolean") st.teamMode = s.teamMode;
|
||||||
|
if (typeof s.numTeams === "number") st.numTeams = Math.min(8, Math.max(2, s.numTeams | 0));
|
||||||
|
if (Array.isArray(s.teamNames)) st.teamNames = s.teamNames.slice(0, 8).map((n: unknown) => sanitize(n, 32));
|
||||||
|
if (typeof s.allowRebuzz === "boolean") st.allowRebuzz = s.allowRebuzz;
|
||||||
|
if (typeof s.buzzerLockout === "boolean") st.buzzerLockout = s.buzzerLockout;
|
||||||
|
if (typeof s.showOrder === "boolean") st.showOrder = s.showOrder;
|
||||||
|
if (typeof s.pointsCorrect === "number") st.pointsCorrect = Math.min(10000, Math.max(0, s.pointsCorrect | 0));
|
||||||
|
if (typeof s.pointsIncorrect === "number") st.pointsIncorrect = Math.min(10000, Math.max(0, s.pointsIncorrect | 0));
|
||||||
|
if (typeof s.pointsNeg === "boolean") st.pointsNeg = s.pointsNeg;
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "settings_updated", room: getPublicRoom(room) });
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "assign_team": {
|
||||||
|
const p = room.players.get(sanitize(msg.playerId, 12));
|
||||||
|
if (p) {
|
||||||
|
p.teamIndex = typeof msg.teamIndex === "number" ? Math.max(0, Math.min(7, msg.teamIndex | 0)) : null;
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "adjust_score": {
|
||||||
|
const p = room.players.get(sanitize(msg.playerId, 12));
|
||||||
|
if (p && typeof msg.delta === "number") {
|
||||||
|
p.score = Math.max(-99999, Math.min(99999, p.score + (msg.delta | 0)));
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "set_score": {
|
||||||
|
const p = room.players.get(sanitize(msg.playerId, 12));
|
||||||
|
if (p && typeof msg.score === "number") {
|
||||||
|
p.score = Math.max(-99999, Math.min(99999, msg.score | 0));
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "mark_correct": {
|
||||||
|
const p = room.players.get(sanitize(msg.playerId, 12));
|
||||||
|
if (p) {
|
||||||
|
p.score += room.settings.pointsCorrect;
|
||||||
|
room.buzzerState = freshBuzzerState();
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "mark_incorrect": {
|
||||||
|
const p = room.players.get(sanitize(msg.playerId, 12));
|
||||||
|
if (p && room.settings.pointsNeg) {
|
||||||
|
p.score -= room.settings.pointsIncorrect;
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "kick_player": {
|
||||||
|
const p = room.players.get(sanitize(msg.playerId, 12));
|
||||||
|
if (p) {
|
||||||
|
if (p.ws) { try { p.ws.send(JSON.stringify({ type: "kicked" })); } catch {} }
|
||||||
|
room.players.delete(p.id);
|
||||||
|
wsToPlayer.delete(p.ws!);
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "lock_room": {
|
||||||
|
room.locked = msg.locked === true;
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "reset_scores": {
|
||||||
|
for (const p of room.players.values()) p.score = 0;
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "reset_teams": {
|
||||||
|
for (const p of room.players.values()) p.teamIndex = null;
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
send(ws, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case "end_room": {
|
||||||
|
broadcastToRoom(room, { type: "room_ended" });
|
||||||
|
for (const p of room.players.values()) {
|
||||||
|
if (p.ws) wsToPlayer.delete(p.ws);
|
||||||
|
}
|
||||||
|
room.modWs = null;
|
||||||
|
rooms.delete(room.id);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function handleClose(ws: WS) {
|
||||||
|
// mod disconnect?
|
||||||
|
for (const room of rooms.values()) {
|
||||||
|
if (room.modWs === ws) {
|
||||||
|
room.modWs = null;
|
||||||
|
broadcastToRoom(room, { type: "mod_disconnected" });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// player disconnect
|
||||||
|
const ctx = wsToPlayer.get(ws);
|
||||||
|
if (ctx) {
|
||||||
|
wsToPlayer.delete(ws);
|
||||||
|
const room = rooms.get(ctx.roomId);
|
||||||
|
if (room) {
|
||||||
|
const p = room.players.get(ctx.playerId);
|
||||||
|
if (p) { p.isConnected = false; p.ws = null; }
|
||||||
|
broadcastToRoom(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
sendToMod(room, { type: "room_update", room: getPublicRoom(room) });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user