Send ConnectedUsersMessages when user requests desiredName change

server.node.ts

@@ -11,6 +11,7 @@ import type {
   TypingMessage,
   ConnectedUser,
   ConnectedUsersMessage,
+  DesiredNameMessage,
 } from "./lib/ServerMessage";
 
 import {
@@ -97,43 +98,78 @@ async function handleCloseConnection(id: string) {
     desiredNames.delete(id);
   }
 
-  for (const to of webSocketServer.clients) {
+  const connected = 
-    to.send(
-      JSON.stringify({
-        type: MessageType.CONNECTED_USERS,
-        __ctx: `${id} left`,
-        date: Date.now(),
-        connected: 
     Array.from(activeConnections,
       (id) => {
         return {
           id,
           desiredName: desiredNames.get(id),
         } as ConnectedUser;
-            }),
+      });
+  for (const to of webSocketServer.clients) {
+    to.send(
+      JSON.stringify({
+        type: MessageType.CONNECTED_USERS,
+        __ctx: `${id} left`,
+        date: Date.now(),
+        connected,
       } as ConnectedUsersMessage)
     );
   }
-
 }
 
 function handleNewConnection(id: string) {
   activeConnections.add(id);
 
+  const connected = 
+    Array.from(activeConnections,
+      (id) => {
+        return {
+          id,
+          desiredName: desiredNames.get(id),
+        } as ConnectedUser;
+      });
+
   for (const to of webSocketServer.clients) {
     to.send(
       JSON.stringify({
         type: MessageType.CONNECTED_USERS,
         __ctx: `${id} joined`,
         date: Date.now(),
-        connected: 
+        connected,
+      } as ConnectedUsersMessage)
+    );
+  }
+}
+
+async function handleDesiredNameMessage(message: DesiredNameMessage, from: string) {
+  if (!activeConnections.has(from) && desiredNames.has(from)) {
+    desiredNames.delete(from);
+    return;
+  }
+
+  console.debug(`${from} set desiredName to ${message.desiredName}`);
+  if (message.desiredName === undefined) {
+    desiredNames.delete(from);
+  } else {
+    desiredNames.set(from, message.desiredName);
+  }
+  
+  const connected = 
     Array.from(activeConnections,
       (id) => {
         return {
           id,
           desiredName: desiredNames.get(id),
         } as ConnectedUser;
-            }),
+      });
+  for (const to of webSocketServer.clients) {
+    to.send(
+      JSON.stringify({
+        type: MessageType.CONNECTED_USERS,
+        __ctx: `${from} changed desiredName to ${message.desiredName}`,
+        date: Date.now(),
+        connected,
       } as ConnectedUsersMessage)
     );
   }
@@ -171,9 +207,10 @@ webSocketServer.on("connection", function connection(socket) {
     } else if (isTypingMessage(message)) {
       handleTypingMessage(message, authorId);
     } else if (isDesiredNameMessage(message)) {
-      desiredNames.set(authorId, message.desiredName);
+      handleDesiredNameMessage(message, authorId);
     }
   });
+
   socket.on("close", function close() {
     console.log("closed a connection");
   });