first-update

easy-dataset-main/app/api/projects/[projectId]/image-datasets/export-zip/route.js

@@ -0,0 +1,85 @@
+import { NextResponse } from 'next/server';
+import { getImageDatasetsForExport } from '@/lib/db/imageDatasets';
+import archiver from 'archiver';
+import { getProjectPath } from '@/lib/db/base';
+import path from 'path';
+import fs from 'fs';
+
+/**
+ * 导出图片文件压缩包
+ */
+export async function GET(request, { params }) {
+  try {
+    const { projectId } = params;
+    const { searchParams } = new URL(request.url);
+    const confirmedOnly = searchParams.get('confirmedOnly') === 'true';
+
+    // 验证项目ID
+    if (!projectId) {
+      return NextResponse.json({ error: 'Project ID cannot be empty' }, { status: 400 });
+    }
+
+    // 获取数据集（用于确定需要哪些图片）
+    const datasets = await getImageDatasetsForExport(projectId, confirmedOnly);
+
+    if (!datasets || datasets.length === 0) {
+      return NextResponse.json({ error: 'No data to export' }, { status: 404 });
+    }
+
+    // 获取所有需要的图片名称
+    const imageNames = new Set(datasets.map(d => d.imageName).filter(Boolean));
+
+    if (imageNames.size === 0) {
+      return NextResponse.json({ error: 'No images to export' }, { status: 404 });
+    }
+
+    // 创建压缩包
+    const archive = archiver('zip', {
+      zlib: { level: 9 }
+    });
+
+    // 设置响应头
+    const dateStr = new Date().toISOString().slice(0, 10);
+    const filename = `images-${projectId}-${dateStr}.zip`;
+
+    // 添加图片文件到压缩包
+    const projectPath = await getProjectPath(projectId);
+    const imageDir = path.join(projectPath, 'images');
+
+    if (!fs.existsSync(imageDir)) {
+      return NextResponse.json({ error: 'Image directory not found' }, { status: 404 });
+    }
+
+    let addedCount = 0;
+    for (const imageName of imageNames) {
+      const imagePath = path.join(imageDir, imageName);
+      if (fs.existsSync(imagePath)) {
+        archive.file(imagePath, { name: imageName });
+        addedCount++;
+      }
+    }
+
+    if (addedCount === 0) {
+      return NextResponse.json({ error: 'No image files found' }, { status: 404 });
+    }
+
+    // 完成压缩
+    archive.finalize();
+
+    // 返回流式响应
+    return new NextResponse(archive, {
+      headers: {
+        'Content-Type': 'application/zip',
+        'Content-Disposition': `attachment; filename="${filename}"`
+      }
+    });
+  } catch (error) {
+    console.error('Failed to export images:', String(error));
+    return NextResponse.json(
+      {
+        error: error.message || 'Failed to export images'
+      },
+      { status: 500 }
+    );
+  }
+}