telegram-ssh-bot 2.0.0 → 2.2.0

package/dist/core/ConnectionPool.js

@@ -0,0 +1,397 @@
+/**
+ * Connection Pool
+ * Manages multiple SSH connections with pooling, timeout, and health monitoring
+ */
+import { EventEmitter } from "events";
+import { ConnectionFailedError, ConnectionTimeoutError, PoolExhaustedError, } from "../errors/index.js";
+import { SSHClient } from "./SSHClient.js";
+/**
+ * Connection pool for managing multiple SSH connections
+ */
+export class ConnectionPool extends EventEmitter {
+    connections = new Map();
+    waitingQueue = [];
+    options;
+    logger;
+    healthCheckTimer = null;
+    idleCheckTimer = null;
+    totalRequestsServed = 0;
+    isDraining = false;
+    constructor(options, logger) {
+        super();
+        this.options = {
+            maxConnections: options.maxConnections ?? 10,
+            connectionTimeout: options.connectionTimeout ?? 30000,
+            commandTimeout: options.commandTimeout ?? 60000,
+            idleTimeout: options.idleTimeout ?? 300000, // 5 minutes
+            healthCheckInterval: options.healthCheckInterval ?? 60000, // 1 minute
+            maxHealthCheckFailures: options.maxHealthCheckFailures ?? 3,
+        };
+        this.logger = logger;
+    }
+    /**
+     * Initialize the connection pool
+     */
+    async initialize() {
+        this.startHealthCheck();
+        this.startIdleCheck();
+        this.logger.info("Connection pool initialized", {
+            maxConnections: this.options.maxConnections,
+        });
+    }
+    /**
+     * Acquire a connection from the pool
+     */
+    async acquire(serverId, config) {
+        if (this.isDraining) {
+            throw new Error("Connection pool is draining");
+        }
+        // Check for existing idle connection
+        const existingConn = this.findIdleConnection(serverId);
+        if (existingConn) {
+            existingConn.inUse = true;
+            existingConn.lastUsedAt = new Date();
+            this.totalRequestsServed++;
+            this.logger.debug("Reusing existing connection", {
+                serverId,
+                connectionId: existingConn.id,
+            });
+            return existingConn;
+        }
+        // Check if we can create a new connection
+        if (this.connections.size < this.options.maxConnections) {
+            return this.createConnection(serverId, config);
+        }
+        // Wait for a connection to become available
+        return this.waitForConnection(serverId, config);
+    }
+    /**
+     * Release a connection back to the pool
+     */
+    release(connection) {
+        const pooled = this.connections.get(connection.id);
+        if (!pooled) {
+            return;
+        }
+        pooled.inUse = false;
+        pooled.lastUsedAt = new Date();
+        // Process waiting queue
+        this.processWaitingQueue();
+        this.logger.debug("Connection released", {
+            connectionId: connection.id,
+            serverId: connection.serverId,
+        });
+    }
+    /**
+     * Remove a connection from the pool
+     */
+    async remove(connectionId) {
+        const connection = this.connections.get(connectionId);
+        if (!connection) {
+            return;
+        }
+        try {
+            await connection.client.disconnect();
+        }
+        catch (error) {
+            this.logger.warn("Error disconnecting during removal", {
+                connectionId,
+                error: error instanceof Error ? error.message : String(error),
+            });
+        }
+        this.connections.delete(connectionId);
+        this.emit("connectionRemoved", connectionId);
+        this.logger.info("Connection removed from pool", { connectionId });
+    }
+    /**
+     * Drain all connections and shutdown pool
+     */
+    async drain() {
+        this.isDraining = true;
+        // Stop timers
+        if (this.healthCheckTimer) {
+            clearInterval(this.healthCheckTimer);
+            this.healthCheckTimer = null;
+        }
+        if (this.idleCheckTimer) {
+            clearInterval(this.idleCheckTimer);
+            this.idleCheckTimer = null;
+        }
+        // Reject all waiting requests
+        for (const request of this.waitingQueue) {
+            clearTimeout(request.timeoutId);
+            request.reject(new Error("Connection pool is draining"));
+        }
+        this.waitingQueue.length = 0;
+        // Close all connections
+        const closePromises = Array.from(this.connections.values()).map(async (conn) => {
+            try {
+                await conn.client.disconnect();
+            }
+            catch (error) {
+                this.logger.warn("Error disconnecting during drain", {
+                    connectionId: conn.id,
+                    error: error instanceof Error ? error.message : String(error),
+                });
+            }
+        });
+        await Promise.allSettled(closePromises);
+        this.connections.clear();
+        this.logger.info("Connection pool drained");
+    }
+    /**
+     * Get pool statistics
+     */
+    getStats() {
+        let activeConnections = 0;
+        let idleConnections = 0;
+        for (const conn of this.connections.values()) {
+            if (conn.inUse) {
+                activeConnections++;
+            }
+            else {
+                idleConnections++;
+            }
+        }
+        return {
+            totalConnections: this.connections.size,
+            activeConnections,
+            idleConnections,
+            waitingRequests: this.waitingQueue.length,
+            totalRequestsServed: this.totalRequestsServed,
+        };
+    }
+    /**
+     * Get active connection count
+     */
+    getActiveCount() {
+        let count = 0;
+        for (const conn of this.connections.values()) {
+            if (conn.inUse) {
+                count++;
+            }
+        }
+        return count;
+    }
+    /**
+     * Execute a command on a pooled connection
+     */
+    async executeCommand(serverId, config, command) {
+        const connection = await this.acquire(serverId, config);
+        try {
+            const result = await connection.client.execute(command);
+            return result;
+        }
+        finally {
+            this.release(connection);
+        }
+    }
+    /**
+     * Find an idle connection for the given server
+     */
+    findIdleConnection(serverId) {
+        for (const conn of this.connections.values()) {
+            if (conn.serverId === serverId && !conn.inUse) {
+                return conn;
+            }
+        }
+        return null;
+    }
+    /**
+     * Create a new connection
+     */
+    async createConnection(serverId, config) {
+        const connectionId = this.generateConnectionId();
+        const client = new SSHClient({
+            connectionTimeout: this.options.connectionTimeout,
+            commandTimeout: this.options.commandTimeout,
+        });
+        try {
+            await client.connect(config);
+            client.setServerId(serverId);
+            const pooled = {
+                id: connectionId,
+                serverId,
+                client,
+                createdAt: new Date(),
+                lastUsedAt: new Date(),
+                inUse: true,
+                healthCheckFailures: 0,
+            };
+            this.connections.set(connectionId, pooled);
+            this.totalRequestsServed++;
+            this.logger.info("New connection created", {
+                connectionId,
+                serverId,
+                totalConnections: this.connections.size,
+            });
+            this.emit("connectionCreated", connectionId, serverId);
+            return pooled;
+        }
+        catch (error) {
+            if (error instanceof ConnectionTimeoutError) {
+                throw error;
+            }
+            throw new ConnectionFailedError(config.host, error instanceof Error ? error : new Error(String(error)));
+        }
+    }
+    /**
+     * Wait for a connection to become available
+     */
+    waitForConnection(serverId, config) {
+        return new Promise((resolve, reject) => {
+            let request;
+            let handled = false;
+            const timeoutId = setTimeout(() => {
+                // Ensure we only handle the timeout once
+                if (handled) {
+                    return;
+                }
+                handled = true;
+                // Find and remove the request from the queue
+                const index = this.waitingQueue.findIndex((r) => r.serverId === serverId && r.createdAt === request.createdAt);
+                if (index !== -1) {
+                    this.waitingQueue.splice(index, 1);
+                }
+                this.logger.debug("Connection request timed out", {
+                    serverId,
+                    queueLength: this.waitingQueue.length,
+                });
+                reject(new PoolExhaustedError(this.options.maxConnections, this.options.connectionTimeout));
+            }, this.options.connectionTimeout);
+            // Create the request object
+            request = {
+                serverId,
+                config,
+                resolve: (conn) => {
+                    if (handled)
+                        return;
+                    handled = true;
+                    clearTimeout(timeoutId);
+                    resolve(conn);
+                },
+                reject: (error) => {
+                    if (handled)
+                        return;
+                    handled = true;
+                    clearTimeout(timeoutId);
+                    reject(error);
+                },
+                timeoutId,
+                createdAt: new Date(),
+            };
+            this.waitingQueue.push(request);
+            this.logger.debug("Connection request queued", {
+                serverId,
+                queueLength: this.waitingQueue.length,
+            });
+        });
+    }
+    /**
+     * Process waiting queue when connections become available
+     */
+    processWaitingQueue() {
+        if (this.waitingQueue.length === 0) {
+            return;
+        }
+        // Find a request that can be served
+        for (let i = 0; i < this.waitingQueue.length; i++) {
+            const request = this.waitingQueue[i];
+            if (!request)
+                continue;
+            // Check for idle connection
+            const idleConn = this.findIdleConnection(request.serverId);
+            if (idleConn) {
+                this.waitingQueue.splice(i, 1);
+                clearTimeout(request.timeoutId);
+                idleConn.inUse = true;
+                idleConn.lastUsedAt = new Date();
+                this.totalRequestsServed++;
+                request.resolve(idleConn);
+                return;
+            }
+            // Create new connection if possible
+            if (this.connections.size < this.options.maxConnections) {
+                this.waitingQueue.splice(i, 1);
+                clearTimeout(request.timeoutId);
+                this.createConnection(request.serverId, request.config)
+                    .then(request.resolve)
+                    .catch(request.reject);
+                return;
+            }
+        }
+    }
+    /**
+     * Start health check timer
+     */
+    startHealthCheck() {
+        this.healthCheckTimer = setInterval(() => {
+            this.performHealthCheck();
+        }, this.options.healthCheckInterval);
+    }
+    /**
+     * Perform health check on all connections
+     */
+    async performHealthCheck() {
+        for (const conn of this.connections.values()) {
+            if (conn.inUse) {
+                continue;
+            }
+            try {
+                // Simple health check - execute echo command
+                await conn.client.execute("echo 'health_check'");
+                conn.healthCheckFailures = 0;
+            }
+            catch (error) {
+                conn.healthCheckFailures++;
+                this.logger.warn("Connection health check failed", {
+                    connectionId: conn.id,
+                    serverId: conn.serverId,
+                    failures: conn.healthCheckFailures,
+                });
+                if (conn.healthCheckFailures >= this.options.maxHealthCheckFailures) {
+                    this.logger.error("Connection removed due to health check failures", undefined, {
+                        connectionId: conn.id,
+                        serverId: conn.serverId,
+                    });
+                    await this.remove(conn.id);
+                }
+            }
+        }
+    }
+    /**
+     * Start idle connection check timer
+     */
+    startIdleCheck() {
+        this.idleCheckTimer = setInterval(() => {
+            this.removeIdleConnections();
+        }, 60000); // Check every minute
+    }
+    /**
+     * Remove idle connections that have exceeded idle timeout
+     */
+    async removeIdleConnections() {
+        const now = Date.now();
+        for (const conn of this.connections.values()) {
+            if (conn.inUse) {
+                continue;
+            }
+            const idleTime = now - conn.lastUsedAt.getTime();
+            if (idleTime > this.options.idleTimeout) {
+                this.logger.info("Removing idle connection", {
+                    connectionId: conn.id,
+                    serverId: conn.serverId,
+                    idleTime: Math.floor(idleTime / 1000),
+                });
+                await this.remove(conn.id);
+            }
+        }
+    }
+    /**
+     * Generate unique connection ID
+     */
+    generateConnectionId() {
+        return `conn_${Date.now()}_${Math.random().toString(36).substring(2, 9)}`;
+    }
+}
+//# sourceMappingURL=ConnectionPool.js.map

