telegram-ssh-bot 2.0.0 → 2.2.0

package/dist/handlers/ServerHandler.js

@@ -0,0 +1,272 @@
+/**
+ * Server Handler
+ * Handles server management commands: /add, /rm, /list, /ssh, /exit, /current
+ */
+import { promises as fs } from "fs";
+import { ConnectionFailedError } from "../errors/index.js";
+import { BaseHandler } from "./BaseHandler.js";
+/**
+ * Handler for /add command - add a new server
+ */
+export class AddServerHandler extends BaseHandler {
+    name = "add";
+    description = "Add a new server configuration";
+    pattern = /^\/add/;
+    constructor(serverManager, sshClient, bot, logger) {
+        super(serverManager, sshClient, bot, logger);
+    }
+    async execute(context) {
+        const args = context.args.trim();
+        if (!args) {
+            await this.send(context.chatId, "Usage: /add <host> <username> [password] [port] [note]\n" +
+                'Example: /add 192.168.1.1 root password123 22 "My Server"');
+            return;
+        }
+        try {
+            const parts = this.parseArgs(args);
+            if (!parts[0] || !parts[1]) {
+                await this.send(context.chatId, "❌ Host and username are required.");
+                return;
+            }
+            // Parse and validate port number
+            let port = 22;
+            if (parts[3]) {
+                const parsedPort = parseInt(parts[3], 10);
+                if (isNaN(parsedPort)) {
+                    await this.send(context.chatId, "❌ Invalid port number. Must be a valid integer.");
+                    return;
+                }
+                if (parsedPort < 1 || parsedPort > 65535) {
+                    await this.send(context.chatId, "❌ Port must be between 1 and 65535.");
+                    return;
+                }
+                port = parsedPort;
+            }
+            const config = {
+                host: parts[0],
+                username: parts[1],
+                password: parts[2],
+                port,
+                note: parts[4],
+            };
+            const server = await this.serverManager.add(config);
+            await this.send(context.chatId, `✅ Server added successfully!\n\n${this.formatServerInfo(server)}`);
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : "Failed to add server";
+            await this.send(context.chatId, `❌ ${message}`);
+        }
+    }
+    parseArgs(args) {
+        // Handle quoted strings for note
+        const result = [];
+        let current = "";
+        let inQuotes = false;
+        for (const char of args) {
+            if (char === '"') {
+                inQuotes = !inQuotes;
+            }
+            else if (char === " " && !inQuotes) {
+                if (current) {
+                    result.push(current);
+                    current = "";
+                }
+            }
+            else {
+                current += char;
+            }
+        }
+        if (current) {
+            result.push(current);
+        }
+        return result;
+    }
+}
+/**
+ * Handler for /rm command - remove a server
+ */
+export class RemoveServerHandler extends BaseHandler {
+    name = "rm";
+    description = "Remove a server by index";
+    pattern = /^\/rm/;
+    constructor(serverManager, sshClient, bot, logger) {
+        super(serverManager, sshClient, bot, logger);
+    }
+    async execute(context) {
+        const args = context.args.trim();
+        if (!args) {
+            await this.send(context.chatId, "Usage: /rm <index>\nUse /list to see server indices.");
+            return;
+        }
+        // BUG-003 FIX: Use parseInt instead of parseFloat
+        const index = parseInt(args, 10);
+        if (isNaN(index) || index < 1) {
+            await this.send(context.chatId, "❌ Invalid index. Please provide a valid number.");
+            return;
+        }
+        try {
+            const server = await this.serverManager.getByIndex(index);
+            if (!server) {
+                await this.send(context.chatId, `❌ Server at index ${index} not found.`);
+                return;
+            }
+            // Disconnect if removing current server
+            if (this.sshClient.isConnected() &&
+                this.serverManager.getCurrent()?.id === server.id) {
+                await this.sshClient.disconnect();
+                this.serverManager.clearCurrent();
+            }
+            await this.serverManager.removeByIndex(index);
+            await this.send(context.chatId, `✅ Server removed: \`${server.username}@${server.host}\``);
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : "Failed to remove server";
+            await this.send(context.chatId, `❌ ${message}`);
+        }
+    }
+}
+/**
+ * Handler for /list command - list all servers
+ */
+export class ListServerHandler extends BaseHandler {
+    name = "list";
+    description = "List all configured servers";
+    pattern = /^\/list/;
+    constructor(serverManager, sshClient, bot, logger) {
+        super(serverManager, sshClient, bot, logger);
+    }
+    async execute(context) {
+        const servers = await this.serverManager.list();
+        const message = this.formatServerList(servers);
+        await this.sendMarkdown(context.chatId, message);
+    }
+}
+/**
+ * Handler for /ssh command - connect to a server
+ */
+export class SSHServerHandler extends BaseHandler {
+    name = "ssh";
+    description = "Connect to a server by index";
+    pattern = /^\/ssh/;
+    constructor(serverManager, sshClient, bot, logger) {
+        super(serverManager, sshClient, bot, logger);
+    }
+    async execute(context) {
+        const args = context.args.trim();
+        if (!args) {
+            await this.send(context.chatId, "Usage: /ssh <index>\nUse /list to see server indices.");
+            return;
+        }
+        // BUG-003 FIX: Use parseInt instead of parseFloat
+        const index = parseInt(args, 10);
+        if (isNaN(index) || index < 1) {
+            await this.send(context.chatId, "❌ Invalid index. Please provide a valid number.");
+            return;
+        }
+        const server = await this.serverManager.getByIndex(index);
+        if (!server) {
+            await this.send(context.chatId, `❌ Server at index ${index} not found.`);
+            return;
+        }
+        // Disconnect current connection
+        if (this.sshClient.isConnected()) {
+            await this.sshClient.disconnect();
+            this.serverManager.clearCurrent();
+        }
+        await this.send(context.chatId, `🔌 Connecting to \`${server.username}@${server.host}\`...`);
+        try {
+            // Get decrypted credentials
+            const credentials = this.serverManager.getDecryptedCredentials(server);
+            // Build SSH config
+            const sshConfig = {
+                host: server.host,
+                port: server.port,
+                username: server.username,
+                password: credentials.password,
+                passphrase: credentials.passphrase,
+            };
+            // Load private key if needed
+            if (server.auth.type === "privateKey" && credentials.privateKeyPath) {
+                try {
+                    const keyBuffer = await fs.readFile(credentials.privateKeyPath);
+                    sshConfig.privateKey = keyBuffer;
+                }
+                catch (error) {
+                    throw new Error(`Failed to read private key: ${credentials.privateKeyPath}`);
+                }
+            }
+            // Connect
+            await this.sshClient.connect(sshConfig);
+            this.sshClient.setServerId(server.id);
+            this.serverManager.setCurrent(server);
+            await this.send(context.chatId, `✅ Connected to \`${server.username}@${server.host}\`\n` +
+                `Send any command to execute on the server.`);
+            this.logger.info("SSH connection established", {
+                serverId: server.id,
+                host: server.host,
+                chatId: context.chatId,
+            });
+        }
+        catch (error) {
+            if (error instanceof ConnectionFailedError) {
+                await this.send(context.chatId, `❌ Connection failed: ${error.message}`);
+            }
+            else {
+                const message = error instanceof Error ? error.message : "Unknown error";
+                await this.send(context.chatId, `❌ Failed to connect: ${message}`);
+            }
+            this.logger.error("SSH connection failed", error instanceof Error ? error : new Error(String(error)), {
+                serverId: server.id,
+                host: server.host,
+            });
+        }
+    }
+}
+/**
+ * Handler for /exit command - disconnect from current server
+ */
+export class ExitServerHandler extends BaseHandler {
+    name = "exit";
+    description = "Disconnect from current SSH session";
+    pattern = /^\/exit/;
+    constructor(serverManager, sshClient, bot, logger) {
+        super(serverManager, sshClient, bot, logger);
+    }
+    async execute(context) {
+        if (!this.sshClient.isConnected()) {
+            await this.send(context.chatId, "❌ Not connected to any server.");
+            return;
+        }
+        const current = this.serverManager.getCurrent();
+        await this.sshClient.disconnect();
+        this.serverManager.clearCurrent();
+        const host = current?.host ?? "unknown";
+        await this.send(context.chatId, `👋 Disconnected from \`${host}\``);
+        this.logger.info("SSH disconnected", {
+            serverId: current?.id,
+            chatId: context.chatId,
+        });
+    }
+}
+/**
+ * Handler for /current command - show current server
+ */
+export class CurrentServerHandler extends BaseHandler {
+    name = "current";
+    description = "Show current connected server";
+    pattern = /^\/current/;
+    constructor(serverManager, sshClient, bot, logger) {
+        super(serverManager, sshClient, bot, logger);
+    }
+    async execute(context) {
+        const current = this.serverManager.getCurrent();
+        if (!current) {
+            await this.send(context.chatId, "❌ Not connected to any server.\nUse /ssh <index> to connect.");
+            return;
+        }
+        const connected = this.sshClient.isConnected();
+        const status = connected ? "🟢 Connected" : "🔴 Disconnected";
+        await this.sendMarkdown(context.chatId, `${status}\n\n${this.formatServerInfo(current)}`);
+    }
+}
+//# sourceMappingURL=ServerHandler.js.map

