rename

src/handlers/MessageHandler.ts

@@ -30,6 +30,8 @@ export class MessageHandler {
         }
 
         const { room_id, username, message } = parsedData
+        let { new_username } = parsedData
+        if (!new_username) new_username = username
         // Normalize action to ensure case-insensitivity matches logic
         const action = parsedData.action.toLowerCase()
 
@@ -64,6 +66,48 @@ export class MessageHandler {
                 }
                 break
 
+            case SocketAction.RENAME: {
+                if (!new_username || new_username === username) {
+                    socket.send(
+                        JSON.stringify({
+                            error: 'Invalid new username',
+                            status: 0,
+                        })
+                    )
+                    return
+                }
+
+                const renamed = roomManager.renameUser(
+                    room_id,
+                    username,
+                    new_username,
+                    socket
+                )
+
+                if (!renamed) {
+                    socket.send(
+                        JSON.stringify({
+                            error: 'Username rename failed',
+                            status: 0,
+                        })
+                    )
+                    return
+                }
+
+                roomManager.broadcast(
+                    room_id,
+                    JSON.stringify({
+                        message: `${username} is now known as ${new_username}`,
+                        username: new_username,
+                        old_username: username,
+                        room_id,
+                        action: 'rename',
+                        status: 1,
+                    })
+                )
+                break
+            }
+
             case SocketAction.MESSAGE:
                 roomManager.broadcast(
                     room_id,

src/services/RoomManager.ts

@@ -19,6 +19,26 @@ export class RoomManager {
         this.rooms.get(roomId)?.set(username, socket)
     }
 
+    renameUser(
+        roomId: string,
+        username: string,
+        newUsername: string,
+        socket: WebSocket
+    ): boolean {
+        const room = this.rooms.get(roomId)
+        if (!room) return false
+
+        if (!room.has(username)) return false
+        if (room.has(newUsername)) return false
+
+        // Remove old username
+        room.delete(username)
+
+        // Add new username with same socket
+        room.set(newUsername, socket)
+        return true
+    }
+
     /**
      * Removes a user from a room.
      * Cleans up the room if it becomes empty.

src/types.ts

@@ -2,6 +2,7 @@ export enum SocketAction {
     JOIN = 'join',
     LEAVE = 'leave',
     MESSAGE = 'message',
+    RENAME = 'rename',
 }
 
 export interface WebSocketMessage {