package/dist/core/ConnectionPool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConnectionPool.js","sourceRoot":"","sources":["../../src/core/ConnectionPool.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EACL,qBAAqB,EACrB,sBAAsB,EACtB,kBAAkB,GACnB,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAkD3C;;GAEG;AACH,MAAM,OAAO,cAAe,SAAQ,YAAY;IAC7B,WAAW,GAAkC,IAAI,GAAG,EAAE,CAAC;IACvD,YAAY,GAAwB,EAAE,CAAC;IACvC,OAAO,CAAwB;IAC/B,MAAM,CAAiB;IAChC,gBAAgB,GAA0B,IAAI,CAAC;IAC/C,cAAc,GAA0B,IAAI,CAAC;IAC7C,mBAAmB,GAAG,CAAC,CAAC;IACxB,UAAU,GAAG,KAAK,CAAC;IAE3B,YAAY,OAAuC,EAAE,MAAsB;QACzE,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,OAAO,GAAG;YACb,cAAc,EAAE,OAAO,CAAC,cAAc,IAAI,EAAE;YAC5C,iBAAiB,EAAE,OAAO,CAAC,iBAAiB,IAAI,KAAK;YACrD,cAAc,EAAE,OAAO,CAAC,cAAc,IAAI,KAAK;YAC/C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,MAAM,EAAE,YAAY;YACxD,mBAAmB,EAAE,OAAO,CAAC,mBAAmB,IAAI,KAAK,EAAE,WAAW;YACtE,sBAAsB,EAAE,OAAO,CAAC,sBAAsB,IAAI,CAAC;SAC5D,CAAC;QACF,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6BAA6B,EAAE;YAC9C,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;SAC5C,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,QAAgB,EAChB,MAA2B;QAE3B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QAED,qCAAqC;QACrC,MAAM,YAAY,GAAG,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QACvD,IAAI,YAAY,EAAE,CAAC;YACjB,YAAY,CAAC,KAAK,GAAG,IAAI,CAAC;YAC1B,YAAY,CAAC,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC;YACrC,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE;gBAC/C,QAAQ;gBACR,YAAY,EAAE,YAAY,CAAC,EAAE;aAC9B,CAAC,CAAC;YACH,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,0CAA0C;QAC1C,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;YACxD,OAAO,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACjD,CAAC;QAED,4CAA4C;QAC5C,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,OAAO,CAAC,UAA4B;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACnD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QAED,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,MAAM,CAAC,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC;QAE/B,wBAAwB;QACxB,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAE3B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,EAAE;YACvC,YAAY,EAAE,UAAU,CAAC,EAAE;YAC3B,QAAQ,EAAE,UAAU,CAAC,QAAQ;SAC9B,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,YAAoB;QAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,UAAU,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;QACvC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oCAAoC,EAAE;gBACrD,YAAY;gBACZ,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACtC,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;QAE7C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,8BAA8B,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QAEvB,cAAc;QACd,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,aAAa,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACrC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;QACD,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QAED,8BAA8B;QAC9B,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACxC,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAChC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC;QAE7B,wBAAwB;QACxB,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAC7D,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YACjC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE;oBACnD,YAAY,EAAE,IAAI,CAAC,EAAE;oBACrB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CACF,CAAC;QAEF,MAAM,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;QACxC,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QAEzB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,IAAI,iBAAiB,GAAG,CAAC,CAAC;QAC1B,IAAI,eAAe,GAAG,CAAC,CAAC;QAExB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,iBAAiB,EAAE,CAAC;YACtB,CAAC;iBAAM,CAAC;gBACN,eAAe,EAAE,CAAC;YACpB,CAAC;QACH,CAAC;QAED,OAAO;YACL,gBAAgB,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;YACvC,iBAAiB;YACjB,eAAe;YACf,eAAe,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM;YACzC,mBAAmB,EAAE,IAAI,CAAC,mBAAmB;SAC9C,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAClB,QAAgB,EAChB,MAA2B,EAC3B,OAAe;QAEf,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAExD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACxD,OAAO,MAAM,CAAC;QAChB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,QAAgB;QACzC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;gBAC9C,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,gBAAgB,CAC5B,QAAgB,EAChB,MAA2B;QAE3B,MAAM,YAAY,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;QACjD,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;YAC3B,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB;YACjD,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;SAC5C,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC7B,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAE7B,MAAM,MAAM,GAAqB;gBAC/B,EAAE,EAAE,YAAY;gBAChB,QAAQ;gBACR,MAAM;gBACN,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,UAAU,EAAE,IAAI,IAAI,EAAE;gBACtB,KAAK,EAAE,IAAI;gBACX,mBAAmB,EAAE,CAAC;aACvB,CAAC;YAEF,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YAC3C,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAE3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE;gBACzC,YAAY;gBACZ,QAAQ;gBACR,gBAAgB,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;aACxC,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC;YAEvD,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,sBAAsB,EAAE,CAAC;gBAC5C,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,IAAI,qBAAqB,CAC7B,MAAM,CAAC,IAAI,EACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAC1D,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CACvB,QAAgB,EAChB,MAA2B;QAE3B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,OAA0B,CAAC;YAC/B,IAAI,OAAO,GAAG,KAAK,CAAC;YAEpB,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;gBAChC,yCAAyC;gBACzC,IAAI,OAAO,EAAE,CAAC;oBACZ,OAAO;gBACT,CAAC;gBACD,OAAO,GAAG,IAAI,CAAC;gBAEf,6CAA6C;gBAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CACvC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,QAAQ,IAAI,CAAC,CAAC,SAAS,KAAK,OAAO,CAAC,SAAS,CACpE,CAAC;gBACF,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;oBACjB,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;gBACrC,CAAC;gBAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE;oBAChD,QAAQ;oBACR,WAAW,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM;iBACtC,CAAC,CAAC;gBAEH,MAAM,CACJ,IAAI,kBAAkB,CACpB,IAAI,CAAC,OAAO,CAAC,cAAc,EAC3B,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAC/B,CACF,CAAC;YACJ,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YAEnC,4BAA4B;YAC5B,OAAO,GAAG;gBACR,QAAQ;gBACR,MAAM;gBACN,OAAO,EAAE,CAAC,IAAsB,EAAE,EAAE;oBAClC,IAAI,OAAO;wBAAE,OAAO;oBACpB,OAAO,GAAG,IAAI,CAAC;oBACf,YAAY,CAAC,SAAS,CAAC,CAAC;oBACxB,OAAO,CAAC,IAAI,CAAC,CAAC;gBAChB,CAAC;gBACD,MAAM,EAAE,CAAC,KAAY,EAAE,EAAE;oBACvB,IAAI,OAAO;wBAAE,OAAO;oBACpB,OAAO,GAAG,IAAI,CAAC;oBACf,YAAY,CAAC,SAAS,CAAC,CAAC;oBACxB,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChB,CAAC;gBACD,SAAS;gBACT,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC;YAEF,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEhC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE;gBAC7C,QAAQ;gBACR,WAAW,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM;aACtC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,mBAAmB;QACzB,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,OAAO;QACT,CAAC;QAED,oCAAoC;QACpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAClD,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,IAAI,CAAC,OAAO;gBAAE,SAAS;YAEvB,4BAA4B;YAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC3D,IAAI,QAAQ,EAAE,CAAC;gBACb,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBAC/B,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAChC,QAAQ,CAAC,KAAK,GAAG,IAAI,CAAC;gBACtB,QAAQ,CAAC,UAAU,GAAG,IAAI,IAAI,EAAE,CAAC;gBACjC,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC3B,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBAC1B,OAAO;YACT,CAAC;YAED,oCAAoC;YACpC,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gBACxD,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBAC/B,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAChC,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC;qBACpD,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC;qBACrB,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBACzB,OAAO;YACT,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,gBAAgB;QACtB,IAAI,CAAC,gBAAgB,GAAG,WAAW,CAAC,GAAG,EAAE;YACvC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC5B,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB;QAC9B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,SAAS;YACX,CAAC;YAED,IAAI,CAAC;gBACH,6CAA6C;gBAC7C,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;gBACjD,IAAI,CAAC,mBAAmB,GAAG,CAAC,CAAC;YAC/B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gCAAgC,EAAE;oBACjD,YAAY,EAAE,IAAI,CAAC,EAAE;oBACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,QAAQ,EAAE,IAAI,CAAC,mBAAmB;iBACnC,CAAC,CAAC;gBAEH,IAAI,IAAI,CAAC,mBAAmB,IAAI,IAAI,CAAC,OAAO,CAAC,sBAAsB,EAAE,CAAC;oBACpE,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,iDAAiD,EACjD,SAAS,EACT;wBACE,YAAY,EAAE,IAAI,CAAC,EAAE;wBACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;qBACxB,CACF,CAAC;oBACF,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc;QACpB,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;YACrC,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC/B,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,qBAAqB;IAClC,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,qBAAqB;QACjC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,SAAS;YACX,CAAC;YAED,MAAM,QAAQ,GAAG,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACjD,IAAI,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;gBACxC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0BAA0B,EAAE;oBAC3C,YAAY,EAAE,IAAI,CAAC,EAAE;oBACrB,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC;iBACtC,CAAC,CAAC;gBACH,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,oBAAoB;QAC1B,OAAO,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IAC5E,CAAC;CACF"}

