@principal-ai/control-tower-core 0.1.19 → 0.1.21

package/dist/index.mjs

@@ -4264,7 +4264,52 @@ class WebSocketServerTransportAdapter {
       return;
     }
     try {
-      const credentials = message.payload;
+      const payload = message.payload;
+      if (payload.token && typeof payload.token === "string") {
+        try {
+          const tokenPayload = await this.authAdapter.validateToken(payload.token);
+          if (client.authTimeout) {
+            clearTimeout(client.authTimeout);
+            client.authTimeout = undefined;
+          }
+          client.authenticated = true;
+          client.userId = tokenPayload.userId;
+          client.metadata = tokenPayload.metadata;
+          this.sendToClient(client, {
+            id: this.generateId(),
+            type: "auth_result",
+            payload: {
+              success: true,
+              userId: client.userId
+            },
+            timestamp: Date.now()
+          });
+          const authMessage = {
+            id: this.generateId(),
+            type: "client_authenticated",
+            payload: {
+              clientId: client.id,
+              userId: client.userId,
+              metadata: client.metadata
+            },
+            timestamp: Date.now()
+          };
+          this.messageHandlers.forEach((handler) => handler(authMessage));
+          return;
+        } catch (error) {
+          this.sendToClient(client, {
+            id: this.generateId(),
+            type: "auth_result",
+            payload: {
+              success: false,
+              error: error.message
+            },
+            timestamp: Date.now()
+          });
+          return;
+        }
+      }
+      const credentials = payload;
       const result = await this.authAdapter.authenticate(credentials);
       if (result.success) {
         if (client.authTimeout) {
@@ -4696,6 +4741,28 @@ class BaseClient extends TypedEventEmitter {
     }
     await this.connect(this.lastConnectionUrl, this.lastCredentials || undefined);
   }
+  async authenticate(options) {
+    if (this.connectionState !== "connected") {
+      throw new Error("Client must be connected to authenticate");
+    }
+    if (!this.authToken) {
+      throw new Error("No auth token available. Connect with credentials first.");
+    }
+    const payload = {
+      token: this.authToken,
+      ...options
+    };
+    const message = {
+      id: this.generateId(),
+      type: "authenticate",
+      payload,
+      timestamp: Date.now()
+    };
+    await this.transport.send(message);
+  }
+  hasAuthToken() {
+    return this.authToken !== null;
+  }
   async joinRoom(roomId) {
     if (this.connectionState !== "connected") {
       throw new Error("Client must be connected to join a room");
@@ -4779,6 +4846,9 @@ class BaseClient extends TypedEventEmitter {
   async handleMessage(message) {
     try {
       switch (message.type) {
+        case "auth_result":
+          await this.handleAuthResult(message.payload);
+          break;
         case "room_joined":
           await this.handleRoomJoined(message.payload);
           break;
@@ -4821,6 +4891,16 @@ class BaseClient extends TypedEventEmitter {
       }
     }
   }
+  async handleAuthResult(payload) {
+    if (payload.success) {
+      if (payload.userId) {
+        this.userId = payload.userId;
+      }
+      await this.emit("authenticated", { userId: this.userId || "" });
+    } else {
+      await this.emit("authentication_failed", { error: payload.error || "Authentication failed" });
+    }
+  }
   async handleRoomJoined(payload) {
     this.currentRoomId = payload.roomId;
     const roomState = deserializeRoomState(payload.state);
@@ -5404,7 +5484,8 @@ class BaseServer extends TypedEventEmitter {
     if (payload.authenticated) {
       await this.emit("client_authenticated", {
         clientId: payload.clientId,
-        userId: payload.userId || ""
+        userId: payload.userId || "",
+        metadata: payload.metadata
       });
     }
   }
@@ -5421,7 +5502,8 @@ class BaseServer extends TypedEventEmitter {
       this.userClientMap.get(payload.userId).add(payload.clientId);
       await this.emit("client_authenticated", {
         clientId: payload.clientId,
-        userId: payload.userId
+        userId: payload.userId,
+        metadata: payload.metadata
       });
     }
   }
@@ -5558,9 +5640,13 @@ class BaseServer extends TypedEventEmitter {
       if (client) {
         client.userId = tokenPayload.userId;
         client.authenticated = true;
-        await this.emit("client_authenticated", { clientId, userId: tokenPayload.userId });
+        await this.emit("client_authenticated", {
+          clientId,
+          userId: tokenPayload.userId,
+          metadata: tokenPayload.metadata
+        });
       }
-      await this.sendToClient(clientId, { type: "auth_result", success: true });
+      await this.sendToClient(clientId, { type: "auth_result", success: true, userId: tokenPayload.userId });
     } catch (error) {
       await this.sendToClient(clientId, { type: "auth_result", success: false, error: error.message });
     }
@@ -5935,4 +6021,4 @@ export {
   BaseClient
 };
 
-//# debugId=16358B5F472DB04F64756E2164756E21
+//# debugId=8D71D2794ACED82564756E2164756E21