ssh-agent-workspace 1.0.0

package/dist/core/TmuxManager.js

@@ -0,0 +1,110 @@
+import { logger } from '../utils/logger.js';
+export const MCP_PROMPT = '__MCP_PROMPT__> ';
+function q(value) {
+    return `'${value.replace(/'/g, "'\\''")}'`;
+}
+export class TmuxManager {
+    sshManager;
+    constructor(sshManager) {
+        this.sshManager = sshManager;
+    }
+    async isInstalled(ssh) {
+        try {
+            const result = await this.sshManager.exec(ssh, 'tmux -V');
+            return result.code === 0;
+        }
+        catch (err) {
+            logger.debug({ error: err.message }, 'tmux check failed');
+            return false;
+        }
+    }
+    async createOrAttachSession(ssh, sessionName) {
+        const result = await this.sshManager.exec(ssh, `tmux new-session -A -s ${q(sessionName)} -d`);
+        if (result.code !== 0) {
+            throw new Error(`Failed to create/attach tmux session '${sessionName}': ${result.stderr || result.stdout}`);
+        }
+        logger.info({ sessionName }, 'Tmux session created/attached');
+    }
+    async applyOptions(ssh, sessionName) {
+        await this.sshManager.exec(ssh, `tmux set-option -t ${q(sessionName)} history-limit 50000`);
+        await this.sshManager.exec(ssh, `tmux set-option -t ${q(sessionName)} mouse off`);
+        logger.info({ sessionName }, 'Applied tmux options (history-limit=50000, mouse=off)');
+    }
+    async detectShell(ssh, sessionName) {
+        const result = await this.sshManager.exec(ssh, `tmux display-message -t ${q(sessionName)} -p '#{pane_current_command}'`);
+        if (result.code !== 0) {
+            return null;
+        }
+        const shell = result.stdout.trim().toLowerCase();
+        logger.info({ sessionName, shell }, 'Shell detected');
+        return shell;
+    }
+    async injectPrompt(ssh, sessionName) {
+        const cmd = `export PS1='${MCP_PROMPT}'`;
+        await this.sendCommand(ssh, sessionName, cmd);
+        logger.info({ sessionName }, 'Injected deterministic prompt');
+    }
+    async healthCheck(ssh, sessionName) {
+        const result = await this.sshManager.exec(ssh, `tmux has-session -t ${q(sessionName)}`);
+        if (result.code !== 0) {
+            logger.warn({ sessionName }, 'Tmux health check failed: session missing');
+            return false;
+        }
+        return true;
+    }
+    async sendInput(ssh, sessionName, input) {
+        const base64 = Buffer.from(input).toString('base64');
+        const command = `printf '%s' ${q(base64)} | base64 -d | tmux load-buffer - && tmux paste-buffer -t ${q(sessionName)} -d`;
+        const result = await this.sshManager.exec(ssh, command);
+        if (result.code !== 0) {
+            throw new Error(`Failed to send input to tmux session '${sessionName}': ${result.stderr || result.stdout}`);
+        }
+    }
+    async sendCommand(ssh, sessionName, command) {
+        const base64 = Buffer.from(command + '\n').toString('base64');
+        const shellCommand = `printf '%s' ${q(base64)} | base64 -d | tmux load-buffer - && tmux paste-buffer -t ${q(sessionName)} -d`;
+        const result = await this.sshManager.exec(ssh, shellCommand);
+        if (result.code !== 0) {
+            throw new Error(`Failed to send command to tmux session '${sessionName}': ${result.stderr || result.stdout}`);
+        }
+    }
+    async sendSignal(ssh, sessionName, signal) {
+        const key = signal === 'SIGINT' ? 'C-c' : 'C-c';
+        const result = await this.sshManager.exec(ssh, `tmux send-keys -t ${q(sessionName)} ${key}`);
+        if (result.code !== 0) {
+            throw new Error(`Failed to send signal ${signal} to tmux session '${sessionName}': ${result.stderr || result.stdout}`);
+        }
+        logger.info({ sessionName, signal }, 'Signal sent to tmux pane');
+    }
+    async capturePane(ssh, sessionName, lines) {
+        const result = await this.sshManager.exec(ssh, `tmux capture-pane -t ${q(sessionName)} -p -S -${lines}`);
+        if (result.code !== 0) {
+            throw new Error(`Failed to capture tmux pane '${sessionName}': ${result.stderr || result.stdout}`);
+        }
+        return result.stdout;
+    }
+    async hasSession(ssh, sessionName) {
+        const result = await this.sshManager.exec(ssh, `tmux has-session -t ${q(sessionName)}`);
+        return result.code === 0;
+    }
+    async listSessions(ssh) {
+        const result = await this.sshManager.exec(ssh, 'tmux list-sessions -F "#{session_name}"');
+        if (result.code !== 0) {
+            return [];
+        }
+        return result.stdout
+            .split('\n')
+            .map((s) => s.trim())
+            .filter(Boolean);
+    }
+    async killSession(ssh, sessionName) {
+        const result = await this.sshManager.exec(ssh, `tmux kill-session -t ${q(sessionName)}`);
+        if (result.code !== 0) {
+            logger.warn({ sessionName, error: result.stderr }, 'Failed to kill tmux session');
+        }
+        else {
+            logger.info({ sessionName }, 'Tmux session killed');
+        }
+    }
+}
+//# sourceMappingURL=TmuxManager.js.map

