refactor(*): Move server into src

src/server/websocket.ts

@@ -0,0 +1,154 @@
+import express from 'express';
+import { WebSocketServer, WebSocket } from 'ws';
+import { Room } from './classes/Room.js';
+import { User } from './classes/User.js';
+import { JoinRoomMessage } from './classes/messages/JoinRoomMessage.js';
+import { ChangeEstimateMessage } from './classes/messages/ChangeEstimateMessage.js';
+import { SelectEstimateMessage } from './classes/messages/SelectEstimateMessage.js';
+
+const app = express();
+const port = process.env.PORT || 3000;
+
+function onSocketPreError(e: Error) {
+	console.log(e);
+}
+
+function onSocketPostError(e: Error) {
+	console.log(e);
+}
+
+console.log(`Attempting to run server on port ${port}`);
+
+const server = app.listen(port, () => {
+	console.log(`Listening on port ${port}`);
+});
+
+const wss = new WebSocketServer({ noServer: true });
+const rooms = new Set<Room>();
+
+server.on('upgrade', (req, socket, head) => {
+	socket.on('error', onSocketPreError);
+
+	// perform auth
+	if (!!req.headers['BadAuth']) {
+		socket.write('HTTP/1.1 401 Unauthorized\r\n\r\n');
+		socket.destroy();
+		return;
+	}
+
+	wss.handleUpgrade(req, socket, head, (ws) => {
+		socket.removeListener('error', onSocketPreError);
+		wss.emit('connection', ws, req);
+	});
+});
+
+wss.on('connection', (ws: WebSocket, req) => {
+	let room: Room | null = null;
+
+	ws.on('error', onSocketPostError);
+
+	ws.on('message', (message) => {
+		const data = JSON.parse(message.toString());
+
+		if (data.type === 'join-room') {
+			const joinRoomMessage: JoinRoomMessage = data as JoinRoomMessage;
+			const { roomId, userId, name } = joinRoomMessage;
+			room = getRoomById(roomId);
+
+			// Create room if it doesn't already exist
+			if (!room) {
+				room = new Room(roomId);
+				rooms.add(room);
+			}
+
+			// Create the user in the room
+			const user = new User(userId, name);
+			room.addUser(ws, user);
+
+			broadcastToRoom(roomId, { type: 'user-joined', message: `${name} has joined the room` });
+		} else if (data.type === 'select-estimate') {
+			const selectEstimateMessage: SelectEstimateMessage = data as SelectEstimateMessage;
+			const { userId, estimate } = selectEstimateMessage;
+
+			if (room) {
+				const user = room.getUserByWebSocket(ws);
+				if (user) {
+					user.estimate = estimate;
+					broadcastToRoom(room.id, { type: 'estimate-selected', userId, estimate });
+
+					const allEstimates = room.getAllEstimates();
+					if (allEstimates.size === room.getUsers().length) {
+						broadcastToRoom(room.id, { type: 'estimation-closed', message: 'Estimation closed' });
+					}
+				}
+			}
+		} else if (data.type === 'change-estimate') {
+			const changeEstimateMessage: ChangeEstimateMessage = data as ChangeEstimateMessage;
+			const { userId, estimate } = changeEstimateMessage;
+
+			if (room) {
+				const user = room.getUserByWebSocket(ws);
+				if (user) {
+					user.estimate = estimate;
+					broadcastToRoom(room.id, { type: 'estimate-changed', userId, estimate });
+				}
+
+				const allEstimates = room.getAllEstimates();
+				if (allEstimates.size === room.getUsers().length) {
+					broadcastToRoom(room.id, { type: 'estimation-closed', message: 'Estimation closed' });
+				}
+			}
+		} else if (data.type === 'restart-estimation') {
+			if (room) {
+				room.clearEstimates();
+				broadcastToRoom(room.id, { type: 'estimation-restarted', message: 'Estimation restarted' });
+			}
+		}
+	});
+
+	ws.on('close', () => {
+		if (room) {
+			const user = room.getUserByWebSocket(ws);
+			if (user) {
+				room.removeUser(ws);
+
+				if (room.getUsers().length === 0) {
+					rooms.delete(room);
+					broadcastToRoom(room.id, {
+						type: 'room-deleted',
+						message: 'Room deleted'
+					});
+				}
+
+				broadcastToRoom(room.id, {
+					type: 'user-left',
+					message: `${user.userId} has left the room`
+				});
+			}
+		}
+	});
+});
+
+function broadcastToRoom(roomId: string, message: any) {
+	const room = getRoomById(roomId);
+	if (room) {
+		room.broadcast(message);
+	}
+}
+
+function getUsersInRoom(roomId: string) {
+	const room = getRoomById(roomId);
+	if (room) {
+		return room.getUsers();
+	}
+	return [];
+}
+
+function getRoomById(roomId: string) {
+	for (const room of rooms) {
+		if (room.id === roomId) {
+			return room;
+		}
+	}
+	return null;
+}