package/dist/handlers/ServerHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ServerHandler.js","sourceRoot":"","sources":["../../src/handlers/ServerHandler.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,IAAI,CAAC;AAGpC,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAI3D,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAE/C;;GAEG;AACH,MAAM,OAAO,gBAAiB,SAAQ,WAAW;IACtC,IAAI,GAAG,KAAK,CAAC;IACb,WAAW,GAAG,gCAAgC,CAAC;IAC/C,OAAO,GAAG,QAAQ,CAAC;IAE5B,YACE,aAA4B,EAC5B,SAAoB,EACpB,GAAS,EACT,MAAsB;QAEtB,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAuB;QACnC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAEjC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,0DAA0D;gBACxD,2DAA2D,CAC9D,CAAC;YACF,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAEnC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3B,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,mCAAmC,CAAC,CAAC;gBACrE,OAAO;YACT,CAAC;YAED,iCAAiC;YACjC,IAAI,IAAI,GAAG,EAAE,CAAC;YACd,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;gBACb,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC1C,IAAI,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;oBACtB,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,iDAAiD,CAClD,CAAC;oBACF,OAAO;gBACT,CAAC;gBACD,IAAI,UAAU,GAAG,CAAC,IAAI,UAAU,GAAG,KAAK,EAAE,CAAC;oBACzC,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,qCAAqC,CACtC,CAAC;oBACF,OAAO;gBACT,CAAC;gBACD,IAAI,GAAG,UAAU,CAAC;YACpB,CAAC;YAED,MAAM,MAAM,GAAG;gBACb,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;gBACd,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC;gBAClB,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC;gBAClB,IAAI;gBACJ,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;aACf,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACpD,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,mCAAmC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CACnE,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,sBAAsB,CAAC;YAClE,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,OAAO,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,IAAY;QAC5B,iCAAiC;QACjC,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,QAAQ,GAAG,KAAK,CAAC;QAErB,KAAK,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YACxB,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjB,QAAQ,GAAG,CAAC,QAAQ,CAAC;YACvB,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACrC,IAAI,OAAO,EAAE,CAAC;oBACZ,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACrB,OAAO,GAAG,EAAE,CAAC;gBACf,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,OAAO,IAAI,IAAI,CAAC;YAClB,CAAC;QACH,CAAC;QAED,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,mBAAoB,SAAQ,WAAW;IACzC,IAAI,GAAG,IAAI,CAAC;IACZ,WAAW,GAAG,0BAA0B,CAAC;IACzC,OAAO,GAAG,OAAO,CAAC;IAE3B,YACE,aAA4B,EAC5B,SAAoB,EACpB,GAAS,EACT,MAAsB;QAEtB,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAuB;QACnC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAEjC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,sDAAsD,CACvD,CAAC;YACF,OAAO;QACT,CAAC;QAED,kDAAkD;QAClD,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAEjC,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,iDAAiD,CAClD,CAAC;YACF,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAC1D,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,qBAAqB,KAAK,aAAa,CACxC,CAAC;gBACF,OAAO;YACT,CAAC;YAED,wCAAwC;YACxC,IACE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE;gBAC5B,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,EAAE,KAAK,MAAM,CAAC,EAAE,EACjD,CAAC;gBACD,MAAM,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;gBAClC,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,CAAC;YACpC,CAAC;YAED,MAAM,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,uBAAuB,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,IAAI,CAC1D,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,yBAAyB,CAAC;YACrE,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,OAAO,EAAE,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,iBAAkB,SAAQ,WAAW;IACvC,IAAI,GAAG,MAAM,CAAC;IACd,WAAW,GAAG,6BAA6B,CAAC;IAC5C,OAAO,GAAG,SAAS,CAAC;IAE7B,YACE,aAA4B,EAC5B,SAAoB,EACpB,GAAS,EACT,MAAsB;QAEtB,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAuB;QACnC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;QAChD,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC/C,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnD,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,gBAAiB,SAAQ,WAAW;IACtC,IAAI,GAAG,KAAK,CAAC;IACb,WAAW,GAAG,8BAA8B,CAAC;IAC7C,OAAO,GAAG,QAAQ,CAAC;IAE5B,YACE,aAA4B,EAC5B,SAAoB,EACpB,GAAS,EACT,MAAsB;QAEtB,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAuB;QACnC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QAEjC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,uDAAuD,CACxD,CAAC;YACF,OAAO;QACT,CAAC;QAED,kDAAkD;QAClD,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAEjC,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,iDAAiD,CAClD,CAAC;YACF,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAC1D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,qBAAqB,KAAK,aAAa,CAAC,CAAC;YACzE,OAAO;QACT,CAAC;QAED,gCAAgC;QAChC,IAAI,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC;YACjC,MAAM,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;YAClC,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,CAAC;QACpC,CAAC;QAED,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,sBAAsB,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,OAAO,CAC5D,CAAC;QAEF,IAAI,CAAC;YACH,4BAA4B;YAC5B,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC;YAEvE,mBAAmB;YACnB,MAAM,SAAS,GAAwB;gBACrC,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,QAAQ,EAAE,WAAW,CAAC,QAAQ;gBAC9B,UAAU,EAAE,WAAW,CAAC,UAAU;aACnC,CAAC;YAEF,6BAA6B;YAC7B,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,KAAK,YAAY,IAAI,WAAW,CAAC,cAAc,EAAE,CAAC;gBACpE,IAAI,CAAC;oBACH,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;oBAChE,SAAS,CAAC,UAAU,GAAG,SAAS,CAAC;gBACnC,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,IAAI,KAAK,CACb,+BAA+B,WAAW,CAAC,cAAc,EAAE,CAC5D,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,UAAU;YACV,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACxC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACtC,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAEtC,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,oBAAoB,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,MAAM;gBACtD,4CAA4C,CAC/C,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,EAAE;gBAC7C,QAAQ,EAAE,MAAM,CAAC,EAAE;gBACnB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,MAAM,EAAE,OAAO,CAAC,MAAM;aACvB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,qBAAqB,EAAE,CAAC;gBAC3C,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,wBAAwB,KAAK,CAAC,OAAO,EAAE,CACxC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;gBAC3D,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,wBAAwB,OAAO,EAAE,CAAC,CAAC;YACrE,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,uBAAuB,EACvB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EACzD;gBACE,QAAQ,EAAE,MAAM,CAAC,EAAE;gBACnB,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB,CACF,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,iBAAkB,SAAQ,WAAW;IACvC,IAAI,GAAG,MAAM,CAAC;IACd,WAAW,GAAG,qCAAqC,CAAC;IACpD,OAAO,GAAG,SAAS,CAAC;IAE7B,YACE,aAA4B,EAC5B,SAAoB,EACpB,GAAS,EACT,MAAsB;QAEtB,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAuB;QACnC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC;YAClC,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,gCAAgC,CAAC,CAAC;YAClE,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;QAChD,MAAM,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;QAClC,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,CAAC;QAElC,MAAM,IAAI,GAAG,OAAO,EAAE,IAAI,IAAI,SAAS,CAAC;QACxC,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,0BAA0B,IAAI,IAAI,CAAC,CAAC;QAEpE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,EAAE;YACnC,QAAQ,EAAE,OAAO,EAAE,EAAE;YACrB,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;IACL,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,oBAAqB,SAAQ,WAAW;IAC1C,IAAI,GAAG,SAAS,CAAC;IACjB,WAAW,GAAG,+BAA+B,CAAC;IAC9C,OAAO,GAAG,YAAY,CAAC;IAEhC,YACE,aAA4B,EAC5B,SAAoB,EACpB,GAAS,EACT,MAAsB;QAEtB,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAuB;QACnC,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;QAEhD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,IAAI,CACb,OAAO,CAAC,MAAM,EACd,8DAA8D,CAC/D,CAAC;YACF,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC;QAC/C,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,iBAAiB,CAAC;QAE9D,MAAM,IAAI,CAAC,YAAY,CACrB,OAAO,CAAC,MAAM,EACd,GAAG,MAAM,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE,CACjD,CAAC;IACJ,CAAC;CACF"}

