mcp-ssh-pty 1.2.0 → 1.2.1

package/dist/shell-manager.d.ts

@@ -3,6 +3,7 @@ import { ShellResult, ShellConfig } from "./types.js";
 export declare class ShellManager {
     private shell;
     private localProcess;
+    private ptyProcess;
     private outputBuffer;
     private outputLines;
     private lastOutputTime;
@@ -14,7 +15,7 @@ export declare class ShellManager {
      */
     open(client: Client): Promise<void>;
     /**
-     * 打开本地 Shell
+     * 打开本地 Shell（使用 node-pty 创建真正的 PTY）
      */
     openLocal(): Promise<void>;
     /**

package/dist/shell-manager.js

@@ -1,4 +1,4 @@
-import { spawn } from "child_process";
+import * as pty from "node-pty";
 const DEFAULT_CONFIG = {
     quickTimeout: 2000,
     maxTimeout: 5000,
@@ -8,6 +8,7 @@ const DEFAULT_CONFIG = {
 export class ShellManager {
     shell = null;
     localProcess = null;
+    ptyProcess = null;
     outputBuffer = "";
     outputLines = [];
     lastOutputTime = 0;
@@ -37,42 +38,39 @@ export class ShellManager {
         });
     }
     /**
-     * 打开本地 Shell
+     * 打开本地 Shell（使用 node-pty 创建真正的 PTY）
      */
     async openLocal() {
         return new Promise((resolve, reject) => {
             try {
                 // 检测系统默认 shell
-                const shellPath = process.env.SHELL || (process.platform === "win32" ? "cmd.exe" : "/bin/sh");
-                const proc = spawn(shellPath, [], {
-                    stdio: ["pipe", "pipe", "pipe"],
-                    env: { ...process.env, TERM: "xterm-256color" },
+                const shellPath = process.env.SHELL || (process.platform === "win32" ? "powershell.exe" : "/bin/sh");
+                // 使用 node-pty 创建真正的 PTY
+                const ptyProc = pty.spawn(shellPath, [], {
+                    name: "xterm-256color",
+                    cols: 120,
+                    rows: 40,
+                    cwd: process.cwd(),
+                    env: process.env,
                 });
-                this.localProcess = proc;
+                this.ptyProcess = ptyProc;
                 this.isLocal = true;
                 // 创建统一的 stream 接口
                 const dataListeners = [];
                 const closeListeners = [];
                 const errorListeners = [];
-                proc.stdout?.on("data", (data) => {
-                    dataListeners.forEach((l) => l(data));
+                ptyProc.onData((data) => {
+                    dataListeners.forEach((l) => l(Buffer.from(data)));
                 });
-                proc.stderr?.on("data", (data) => {
-                    dataListeners.forEach((l) => l(data));
-                });
-                proc.on("close", () => {
+                ptyProc.onExit(() => {
                     closeListeners.forEach((l) => l());
                 });
-                proc.on("error", (err) => {
-                    errorListeners.forEach((l) => l(err));
-                });
                 const stream = {
                     write: (data) => {
-                        proc.stdin?.write(data);
+                        ptyProc.write(data);
                     },
                     end: () => {
-                        proc.stdin?.end();
-                        proc.kill();
+                        ptyProc.kill();
                     },
                     on: ((event, listener) => {
                         if (event === "data") {
@@ -88,10 +86,10 @@ export class ShellManager {
                 };
                 this.shell = stream;
                 this.setupStream(stream);
-                // 等待初始化
+                // 等待初始提示符
                 setTimeout(() => {
                     resolve();
-                }, 300);
+                }, 500);
             }
             catch (error) {
                 reject(new Error(`无法打开本地 shell: ${error instanceof Error ? error.message : String(error)}`));
@@ -124,6 +122,7 @@ export class ShellManager {
         stream.on("close", () => {
             this.shell = null;
             this.localProcess = null;
+            this.ptyProcess = null;
         });
         stream.on("error", (err) => {
             console.error("Shell error:", err.message);
@@ -223,6 +222,10 @@ export class ShellManager {
             this.localProcess.kill();
             this.localProcess = null;
         }
+        if (this.ptyProcess) {
+            this.ptyProcess.kill();
+            this.ptyProcess = null;
+        }
         this.outputLines = [];
         this.outputBuffer = "";
         this.isLocal = false;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-ssh-pty",
-  "version": "1.2.0",
+  "version": "1.2.1",
   "description": "MCP Server for SSH remote command execution with PTY shell support",
   "type": "module",
   "main": "dist/index.js",
@@ -33,6 +33,7 @@
     "@inquirer/prompts": "^8.1.0",
     "@modelcontextprotocol/sdk": "^1.0.0",
     "commander": "^14.0.2",
+    "node-pty": "^1.1.0",
     "ssh2": "^1.16.0",
     "zod": "^3.24.0"
   },