@principal-ai/control-tower-core 0.1.1 → 0.1.2

package/dist/abstractions/TransportAdapter.d.ts

@@ -1,4 +1,7 @@
 import type { ConnectionState, ConnectionOptions, Message, MessageHandler, ErrorHandler, CloseHandler } from '../types';
+import type { Server as HttpServer } from 'http';
+import type { Server as HttpsServer } from 'https';
+import type { WebSocketServer } from 'ws';
 export interface ITransportAdapter {
     connect(url: string, options?: ConnectionOptions): Promise<void>;
     disconnect(): Promise<void>;
@@ -8,5 +11,7 @@ export interface ITransportAdapter {
     onClose(handler: CloseHandler): void;
     getState(): ConnectionState;
     isConnected(): boolean;
+    attach?(server: HttpServer | HttpsServer, path?: string): Promise<void>;
+    attachToWebSocketServer?(wss: WebSocketServer): Promise<void>;
 }
 //# sourceMappingURL=TransportAdapter.d.ts.map

package/dist/abstractions/TransportAdapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TransportAdapter.d.ts","sourceRoot":"","sources":["../../src/abstractions/TransportAdapter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,iBAAiB,EACjB,OAAO,EACP,cAAc,EACd,YAAY,EACZ,YAAY,EACb,MAAM,UAAU,CAAC;AAElB,MAAM,WAAW,iBAAiB;IAChC,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACjE,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5B,IAAI,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACtC,SAAS,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI,CAAC;IACzC,OAAO,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI,CAAC;IACrC,OAAO,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI,CAAC;IACrC,QAAQ,IAAI,eAAe,CAAC;IAC5B,WAAW,IAAI,OAAO,CAAC;CACxB"}
+{"version":3,"file":"TransportAdapter.d.ts","sourceRoot":"","sources":["../../src/abstractions/TransportAdapter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,iBAAiB,EACjB,OAAO,EACP,cAAc,EACd,YAAY,EACZ,YAAY,EACb,MAAM,UAAU,CAAC;AAClB,OAAO,KAAK,EAAE,MAAM,IAAI,UAAU,EAAE,MAAM,MAAM,CAAC;AACjD,OAAO,KAAK,EAAE,MAAM,IAAI,WAAW,EAAE,MAAM,OAAO,CAAC;AACnD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC;AAE1C,MAAM,WAAW,iBAAiB;IAEhC,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACjE,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5B,IAAI,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACtC,SAAS,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI,CAAC;IACzC,OAAO,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI,CAAC;IACrC,OAAO,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI,CAAC;IACrC,QAAQ,IAAI,eAAe,CAAC;IAC5B,WAAW,IAAI,OAAO,CAAC;IAGvB,MAAM,CAAC,CAAC,MAAM,EAAE,UAAU,GAAG,WAAW,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxE,uBAAuB,CAAC,CAAC,GAAG,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC/D"}

package/dist/adapters/websocket/WebSocketTransportAdapter.d.ts

@@ -0,0 +1,40 @@
+import type { ITransportAdapter } from '../../abstractions/TransportAdapter';
+import type { ConnectionState, ConnectionOptions, Message, MessageHandler, ErrorHandler, CloseHandler } from '../../types';
+import type { Server as HttpServer } from 'http';
+import type { Server as HttpsServer } from 'https';
+import { WebSocketServer, WebSocket } from 'ws';
+interface ClientConnection {
+    id: string;
+    ws: WebSocket;
+    userId?: string;
+}
+export declare class WebSocketTransportAdapter implements ITransportAdapter {
+    private state;
+    private messageHandlers;
+    private errorHandlers;
+    private closeHandlers;
+    private wss?;
+    private serverUrl?;
+    private attachedServer?;
+    private attachedWss?;
+    private webSocketPath?;
+    private clients;
+    private mode;
+    connect(url: string, _options?: ConnectionOptions): Promise<void>;
+    attach(server: HttpServer | HttpsServer, path?: string): Promise<void>;
+    attachToWebSocketServer(wss: WebSocketServer): Promise<void>;
+    private handleConnection;
+    disconnect(): Promise<void>;
+    send(message: Message): Promise<void>;
+    onMessage(handler: MessageHandler): void;
+    onError(handler: ErrorHandler): void;
+    onClose(handler: CloseHandler): void;
+    getState(): ConnectionState;
+    isConnected(): boolean;
+    getConnectedClients(): ClientConnection[];
+    getClientCount(): number;
+    getMode(): 'standalone' | 'integration';
+    private generateId;
+}
+export {};
+//# sourceMappingURL=WebSocketTransportAdapter.d.ts.map

package/dist/adapters/websocket/WebSocketTransportAdapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WebSocketTransportAdapter.d.ts","sourceRoot":"","sources":["../../../src/adapters/websocket/WebSocketTransportAdapter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qCAAqC,CAAC;AAC7E,OAAO,KAAK,EACV,eAAe,EACf,iBAAiB,EACjB,OAAO,EACP,cAAc,EACd,YAAY,EACZ,YAAY,EACb,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,MAAM,IAAI,UAAU,EAAE,MAAM,MAAM,CAAC;AACjD,OAAO,KAAK,EAAE,MAAM,IAAI,WAAW,EAAE,MAAM,OAAO,CAAC;AACnD,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAEhD,UAAU,gBAAgB;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,EAAE,EAAE,SAAS,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,qBAAa,yBAA0B,YAAW,iBAAiB;IACjE,OAAO,CAAC,KAAK,CAAmC;IAChD,OAAO,CAAC,eAAe,CAAkC;IACzD,OAAO,CAAC,aAAa,CAAgC;IACrD,OAAO,CAAC,aAAa,CAAgC;IAGrD,OAAO,CAAC,GAAG,CAAC,CAAkB;IAC9B,OAAO,CAAC,SAAS,CAAC,CAAS;IAG3B,OAAO,CAAC,cAAc,CAAC,CAA2B;IAClD,OAAO,CAAC,WAAW,CAAC,CAAkB;IACtC,OAAO,CAAC,aAAa,CAAC,CAAS;IAG/B,OAAO,CAAC,OAAO,CAAuC;IACtD,OAAO,CAAC,IAAI,CAA8C;IAGpD,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAsCjE,MAAM,CAAC,MAAM,EAAE,UAAU,GAAG,WAAW,EAAE,IAAI,GAAE,MAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAiC7E,uBAAuB,CAAC,GAAG,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;IA2BlE,OAAO,CAAC,gBAAgB;IAqClB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IA8B3B,IAAI,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IA8B3C,SAAS,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI;IAIxC,OAAO,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAIpC,OAAO,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAIpC,QAAQ,IAAI,eAAe;IAI3B,WAAW,IAAI,OAAO;IAKtB,mBAAmB,IAAI,gBAAgB,EAAE;IAIzC,cAAc,IAAI,MAAM;IAIxB,OAAO,IAAI,YAAY,GAAG,aAAa;IAIvC,OAAO,CAAC,UAAU;CAGnB"}

package/dist/adapters/websocket/WebSocketTransportAdapter.js

@@ -0,0 +1,204 @@
+import { WebSocketServer, WebSocket } from 'ws';
+export class WebSocketTransportAdapter {
+    constructor() {
+        this.state = 'disconnected';
+        this.messageHandlers = new Set();
+        this.errorHandlers = new Set();
+        this.closeHandlers = new Set();
+        // Client management
+        this.clients = new Map();
+        this.mode = 'standalone';
+    }
+    // Standalone mode implementation
+    async connect(url, _options) {
+        if (this.state === 'connected' || this.state === 'connecting') {
+            throw new Error('Already connected or connecting');
+        }
+        this.state = 'connecting';
+        this.serverUrl = url;
+        this.mode = 'standalone';
+        try {
+            // Extract port from URL
+            const urlObj = new URL(url);
+            const port = parseInt(urlObj.port) || (urlObj.protocol === 'wss:' ? 443 : 80);
+            // Create WebSocket server
+            this.wss = new WebSocketServer({ port });
+            this.wss.on('connection', (ws, req) => {
+                this.handleConnection(ws, req);
+            });
+            this.wss.on('error', (error) => {
+                this.errorHandlers.forEach(handler => handler(error));
+            });
+            this.wss.on('close', () => {
+                this.state = 'disconnected';
+                this.closeHandlers.forEach(handler => handler(1000, 'Server closed'));
+            });
+            this.state = 'connected';
+        }
+        catch (error) {
+            this.state = 'disconnected';
+            throw error;
+        }
+    }
+    // Integration mode implementation
+    async attach(server, path = '/ws') {
+        if (this.state === 'connected' || this.state === 'connecting') {
+            throw new Error('Already connected or connecting');
+        }
+        this.state = 'connecting';
+        this.attachedServer = server;
+        this.webSocketPath = path;
+        this.mode = 'integration';
+        try {
+            // Create WebSocket server attached to the existing HTTP server
+            this.wss = new WebSocketServer({
+                server,
+                path,
+                perMessageDeflate: false
+            });
+            this.wss.on('connection', (ws, req) => {
+                this.handleConnection(ws, req);
+            });
+            this.wss.on('error', (error) => {
+                this.errorHandlers.forEach(handler => handler(error));
+            });
+            this.state = 'connected';
+        }
+        catch (error) {
+            this.state = 'disconnected';
+            throw error;
+        }
+    }
+    async attachToWebSocketServer(wss) {
+        if (this.state === 'connected' || this.state === 'connecting') {
+            throw new Error('Already connected or connecting');
+        }
+        this.state = 'connecting';
+        this.attachedWss = wss;
+        this.mode = 'integration';
+        try {
+            this.wss = wss;
+            this.wss.on('connection', (ws, req) => {
+                this.handleConnection(ws, req);
+            });
+            this.wss.on('error', (error) => {
+                this.errorHandlers.forEach(handler => handler(error));
+            });
+            this.state = 'connected';
+        }
+        catch (error) {
+            this.state = 'disconnected';
+            throw error;
+        }
+    }
+    handleConnection(ws, req) {
+        const clientId = this.generateId();
+        const client = {
+            id: clientId,
+            ws,
+            userId: typeof req.headers['x-user-id'] === 'string' ? req.headers['x-user-id'] : undefined
+        };
+        this.clients.set(clientId, client);
+        ws.on('message', (data) => {
+            try {
+                const message = JSON.parse(data.toString());
+                // Add clientId to message payload for routing
+                const enrichedMessage = {
+                    ...message,
+                    payload: {
+                        ...message.payload,
+                        clientId
+                    }
+                };
+                this.messageHandlers.forEach(handler => handler(enrichedMessage));
+            }
+            catch (error) {
+                this.errorHandlers.forEach(handler => handler(error));
+            }
+        });
+        ws.on('error', (error) => {
+            this.errorHandlers.forEach(handler => handler(error));
+        });
+        ws.on('close', (_code, _reason) => {
+            this.clients.delete(clientId);
+            // Don't emit close for individual clients, only for server shutdown
+        });
+    }
+    async disconnect() {
+        if (this.state === 'disconnected') {
+            return;
+        }
+        this.state = 'disconnecting';
+        // Close all client connections
+        for (const [_clientId, client] of this.clients) {
+            try {
+                client.ws.close(1000, 'Server shutting down');
+            }
+            catch {
+                // Ignore errors when closing individual connections
+            }
+        }
+        this.clients.clear();
+        // Close the WebSocket server (only if we created it)
+        if (this.wss && this.mode === 'standalone') {
+            await new Promise((resolve) => {
+                this.wss.close(() => resolve());
+            });
+        }
+        this.state = 'disconnected';
+        this.wss = undefined;
+        this.attachedServer = undefined;
+        this.attachedWss = undefined;
+    }
+    async send(message) {
+        if (this.state !== 'connected') {
+            throw new Error('Not connected');
+        }
+        // Extract clientId from message payload for routing
+        const payload = message.payload;
+        const { clientId, ...clientMessage } = payload;
+        if (!clientId) {
+            throw new Error('Message must contain clientId in payload for routing');
+        }
+        const client = this.clients.get(clientId);
+        if (!client) {
+            throw new Error(`Client ${clientId} not found`);
+        }
+        if (client.ws.readyState !== WebSocket.OPEN) {
+            throw new Error(`Client ${clientId} connection is not open`);
+        }
+        const messageToSend = {
+            ...message,
+            payload: clientMessage
+        };
+        client.ws.send(JSON.stringify(messageToSend));
+    }
+    onMessage(handler) {
+        this.messageHandlers.add(handler);
+    }
+    onError(handler) {
+        this.errorHandlers.add(handler);
+    }
+    onClose(handler) {
+        this.closeHandlers.add(handler);
+    }
+    getState() {
+        return this.state;
+    }
+    isConnected() {
+        return this.state === 'connected';
+    }
+    // Utility methods
+    getConnectedClients() {
+        return Array.from(this.clients.values());
+    }
+    getClientCount() {
+        return this.clients.size;
+    }
+    getMode() {
+        return this.mode;
+    }
+    generateId() {
+        return Math.random().toString(36).substring(2) + Date.now().toString(36);
+    }
+}

package/dist/adapters/websocket/index.d.ts

@@ -0,0 +1,2 @@
+export { WebSocketTransportAdapter } from './WebSocketTransportAdapter';
+//# sourceMappingURL=index.d.ts.map

package/dist/adapters/websocket/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/adapters/websocket/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,MAAM,6BAA6B,CAAC"}

package/dist/adapters/websocket/index.js

@@ -0,0 +1 @@
+export { WebSocketTransportAdapter } from './WebSocketTransportAdapter';

package/dist/index.d.ts

@@ -1,6 +1,7 @@
 export { EventType, BaseEvent, FileAction, FileChangeEvent, GitStatusEvent, CommitEvent, BranchChangeEvent, CursorPositionEvent, Event, TokenPayload, AuthResult, CredentialType, OAuthProvider, Credentials, JWTConfig, OAuthConfig, AuthProvider, RoomPermission, Room, RoomUser, UserStatus, RoomState, RoomConfig, LockType, LockPriority, Lock, LockRequest, LockQueueItem, LockState, ConnectionState, ConnectionOptions, Message, MessageHandler, ErrorHandler, CloseHandler } from './types';
 export { ITransportAdapter, IStorageAdapter, StorageOperation, IAuthAdapter, TypedEventEmitter, EventListener, UnsubscribeFn, RoomManager, LockManager, DefaultRoomManager, DefaultLockManager } from './abstractions';
 export { MockTransportAdapter, MockStorageAdapter, MockAuthAdapter } from './adapters/mock';
+export { WebSocketTransportAdapter } from './adapters/websocket';
 export { BaseClient, ClientBuilder, type ClientConfig, type ClientEvents } from './client';
 export { BaseServer, ServerBuilder, type ServerConfig, type ServerEvents, type ConnectedClient } from './server';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAEL,SAAS,EACT,SAAS,EACT,UAAU,EACV,eAAe,EACf,cAAc,EACd,WAAW,EACX,iBAAiB,EACjB,mBAAmB,EACnB,KAAK,EAEL,YAAY,EACZ,UAAU,EACV,cAAc,EACd,aAAa,EACb,WAAW,EACX,SAAS,EACT,WAAW,EACX,YAAY,EAEZ,cAAc,EACd,IAAI,EACJ,QAAQ,EACR,UAAU,EACV,SAAS,EACT,UAAU,EAEV,QAAQ,EACR,YAAY,EACZ,IAAI,EACJ,WAAW,EACX,aAAa,EACb,SAAS,EAET,eAAe,EACf,iBAAiB,EACjB,OAAO,EACP,cAAc,EACd,YAAY,EACZ,YAAY,EACb,MAAM,SAAS,CAAC;AAGjB,OAAO,EACL,iBAAiB,EACjB,eAAe,EACf,gBAAgB,EAChB,YAAY,EACZ,iBAAiB,EACjB,aAAa,EACb,aAAa,EACb,WAAW,EACX,WAAW,EACX,kBAAkB,EAClB,kBAAkB,EACnB,MAAM,gBAAgB,CAAC;AAGxB,OAAO,EACL,oBAAoB,EACpB,kBAAkB,EAClB,eAAe,EAChB,MAAM,iBAAiB,CAAC;AAGzB,OAAO,EACL,UAAU,EACV,aAAa,EACb,KAAK,YAAY,EACjB,KAAK,YAAY,EAClB,MAAM,UAAU,CAAC;AAGlB,OAAO,EACL,UAAU,EACV,aAAa,EACb,KAAK,YAAY,EACjB,KAAK,YAAY,EACjB,KAAK,eAAe,EACrB,MAAM,UAAU,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAEL,SAAS,EACT,SAAS,EACT,UAAU,EACV,eAAe,EACf,cAAc,EACd,WAAW,EACX,iBAAiB,EACjB,mBAAmB,EACnB,KAAK,EAEL,YAAY,EACZ,UAAU,EACV,cAAc,EACd,aAAa,EACb,WAAW,EACX,SAAS,EACT,WAAW,EACX,YAAY,EAEZ,cAAc,EACd,IAAI,EACJ,QAAQ,EACR,UAAU,EACV,SAAS,EACT,UAAU,EAEV,QAAQ,EACR,YAAY,EACZ,IAAI,EACJ,WAAW,EACX,aAAa,EACb,SAAS,EAET,eAAe,EACf,iBAAiB,EACjB,OAAO,EACP,cAAc,EACd,YAAY,EACZ,YAAY,EACb,MAAM,SAAS,CAAC;AAGjB,OAAO,EACL,iBAAiB,EACjB,eAAe,EACf,gBAAgB,EAChB,YAAY,EACZ,iBAAiB,EACjB,aAAa,EACb,aAAa,EACb,WAAW,EACX,WAAW,EACX,kBAAkB,EAClB,kBAAkB,EACnB,MAAM,gBAAgB,CAAC;AAGxB,OAAO,EACL,oBAAoB,EACpB,kBAAkB,EAClB,eAAe,EAChB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,yBAAyB,EAC1B,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EACL,UAAU,EACV,aAAa,EACb,KAAK,YAAY,EACjB,KAAK,YAAY,EAClB,MAAM,UAAU,CAAC;AAGlB,OAAO,EACL,UAAU,EACV,aAAa,EACb,KAAK,YAAY,EACjB,KAAK,YAAY,EACjB,KAAK,eAAe,EACrB,MAAM,UAAU,CAAC"}

package/dist/index.js

@@ -1,7 +1,8 @@
 // Abstractions
 export { TypedEventEmitter, RoomManager, LockManager, DefaultRoomManager, DefaultLockManager } from './abstractions';
-// Mock Adapters (for testing)
+// Adapters
 export { MockTransportAdapter, MockStorageAdapter, MockAuthAdapter } from './adapters/mock';
+export { WebSocketTransportAdapter } from './adapters/websocket';
 // Client
 export { BaseClient, ClientBuilder } from './client';
 // Server