质量管理/安全知识库

src/plugins/download.ts

@@ -1,5 +1,6 @@
 import axios from 'axios';
 import FileSaver from 'file-saver';
+import JSZip from 'jszip';
 import errorCode from '@/utils/errorCode';
 import { blobValidate } from '@/utils/ruoyi';
 import { LoadingInstance } from 'element-plus/es/components/loading/src/loading';
@@ -80,5 +81,49 @@ export default {
     } finally {
       downloadLoadingInstance.close();
     }
+  },
+  /**
+   * 下载多个文件并打包成 zip
+   * @param files 文件信息列表，包含 { url, name }
+   * @param zipName 压缩包名称
+   */
+  async downloadFilesAsZip(
+    files: any[],
+    options: {
+      urlKey?: string; // 默认为 'url'
+      nameKey?: string; // 默认为 'name'
+      zipName?: string; // 默认为 '打包下载.zip'
+    } = {}
+  ) {
+    const { urlKey = 'url', nameKey = 'name', zipName = '打包下载.zip' } = options;
+
+    if (!files.length) {
+      ElMessage.warning('没有可下载的文件');
+      return;
+    }
+
+    const zip = new JSZip();
+    const folder = zip.folder('files');
+    const loading = ElLoading.service({ text: '正在打包文件，请稍候...', background: 'rgba(0,0,0,0.7)' });
+
+    try {
+      for (const file of files) {
+        const fileUrl = file[urlKey];
+        const fileName = file[nameKey] || '未命名文件';
+
+        if (!fileUrl) continue;
+
+        const res = await axios.get(fileUrl, { responseType: 'blob' });
+        folder?.file(fileName, res.data);
+      }
+
+      const content = await zip.generateAsync({ type: 'blob' });
+      FileSaver.saveAs(content, zipName);
+    } catch (error) {
+      console.error(error);
+      ElMessage.error('打包下载失败，请联系管理员');
+    } finally {
+      loading.close();
+    }
   }
 };