add chat system

Overlord-Server/src/sessions/sessionManager.ts

@@ -7,6 +7,7 @@ import type {
   VoiceViewer,
   NotificationsViewer,
   KeyloggerViewer,
+  ChatViewer,
   SocketData,
 } from "./types";
 
@@ -36,6 +37,7 @@ const keyloggerSessionsByClient = new Map<string, Set<string>>();
 const voiceSessions = new Map<string, VoiceViewer>();
 const voiceSessionsByClient = new Map<string, Set<string>>();
 const dashboardSessions = new Map<string, DashboardViewer>();
+const chatSessions = new Map<string, ChatViewer>();
 
 function addSessionToClientIndex(
   index: Map<string, Set<string>>,
@@ -495,3 +497,29 @@ export function notifyDashboardViewers(): void {
     }
   }, DASHBOARD_DEBOUNCE_MS);
 }
+
+export function addChatSession(session: ChatViewer): void {
+  chatSessions.set(session.id, session);
+}
+
+export function deleteChatSession(sessionId: string): boolean {
+  return chatSessions.delete(sessionId);
+}
+
+export function getAllChatSessions(): Map<string, ChatViewer> {
+  return chatSessions;
+}
+
+export function getChatSessionCount(): number {
+  return chatSessions.size;
+}
+
+export function broadcastChatMessage(msg: string): void {
+  for (const [id, session] of chatSessions) {
+    try {
+      session.viewer.send(msg);
+    } catch {
+      chatSessions.delete(id);
+    }
+  }
+}

Overlord-Server/src/sessions/types.ts

@@ -11,7 +11,8 @@ export type SocketRole =
   | "notifications_viewer"
   | "keylogger_viewer"
   | "voice_viewer"
-  | "dashboard_viewer";
+  | "dashboard_viewer"
+  | "chat_viewer";
 
 export type SocketData = {
   role: SocketRole;
@@ -77,3 +78,12 @@ export type VoiceViewer = {
   viewer: ServerWebSocket<SocketData>;
   createdAt: number;
 };
+
+export type ChatViewer = {
+  id: string;
+  viewer: ServerWebSocket<SocketData>;
+  createdAt: number;
+  userId: number;
+  username: string;
+  userRole: string;
+};