package/dist/core/SSHClient.d.ts

@@ -0,0 +1,112 @@
+/**
+ * SSH Client
+ * Provides SSH connection management and command execution with timeout and cancellation support
+ */
+import type { CommandResult, ISSHClient, SSHConnectionConfig, SSHConnectionState } from "../types/index.js";
+/**
+ * Cancellation token for aborting operations
+ */
+export declare class CancellationToken {
+    private cancelled;
+    private readonly listeners;
+    /**
+     * Check if cancellation has been requested
+     */
+    get isCancelled(): boolean;
+    /**
+     * Request cancellation
+     */
+    cancel(): void;
+    /**
+     * Register a callback for cancellation
+     */
+    onCancel(callback: () => void): void;
+    /**
+     * Throw if cancelled
+     */
+    throwIfCancelled(): void;
+}
+/**
+ * Cancellation token source for creating tokens
+ */
+export declare class CancellationTokenSource {
+    readonly token: CancellationToken;
+    constructor();
+    /**
+     * Cancel the operation
+     */
+    cancel(): void;
+}
+/**
+ * Execute options for command execution
+ */
+export interface ExecuteOptions {
+    timeout?: number;
+    cancellationToken?: CancellationToken;
+    maxBufferSize?: number;
+    chunkCallback?: (chunk: string, isStderr: boolean) => void;
+}
+/**
+ * Stream chunk for large outputs
+ */
+export interface StreamChunk {
+    data: string;
+    isStderr: boolean;
+    timestamp: Date;
+}
+/**
+ * SSH Client implementation with enhanced timeout and cancellation support
+ */
+export declare class SSHClient implements ISSHClient {
+    private client;
+    private serverId;
+    private state;
+    private readonly connectionTimeout;
+    private readonly commandTimeout;
+    private activeExecutions;
+    constructor(options?: {
+        connectionTimeout?: number;
+        commandTimeout?: number;
+    });
+    /**
+     * Connect to SSH server
+     */
+    connect(config: SSHConnectionConfig): Promise<void>;
+    /**
+     * Disconnect from SSH server
+     */
+    disconnect(): Promise<void>;
+    /**
+     * Check if connected to SSH server
+     */
+    isConnected(): boolean;
+    /**
+     * Execute a command on the SSH server
+     */
+    execute(command: string, options?: ExecuteOptions): Promise<CommandResult>;
+    /**
+     * Execute a command with streaming output
+     */
+    executeStreaming(command: string, onChunk: (chunk: StreamChunk) => void, options?: ExecuteOptions): Promise<CommandResult>;
+    /**
+     * Cancel all active executions
+     */
+    private cancelAllExecutions;
+    /**
+     * Get the server ID if connected
+     */
+    getServerId(): string | undefined;
+    /**
+     * Set the server ID
+     */
+    setServerId(id: string): void;
+    /**
+     * Get current connection state
+     */
+    getState(): SSHConnectionState;
+    /**
+     * Get active execution count
+     */
+    getActiveExecutionCount(): number;
+}
+//# sourceMappingURL=SSHClient.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"SSHClient.d.ts","sourceRoot":"","sources":["../../src/core/SSHClient.ts"],"names":[],"mappings":"AAAA;;;GAGG;AASH,OAAO,KAAK,EACV,aAAa,EACb,UAAU,EACV,mBAAmB,EACnB,kBAAkB,EACnB,MAAM,mBAAmB,CAAC;AAE3B;;GAEG;AACH,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAyB;IAEnD;;OAEG;IACH,IAAI,WAAW,IAAI,OAAO,CAEzB;IAED;;OAEG;IACH,MAAM,IAAI,IAAI;IAed;;OAEG;IACH,QAAQ,CAAC,QAAQ,EAAE,MAAM,IAAI,GAAG,IAAI;IAQpC;;OAEG;IACH,gBAAgB,IAAI,IAAI;CAKzB;AAED;;GAEG;AACH,qBAAa,uBAAuB;IAClC,QAAQ,CAAC,KAAK,EAAE,iBAAiB,CAAC;;IAMlC;;OAEG;IACH,MAAM,IAAI,IAAI;CAGf;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IACtC,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,KAAK,IAAI,CAAC;CAC5D;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,OAAO,CAAC;IAClB,SAAS,EAAE,IAAI,CAAC;CACjB;AAED;;GAEG;AACH,qBAAa,SAAU,YAAW,UAAU;IAC1C,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,QAAQ,CAAqB;IACrC,OAAO,CAAC,KAAK,CAAkD;IAC/D,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAS;IAC3C,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAS;IACxC,OAAO,CAAC,gBAAgB,CAGV;gBAEF,OAAO,CAAC,EAAE;QACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB;IAKD;;OAEG;IACG,OAAO,CAAC,MAAM,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IA8EzD;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAsBjC;;OAEG;IACH,WAAW,IAAI,OAAO;IAItB;;OAEG;IACG,OAAO,CACX,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,aAAa,CAAC;IAqLzB;;OAEG;IACG,gBAAgB,CACpB,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,EACrC,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,aAAa,CAAC;IAazB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAY3B;;OAEG;IACH,WAAW,IAAI,MAAM,GAAG,SAAS;IAIjC;;OAEG;IACH,WAAW,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI;IAI7B;;OAEG;IACH,QAAQ,IAAI,kBAAkB;IAI9B;;OAEG;IACH,uBAAuB,IAAI,MAAM;CAGlC"}