@principal-ai/control-tower-core 0.2.0 → 0.2.1

package/dist/index.mjs

@@ -2856,6 +2856,134 @@ var require_websocket_server = __commonJS((exports, module) => {
     }
   }
 });
+// src/abstractions/RoomManager.ts
+class RoomManager {
+  rooms = new Map;
+}
+
+// src/abstractions/DefaultRoomManager.ts
+class DefaultRoomManager extends RoomManager {
+  async createRoom(id, config) {
+    const room = {
+      id,
+      name: config.name || id,
+      createdAt: Date.now(),
+      maxUsers: config.maxUsers,
+      maxHistory: config.maxHistory,
+      permissions: config.permissions,
+      metadata: config.metadata || {}
+    };
+    const roomState = {
+      room,
+      users: new Map,
+      eventHistory: [],
+      locks: new Map
+    };
+    this.rooms.set(id, roomState);
+    return room;
+  }
+  async deleteRoom(id) {
+    this.rooms.delete(id);
+  }
+  async joinRoom(roomId, user) {
+    const roomState = this.rooms.get(roomId);
+    if (!roomState) {
+      throw new Error(`Room ${roomId} not found`);
+    }
+    if (roomState.room.maxUsers && roomState.users.size >= roomState.room.maxUsers) {
+      throw new Error(`Room ${roomId} is full`);
+    }
+    roomState.users.set(user.id, user);
+  }
+  async leaveRoom(roomId, userId) {
+    const roomState = this.rooms.get(roomId);
+    if (!roomState) {
+      return;
+    }
+    roomState.users.delete(userId);
+  }
+  async broadcastToRoom(roomId, event, _excludeUserId) {
+    const roomState = this.rooms.get(roomId);
+    if (!roomState) {
+      throw new Error(`Room ${roomId} not found`);
+    }
+    await this.addEventToHistory(roomId, event);
+  }
+  async getRoomState(roomId) {
+    return this.rooms.get(roomId) || null;
+  }
+  async getUsersInRoom(roomId) {
+    const roomState = this.rooms.get(roomId);
+    if (!roomState) {
+      return [];
+    }
+    return Array.from(roomState.users.values());
+  }
+  async addEventToHistory(roomId, event) {
+    const roomState = this.rooms.get(roomId);
+    if (!roomState) {
+      throw new Error(`Room ${roomId} not found`);
+    }
+    roomState.eventHistory.push(event);
+    if (roomState.room.maxHistory && roomState.eventHistory.length > roomState.room.maxHistory) {
+      roomState.eventHistory = roomState.eventHistory.slice(-roomState.room.maxHistory);
+    }
+  }
+  async getEventHistory(roomId, limit) {
+    const roomState = this.rooms.get(roomId);
+    if (!roomState) {
+      return [];
+    }
+    const history = roomState.eventHistory;
+    return limit ? history.slice(-limit) : history;
+  }
+  async updateUserStatus(roomId, userId, status) {
+    const roomState = this.rooms.get(roomId);
+    if (!roomState) {
+      throw new Error(`Room ${roomId} not found`);
+    }
+    const user = roomState.users.get(userId);
+    if (user) {
+      user.status = status;
+      user.lastActivity = Date.now();
+    }
+  }
+}
+
+// src/abstractions/ConnectedRoomManager.ts
+class ConnectedRoomManager extends DefaultRoomManager {
+  sendToClient;
+  getClientIdsInRoom;
+  constructor(sendToClient, getClientIdsInRoom) {
+    super();
+    this.sendToClient = sendToClient;
+    this.getClientIdsInRoom = getClientIdsInRoom;
+  }
+  async broadcastToRoom(roomId, event, excludeUserId) {
+    const roomState = this.rooms.get(roomId);
+    if (!roomState) {
+      throw new Error(`Room ${roomId} not found`);
+    }
+    await this.addEventToHistory(roomId, event);
+    const users = Array.from(roomState.users.values());
+    const sendPromises = [];
+    for (const user of users) {
+      if (excludeUserId && user.id === excludeUserId) {
+        continue;
+      }
+      const clientIds = this.getClientIdsInRoom(roomId, user.id);
+      for (const clientId of clientIds) {
+        sendPromises.push(this.sendToClient(clientId, {
+          type: "event_received",
+          event
+        }).catch((error) => {
+          console.error(`[ConnectedRoomManager] Failed to send event to client ${clientId}:`, error);
+        }));
+      }
+    }
+    await Promise.all(sendPromises);
+  }
+}
 // src/abstractions/LockManager.ts
 class LockManager {
   lockState = {
@@ -3374,99 +3502,6 @@ class DefaultPresenceManager extends PresenceManager {
     return this.userPresences.size + this.gracePeriodEntries.size;
   }
 }
-// src/abstractions/RoomManager.ts
-class RoomManager {
-  rooms = new Map;
-}
-
-// src/abstractions/DefaultRoomManager.ts
-class DefaultRoomManager extends RoomManager {
-  async createRoom(id, config) {
-    const room = {
-      id,
-      name: config.name || id,
-      createdAt: Date.now(),
-      maxUsers: config.maxUsers,
-      maxHistory: config.maxHistory,
-      permissions: config.permissions,
-      metadata: config.metadata || {}
-    };
-    const roomState = {
-      room,
-      users: new Map,
-      eventHistory: [],
-      locks: new Map
-    };
-    this.rooms.set(id, roomState);
-    return room;
-  }
-  async deleteRoom(id) {
-    this.rooms.delete(id);
-  }
-  async joinRoom(roomId, user) {
-    const roomState = this.rooms.get(roomId);
-    if (!roomState) {
-      throw new Error(`Room ${roomId} not found`);
-    }
-    if (roomState.room.maxUsers && roomState.users.size >= roomState.room.maxUsers) {
-      throw new Error(`Room ${roomId} is full`);
-    }
-    roomState.users.set(user.id, user);
-  }
-  async leaveRoom(roomId, userId) {
-    const roomState = this.rooms.get(roomId);
-    if (!roomState) {
-      return;
-    }
-    roomState.users.delete(userId);
-  }
-  async broadcastToRoom(roomId, event, _excludeUserId) {
-    const roomState = this.rooms.get(roomId);
-    if (!roomState) {
-      throw new Error(`Room ${roomId} not found`);
-    }
-    await this.addEventToHistory(roomId, event);
-  }
-  async getRoomState(roomId) {
-    return this.rooms.get(roomId) || null;
-  }
-  async getUsersInRoom(roomId) {
-    const roomState = this.rooms.get(roomId);
-    if (!roomState) {
-      return [];
-    }
-    return Array.from(roomState.users.values());
-  }
-  async addEventToHistory(roomId, event) {
-    const roomState = this.rooms.get(roomId);
-    if (!roomState) {
-      throw new Error(`Room ${roomId} not found`);
-    }
-    roomState.eventHistory.push(event);
-    if (roomState.room.maxHistory && roomState.eventHistory.length > roomState.room.maxHistory) {
-      roomState.eventHistory = roomState.eventHistory.slice(-roomState.room.maxHistory);
-    }
-  }
-  async getEventHistory(roomId, limit) {
-    const roomState = this.rooms.get(roomId);
-    if (!roomState) {
-      return [];
-    }
-    const history = roomState.eventHistory;
-    return limit ? history.slice(-limit) : history;
-  }
-  async updateUserStatus(roomId, userId, status) {
-    const roomState = this.rooms.get(roomId);
-    if (!roomState) {
-      throw new Error(`Room ${roomId} not found`);
-    }
-    const user = roomState.users.get(userId);
-    if (user) {
-      user.status = status;
-      user.lastActivity = Date.now();
-    }
-  }
-}
 // src/abstractions/EventEmitter.ts
 class TypedEventEmitter {
   listeners = new Map;
@@ -6764,6 +6799,13 @@ class BaseServer extends TypedEventEmitter {
   getClientIdsForUser(userId) {
     return Array.from(this.userClientMap.get(userId) || []);
   }
+  getClientIdsInRoom(roomId, userId) {
+    const allClientIds = this.getClientIdsForUser(userId);
+    return allClientIds.filter((clientId) => {
+      const client = this.clients.get(clientId);
+      return client && client.roomIds.has(roomId);
+    });
+  }
   getDeviceIdFromClientId(clientId) {
     return clientId;
   }
@@ -6857,6 +6899,11 @@ class ServerBuilder {
       }
     };
     const server = new BaseServer(config);
+    if (this.roomManager.constructor.name === "DefaultRoomManager") {
+      const connectedManager = new ConnectedRoomManager((clientId, message) => server.sendToClient(clientId, message), (roomId, userId) => server.getClientIdsInRoom(roomId, userId));
+      connectedManager.rooms = this.roomManager.rooms;
+      server.roomManager = connectedManager;
+    }
     if (this.presenceManager instanceof DefaultPresenceManager) {
       this.presenceManager.setServer(server);
     }
@@ -6916,10 +6963,11 @@ export {
   DefaultRoomManager,
   DefaultPresenceManager,
   DefaultLockManager,
+  ConnectedRoomManager,
   ClientBuilder,
   BrowserWebSocketTransportAdapter,
   BaseServer,
   BaseClient
 };
 
-//# debugId=94389897A03B4D7064756E2164756E21
+//# debugId=AC1FE1DD89B6DDC464756E2164756E21