package/dist/handlers/index.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Handler modules index - exports all handlers
+ */
+export { BaseHandler } from "./BaseHandler.js";
+export { SSHCommandHandler } from "./CommandHandler.js";
+export { HealthHandler } from "./HealthHandler.js";
+export { HelpHandler, StartHandler } from "./HelpHandler.js";
+export { AddServerHandler, CurrentServerHandler, ExitServerHandler, ListServerHandler, RemoveServerHandler, SSHServerHandler, } from "./ServerHandler.js";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/handlers/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAC7D,OAAO,EACL,gBAAgB,EAChB,oBAAoB,EACpB,iBAAiB,EACjB,iBAAiB,EACjB,mBAAmB,EACnB,gBAAgB,GACjB,MAAM,oBAAoB,CAAC"}

package/dist/handlers/index.js

@@ -0,0 +1,9 @@
+/**
+ * Handler modules index - exports all handlers
+ */
+export { BaseHandler } from "./BaseHandler.js";
+export { SSHCommandHandler } from "./CommandHandler.js";
+export { HealthHandler } from "./HealthHandler.js";
+export { HelpHandler, StartHandler } from "./HelpHandler.js";
+export { AddServerHandler, CurrentServerHandler, ExitServerHandler, ListServerHandler, RemoveServerHandler, SSHServerHandler, } from "./ServerHandler.js";
+//# sourceMappingURL=index.js.map

