Update server.js

server.js

@@ -1,4 +1,44 @@
-socket.on('chat message', (msg) => {
-    console.log(`Message received: ${msg}`);
-    io.emit('chat message', msg);
+const express = require('express');
+const http = require('http');
+const { Server } = require('socket.io');
+
+const app = express();
+const server = http.createServer(app);
+const io = new Server(server);
+
+// Serve static files from the public directory
+app.use(express.static('public'));
+
+// Store usernames for connected clients
+const users = {}; // Format: { socketId: username }
+
+io.on('connection', (socket) => {
+    console.log('A user connected:', socket.id);
+
+    // Set a default username when the user connects
+    users[socket.id] = "Anonymous";
+
+    // Handle username updates from clients
+    socket.on('set username', (username) => {
+        users[socket.id] = username || "Anonymous"; // Update username or default to "Anonymous"
+        console.log(`User ${socket.id} set their username to: ${users[socket.id]}`);
+    });
+
+    // Handle chat messages
+    socket.on('chat message', (msg) => {
+        const username = users[socket.id] || "Anonymous"; // Get the username for this socket
+        const formattedMessage = `${username}: ${msg}`; // Format message as "username: message"
+        io.emit('chat message', formattedMessage); // Broadcast the message to all clients
+    });
+
+    // Handle user disconnection
+    socket.on('disconnect', () => {
+        console.log('A user disconnected:', socket.id);
+        delete users[socket.id]; // Remove the user from the list
+    });
+});
+
+// Start the server
+server.listen(3000, () => {
+    console.log('Server is running on http://localhost:3000');
 });