@pyrokine/mcp-ssh 1.1.0 → 1.1.2

package/dist/index.js

@@ -15,10 +15,10 @@
  */
 import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import { CallToolRequestSchema, ListToolsRequestSchema, } from '@modelcontextprotocol/sdk/types.js';
-import { sessionManager } from './session-manager.js';
+import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
 import * as fileOps from './file-ops.js';
-import { parseSSHConfig, parseProxyJump } from './ssh-config.js';
+import { sessionManager } from './session-manager.js';
+import { parseProxyJump, parseSSHConfig } from './ssh-config.js';
 // 创建 MCP Server
 const server = new Server({
     name: 'ssh-mcp-pro',
@@ -133,6 +133,9 @@ const tools = [
 
 适用场景: SSH 以 root 登录，但需要以其他用户（如 caros）执行命令。
 
+默认加载目标用户的 shell 配置以获取环境变量（su -c 创建非交互式 shell，不会自动执行 rc 文件）。
+支持 bash(.bashrc)、zsh(.zshrc) 及其他 shell(.profile)。
+
 示例: ssh_exec_as_user(alias="server", command="whoami", targetUser="caros")`,
         inputSchema: {
             type: 'object',
@@ -141,6 +144,7 @@ const tools = [
                 command: { type: 'string', description: '要执行的命令' },
                 targetUser: { type: 'string', description: '目标用户名' },
                 timeout: { type: 'number', description: '超时（毫秒）' },
+                loadProfile: { type: 'boolean', description: '是否加载 .bashrc（默认 true）' },
             },
             required: ['alias', 'command', 'targetUser'],
         },
@@ -396,7 +400,11 @@ rsync 可实现增量传输，对大目录同步效率更高。
             type: 'object',
             properties: {
                 ptyId: { type: 'string', description: 'PTY 会话 ID' },
-                mode: { type: 'string', enum: ['screen', 'raw'], description: '输出模式：screen（当前屏幕）或 raw（原始流），默认 screen' },
+                mode: {
+                    type: 'string',
+                    enum: ['screen', 'raw'],
+                    description: '输出模式：screen（当前屏幕）或 raw（原始流），默认 screen',
+                },
                 clear: { type: 'boolean', description: '(仅 raw 模式) 读取后是否清空缓冲区，默认 true' },
             },
             required: ['ptyId'],
@@ -607,7 +615,8 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                     username: user,
                     password: args.password,
                     privateKeyPath: keyPath,
-                    alias: args.alias || args.configHost,
+                    alias: args.alias ||
+                        args.configHost,
                     env: args.env,
                     keepaliveInterval: args.keepaliveInterval,
                     jumpHost,
@@ -655,7 +664,10 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                 break;
             }
             case 'ssh_exec_as_user': {
-                const execResult = await sessionManager.execAsUser(args.alias, args.command, args.targetUser, { timeout: args.timeout });
+                const execResult = await sessionManager.execAsUser(args.alias, args.command, args.targetUser, {
+                    timeout: args.timeout,
+                    loadProfile: args.loadProfile,
+                });
                 result = execResult;
                 break;
             }
@@ -688,8 +700,9 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                             success: false,
                             error: err.message,
                         });
-                        if (stopOnError)
+                        if (stopOnError) {
                             break;
+                        }
                     }
                 }
                 result = {
@@ -841,7 +854,8 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                     success: true,
                     forwardId,
                     type: 'local',
-                    message: `Local forward: ${args.localHost || '127.0.0.1'}:${args.localPort} -> ${args.remoteHost}:${args.remotePort}`,
+                    message: `Local forward: ${args.localHost ||
+                        '127.0.0.1'}:${args.localPort} -> ${args.remoteHost}:${args.remotePort}`,
                 };
                 break;
             }
@@ -851,7 +865,8 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                     success: true,
                     forwardId,
                     type: 'remote',
-                    message: `Remote forward: ${args.remoteHost || '127.0.0.1'}:${args.remotePort} -> ${args.localHost}:${args.localPort}`,
+                    message: `Remote forward: ${args.remoteHost ||
+                        '127.0.0.1'}:${args.remotePort} -> ${args.localHost}:${args.localPort}`,
                 };
                 break;
             }

package/dist/session-manager.d.ts

@@ -8,7 +8,7 @@
  * - 会话持久化
  */
 import { Client, ClientChannel, SFTPWrapper } from 'ssh2';
-import { SSHConnectionConfig, SSHSessionInfo, ExecOptions, ExecResult, PersistedSession, PtyOptions, PtySessionInfo, PortForwardInfo } from './types.js';
+import { ExecOptions, ExecResult, PersistedSession, PortForwardInfo, PtyOptions, PtySessionInfo, SSHConnectionConfig, SSHSessionInfo } from './types.js';
 interface SSHSession {
     client: Client;
     config: SSHConnectionConfig;
@@ -37,23 +37,10 @@ export declare class SessionManager {
     private maxReconnectAttempts;
     private defaultPtyBufferSize;
     constructor(persistPath?: string);
-    private ensurePersistDir;
-    /**
-     * 生成连接别名
-     */
-    private generateAlias;
     /**
      * 建立 SSH 连接
      */
     connect(config: SSHConnectionConfig): Promise<string>;
-    /**
-     * 通过跳板机转发连接
-     */
-    private forwardConnection;
-    /**
-     * 检查连接是否存活
-     */
-    private isAlive;
     /**
      * 重新连接
      */
@@ -74,26 +61,20 @@ export declare class SessionManager {
      * 列出所有会话
      */
     listSessions(): SSHSessionInfo[];
-    /**
-     * 转义 shell 参数（使用单引号方式）
-     */
-    private escapeShellArg;
     /**
      * 执行命令
      */
     exec(alias: string, command: string, options?: ExecOptions): Promise<ExecResult>;
-    /**
-     * 校验用户名（只允许字母、数字、下划线、连字符）
-     */
-    private isValidUsername;
-    /**
-     * 校验环境变量名（只允许字母、数字、下划线，不能以数字开头）
-     */
-    private isValidEnvKey;
     /**
      * 以其他用户身份执行命令
+     * @param loadProfile 是否加载用户的 shell 配置（默认 true）。
+     *   su -c 创建非交互式 shell，不会自动执行 rc 文件，
+     *   但大多数用户的环境变量设置在 rc 文件中，因此默认加载。
+     *   支持 bash(.bashrc)、zsh(.zshrc) 及其他 shell(.profile)。
      */
-    execAsUser(alias: string, command: string, targetUser: string, options?: ExecOptions): Promise<ExecResult>;
+    execAsUser(alias: string, command: string, targetUser: string, options?: ExecOptions & {
+        loadProfile?: boolean;
+    }): Promise<ExecResult>;
     /**
      * 使用 sudo 执行命令
      */
@@ -102,18 +83,10 @@ export declare class SessionManager {
      * 获取 SFTP 客户端
      */
     getSftp(alias: string): Promise<SFTPWrapper>;
-    /**
-     * 持久化会话信息
-     */
-    private persistSessions;
     /**
      * 加载持久化的会话信息（仅用于显示，不自动重连）
      */
     loadPersistedSessions(): PersistedSession[];
-    /**
-     * 生成 PTY 会话 ID
-     */
-    private generatePtyId;
     /**
      * 启动持久化 PTY 会话
      */
@@ -122,10 +95,6 @@ export declare class SessionManager {
      * 向 PTY 写入数据
      */
     ptyWrite(ptyId: string, data: string): boolean;
-    /**
-     * 从终端仿真器获取当前屏幕内容
-     */
-    private getScreenContent;
     /**
      * 读取 PTY 输出
      * @param mode 'screen' 返回当前屏幕内容，'raw' 返回原始 ANSI 流
@@ -155,10 +124,6 @@ export declare class SessionManager {
      * 关闭所有 PTY 会话
      */
     ptyCloseAll(): number;
-    /**
-     * 生成端口转发 ID
-     */
-    private generateForwardId;
     /**
      * 创建本地端口转发
      * 本地监听 localHost:localPort，转发到远程 remoteHost:remotePort
@@ -169,6 +134,60 @@ export declare class SessionManager {
      * 远程监听 remoteHost:remotePort，转发到本地 localHost:localPort
      */
     forwardRemote(alias: string, remotePort: number, localHost: string, localPort: number, remoteHost?: string): Promise<string>;
+    /**
+     * 关闭端口转发
+     */
+    forwardClose(forwardId: string): boolean;
+    /**
+     * 列出所有端口转发
+     */
+    forwardList(): PortForwardInfo[];
+    private ensurePersistDir;
+    /**
+     * 生成连接别名
+     */
+    private generateAlias;
+    /**
+     * 通过跳板机转发连接
+     */
+    private forwardConnection;
+    /**
+     * 检查连接是否存活
+     */
+    private isAlive;
+    /**
+     * 转义 shell 参数（使用单引号方式）
+     */
+    private escapeShellArg;
+    /**
+     * 校验用户名（只允许字母、数字、下划线、连字符）
+     */
+    private isValidUsername;
+    /**
+     * 校验环境变量名（只允许字母、数字、下划线，不能以数字开头）
+     */
+    private isValidEnvKey;
+    /**
+     * 根据用户 shell 类型生成加载配置文件的命令
+     * bash → .bashrc, zsh → .zshrc, 其他 → .profile
+     */
+    private getLoadProfileCommand;
+    /**
+     * 持久化会话信息
+     */
+    private persistSessions;
+    /**
+     * 生成 PTY 会话 ID
+     */
+    private generatePtyId;
+    /**
+     * 从终端仿真器获取当前屏幕内容
+     */
+    private getScreenContent;
+    /**
+     * 生成端口转发 ID
+     */
+    private generateForwardId;
     /**
      * 确保 SSH session 有共享的 tcp connection dispatcher
      * 所有 remote forward 共用一个 dispatcher，根据 destIP/destPort 路由
@@ -180,14 +199,6 @@ export declare class SessionManager {
      * @param alias - session 的 map key
      */
     private removeTcpDispatcherIfEmpty;
-    /**
-     * 关闭端口转发
-     */
-    forwardClose(forwardId: string): boolean;
-    /**
-     * 列出所有端口转发
-     */
-    forwardList(): PortForwardInfo[];
 }
 export declare const sessionManager: SessionManager;
 export {};