package/dist/core/TmuxManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TmuxManager.js","sourceRoot":"","sources":["../../src/core/TmuxManager.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,MAAM,CAAC,MAAM,UAAU,GAAG,kBAAkB,CAAC;AAE7C,SAAS,CAAC,CAAC,KAAa;IACtB,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC;AAC7C,CAAC;AAED,MAAM,OAAO,WAAW;IACd,UAAU,CAAa;IAE/B,YAAY,UAAsB;QAChC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,GAAW;QAC3B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YAC1D,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,CAAC;QAC3B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,EAAE,mBAAmB,CAAC,CAAC;YACrE,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,KAAK,CAAC,qBAAqB,CACzB,GAAW,EACX,WAAmB;QAEnB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACvC,GAAG,EACH,0BAA0B,CAAC,CAAC,WAAW,CAAC,KAAK,CAC9C,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,yCAAyC,WAAW,MAAM,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,EAAE,CAC3F,CAAC;QACJ,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,EAAE,+BAA+B,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAW,EAAE,WAAmB;QACjD,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACxB,GAAG,EACH,sBAAsB,CAAC,CAAC,WAAW,CAAC,sBAAsB,CAC3D,CAAC;QACF,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACxB,GAAG,EACH,sBAAsB,CAAC,CAAC,WAAW,CAAC,YAAY,CACjD,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,EAAE,uDAAuD,CAAC,CAAC;IACxF,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,GAAW,EAAE,WAAmB;QAChD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACvC,GAAG,EACH,2BAA2B,CAAC,CAAC,WAAW,CAAC,+BAA+B,CACzE,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACjD,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,gBAAgB,CAAC,CAAC;QACtD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAW,EAAE,WAAmB;QACjD,MAAM,GAAG,GAAG,eAAe,UAAU,GAAG,CAAC;QACzC,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,WAAW,EAAE,GAAG,CAAC,CAAC;QAC9C,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,EAAE,+BAA+B,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,GAAW,EAAE,WAAmB;QAChD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACvC,GAAG,EACH,uBAAuB,CAAC,CAAC,WAAW,CAAC,EAAE,CACxC,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,EAAE,2CAA2C,CAAC,CAAC;YAC1E,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,SAAS,CACb,GAAW,EACX,WAAmB,EACnB,KAAa;QAEb,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACrD,MAAM,OAAO,GAAG,eAAe,CAAC,CAAC,MAAM,CAAC,6DAA6D,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC;QACzH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACxD,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,yCAAyC,WAAW,MAAM,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,EAAE,CAC3F,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CACf,GAAW,EACX,WAAmB,EACnB,OAAe;QAEf,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC9D,MAAM,YAAY,GAAG,eAAe,CAAC,CAAC,MAAM,CAAC,6DAA6D,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC;QAC9H,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;QAC7D,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,2CAA2C,WAAW,MAAM,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,EAAE,CAC7F,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CACd,GAAW,EACX,WAAmB,EACnB,MAA4B;QAE5B,MAAM,GAAG,GAAG,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;QAChD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACvC,GAAG,EACH,qBAAqB,CAAC,CAAC,WAAW,CAAC,IAAI,GAAG,EAAE,CAC7C,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,yBAAyB,MAAM,qBAAqB,WAAW,MAAM,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,EAAE,CACtG,CAAC;QACJ,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,0BAA0B,CAAC,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,WAAW,CACf,GAAW,EACX,WAAmB,EACnB,KAAa;QAEb,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACvC,GAAG,EACH,wBAAwB,CAAC,CAAC,WAAW,CAAC,WAAW,KAAK,EAAE,CACzD,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CACb,gCAAgC,WAAW,MAAM,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,EAAE,CAClF,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,GAAW,EAAE,WAAmB;QAC/C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACvC,GAAG,EACH,uBAAuB,CAAC,CAAC,WAAW,CAAC,EAAE,CACxC,CAAC;QACF,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAW;QAC5B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,yCAAyC,CAAC,CAAC;QAC1F,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,OAAO,MAAM,CAAC,MAAM;aACjB,KAAK,CAAC,IAAI,CAAC;aACX,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;aACpB,MAAM,CAAC,OAAO,CAAC,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,GAAW,EAAE,WAAmB;QAChD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACvC,GAAG,EACH,wBAAwB,CAAC,CAAC,WAAW,CAAC,EAAE,CACzC,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM,CAAC,IAAI,CACT,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,EACrC,6BAA6B,CAC9B,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,EAAE,WAAW,EAAE,EAAE,qBAAqB,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;CACF"}

package/dist/core/ToolConfigManager.d.ts

@@ -0,0 +1,15 @@
+export interface ToolConfig {
+    [toolName: string]: boolean;
+}
+export declare class ToolConfigManager {
+    private config;
+    private configPath;
+    constructor(configDir?: string);
+    private load;
+    private save;
+    isEnabled(toolName: string): boolean;
+    setEnabled(toolName: string, enabled: boolean): void;
+    reset(): void;
+    getAll(): ToolConfig;
+}
+//# sourceMappingURL=ToolConfigManager.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ToolConfigManager.d.ts","sourceRoot":"","sources":["../../src/core/ToolConfigManager.ts"],"names":[],"mappings":"AAKA,MAAM,WAAW,UAAU;IACzB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;CAC7B;AAKD,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,MAAM,CAAa;IAC3B,OAAO,CAAC,UAAU,CAAS;gBAEf,SAAS,CAAC,EAAE,MAAM;IAK9B,OAAO,CAAC,IAAI;IAYZ,OAAO,CAAC,IAAI;IAYZ,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAMpC,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;IAKpD,KAAK,IAAI,IAAI;IAKb,MAAM,IAAI,UAAU;CAGrB"}

