@nxuss/lemma 0.1.0

package/dist/core/WebSocketServer.js

@@ -0,0 +1,416 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WebSocketServer = void 0;
+const ws_1 = __importStar(require("ws"));
+const events_1 = require("events");
+const security_1 = require("../security");
+/**
+ * High-performance WebSocket server for agent connections
+ * Handles connection lifecycle, heartbeat, reconnection logic, and security
+ */
+class WebSocketServer extends events_1.EventEmitter {
+    constructor(config, securityConfig) {
+        super();
+        this.wss = null;
+        this.connections = new Map();
+        this.heartbeatInterval = null;
+        this.security = null;
+        this.stats = {
+            totalConnections: 0,
+            activeConnections: 0,
+            messagesReceived: 0,
+            messagesSent: 0,
+            reconnections: 0,
+        };
+        this.config = {
+            port: config.port || 8080,
+            heartbeatInterval: config.heartbeatInterval || 30000,
+            connectionTimeout: config.connectionTimeout || 60000,
+            maxConnections: config.maxConnections || 100,
+            ...config,
+        };
+        // Initialize security if config provided
+        if (securityConfig) {
+            this.security = new security_1.SecurityMiddleware(securityConfig);
+            console.log('🔒 Security enabled:', {
+                auth: securityConfig.auth.enabled,
+                rateLimit: securityConfig.rateLimit.enabled,
+                sanitization: securityConfig.sanitization.enabled,
+            });
+        }
+    }
+    /**
+     * Start the WebSocket server
+     */
+    start() {
+        return new Promise((resolve, reject) => {
+            try {
+                this.wss = new ws_1.WebSocketServer({
+                    port: this.config.port,
+                    maxPayload: 10 * 1024 * 1024, // 10MB max message size
+                });
+                this.wss.on('connection', (ws, req) => {
+                    this.handleConnection(ws, req);
+                });
+                this.wss.on('error', (error) => {
+                    this.emit('error', error);
+                    reject(error);
+                });
+                this.wss.on('listening', () => {
+                    console.log(`WebSocket server listening on port ${this.config.port}`);
+                    this.startHeartbeat();
+                    this.emit('ready');
+                    resolve();
+                });
+            }
+            catch (error) {
+                reject(error);
+            }
+        });
+    }
+    /**
+     * Handle new WebSocket connection
+     */
+    handleConnection(ws, req) {
+        // Check max connections limit
+        if (this.connections.size >= this.config.maxConnections) {
+            ws.close(1008, 'Maximum connections reached');
+            return;
+        }
+        // Extract security headers
+        const apiKey = req.headers['authorization']?.replace('Bearer ', '');
+        const origin = req.headers['origin'];
+        // Validate connection if security is enabled
+        if (this.security) {
+            const securityCheck = this.security.checkRequest(apiKey, origin, { type: 'connection', timestamp: Date.now() }, req.socket.remoteAddress);
+            if (!securityCheck.allowed) {
+                console.log(`🚫 Connection rejected: ${securityCheck.reason}`);
+                ws.close(1008, securityCheck.reason);
+                this.emit('securityViolation', {
+                    type: 'connection_rejected',
+                    reason: securityCheck.reason,
+                    ip: req.socket.remoteAddress,
+                    origin,
+                });
+                return;
+            }
+            console.log('✅ Connection authenticated:', securityCheck.apiKey?.agentId);
+        }
+        const connectionId = this.generateConnectionId();
+        const connection = {
+            id: connectionId,
+            ws,
+            agentId: null,
+            capabilities: [],
+            isAlive: true,
+            connectedAt: Date.now(),
+            lastPing: Date.now(),
+            metadata: {
+                ip: req.socket.remoteAddress,
+                userAgent: req.headers['user-agent'],
+                apiKey: apiKey,
+                origin: origin,
+            },
+        };
+        this.connections.set(connectionId, connection);
+        this.stats.totalConnections++;
+        this.stats.activeConnections++;
+        console.log(`New connection: ${connectionId} (${this.stats.activeConnections} active)`);
+        // Setup WebSocket event handlers
+        ws.on('message', (data) => {
+            this.handleMessage(connectionId, data);
+        });
+        ws.on('pong', () => {
+            this.handlePong(connectionId);
+        });
+        ws.on('close', (code, reason) => {
+            this.handleDisconnection(connectionId, code, reason.toString());
+        });
+        ws.on('error', (error) => {
+            this.handleError(connectionId, error);
+        });
+        this.emit('connection', connection);
+    }
+    /**
+     * Handle incoming message from agent
+     */
+    handleMessage(connectionId, data) {
+        const connection = this.connections.get(connectionId);
+        if (!connection)
+            return;
+        try {
+            const message = JSON.parse(data.toString());
+            // Security check if enabled
+            if (this.security) {
+                const securityCheck = this.security.checkRequest(connection.metadata?.apiKey, connection.metadata?.origin, message, connection.agentId || connection.metadata?.ip || connectionId);
+                if (!securityCheck.allowed) {
+                    console.log(`🚫 Message rejected from ${connectionId}: ${securityCheck.reason}`);
+                    this.sendToConnection(connectionId, {
+                        type: 'ERROR',
+                        id: message.id || 'unknown',
+                        timestamp: Date.now(),
+                        payload: {
+                            code: 'SECURITY_ERROR',
+                            message: securityCheck.reason,
+                            rateLimit: securityCheck.rateLimit,
+                        },
+                    });
+                    this.emit('securityViolation', {
+                        type: 'message_rejected',
+                        connectionId,
+                        agentId: connection.agentId,
+                        reason: securityCheck.reason,
+                        message: message.type,
+                    });
+                    return;
+                }
+                // Use sanitized message
+                if (securityCheck.sanitizedMessage) {
+                    Object.assign(message, securityCheck.sanitizedMessage);
+                }
+                // Add rate limit info to response headers
+                if (securityCheck.rateLimit) {
+                    connection.metadata = {
+                        ...connection.metadata,
+                        rateLimit: securityCheck.rateLimit,
+                    };
+                }
+            }
+            this.stats.messagesReceived++;
+            console.log('📨 WebSocketServer received message:');
+            console.log('   Connection ID:', connectionId);
+            console.log('   Agent ID:', connection.agentId);
+            console.log('   Message type:', message.type);
+            // Update last activity
+            connection.lastPing = Date.now();
+            this.emit('message', {
+                connectionId,
+                agentId: connection.agentId,
+                message,
+            });
+        }
+        catch (error) {
+            console.error(`Failed to parse message from ${connectionId}:`, error);
+            this.sendError(connectionId, 'Invalid JSON message');
+        }
+    }
+    /**
+     * Handle pong response (heartbeat)
+     */
+    handlePong(connectionId) {
+        const connection = this.connections.get(connectionId);
+        if (connection) {
+            connection.isAlive = true;
+            connection.lastPing = Date.now();
+        }
+    }
+    /**
+     * Handle connection close
+     */
+    handleDisconnection(connectionId, code, reason) {
+        const connection = this.connections.get(connectionId);
+        if (!connection)
+            return;
+        console.log(`Connection closed: ${connectionId} (code: ${code}, reason: ${reason})`);
+        this.emit('disconnection', {
+            connectionId,
+            agentId: connection.agentId,
+            code,
+            reason,
+        });
+        this.connections.delete(connectionId);
+        this.stats.activeConnections--;
+    }
+    /**
+     * Handle WebSocket error
+     */
+    handleError(connectionId, error) {
+        console.error(`WebSocket error on ${connectionId}:`, error);
+        this.emit('connectionError', { connectionId, error });
+    }
+    /**
+     * Start heartbeat mechanism (ping/pong)
+     */
+    startHeartbeat() {
+        this.heartbeatInterval = setInterval(() => {
+            this.connections.forEach((connection, connectionId) => {
+                // Check if connection is still alive
+                if (!connection.isAlive) {
+                    console.log(`Terminating dead connection: ${connectionId}`);
+                    connection.ws.terminate();
+                    return;
+                }
+                // Check for timeout
+                const timeSinceLastPing = Date.now() - connection.lastPing;
+                if (timeSinceLastPing > this.config.connectionTimeout) {
+                    console.log(`Connection timeout: ${connectionId}`);
+                    connection.ws.close(1000, 'Connection timeout');
+                    return;
+                }
+                // Send ping
+                connection.isAlive = false;
+                connection.ws.ping();
+            });
+        }, this.config.heartbeatInterval);
+    }
+    /**
+     * Send message to specific connection
+     */
+    sendToConnection(connectionId, message) {
+        const connection = this.connections.get(connectionId);
+        if (!connection || connection.ws.readyState !== ws_1.default.OPEN) {
+            return false;
+        }
+        try {
+            connection.ws.send(JSON.stringify(message));
+            this.stats.messagesSent++;
+            return true;
+        }
+        catch (error) {
+            console.error(`Failed to send message to ${connectionId}:`, error);
+            return false;
+        }
+    }
+    /**
+     * Send message to specific agent
+     */
+    sendToAgent(agentId, message) {
+        const connection = Array.from(this.connections.values()).find(conn => conn.agentId === agentId);
+        if (!connection) {
+            return false;
+        }
+        return this.sendToConnection(connection.id, message);
+    }
+    /**
+     * Broadcast message to all connected agents
+     */
+    broadcast(message, excludeConnectionId) {
+        const payload = JSON.stringify(message);
+        this.connections.forEach((connection, connectionId) => {
+            if (excludeConnectionId && connectionId === excludeConnectionId) {
+                return;
+            }
+            if (connection.ws.readyState === ws_1.default.OPEN) {
+                connection.ws.send(payload);
+                this.stats.messagesSent++;
+            }
+        });
+    }
+    /**
+     * Send error message to connection
+     */
+    sendError(connectionId, errorMessage) {
+        this.sendToConnection(connectionId, {
+            type: 'error',
+            error: errorMessage,
+            timestamp: Date.now(),
+        });
+    }
+    /**
+     * Update agent information for a connection
+     */
+    updateAgentInfo(connectionId, agentId, capabilities) {
+        const connection = this.connections.get(connectionId);
+        if (!connection) {
+            return false;
+        }
+        connection.agentId = agentId;
+        connection.capabilities = capabilities;
+        return true;
+    }
+    /**
+     * Get connection by ID
+     */
+    getConnection(connectionId) {
+        return this.connections.get(connectionId);
+    }
+    /**
+     * Get all active connections
+     */
+    getConnections() {
+        return Array.from(this.connections.values());
+    }
+    /**
+     * Get connection statistics
+     */
+    getStats() {
+        return { ...this.stats };
+    }
+    /**
+     * Generate unique connection ID
+     */
+    generateConnectionId() {
+        return `conn_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
+    }
+    /**
+     * Stop the WebSocket server
+     */
+    async stop() {
+        return new Promise((resolve) => {
+            // Stop heartbeat
+            if (this.heartbeatInterval) {
+                clearInterval(this.heartbeatInterval);
+                this.heartbeatInterval = null;
+            }
+            // Destroy security middleware
+            if (this.security) {
+                this.security.destroy();
+            }
+            // Close all connections
+            this.connections.forEach((connection) => {
+                connection.ws.close(1000, 'Server shutting down');
+            });
+            // Close server
+            if (this.wss) {
+                this.wss.close(() => {
+                    console.log('WebSocket server stopped');
+                    resolve();
+                });
+            }
+            else {
+                resolve();
+            }
+        });
+    }
+    /**
+     * Get security middleware (for key management)
+     */
+    getSecurity() {
+        return this.security;
+    }
+}
+exports.WebSocketServer = WebSocketServer;
+//# sourceMappingURL=WebSocketServer.js.map

package/dist/core/WebSocketServer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WebSocketServer.js","sourceRoot":"","sources":["../../src/core/WebSocketServer.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,yCAA4D;AAC5D,mCAAsC;AAEtC,0CAAiE;AAEjE;;;GAGG;AACH,MAAa,eAAgB,SAAQ,qBAAY;IAc/C,YAAY,MAAoB,EAAE,cAA+B;QAC/D,KAAK,EAAE,CAAC;QAdF,QAAG,GAAoB,IAAI,CAAC;QAC5B,gBAAW,GAAiC,IAAI,GAAG,EAAE,CAAC;QACtD,sBAAiB,GAA0B,IAAI,CAAC;QAEhD,aAAQ,GAA8B,IAAI,CAAC;QAC3C,UAAK,GAAoB;YAC/B,gBAAgB,EAAE,CAAC;YACnB,iBAAiB,EAAE,CAAC;YACpB,gBAAgB,EAAE,CAAC;YACnB,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;SACjB,CAAC;QAIA,IAAI,CAAC,MAAM,GAAG;YACZ,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,IAAI;YACzB,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,KAAK;YACpD,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,KAAK;YACpD,cAAc,EAAE,MAAM,CAAC,cAAc,IAAI,GAAG;YAC5C,GAAG,MAAM;SACV,CAAC;QAEF,yCAAyC;QACzC,IAAI,cAAc,EAAE,CAAC;YACnB,IAAI,CAAC,QAAQ,GAAG,IAAI,6BAAkB,CAAC,cAAc,CAAC,CAAC;YACvD,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE;gBAClC,IAAI,EAAE,cAAc,CAAC,IAAI,CAAC,OAAO;gBACjC,SAAS,EAAE,cAAc,CAAC,SAAS,CAAC,OAAO;gBAC3C,YAAY,EAAE,cAAc,CAAC,YAAY,CAAC,OAAO;aAClD,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK;QACV,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC;gBACH,IAAI,CAAC,GAAG,GAAG,IAAI,oBAAQ,CAAC;oBACtB,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;oBACtB,UAAU,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI,EAAE,wBAAwB;iBACvD,CAAC,CAAC;gBAEH,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAa,EAAE,GAAG,EAAE,EAAE;oBAC/C,IAAI,CAAC,gBAAgB,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;gBACjC,CAAC,CAAC,CAAC;gBAEH,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBAC7B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;oBAC1B,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChB,CAAC,CAAC,CAAC;gBAEH,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE;oBAC5B,OAAO,CAAC,GAAG,CAAC,sCAAsC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;oBACtE,IAAI,CAAC,cAAc,EAAE,CAAC;oBACtB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACnB,OAAO,EAAE,CAAC;gBACZ,CAAC,CAAC,CAAC;YAEL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,EAAa,EAAE,GAAQ;QAC9C,8BAA8B;QAC9B,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,cAAe,EAAE,CAAC;YACzD,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,6BAA6B,CAAC,CAAC;YAC9C,OAAO;QACT,CAAC;QAED,2BAA2B;QAC3B,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QACpE,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAErC,6CAA6C;QAC7C,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAC9C,MAAM,EACN,MAAM,EACN,EAAE,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAC7C,GAAG,CAAC,MAAM,CAAC,aAAa,CACzB,CAAC;YAEF,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;gBAC3B,OAAO,CAAC,GAAG,CAAC,2BAA2B,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC/D,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;gBACrC,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE;oBAC7B,IAAI,EAAE,qBAAqB;oBAC3B,MAAM,EAAE,aAAa,CAAC,MAAM;oBAC5B,EAAE,EAAE,GAAG,CAAC,MAAM,CAAC,aAAa;oBAC5B,MAAM;iBACP,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,6BAA6B,EAAE,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;QACjD,MAAM,UAAU,GAAoB;YAClC,EAAE,EAAE,YAAY;YAChB,EAAE;YACF,OAAO,EAAE,IAAI;YACb,YAAY,EAAE,EAAE;YAChB,OAAO,EAAE,IAAI;YACb,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;YACvB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE;YACpB,QAAQ,EAAE;gBACR,EAAE,EAAE,GAAG,CAAC,MAAM,CAAC,aAAa;gBAC5B,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC;gBACpC,MAAM,EAAE,MAAM;gBACd,MAAM,EAAE,MAAM;aACf;SACF,CAAC;QAEF,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QAC/C,IAAI,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC;QAC9B,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC;QAE/B,OAAO,CAAC,GAAG,CAAC,mBAAmB,YAAY,KAAK,IAAI,CAAC,KAAK,CAAC,iBAAiB,UAAU,CAAC,CAAC;QAExF,iCAAiC;QACjC,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAY,EAAE,EAAE;YAChC,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACjB,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAY,EAAE,MAAc,EAAE,EAAE;YAC9C,IAAI,CAAC,mBAAmB,CAAC,YAAY,EAAE,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAY,EAAE,EAAE;YAC9B,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,YAAoB,EAAE,IAAY;QACtD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU;YAAE,OAAO;QAExB,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;YAE5C,4BAA4B;YAC5B,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAC9C,UAAU,CAAC,QAAQ,EAAE,MAAM,EAC3B,UAAU,CAAC,QAAQ,EAAE,MAAM,EAC3B,OAAO,EACP,UAAU,CAAC,OAAO,IAAI,UAAU,CAAC,QAAQ,EAAE,EAAE,IAAI,YAAY,CAC9D,CAAC;gBAEF,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;oBAC3B,OAAO,CAAC,GAAG,CAAC,4BAA4B,YAAY,KAAK,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;oBAEjF,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE;wBAClC,IAAI,EAAE,OAAO;wBACb,EAAE,EAAE,OAAO,CAAC,EAAE,IAAI,SAAS;wBAC3B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;wBACrB,OAAO,EAAE;4BACP,IAAI,EAAE,gBAAgB;4BACtB,OAAO,EAAE,aAAa,CAAC,MAAM;4BAC7B,SAAS,EAAE,aAAa,CAAC,SAAS;yBACnC;qBACF,CAAC,CAAC;oBAEH,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE;wBAC7B,IAAI,EAAE,kBAAkB;wBACxB,YAAY;wBACZ,OAAO,EAAE,UAAU,CAAC,OAAO;wBAC3B,MAAM,EAAE,aAAa,CAAC,MAAM;wBAC5B,OAAO,EAAE,OAAO,CAAC,IAAI;qBACtB,CAAC,CAAC;oBAEH,OAAO;gBACT,CAAC;gBAED,wBAAwB;gBACxB,IAAI,aAAa,CAAC,gBAAgB,EAAE,CAAC;oBACnC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,aAAa,CAAC,gBAAgB,CAAC,CAAC;gBACzD,CAAC;gBAED,0CAA0C;gBAC1C,IAAI,aAAa,CAAC,SAAS,EAAE,CAAC;oBAC5B,UAAU,CAAC,QAAQ,GAAG;wBACpB,GAAG,UAAU,CAAC,QAAQ;wBACtB,SAAS,EAAE,aAAa,CAAC,SAAS;qBACnC,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,gBAAgB,EAAE,CAAC;YAE9B,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;YACpD,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;YAC/C,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;YAChD,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAE9C,uBAAuB;YACvB,UAAU,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAEjC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;gBACnB,YAAY;gBACZ,OAAO,EAAE,UAAU,CAAC,OAAO;gBAC3B,OAAO;aACR,CAAC,CAAC;QAEL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,gCAAgC,YAAY,GAAG,EAAE,KAAK,CAAC,CAAC;YACtE,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,sBAAsB,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,UAAU,CAAC,YAAoB;QACrC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,UAAU,EAAE,CAAC;YACf,UAAU,CAAC,OAAO,GAAG,IAAI,CAAC;YAC1B,UAAU,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACnC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,YAAoB,EAAE,IAAY,EAAE,MAAc;QAC5E,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU;YAAE,OAAO;QAExB,OAAO,CAAC,GAAG,CAAC,sBAAsB,YAAY,WAAW,IAAI,aAAa,MAAM,GAAG,CAAC,CAAC;QAErF,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;YACzB,YAAY;YACZ,OAAO,EAAE,UAAU,CAAC,OAAO;YAC3B,IAAI;YACJ,MAAM;SACP,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC;IACjC,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,YAAoB,EAAE,KAAY;QACpD,OAAO,CAAC,KAAK,CAAC,sBAAsB,YAAY,GAAG,EAAE,KAAK,CAAC,CAAC;QAC5D,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;IACxD,CAAC;IAED;;OAEG;IACK,cAAc;QACpB,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;YACxC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE;gBACpD,qCAAqC;gBACrC,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;oBACxB,OAAO,CAAC,GAAG,CAAC,gCAAgC,YAAY,EAAE,CAAC,CAAC;oBAC5D,UAAU,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC;oBAC1B,OAAO;gBACT,CAAC;gBAED,oBAAoB;gBACpB,MAAM,iBAAiB,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC;gBAC3D,IAAI,iBAAiB,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAkB,EAAE,CAAC;oBACvD,OAAO,CAAC,GAAG,CAAC,uBAAuB,YAAY,EAAE,CAAC,CAAC;oBACnD,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,oBAAoB,CAAC,CAAC;oBAChD,OAAO;gBACT,CAAC;gBAED,YAAY;gBACZ,UAAU,CAAC,OAAO,GAAG,KAAK,CAAC;gBAC3B,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YACvB,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACI,gBAAgB,CAAC,YAAoB,EAAE,OAAY;QACxD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,EAAE,CAAC,UAAU,KAAK,YAAS,CAAC,IAAI,EAAE,CAAC;YAC/D,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC;YACH,UAAU,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,YAAY,GAAG,EAAE,KAAK,CAAC,CAAC;YACnE,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACI,WAAW,CAAC,OAAe,EAAE,OAAY;QAC9C,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAC3D,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,KAAK,OAAO,CACjC,CAAC;QAEF,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;IAED;;OAEG;IACI,SAAS,CAAC,OAAY,EAAE,mBAA4B;QACzD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAExC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE;YACpD,IAAI,mBAAmB,IAAI,YAAY,KAAK,mBAAmB,EAAE,CAAC;gBAChE,OAAO;YACT,CAAC;YAED,IAAI,UAAU,CAAC,EAAE,CAAC,UAAU,KAAK,YAAS,CAAC,IAAI,EAAE,CAAC;gBAChD,UAAU,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC5B,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,SAAS,CAAC,YAAoB,EAAE,YAAoB;QAC1D,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE;YAClC,IAAI,EAAE,OAAO;YACb,KAAK,EAAE,YAAY;YACnB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACI,eAAe,CAAC,YAAoB,EAAE,OAAe,EAAE,YAAsB;QAClF,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,UAAU,CAAC,OAAO,GAAG,OAAO,CAAC;QAC7B,UAAU,CAAC,YAAY,GAAG,YAAY,CAAC;QAEvC,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACI,aAAa,CAAC,YAAoB;QACvC,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACI,cAAc;QACnB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED;;OAEG;IACI,QAAQ;QACb,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED;;OAEG;IACK,oBAAoB;QAC1B,OAAO,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IACzE,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,IAAI;QACf,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,iBAAiB;YACjB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3B,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;gBACtC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;YAChC,CAAC;YAED,8BAA8B;YAC9B,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC;YAED,wBAAwB;YACxB,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE;gBACtC,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;YAEH,eAAe;YACf,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE;oBAClB,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;oBACxC,OAAO,EAAE,CAAC;gBACZ,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACI,WAAW;QAChB,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;CACF;AAzbD,0CAybC"}

package/dist/core/index.d.ts

@@ -0,0 +1,15 @@
+/**
+ * Core Hub WebSocket - Main exports
+ *
+ * This module provides the core infrastructure for the Subconscious Router:
+ * - WebSocketServer: High-performance WebSocket server with connection management
+ * - AgentRegistry: Dynamic agent registration and capability tracking
+ * - Router: I.A.P. message routing and task orchestration
+ */
+export { WebSocketServer } from './WebSocketServer';
+export { AgentRegistry } from './AgentRegistry';
+export { Router } from './router';
+export { OrchestrationEngine } from './OrchestrationEngine';
+export { SubconsciousEngine } from './SubconsciousEngine';
+export { DashboardBroadcaster } from './DashboardBroadcaster';
+//# sourceMappingURL=index.d.ts.map

package/dist/core/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/core/index.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC"}

package/dist/core/index.js

@@ -0,0 +1,24 @@
+"use strict";
+/**
+ * Core Hub WebSocket - Main exports
+ *
+ * This module provides the core infrastructure for the Subconscious Router:
+ * - WebSocketServer: High-performance WebSocket server with connection management
+ * - AgentRegistry: Dynamic agent registration and capability tracking
+ * - Router: I.A.P. message routing and task orchestration
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DashboardBroadcaster = exports.SubconsciousEngine = exports.OrchestrationEngine = exports.Router = exports.AgentRegistry = exports.WebSocketServer = void 0;
+var WebSocketServer_1 = require("./WebSocketServer");
+Object.defineProperty(exports, "WebSocketServer", { enumerable: true, get: function () { return WebSocketServer_1.WebSocketServer; } });
+var AgentRegistry_1 = require("./AgentRegistry");
+Object.defineProperty(exports, "AgentRegistry", { enumerable: true, get: function () { return AgentRegistry_1.AgentRegistry; } });
+var router_1 = require("./router");
+Object.defineProperty(exports, "Router", { enumerable: true, get: function () { return router_1.Router; } });
+var OrchestrationEngine_1 = require("./OrchestrationEngine");
+Object.defineProperty(exports, "OrchestrationEngine", { enumerable: true, get: function () { return OrchestrationEngine_1.OrchestrationEngine; } });
+var SubconsciousEngine_1 = require("./SubconsciousEngine");
+Object.defineProperty(exports, "SubconsciousEngine", { enumerable: true, get: function () { return SubconsciousEngine_1.SubconsciousEngine; } });
+var DashboardBroadcaster_1 = require("./DashboardBroadcaster");
+Object.defineProperty(exports, "DashboardBroadcaster", { enumerable: true, get: function () { return DashboardBroadcaster_1.DashboardBroadcaster; } });
+//# sourceMappingURL=index.js.map

package/dist/core/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/core/index.ts"],"names":[],"mappings":";AAAA;;;;;;;GAOG;;;AAEH,qDAAoD;AAA3C,kHAAA,eAAe,OAAA;AACxB,iDAAgD;AAAvC,8GAAA,aAAa,OAAA;AACtB,mCAAkC;AAAzB,gGAAA,MAAM,OAAA;AACf,6DAA4D;AAAnD,0HAAA,mBAAmB,OAAA;AAC5B,2DAA0D;AAAjD,wHAAA,kBAAkB,OAAA;AAC3B,+DAA8D;AAArD,4HAAA,oBAAoB,OAAA"}

package/dist/core/router.d.ts

@@ -0,0 +1,105 @@
+import { EventEmitter } from 'events';
+import { AgentRegistry } from './AgentRegistry';
+import { WebSocketServer } from './WebSocketServer';
+import { RouterConfig } from '../types';
+/**
+ * Core routing engine for Inter-Agent Protocol (I.A.P.) messages
+ * Handles message reception, agent selection, and error management
+ */
+export declare class Router extends EventEmitter {
+    private wsServer;
+    private agentRegistry;
+    private config;
+    private pendingTasks;
+    private taskTimeouts;
+    constructor(wsServer: WebSocketServer, agentRegistry: AgentRegistry, config: RouterConfig);
+    /**
+     * Setup event handlers for WebSocket server
+     */
+    private setupEventHandlers;
+    /**
+     * Handle incoming I.A.P. message
+     */
+    private handleMessage;
+    /**
+     * Handle agent handshake (registration)
+     */
+    private handleHandshake;
+    /**
+     * Handle task request from agent or external source
+     */
+    private handleTaskRequest;
+    /**
+     * Handle task response from agent
+     */
+    private handleTaskResponse;
+    /**
+     * Handle capability query
+     */
+    private handleCapabilityQuery;
+    /**
+     * Handle heartbeat message
+     */
+    private handleHeartbeat;
+    /**
+     * Handle error message from agent
+     */
+    private handleErrorMessage;
+    /**
+     * Select best agent for a task
+     */
+    private selectAgent;
+    /**
+     * Handle agent disconnection
+     */
+    private handleAgentDisconnection;
+    /**
+     * Handle task failure
+     */
+    private handleTaskFailure;
+    /**
+     * Retry a failed task
+     */
+    private retryTask;
+    /**
+     * Set timeout for a task
+     */
+    private setTaskTimeout;
+    /**
+     * Clear timeout for a task
+     */
+    private clearTaskTimeout;
+    /**
+     * Send error message to connection
+     */
+    private sendError;
+    /**
+     * Validate I.A.P. message structure
+     */
+    private validateMessage;
+    /**
+     * Generate unique message ID
+     */
+    private generateMessageId;
+    /**
+     * Get router statistics
+     */
+    getStats(): {
+        pendingTasks: number;
+        agentStats: {
+            totalAgents: number;
+            onlineAgents: number;
+            offlineAgents: number;
+            totalCapabilities: number;
+            totalTasksInProgress: number;
+            totalTasksCompleted: number;
+            totalTasksFailed: number;
+        };
+        connectionStats: import("../types").ConnectionStats;
+    };
+    /**
+     * Stop the router
+     */
+    stop(): void;
+}
+//# sourceMappingURL=router.d.ts.map

package/dist/core/router.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"router.d.ts","sourceRoot":"","sources":["../../src/core/router.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAKL,YAAY,EAEb,MAAM,UAAU,CAAC;AAElB;;;GAGG;AACH,qBAAa,MAAO,SAAQ,YAAY;IACtC,OAAO,CAAC,QAAQ,CAAkB;IAClC,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,YAAY,CAAuC;IAC3D,OAAO,CAAC,YAAY,CAA0C;gBAG5D,QAAQ,EAAE,eAAe,EACzB,aAAa,EAAE,aAAa,EAC5B,MAAM,EAAE,YAAY;IAetB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAmB1B;;OAEG;YACW,aAAa;IAsD3B;;OAEG;YACW,eAAe;IAsC7B;;OAEG;YACW,iBAAiB;IA2D/B;;OAEG;YACW,kBAAkB;IA4ChC;;OAEG;YACW,qBAAqB;IA0BnC;;OAEG;YACW,eAAe;IAgB7B;;OAEG;YACW,kBAAkB;IAWhC;;OAEG;IACH,OAAO,CAAC,WAAW;IAsBnB;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAchC;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAwCzB;;OAEG;YACW,SAAS;IA6BvB;;OAEG;IACH,OAAO,CAAC,cAAc;IAUtB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAQxB;;OAEG;IACH,OAAO,CAAC,SAAS;IAWjB;;OAEG;IACH,OAAO,CAAC,eAAe;IAUvB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAIzB;;OAEG;IACI,QAAQ;;;;;;;;;;;;;IAQf;;OAEG;IACI,IAAI,IAAI,IAAI;CAMpB"}