update

2025-04-29 21:28:00 +08:00
parent 4ce5d90512
commit e0a234210d
8 changed files with 1075 additions and 19 deletions
@@ -1,6 +1,9 @@
 import { Request, Response } from 'express';
 import path from 'path';
-import { clientStates } from '../websocket'; 
+import { clientStates } from '../websocket';
+import archiver from 'archiver'; // +++ 引入 archiver +++
+import { SFTPWrapper, Stats } from 'ssh2'; // +++ 移除 Dirent 导入 +++
+// 移除 ssh2-streams 导入

 /**
 * 处理文件下载请求 (GET /api/v1/sftp/download)
@@ -103,3 +106,152 @@ export const downloadFile = async (req: Request, res: Response): Promise<void> =
 };


+/**
+ * 处理文件夹下载请求 (GET /api/v1/sftp/download-directory)
+ */
+export const downloadDirectory = async (req: Request, res: Response): Promise<void> => {
+    const userId = req.session.userId;
+    const connectionId = req.query.connectionId as string; // 从查询参数获取
+    const remotePath = req.query.remotePath as string;   // 从查询参数获取
+
+    // 参数验证
+    if (!userId) {
+        res.status(401).json({ message: '未授权：需要登录。' });
+        return;
+    }
+    if (!connectionId || !remotePath) {
+        res.status(400).json({ message: '缺少必要的查询参数 (connectionId, remotePath)。' });
+        return;
+    }
+
+    console.log(`SFTP 文件夹下载请求：用户 ${userId}, 连接 ${connectionId}, 路径 ${remotePath}`);
+
+    // 查找与当前用户会话关联的活动 WebSocket 连接和 SFTP 会话 (与 downloadFile 逻辑相同)
+    let userSftpSession = null;
+    for (const [sessionId, state] of clientStates.entries()) {
+        const ws = state.ws;
+        const connData = state;
+        if (ws.userId === userId && connData.sftp) {
+            // TODO: 关联特定的 connectionId
+            userSftpSession = connData.sftp;
+            console.log(`找到用户 ${userId} 的活动 SFTP 会话。`);
+            break;
+        }
+    }
+
+    if (!userSftpSession) {
+        console.warn(`SFTP 文件夹下载失败：未找到用户 ${userId} 的活动 SFTP 会话。`);
+        res.status(404).json({ message: '未找到活动的 SFTP 会话。请确保您已通过 WebSocket 连接到目标服务器。' });
+        return;
+    }
+
+    try {
+        // 1. 验证路径是否为目录
+        const stats = await new Promise<import('ssh2').Stats>((resolve, reject) => {
+            userSftpSession!.lstat(remotePath, (err, stats) => {
+                if (err) return reject(err);
+                resolve(stats);
+            });
+        });
+
+        if (!stats.isDirectory()) {
+            res.status(400).json({ message: '指定的路径不是一个目录。' });
+            return;
+        }
+
+        // 2. 设置响应头
+        // --- 修正：更健壮地生成压缩包名称 ---
+        let baseName = path.basename(remotePath);
+        // 处理根目录或路径以斜杠结尾的特殊情况
+        if (!baseName || baseName === '/') {
+            baseName = 'download'; // 如果 basename 为空或只是 '/'，使用默认名称
+        }
+        // 确保 basename 本身不包含尾部斜杠（尽管 path.basename 通常会处理）
+        baseName = baseName.replace(/\/$/, '');
+        const archiveName = `${baseName}.zip`; // 拼接 .zip 后缀
+        // --- 结束修正 ---
+        res.setHeader('Content-Type', 'application/zip');
+        res.setHeader('Content-Disposition', `attachment; filename="${archiveName}"`); // 使用修正后的名称
+
+        // 3. 创建 Archiver 实例
+        const archive = archiver('zip', {
+            zlib: { level: 9 } // 设置压缩级别 (可选)
+        });
+
+        // 监听错误事件
+        archive.on('warning', (err) => {
+            console.warn(`Archiver warning (用户 ${userId}, 路径 ${remotePath}):`, err);
+        });
+        archive.on('error', (err) => {
+            console.error(`Archiver error (用户 ${userId}, 路径 ${remotePath}):`, err);
+            // 尝试发送错误响应，如果头还没发送
+            if (!res.headersSent) {
+                res.status(500).json({ message: `创建压缩文件时出错: ${err.message}` });
+            } else {
+                res.end(); // 否则尝试结束响应
+            }
+        });
+
+        // 将 Archiver 输出流 pipe 到 HTTP 响应流
+        archive.pipe(res);
+
+        // 4. 递归添加文件/目录到 archive (核心逻辑)
+        //    这部分需要一个辅助函数来处理 SFTP 递归和 Archiver 添加
+        const addDirectoryToArchive = async (sftp: SFTPWrapper, dirPath: string, archivePath: string) => { // 使用导入的 SFTPWrapper
+            // 移除 list 的显式类型注解 FileEntry[]，让 TypeScript 推断
+            const entries = await new Promise<any[]>((resolve, reject) => { // 使用 any[] 作为 Promise 类型，或更具体的推断类型
+                sftp.readdir(dirPath, (err: Error | undefined, list) => { // 移除 list 的类型注解
+                    if (err) return reject(err);
+                    // 可以在这里检查 list 的结构，但暂时依赖推断
+                    resolve(list);
+                });
+            });
+
+            for (const entry of entries) {
+                const currentRemotePath = path.posix.join(dirPath, entry.filename); // 使用 posix.join 处理路径
+                const currentArchivePath = path.posix.join(archivePath, entry.filename);
+
+                if (entry.attrs.isDirectory()) {
+                    // 递归添加子目录
+                    // 使用 Buffer.from('') 代替 null
+                    archive.append(Buffer.from(''), { name: currentArchivePath + '/' });
+                    await addDirectoryToArchive(sftp, currentRemotePath, currentArchivePath);
+                } else if (entry.attrs.isFile()) {
+                    // 添加文件流
+                    const fileStream = sftp.createReadStream(currentRemotePath);
+                    archive.append(fileStream, { name: currentArchivePath });
+                    // 注意：需要处理 fileStream 的错误事件吗？Archiver 应该会处理？待验证。
+                     fileStream.on('error', (streamErr: Error) => { // 添加类型注解
+                         console.error(`Error reading file stream ${currentRemotePath}:`, streamErr);
+                         // 如何通知 Archiver 或中断？ Archiver 的 error 事件应该会捕获？
+                         if (!archive.destroyed) { // 检查 archive 是否已被销毁
+                            archive.abort(); // 尝试中止 archive
+                         }
+                     });
+                }
+            }
+        };
+
+        // 开始添加根目录内容
+        await addDirectoryToArchive(userSftpSession, remotePath, ''); // 归档路径从根开始
+
+        // 5. 完成归档
+        await archive.finalize();
+
+        console.log(`SFTP 文件夹下载完成 (用户 ${userId}, 路径 ${remotePath})`);
+
+    } catch (error: any) {
+        console.error(`SFTP 文件夹下载处理失败 (用户 ${userId}, 路径 ${remotePath}):`, error);
+        if (!res.headersSent) {
+            if (error.code === 'ENOENT' || error.message?.includes('No such file')) { // 检查 SFTP 错误码或消息
+                 res.status(404).json({ message: '远程目录未找到。' });
+            } else {
+                 res.status(500).json({ message: `处理文件夹下载请求时出错: ${error.message}` });
+            }
+        } else {
+            res.end(); // 如果头已发送，尝试结束响应
+        }
+    }
+};
+
+
@@ -1,6 +1,6 @@
 import { Router } from 'express';
 import { isAuthenticated } from '../auth/auth.middleware';
-import { downloadFile } from './sftp.controller';
+import { downloadFile, downloadDirectory } from './sftp.controller'; // +++ 导入 downloadDirectory +++

 const router = Router();

@@ -10,5 +10,8 @@ router.use(isAuthenticated);
 // GET /api/v1/sftp/download?connectionId=...&remotePath=...
 router.get('/download', downloadFile);

+// +++ 新增：GET /api/v1/sftp/download-directory?connectionId=...&remotePath=... +++
+router.get('/download-directory', downloadDirectory);
+

 export default router;