package/dist/core/ToolConfigManager.js

@@ -0,0 +1,57 @@
+import { logger } from '../utils/logger.js';
+import * as fs from 'fs';
+import * as path from 'path';
+import * as os from 'os';
+const DEFAULT_DIR = path.join(os.homedir(), '.dynamic-ssh-mcp');
+const CONFIG_FILE = 'tools.json';
+export class ToolConfigManager {
+    config;
+    configPath;
+    constructor(configDir) {
+        this.configPath = path.join(configDir || DEFAULT_DIR, CONFIG_FILE);
+        this.config = this.load();
+    }
+    load() {
+        try {
+            if (fs.existsSync(this.configPath)) {
+                const raw = fs.readFileSync(this.configPath, 'utf-8');
+                return JSON.parse(raw);
+            }
+        }
+        catch (err) {
+            logger.warn({ err }, 'Failed to load tool config, using defaults');
+        }
+        return {};
+    }
+    save() {
+        try {
+            const dir = path.dirname(this.configPath);
+            if (!fs.existsSync(dir)) {
+                fs.mkdirSync(dir, { recursive: true });
+            }
+            fs.writeFileSync(this.configPath, JSON.stringify(this.config, null, 2));
+        }
+        catch (err) {
+            logger.error({ err }, 'Failed to save tool config');
+        }
+    }
+    isEnabled(toolName) {
+        if (toolName === 'tools_config')
+            return true;
+        if (!(toolName in this.config))
+            return true;
+        return this.config[toolName];
+    }
+    setEnabled(toolName, enabled) {
+        this.config[toolName] = enabled;
+        this.save();
+    }
+    reset() {
+        this.config = {};
+        this.save();
+    }
+    getAll() {
+        return { ...this.config };
+    }
+}
+//# sourceMappingURL=ToolConfigManager.js.map

package/dist/core/ToolConfigManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ToolConfigManager.js","sourceRoot":"","sources":["../../src/core/ToolConfigManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AAMzB,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,kBAAkB,CAAC,CAAC;AAChE,MAAM,WAAW,GAAG,YAAY,CAAC;AAEjC,MAAM,OAAO,iBAAiB;IACpB,MAAM,CAAa;IACnB,UAAU,CAAS;IAE3B,YAAY,SAAkB;QAC5B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,IAAI,WAAW,EAAE,WAAW,CAAC,CAAC;QACnE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC5B,CAAC;IAEO,IAAI;QACV,IAAI,CAAC;YACH,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;gBACnC,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBACtD,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,EAAE,4CAA4C,CAAC,CAAC;QACrE,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,IAAI;QACV,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC1C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBACxB,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACzC,CAAC;YACD,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1E,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,4BAA4B,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,SAAS,CAAC,QAAgB;QACxB,IAAI,QAAQ,KAAK,cAAc;YAAE,OAAO,IAAI,CAAC;QAC7C,IAAI,CAAC,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC;YAAE,OAAO,IAAI,CAAC;QAC5C,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAED,UAAU,CAAC,QAAgB,EAAE,OAAgB;QAC3C,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC;QAChC,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;QACjB,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAED,MAAM;QACJ,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;CACF"}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+export {};
+//# 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":""}

package/dist/index.js

