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

package/dist/adapters/websocket/WebSocketTransportAdapter.js

@@ -1,204 +0,0 @@
-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

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

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

@@ -1 +0,0 @@
-{"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

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