package/dist/handlers/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/handlers/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAC7D,OAAO,EACL,gBAAgB,EAChB,oBAAoB,EACpB,iBAAiB,EACjB,iBAAiB,EACjB,mBAAmB,EACnB,gBAAgB,GACjB,MAAM,oBAAoB,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,10 @@
+#!/usr/bin/env node
+/**
+ * Telegram SSH Bot - Entry Point
+ *
+ * A secure Telegram bot for managing SSH connections
+ *
+ * @version 2.0.0
+ */
+import "dotenv/config";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA;;;;;;GAMG;AAEH,OAAO,eAAe,CAAC"}

package/dist/index.js

@@ -0,0 +1,348 @@
+#!/usr/bin/env node
+/**
+ * Telegram SSH Bot - Entry Point
+ *
+ * A secure Telegram bot for managing SSH connections
+ *
+ * @version 2.0.0
+ */
+import "dotenv/config";
+import { ConfigReloader, loadConfig, validateOwnerIds, } from "./config/index.js";
+import { Bot, ConnectionPool, SSHClient, ServerManager } from "./core/index.js";
+import { HealthHandler } from "./handlers/HealthHandler.js";
+import { AddServerHandler, CurrentServerHandler, ExitServerHandler, HelpHandler, ListServerHandler, RemoveServerHandler, SSHCommandHandler, SSHServerHandler, StartHandler, } from "./handlers/index.js";
+import { AuthMiddleware, RateLimitMiddleware } from "./middleware/index.js";
+import { BackupService, CryptoService, HealthService, LoggingService, MonitoringService, NotificationService, RateLimiter, ValidationService, } from "./services/index.js";
+/**
+ * Application class with enhanced graceful shutdown
+ */
+class Application {
+    config;
+    configReloader;
+    logger;
+    cryptoService;
+    validationService;
+    rateLimiter;
+    sshClient;
+    connectionPool;
+    serverManager;
+    bot;
+    healthService;
+    monitoringService;
+    backupService;
+    notificationService;
+    state = "idle";
+    startTime;
+    shutdownTimeout = null;
+    isShuttingDown = false;
+    constructor() {
+        this.startTime = new Date();
+    }
+    /**
+     * Initialize the application
+     */
+    async initialize() {
+        if (this.state !== "idle") {
+            throw new Error(`Cannot initialize from state: ${this.state}`);
+        }
+        this.state = "initializing";
+        try {
+            // Load configuration
+            this.config = await loadConfig();
+            validateOwnerIds(this.config);
+            // Initialize config reloader for hot reload
+            this.configReloader = new ConfigReloader(this.config);
+            this.setupConfigReload();
+            // Initialize services
+            this.logger = new LoggingService(this.config.logging);
+            this.cryptoService = new CryptoService(this.config.security.encryptionKey);
+            this.validationService = new ValidationService({
+                allowedCommands: this.config.security.allowedCommands,
+                blockedCommands: this.config.security.blockedCommands,
+            });
+            this.rateLimiter = new RateLimiter(this.config.security.rateLimit);
+            this.rateLimiter.startAutoCleanup();
+            // Initialize health service
+            this.healthService = new HealthService({
+                version: "2.0.0",
+                startTime: this.startTime,
+            });
+            // Initialize notification service
+            this.notificationService = new NotificationService({
+                enabled: true,
+                notifyOnStartup: true,
+                notifyOnShutdown: true,
+                notifyOnErrors: true,
+                notifyOnSecurity: true,
+                adminChatId: this.config.telegram.chatId,
+            }, this.logger);
+            // Initialize backup service
+            this.backupService = new BackupService({
+                enabled: this.config.backup.enabled,
+                backupDir: "./backups",
+                maxBackups: this.config.backup.maxCount,
+                backupInterval: this.config.backup.intervalMs,
+                backupBeforeWrite: true,
+            }, this.logger);
+            await this.backupService.initialize();
+            // Initialize monitoring service
+            this.monitoringService = new MonitoringService({
+                enabled: this.config.monitoring.enabled,
+                checkInterval: this.config.monitoring.intervalMs,
+                timeout: 10000,
+                alertThreshold: 3,
+                cooldownPeriod: 300000,
+            }, this.logger);
+            this.setupMonitoring();
+            // Initialize core components
+            this.sshClient = new SSHClient({
+                connectionTimeout: this.config.ssh.connectionTimeout,
+                commandTimeout: this.config.ssh.commandTimeout,
+            });
+            // Initialize connection pool
+            this.connectionPool = new ConnectionPool({
+                maxConnections: this.config.ssh.maxConnections,
+                connectionTimeout: this.config.ssh.connectionTimeout,
+                commandTimeout: this.config.ssh.commandTimeout,
+                idleTimeout: 300000, // 5 minutes
+                healthCheckInterval: 60000, // 1 minute
+                maxHealthCheckFailures: 3,
+            }, this.logger);
+            await this.connectionPool.initialize();
+            this.serverManager = new ServerManager(this.config.storage.serversFile, this.cryptoService, this.logger);
+            // Initialize server manager
+            await this.serverManager.initialize();
+            // Initialize bot
+            this.bot = new Bot({
+                token: this.config.telegram.token,
+                ownerIds: this.config.telegram.ownerIds,
+                polling: this.config.telegram.polling,
+            }, this.logger);
+            // Set bot for notification service
+            this.notificationService.setBot(this.bot);
+            // Setup middleware
+            this.setupMiddleware();
+            // Register handlers
+            this.registerHandlers();
+            // Update health service
+            this.healthService.setBotHealth(true);
+            this.healthService.setStorageHealth(true);
+            this.state = "running";
+            this.logger.info("Application initialized", {
+                ownerCount: this.config.telegram.ownerIds.length,
+                serversFile: this.config.storage.serversFile,
+            });
+        }
+        catch (error) {
+            this.state = "idle";
+            throw error;
+        }
+    }
+    /**
+     * Setup configuration hot reload
+     */
+    setupConfigReload() {
+        this.configReloader.on("reload", (event) => {
+            this.logger.info("Configuration reloaded", {
+                changedKeys: event.changedKeys,
+            });
+            this.notificationService.notifyConfigReloaded(true);
+        });
+        this.configReloader.on("error", (error) => {
+            this.logger.error("Configuration reload failed", error);
+            this.notificationService.notifyConfigReloaded(false);
+        });
+        this.configReloader.startWatching();
+    }
+    /**
+     * Setup monitoring service
+     */
+    setupMonitoring() {
+        this.monitoringService.on("alert:triggered", (payload) => {
+            this.logger.warn("Monitoring alert triggered", {
+                serverId: payload.server.id,
+                status: payload.status,
+            });
+            this.notificationService.notifyConnectionLost(payload.server, payload.status.error);
+        });
+        this.monitoringService.on("server:recovered", (payload) => {
+            this.logger.info("Server recovered", {
+                serverId: payload.server.id,
+            });
+            this.notificationService.notifyConnectionRestored(payload.server);
+        });
+    }
+    /**
+     * Setup middleware stack
+     */
+    setupMiddleware() {
+        // Auth middleware - only allow authorized users
+        const authMiddleware = new AuthMiddleware(this.config.telegram.ownerIds, this.logger);
+        this.bot.use(authMiddleware);
+        // Rate limiting middleware
+        const rateLimitMiddleware = new RateLimitMiddleware(this.rateLimiter);
+        this.bot.use(rateLimitMiddleware);
+        this.logger.debug("Middleware configured");
+    }
+    /**
+     * Register command handlers
+     */
+    registerHandlers() {
+        // Server management handlers
+        this.bot.registerCommand(new AddServerHandler(this.serverManager, this.sshClient, this.bot, this.logger));
+        this.bot.registerCommand(new RemoveServerHandler(this.serverManager, this.sshClient, this.bot, this.logger));
+        this.bot.registerCommand(new ListServerHandler(this.serverManager, this.sshClient, this.bot, this.logger));
+        this.bot.registerCommand(new SSHServerHandler(this.serverManager, this.sshClient, this.bot, this.logger));
+        this.bot.registerCommand(new ExitServerHandler(this.serverManager, this.sshClient, this.bot, this.logger));
+        this.bot.registerCommand(new CurrentServerHandler(this.serverManager, this.sshClient, this.bot, this.logger));
+        // SSH command handler (for non-command messages)
+        this.bot.registerCommand(new SSHCommandHandler(this.serverManager, this.sshClient, this.bot, this.logger, this.validationService));
+        // Help handlers
+        this.bot.registerCommand(new HelpHandler(this.serverManager, this.sshClient, this.bot, this.logger));
+        this.bot.registerCommand(new StartHandler(this.serverManager, this.sshClient, this.bot, this.logger));
+        // Health handler
+        this.bot.registerCommand(new HealthHandler(this.serverManager, this.sshClient, this.bot, this.logger, this.healthService));
+        this.logger.debug("Handlers registered");
+    }
+    /**
+     * Start the application
+     */
+    async start() {
+        if (this.state !== "running") {
+            throw new Error(`Cannot start from state: ${this.state}`);
+        }
+        await this.bot.start();
+        // Send startup notification
+        await this.notificationService.notifyStartup("2.0.0");
+        this.logger.info("Bot started successfully");
+    }
+    /**
+     * Stop the application gracefully
+     */
+    async stop(options) {
+        const opts = {
+            reason: options?.reason ?? "Manual shutdown",
+            timeout: options?.timeout ?? 30000, // 30 seconds default
+            notifyAdmin: options?.notifyAdmin ?? true,
+        };
+        if (this.isShuttingDown) {
+            this.logger.warn("Shutdown already in progress");
+            return;
+        }
+        this.isShuttingDown = true;
+        this.state = "shutting_down";
+        this.logger.info("Starting graceful shutdown", { reason: opts.reason });
+        // Set timeout for forced shutdown
+        this.shutdownTimeout = setTimeout(() => {
+            this.logger.error("Graceful shutdown timeout exceeded, forcing exit");
+            process.exit(1);
+        }, opts.timeout);
+        try {
+            // Notify admin about shutdown
+            if (opts.notifyAdmin) {
+                await this.notificationService.notifyShutdown(opts.reason);
+            }
+            // Stop accepting new requests
+            this.healthService.setBotHealth(false);
+            // Stop monitoring
+            this.monitoringService.stop();
+            // Stop rate limiter auto cleanup
+            this.rateLimiter.stopAutoCleanup();
+            // Stop config watcher
+            this.configReloader.stopWatching();
+            // Disconnect SSH client
+            if (this.sshClient.isConnected()) {
+                this.logger.info("Disconnecting SSH client");
+                await this.sshClient.disconnect();
+            }
+            // Drain connection pool
+            this.logger.info("Draining connection pool");
+            await this.connectionPool.drain();
+            // Stop backup service
+            this.backupService.stop();
+            // Stop bot
+            this.logger.info("Stopping bot");
+            await this.bot.stop();
+            // Clear shutdown timeout
+            if (this.shutdownTimeout) {
+                clearTimeout(this.shutdownTimeout);
+                this.shutdownTimeout = null;
+            }
+            this.state = "stopped";
+            this.logger.info("Application stopped gracefully");
+        }
+        catch (error) {
+            this.logger.error("Error during shutdown", error);
+            throw error;
+        }
+    }
+    /**
+     * Get application state
+     */
+    getState() {
+        return this.state;
+    }
+    /**
+     * Get health service
+     */
+    getHealthService() {
+        return this.healthService;
+    }
+}
+/**
+ * Main entry point
+ */
+async function main() {
+    const app = new Application();
+    // Track if we're already shutting down
+    let isShuttingDown = false;
+    /**
+     * Handle shutdown signals gracefully
+     */
+    const handleShutdown = async (signal) => {
+        if (isShuttingDown) {
+            console.log(`\nAlready shutting down, please wait...`);
+            return;
+        }
+        isShuttingDown = true;
+        console.log(`\nReceived ${signal}, initiating graceful shutdown...`);
+        try {
+            await app.stop({ reason: `Received ${signal}` });
+            process.exit(0);
+        }
+        catch (error) {
+            console.error("Error during shutdown:", error);
+            process.exit(1);
+        }
+    };
+    // Register signal handlers
+    process.on("SIGINT", () => handleShutdown("SIGINT"));
+    process.on("SIGTERM", () => handleShutdown("SIGTERM"));
+    process.on("SIGHUP", () => handleShutdown("SIGHUP"));
+    // Handle uncaught exceptions
+    process.on("uncaughtException", async (error) => {
+        console.error("Uncaught exception:", error);
+        try {
+            await handleShutdown("uncaughtException");
+        }
+        catch (e) {
+            console.error("Error during shutdown:", e);
+        }
+        process.exit(1);
+    });
+    // Handle unhandled promise rejections
+    process.on("unhandledRejection", (reason, promise) => {
+        console.error("Unhandled rejection at:", promise, "reason:", reason);
+    });
+    try {
+        await app.initialize();
+        await app.start();
+    }
+    catch (error) {
+        console.error("Failed to start application:", error);
+        process.exit(1);
+    }
+}
+// Run the application
+main();
+//# sourceMappingURL=index.js.map