@@ -0,0 +1,169 @@
+#!/usr/bin/env node
+import { createServer } from './server.js';
+import { StorageManager } from './core/StorageManager.js';
+import { SSHManager } from './core/SSHManager.js';
+import { TmuxManager, MCP_PROMPT } from './core/TmuxManager.js';
+import { listHostAliases, getHostConfig } from './utils/sshConfig.js';
+import { buildSshConfig } from './utils/ssh.js';
+import { isHostAllowed } from './utils/security.js';
+import { logger } from './utils/logger.js';
+const RESTORE_ENABLED = process.env.MCP_SSH_RESTORE_SESSIONS !== 'false';
+async function restoreSessions(storage, sshManager, tmuxManager, sessionManager) {
+    if (!RESTORE_ENABLED) {
+        logger.info('Session restore disabled (MCP_SSH_RESTORE_SESSIONS=false)');
+        return 0;
+    }
+    // Strategy 1: Restore from persistent storage (known sessions from last run)
+    const storedSessions = storage.list();
+    const hosts = [...new Set(storedSessions.map((s) => s.host))];
+    // Also scan all configured hosts in case storage is empty or incomplete
+    const allAliases = listHostAliases();
+    for (const alias of allAliases) {
+        if (!hosts.includes(alias) && isHostAllowed(alias)) {
+            hosts.push(alias);
+        }
+    }
+    if (hosts.length === 0) {
+        logger.debug('No hosts to restore');
+        return 0;
+    }
+    logger.info({ hosts }, 'Attempting session restore');
+    let restored = 0;
+    const knownTmuxSessions = new Set(storage.listTmuxSessions());
+    for (const host of hosts) {
+        if (!isHostAllowed(host))
+            continue;
+        const hostConfig = getHostConfig(host);
+        if (!hostConfig) {
+            logger.debug({ host }, 'Skip restore: no SSH config');
+            continue;
+        }
+        const baseSshConfig = buildSshConfig(hostConfig, 15000);
+        if (!baseSshConfig) {
+            logger.warn({ host }, 'Skip restore: failed to build SSH config');
+            continue;
+        }
+        // First, connect temporarily to discover mcp_* tmux sessions
+        let probeSsh;
+        try {
+            probeSsh = await sshManager.connect({ ...baseSshConfig });
+        }
+        catch (err) {
+            logger.warn({ host, error: err.message }, 'Restore: SSH probe failed');
+            continue;
+        }
+        const tmuxInstalled = await tmuxManager.isInstalled(probeSsh);
+        if (!tmuxInstalled) {
+            sshManager.disconnect(probeSsh);
+            logger.debug({ host }, 'Restore: tmux not installed, skipping');
+            continue;
+        }
+        let tmuxSessions = [];
+        try {
+            tmuxSessions = await tmuxManager.listSessions(probeSsh);
+        }
+        catch (err) {
+            sshManager.disconnect(probeSsh);
+            logger.warn({ host, error: err.message }, 'Restore: list sessions failed');
+            continue;
+        }
+        // Filter for MCP-managed sessions (mcp_ prefix)
+        const mcpSessions = tmuxSessions.filter((s) => s.startsWith('mcp_'));
+        sshManager.disconnect(probeSsh);
+        if (mcpSessions.length === 0) {
+            logger.debug({ host }, 'Restore: no mcp_* tmux sessions found');
+            continue;
+        }
+        // Reconnect to each MCP tmux session with its own SSH connection
+        for (const tmuxSession of mcpSessions) {
+            const storedSession = storedSessions.find((s) => s.host === host && s.tmuxSession === tmuxSession);
+            // Each restored session gets its own SSH connection
+            let ssh;
+            try {
+                ssh = await sshManager.connect({ ...baseSshConfig });
+            }
+            catch (err) {
+                logger.warn({ host, tmuxSession, error: err.message }, 'Restore: SSH session connect failed');
+                continue;
+            }
+            try {
+                const exists = await tmuxManager.hasSession(ssh, tmuxSession);
+                if (!exists) {
+                    sshManager.disconnect(ssh);
+                    if (storedSession)
+                        storage.remove(storedSession.id);
+                    continue;
+                }
+                // Check and inject prompt if missing
+                try {
+                    const output = await tmuxManager.capturePane(ssh, tmuxSession, 5);
+                    if (!output.includes(MCP_PROMPT)) {
+                        await tmuxManager.injectPrompt(ssh, tmuxSession);
+                        await new Promise((r) => setTimeout(r, 300));
+                    }
+                }
+                catch {
+                    logger.warn({ host, tmuxSession }, 'Restore: prompt check failed, continuing');
+                }
+                const session = sessionManager.create(host, ssh, tmuxSession, storedSession?.shell, storedSession?.id);
+                logger.info({ sessionId: session.id, host, tmuxSession }, 'Session restored');
+                restored++;
+                knownTmuxSessions.add(tmuxSession);
+            }
+            catch (err) {
+                sshManager.disconnect(ssh);
+                logger.warn({ host, tmuxSession, error: err.message }, 'Restore: session skipped');
+            }
+        }
+    }
+    // Clean up stale storage entries for sessions that no longer exist
+    for (const stored of storedSessions) {
+        if (!knownTmuxSessions.has(stored.tmuxSession)) {
+            storage.remove(stored.id);
+            logger.debug({ sessionId: stored.id, tmuxSession: stored.tmuxSession }, 'Cleaned up stale session record');
+        }
+    }
+    logger.info({ restored }, 'Session restore complete');
+    return restored;
+}
+async function main() {
+    const storage = new StorageManager();
+    const { server, sessionManager, transport } = createServer(storage);
+    const sshManager = new SSHManager();
+    const tmuxManager = new TmuxManager(sshManager);
+    // Prevent silent crashes from unhandled errors
+    process.on('uncaughtException', (err) => {
+        logger.fatal({ error: err.message, stack: err.stack }, 'Uncaught exception');
+        process.exit(1);
+    });
+    process.on('unhandledRejection', (reason) => {
+        logger.fatal({ error: reason?.message || String(reason) }, 'Unhandled rejection');
+    });
+    // Restore previous sessions on startup (fire-and-forget, doesn't block MCP)
+    restoreSessions(storage, sshManager, tmuxManager, sessionManager).catch((err) => {
+        logger.error({ error: err.message }, 'Session restore failed');
+    });
+    // Graceful shutdown
+    const cleanup = () => {
+        logger.info('Shutting down...');
+        sessionManager.disconnectAll();
+        storage.shutdown();
+        server
+            .close()
+            .then(() => {
+            process.exit(0);
+        })
+            .catch(() => {
+            process.exit(1);
+        });
+    };
+    process.on('SIGINT', cleanup);
+    process.on('SIGTERM', cleanup);
+    await server.connect(transport);
+    logger.info('Dynamic SSH MCP server running on stdio');
+}
+main().catch((err) => {
+    logger.error({ error: err.message }, 'Fatal error');
+    process.exit(1);
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACtE,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAE3C,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,KAAK,OAAO,CAAC;AAEzE,KAAK,UAAU,eAAe,CAC5B,OAAuB,EACvB,UAAsB,EACtB,WAAwB,EACxB,cAAiE;IAEjE,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,MAAM,CAAC,IAAI,CAAC,2DAA2D,CAAC,CAAC;QACzE,OAAO,CAAC,CAAC;IACX,CAAC;IAED,6EAA6E;IAC7E,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IACtC,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC9D,wEAAwE;IACxE,MAAM,UAAU,GAAG,eAAe,EAAE,CAAC;IACrC,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;QAC/B,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;YACnD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;IAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACpC,OAAO,CAAC,CAAC;IACX,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,4BAA4B,CAAC,CAAC;IAErD,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC;IAE9D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;YAAE,SAAS;QAEnC,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,EAAE,6BAA6B,CAAC,CAAC;YACtD,SAAS;QACX,CAAC;QAED,MAAM,aAAa,GAAG,cAAc,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QACxD,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,0CAA0C,CAAC,CAAC;YAClE,SAAS;QACX,CAAC;QAED,6DAA6D;QAC7D,IAAI,QAAQ,CAAC;QACb,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,EAAE,GAAG,aAAa,EAAE,CAAC,CAAC;QAC5D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,EAAE,2BAA2B,CAAC,CAAC;YAClF,SAAS;QACX,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,WAAW,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC9D,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,UAAU,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAChC,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,EAAE,uCAAuC,CAAC,CAAC;YAChE,SAAS;QACX,CAAC;QAED,IAAI,YAAY,GAAa,EAAE,CAAC;QAChC,IAAI,CAAC;YACH,YAAY,GAAG,MAAM,WAAW,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAC1D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,UAAU,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,EAAE,+BAA+B,CAAC,CAAC;YACtF,SAAS;QACX,CAAC;QAED,gDAAgD;QAChD,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;QACrE,UAAU,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEhC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,EAAE,uCAAuC,CAAC,CAAC;YAChE,SAAS;QACX,CAAC;QAED,iEAAiE;QACjE,KAAK,MAAM,WAAW,IAAI,WAAW,EAAE,CAAC;YACtC,MAAM,aAAa,GAAG,cAAc,CAAC,IAAI,CACvC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,CAAC,WAAW,KAAK,WAAW,CACxD,CAAC;YAEF,oDAAoD;YACpD,IAAI,GAAG,CAAC;YACR,IAAI,CAAC;gBACH,GAAG,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,EAAE,GAAG,aAAa,EAAE,CAAC,CAAC;YACvD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,EAAE,qCAAqC,CAAC,CAAC;gBACzG,SAAS;YACX,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,UAAU,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;gBAC9D,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;oBAC3B,IAAI,aAAa;wBAAE,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;oBACpD,SAAS;gBACX,CAAC;gBAED,qCAAqC;gBACrC,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,WAAW,CAAC,GAAG,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;oBAClE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;wBACjC,MAAM,WAAW,CAAC,YAAY,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;wBACjD,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;oBAC/C,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,0CAA0C,CAAC,CAAC;gBACjF,CAAC;gBAED,MAAM,OAAO,GAAG,cAAc,CAAC,MAAM,CACnC,IAAI,EACJ,GAAG,EACH,WAAW,EACX,aAAa,EAAE,KAAK,EACpB,aAAa,EAAE,EAAE,CAClB,CAAC;gBAEF,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,kBAAkB,CAAC,CAAC;gBAC9E,QAAQ,EAAE,CAAC;gBACX,iBAAiB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YACrC,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;gBAC3B,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,EAAE,0BAA0B,CAAC,CAAC;YAChG,CAAC;QACH,CAAC;IACH,CAAC;IAED,mEAAmE;IACnE,KAAK,MAAM,MAAM,IAAI,cAAc,EAAE,CAAC;QACpC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;YAC/C,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,EAAE,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,EAAE,iCAAiC,CAAC,CAAC;QAC7G,CAAC;IACH,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,EAAE,0BAA0B,CAAC,CAAC;IACtD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,KAAK,UAAU,IAAI;IACjB,MAAM,OAAO,GAAG,IAAI,cAAc,EAAE,CAAC;IACrC,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,SAAS,EAAE,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;IACpE,MAAM,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;IACpC,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,UAAU,CAAC,CAAC;IAEhD,+CAA+C;IAC/C,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC,GAAG,EAAE,EAAE;QACtC,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,EAAE,oBAAoB,CAAC,CAAC;QAC7E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,EAAE;QAC1C,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAG,MAAgB,EAAE,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,qBAAqB,CAAC,CAAC;IAC/F,CAAC,CAAC,CAAC;IAEH,4EAA4E;IAC5E,eAAe,CAAC,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;QAC9E,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,wBAAwB,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,oBAAoB;IACpB,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAChC,cAAc,CAAC,aAAa,EAAE,CAAC;QAC/B,OAAO,CAAC,QAAQ,EAAE,CAAC;QACnB,MAAM;aACH,KAAK,EAAE;aACP,IAAI,CAAC,GAAG,EAAE;YACT,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC;aACD,KAAK,CAAC,GAAG,EAAE;YACV,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;IACP,CAAC,CAAC;IAEF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC9B,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAE/B,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,MAAM,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;AACzD,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;IACnB,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,EAAE,aAAa,CAAC,CAAC;IACpD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/dist/server.d.ts

@@ -0,0 +1,44 @@
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { SessionManager } from './core/SessionManager.js';
+import { StorageManager } from './core/StorageManager.js';
+export declare function createServer(storage?: StorageManager): {
+    server: Server<{
+        method: string;
+        params?: {
+            [x: string]: unknown;
+            _meta?: {
+                [x: string]: unknown;
+                progressToken?: string | number | undefined;
+                "io.modelcontextprotocol/related-task"?: {
+                    taskId: string;
+                } | undefined;
+            } | undefined;
+        } | undefined;
+    }, {
+        method: string;
+        params?: {
+            [x: string]: unknown;
+            _meta?: {
+                [x: string]: unknown;
+                progressToken?: string | number | undefined;
+                "io.modelcontextprotocol/related-task"?: {
+                    taskId: string;
+                } | undefined;
+            } | undefined;
+        } | undefined;
+    }, {
+        [x: string]: unknown;
+        _meta?: {
+            [x: string]: unknown;
+            progressToken?: string | number | undefined;
+            "io.modelcontextprotocol/related-task"?: {
+                taskId: string;
+            } | undefined;
+        } | undefined;
+    }>;
+    sessionManager: SessionManager;
+    storageManager: StorageManager;
+    transport: StdioServerTransport;
+};
+//# sourceMappingURL=server.d.ts.map

package/dist/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAOjF,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAG1D,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAsF1D,wBAAgB,YAAY,CAAC,OAAO,CAAC,EAAE,cAAc;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAoIpD"}

package/dist/server.js

@@ -0,0 +1,152 @@
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ListToolsRequestSchema, McpError, ErrorCode, } from '@modelcontextprotocol/sdk/types.js';
+import { SessionManager } from './core/SessionManager.js';
+import { SSHManager } from './core/SSHManager.js';
+import { TmuxManager } from './core/TmuxManager.js';
+import { StorageManager } from './core/StorageManager.js';
+import { ToolConfigManager } from './core/ToolConfigManager.js';
+import { HostSecurityManager } from './core/HostSecurityManager.js';
+import { setHostSecurityManager } from './utils/security.js';
+import { logger } from './utils/logger.js';
+import { listHostsTool, handleListHosts, connectTool, handleConnect, reconnectTool, handleReconnect, sendInputTool, handleSendInput, readOutputTool, handleReadOutput, execTool, handleExec, interruptTool, handleInterrupt, disconnectTool, handleDisconnect, listSessionsTool, handleListSessions, sftpUploadTool, handleSftpUpload, sftpDownloadTool, handleSftpDownload, sftpListTool, handleSftpList, connectionStatusTool, handleConnectionStatus, healthCheckTool, handleHealthCheck, tailLogTool, handleTailLog, tunnelOpenTool, tunnelCloseTool, tunnelListTool, handleTunnelOpen, handleTunnelClose, handleTunnelList, deployTool, handleDeploy, backupTool, handleBackup, syncTool, handleSync, groupExecTool, handleGroupExec, dbQueryTool, handleDbQuery, toolsConfigTool, handleToolsConfig, hostSecurityTool, handleHostSecurity, } from './tools/index.js';
+const allTools = [
+    listHostsTool,
+    connectTool,
+    reconnectTool,
+    sendInputTool,
+    readOutputTool,
+    execTool,
+    interruptTool,
+    disconnectTool,
+    listSessionsTool,
+    sftpUploadTool,
+    sftpDownloadTool,
+    sftpListTool,
+    connectionStatusTool,
+    healthCheckTool,
+    tailLogTool,
+    tunnelOpenTool,
+    tunnelCloseTool,
+    tunnelListTool,
+    deployTool,
+    backupTool,
+    syncTool,
+    groupExecTool,
+    dbQueryTool,
+    toolsConfigTool,
+    hostSecurityTool,
+];
+export function createServer(storage) {
+    const storageManager = storage || new StorageManager();
+    const server = new Server({
+        name: 'dynamic-ssh-mcp',
+        version: '1.0.0',
+    }, {
+        capabilities: {
+            tools: {},
+        },
+    });
+    const sessionManager = new SessionManager(storageManager);
+    const sshManager = new SSHManager();
+    const tmuxManager = new TmuxManager(sshManager);
+    const toolConfigManager = new ToolConfigManager();
+    const hostSecurityManager = new HostSecurityManager();
+    setHostSecurityManager(hostSecurityManager);
+    server.setRequestHandler(ListToolsRequestSchema, async () => ({
+        tools: allTools.filter((t) => toolConfigManager.isEnabled(t.name)),
+    }));
+    server.setRequestHandler(CallToolRequestSchema, async (request) => {
+        const { name, arguments: args } = request.params;
+        logger.info({ tool: name }, 'Tool called');
+        try {
+            switch (name) {
+                case 'list_hosts': {
+                    return await handleListHosts();
+                }
+                case 'connect': {
+                    return await handleConnect(args, sessionManager, sshManager, tmuxManager);
+                }
+                case 'reconnect_to_tmux': {
+                    return await handleReconnect(args, sessionManager, sshManager, tmuxManager);
+                }
+                case 'send_input': {
+                    return await handleSendInput(args, sessionManager, tmuxManager);
+                }
+                case 'read_output': {
+                    return await handleReadOutput(args, sessionManager, tmuxManager);
+                }
+                case 'exec': {
+                    return await handleExec(args, sessionManager, tmuxManager);
+                }
+                case 'interrupt': {
+                    return await handleInterrupt(args, sessionManager, tmuxManager);
+                }
+                case 'disconnect': {
+                    return await handleDisconnect(args, sessionManager, tmuxManager);
+                }
+                case 'list_sessions': {
+                    return await handleListSessions(sessionManager);
+                }
+                case 'sftp_upload': {
+                    return await handleSftpUpload(args, sessionManager, sshManager);
+                }
+                case 'sftp_download': {
+                    return await handleSftpDownload(args, sessionManager, sshManager);
+                }
+                case 'sftp_list': {
+                    return await handleSftpList(args, sessionManager, sshManager);
+                }
+                case 'connection_status': {
+                    return await handleConnectionStatus(args, sessionManager, sshManager, tmuxManager);
+                }
+                case 'health_check': {
+                    return await handleHealthCheck(args, sessionManager, sshManager);
+                }
+                case 'tail_log': {
+                    return await handleTailLog(args, sessionManager, sshManager);
+                }
+                case 'ssh_tunnel_open': {
+                    return await handleTunnelOpen(args, sessionManager, sshManager);
+                }
+                case 'ssh_tunnel_close': {
+                    return await handleTunnelClose(args);
+                }
+                case 'ssh_tunnel_list': {
+                    return await handleTunnelList();
+                }
+                case 'deploy': {
+                    return await handleDeploy(args, sessionManager, sshManager);
+                }
+                case 'backup': {
+                    return await handleBackup(args, sessionManager, sshManager);
+                }
+                case 'sync': {
+                    return await handleSync(args, sessionManager, sshManager);
+                }
+                case 'group_exec': {
+                    return await handleGroupExec(args, sessionManager, tmuxManager);
+                }
+                case 'db_query': {
+                    return await handleDbQuery(args, sessionManager, sshManager);
+                }
+                case 'tools_config': {
+                    return await handleToolsConfig(args, toolConfigManager);
+                }
+                case 'host_security': {
+                    return await handleHostSecurity(args, hostSecurityManager);
+                }
+                default: {
+                    throw new McpError(ErrorCode.MethodNotFound, `Unknown tool: ${name}`);
+                }
+            }
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            logger.error({ tool: name, error: message }, 'Unhandled tool error');
+            throw new McpError(ErrorCode.InternalError, `Tool execution failed: ${message}`);
+        }
+    });
+    return { server, sessionManager, storageManager, transport: new StdioServerTransport() };
+}
+//# sourceMappingURL=server.js.map

package/dist/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EACL,qBAAqB,EACrB,sBAAsB,EACtB,QAAQ,EACR,SAAS,GACV,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,6BAA6B,CAAC;AAChE,OAAO,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AACpE,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EACL,aAAa,EACb,eAAe,EACf,WAAW,EACX,aAAa,EACb,aAAa,EACb,eAAe,EACf,aAAa,EACb,eAAe,EACf,cAAc,EACd,gBAAgB,EAChB,QAAQ,EACR,UAAU,EACV,aAAa,EACb,eAAe,EACf,cAAc,EACd,gBAAgB,EAChB,gBAAgB,EAChB,kBAAkB,EAClB,cAAc,EACd,gBAAgB,EAChB,gBAAgB,EAChB,kBAAkB,EAClB,YAAY,EACZ,cAAc,EACd,oBAAoB,EACpB,sBAAsB,EACtB,eAAe,EACf,iBAAiB,EACjB,WAAW,EACX,aAAa,EACb,cAAc,EACd,eAAe,EACf,cAAc,EACd,gBAAgB,EAChB,iBAAiB,EACjB,gBAAgB,EAChB,UAAU,EACV,YAAY,EACZ,UAAU,EACV,YAAY,EACZ,QAAQ,EACR,UAAU,EACV,aAAa,EACb,eAAe,EACf,WAAW,EACX,aAAa,EACb,eAAe,EACf,iBAAiB,EACjB,gBAAgB,EAChB,kBAAkB,GACnB,MAAM,kBAAkB,CAAC;AAE1B,MAAM,QAAQ,GAAG;IACf,aAAa;IACb,WAAW;IACX,aAAa;IACb,aAAa;IACb,cAAc;IACd,QAAQ;IACR,aAAa;IACb,cAAc;IACd,gBAAgB;IAChB,cAAc;IACd,gBAAgB;IAChB,YAAY;IACZ,oBAAoB;IACpB,eAAe;IACf,WAAW;IACX,cAAc;IACd,eAAe;IACf,cAAc;IACd,UAAU;IACV,UAAU;IACV,QAAQ;IACR,aAAa;IACb,WAAW;IACX,eAAe;IACf,gBAAgB;CACjB,CAAC;AAEF,MAAM,UAAU,YAAY,CAAC,OAAwB;IACnD,MAAM,cAAc,GAAG,OAAO,IAAI,IAAI,cAAc,EAAE,CAAC;IACvD,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB;QACE,IAAI,EAAE,iBAAiB;QACvB,OAAO,EAAE,OAAO;KACjB,EACD;QACE,YAAY,EAAE;YACZ,KAAK,EAAE,EAAE;SACV;KACF,CACF,CAAC;IAEF,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,cAAc,CAAC,CAAC;IAC1D,MAAM,UAAU,GAAG,IAAI,UAAU,EAAE,CAAC;IACpC,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,UAAU,CAAC,CAAC;IAChD,MAAM,iBAAiB,GAAG,IAAI,iBAAiB,EAAE,CAAC;IAClD,MAAM,mBAAmB,GAAG,IAAI,mBAAmB,EAAE,CAAC;IAEtD,sBAAsB,CAAC,mBAAmB,CAAC,CAAC;IAE5C,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;QAC5D,KAAK,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;KACnE,CAAC,CAAC,CAAC;IAEJ,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;QACjD,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,aAAa,CAAC,CAAC;QAE3C,IAAI,CAAC;YACH,QAAQ,IAAI,EAAE,CAAC;gBACb,KAAK,YAAY,CAAC,CAAC,CAAC;oBAClB,OAAO,MAAM,eAAe,EAAE,CAAC;gBACjC,CAAC;gBACD,KAAK,SAAS,CAAC,CAAC,CAAC;oBACf,OAAO,MAAM,aAAa,CACxB,IAAI,EACJ,cAAc,EACd,UAAU,EACV,WAAW,CACZ,CAAC;gBACJ,CAAC;gBACD,KAAK,mBAAmB,CAAC,CAAC,CAAC;oBACzB,OAAO,MAAM,eAAe,CAC1B,IAAI,EACJ,cAAc,EACd,UAAU,EACV,WAAW,CACZ,CAAC;gBACJ,CAAC;gBACD,KAAK,YAAY,CAAC,CAAC,CAAC;oBAClB,OAAO,MAAM,eAAe,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;gBAClE,CAAC;gBACD,KAAK,aAAa,CAAC,CAAC,CAAC;oBACnB,OAAO,MAAM,gBAAgB,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;gBACnE,CAAC;gBACD,KAAK,MAAM,CAAC,CAAC,CAAC;oBACZ,OAAO,MAAM,UAAU,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;gBAC7D,CAAC;gBACD,KAAK,WAAW,CAAC,CAAC,CAAC;oBACjB,OAAO,MAAM,eAAe,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;gBAClE,CAAC;gBACD,KAAK,YAAY,CAAC,CAAC,CAAC;oBAClB,OAAO,MAAM,gBAAgB,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;gBACnE,CAAC;gBACD,KAAK,eAAe,CAAC,CAAC,CAAC;oBACrB,OAAO,MAAM,kBAAkB,CAAC,cAAc,CAAC,CAAC;gBAClD,CAAC;gBACD,KAAK,aAAa,CAAC,CAAC,CAAC;oBACnB,OAAO,MAAM,gBAAgB,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;gBAClE,CAAC;gBACD,KAAK,eAAe,CAAC,CAAC,CAAC;oBACrB,OAAO,MAAM,kBAAkB,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;gBACpE,CAAC;gBACD,KAAK,WAAW,CAAC,CAAC,CAAC;oBACjB,OAAO,MAAM,cAAc,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;gBAChE,CAAC;gBACD,KAAK,mBAAmB,CAAC,CAAC,CAAC;oBACzB,OAAO,MAAM,sBAAsB,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;gBACrF,CAAC;gBACD,KAAK,cAAc,CAAC,CAAC,CAAC;oBACpB,OAAO,MAAM,iBAAiB,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;gBACnE,CAAC;gBACD,KAAK,UAAU,CAAC,CAAC,CAAC;oBAChB,OAAO,MAAM,aAAa,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;gBAC/D,CAAC;gBACD,KAAK,iBAAiB,CAAC,CAAC,CAAC;oBACvB,OAAO,MAAM,gBAAgB,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;gBAClE,CAAC;gBACD,KAAK,kBAAkB,CAAC,CAAC,CAAC;oBACxB,OAAO,MAAM,iBAAiB,CAAC,IAAI,CAAC,CAAC;gBACvC,CAAC;gBACD,KAAK,iBAAiB,CAAC,CAAC,CAAC;oBACvB,OAAO,MAAM,gBAAgB,EAAE,CAAC;gBAClC,CAAC;gBACD,KAAK,QAAQ,CAAC,CAAC,CAAC;oBACd,OAAO,MAAM,YAAY,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;gBAC9D,CAAC;gBACD,KAAK,QAAQ,CAAC,CAAC,CAAC;oBACd,OAAO,MAAM,YAAY,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;gBAC9D,CAAC;gBACD,KAAK,MAAM,CAAC,CAAC,CAAC;oBACZ,OAAO,MAAM,UAAU,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;gBAC5D,CAAC;gBACD,KAAK,YAAY,CAAC,CAAC,CAAC;oBAClB,OAAO,MAAM,eAAe,CAAC,IAAI,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;gBAClE,CAAC;gBACD,KAAK,UAAU,CAAC,CAAC,CAAC;oBAChB,OAAO,MAAM,aAAa,CAAC,IAAI,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;gBAC/D,CAAC;gBACD,KAAK,cAAc,CAAC,CAAC,CAAC;oBACpB,OAAO,MAAM,iBAAiB,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;gBAC1D,CAAC;gBACD,KAAK,eAAe,CAAC,CAAC,CAAC;oBACrB,OAAO,MAAM,kBAAkB,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;gBAC7D,CAAC;gBACD,OAAO,CAAC,CAAC,CAAC;oBACR,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,cAAc,EAAE,iBAAiB,IAAI,EAAE,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,sBAAsB,CAAC,CAAC;YACrE,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,0BAA0B,OAAO,EAAE,CACpC,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,cAAc,EAAE,SAAS,EAAE,IAAI,oBAAoB,EAAE,EAAE,CAAC;AAC3F,CAAC"}