ssh-mcp-devops 2.0.0 → 2.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ssh-mcp-devops",
-  "version": "2.0.0",
+  "version": "2.0.2",
   "description": "SSH MCP Server for AI-powered DevOps with Docker and System Monitoring",
   "type": "module",
   "main": "index-v2.js",

package/ssh-manager-v2.js

@@ -139,20 +139,44 @@ export class SSHManager {
       throw new Error(`本地文件不存在: ${localPath}`);
     }
 
+    // 获取文件大小
+    const stats = await fs.stat(localPath);
+    const fileSizeMB = (stats.size / 1024 / 1024).toFixed(2);
+    console.error(`开始上传文件: ${localPath} (${fileSizeMB} MB)`);
+
     return new Promise((resolve, reject) => {
+      // 根据文件大小设置超时（每 MB 10 秒，最少 60 秒）
+      const timeout = Math.max(60000, stats.size / 1024 / 1024 * 10000);
+      let uploadTimer;
+
       this.client.sftp((err, sftp) => {
         if (err) return reject(err);
 
+        // 设置超时
+        uploadTimer = setTimeout(() => {
+          sftp.end();
+          reject(new Error(`上传超时 (${(timeout / 1000).toFixed(0)}秒): 文件可能太大`));
+        }, timeout);
+
         const transferOptions = {
           concurrency: 64,
           chunkSize: 32768,
-          step: options.onProgress || undefined,
+          step: (total, nb, fsize) => {
+            // 进度回调
+            const percent = ((nb / fsize) * 100).toFixed(1);
+            console.error(`上传进度: ${percent}% (${(nb / 1024 / 1024).toFixed(2)} MB / ${fileSizeMB} MB)`);
+            if (options.onProgress) {
+              options.onProgress(total, nb, fsize);
+            }
+          },
         };
 
         sftp.fastPut(localPath, remotePath, transferOptions, (err) => {
+          clearTimeout(uploadTimer);
           sftp.end();
           if (err) return reject(new Error(`上传失败: ${err.message}`));
-          resolve(`✅ 文件上传成功: ${localPath} -> ${remotePath}`);
+          console.error(`✅ 上传完成: ${fileSizeMB} MB`);
+          resolve(`✅ 文件上传成功: ${localPath} -> ${remotePath} (${fileSizeMB} MB)`);
         });
       });
     });
@@ -166,19 +190,48 @@ export class SSHManager {
     await fs.mkdir(localDir, { recursive: true });
 
     return new Promise((resolve, reject) => {
+      let downloadTimer;
+
       this.client.sftp((err, sftp) => {
         if (err) return reject(err);
 
-        const transferOptions = {
-          concurrency: 64,
-          chunkSize: 32768,
-          step: options.onProgress || undefined,
-        };
+        // 先获取文件大小
+        sftp.stat(remotePath, (err, stats) => {
+          if (err) {
+            sftp.end();
+            return reject(new Error(`无法获取远程文件信息: ${err.message}`));
+          }
 
-        sftp.fastGet(remotePath, localPath, transferOptions, (err) => {
-          sftp.end();
-          if (err) return reject(new Error(`下载失败: ${err.message}`));
-          resolve(`✅ 文件下载成功: ${remotePath} -> ${localPath}`);
+          const fileSizeMB = (stats.size / 1024 / 1024).toFixed(2);
+          console.error(`开始下载文件: ${remotePath} (${fileSizeMB} MB)`);
+
+          // 根据文件大小设置超时（每 MB 10 秒，最少 60 秒）
+          const timeout = Math.max(60000, stats.size / 1024 / 1024 * 10000);
+
+          downloadTimer = setTimeout(() => {
+            sftp.end();
+            reject(new Error(`下载超时 (${(timeout / 1000).toFixed(0)}秒): 文件可能太大`));
+          }, timeout);
+
+          const transferOptions = {
+            concurrency: 64,
+            chunkSize: 32768,
+            step: (total, nb, fsize) => {
+              const percent = ((nb / fsize) * 100).toFixed(1);
+              console.error(`下载进度: ${percent}% (${(nb / 1024 / 1024).toFixed(2)} MB / ${fileSizeMB} MB)`);
+              if (options.onProgress) {
+                options.onProgress(total, nb, fsize);
+              }
+            },
+          };
+
+          sftp.fastGet(remotePath, localPath, transferOptions, (err) => {
+            clearTimeout(downloadTimer);
+            sftp.end();
+            if (err) return reject(new Error(`下载失败: ${err.message}`));
+            console.error(`✅ 下载完成: ${fileSizeMB} MB`);
+            resolve(`✅ 文件下载成功: ${remotePath} -> ${localPath} (${fileSizeMB} MB)`);
+          });
         });
       });
     });