AnimationManager/server/zip.js

// ZIP 打包处理模块
// 负责接收图片和 JSON 数据，打包成 ZIP 文件并返回

const archiver = require('archiver');
const http = require('http');

class ZipHandler {
  /**
   * 处理打包请求
   * @param {http.IncomingMessage} req - HTTP 请求对象
   * @param {http.ServerResponse} res - HTTP 响应对象
   */
  static handlePackRequest(req, res) {
    if (req.method !== 'POST') {
      res.writeHead(405, { 'Content-Type': 'application/json' });
      res.end(JSON.stringify({ error: 'Method not allowed' }));
      return;
    }

    let body = '';
    
    req.on('data', (chunk) => {
      body += chunk.toString();
    });

    req.on('end', () => {
      try {
        const data = JSON.parse(body);
        const { folderName, imageData, jsonData } = data;

        if (!folderName || !imageData || !jsonData) {
          res.writeHead(400, { 'Content-Type': 'application/json' });
          res.end(JSON.stringify({ error: 'Missing required fields: folderName, imageData, jsonData' }));
          return;
        }

        // 创建 ZIP 文件
        this.createZip(folderName, imageData, jsonData, res);
      } catch (error) {
        console.error('[ZipHandler] 解析请求数据失败:', error);
        res.writeHead(400, { 'Content-Type': 'application/json' });
        res.end(JSON.stringify({ error: 'Invalid JSON data', details: error.message }));
      }
    });

    req.on('error', (error) => {
      console.error('[ZipHandler] 请求错误:', error);
      res.writeHead(500, { 'Content-Type': 'application/json' });
      res.end(JSON.stringify({ error: 'Request error', details: error.message }));
    });
  }

  /**
   * 创建 ZIP 文件并返回
   * @param {string} folderName - 文件夹名称
   * @param {string} imageData - Base64 编码的图片数据
   * @param {string} jsonData - JSON 字符串数据
   * @param {http.ServerResponse} res - HTTP 响应对象
   */
  static createZip(folderName, imageData, jsonData, res) {
    try {
      // 设置响应头
      res.writeHead(200, {
        'Content-Type': 'application/zip',
        'Content-Disposition': `attachment; filename="${folderName}.zip"`,
        'Access-Control-Allow-Origin': '*'
      });

      // 创建 archiver 实例
      const archive = archiver('zip', {
        zlib: { level: 6 } // 压缩级别
      });

      // 监听错误
      archive.on('error', (err) => {
        console.error('[ZipHandler] ZIP 创建错误:', err);
        if (!res.headersSent) {
          res.writeHead(500, { 'Content-Type': 'application/json' });
          res.end(JSON.stringify({ error: 'Failed to create zip', details: err.message }));
        }
      });

      // 将 ZIP 流连接到响应
      archive.pipe(res);

      // 将 Base64 图片数据转换为 Buffer
      const imageBuffer = Buffer.from(imageData, 'base64');
      
      // 添加文件到 ZIP
      archive.append(imageBuffer, { name: `${folderName}.png` });
      archive.append(jsonData, { name: `${folderName}.json` });

      // 完成打包
      archive.finalize();

      console.log(`[ZipHandler] ✓ 成功创建 ZIP: ${folderName}.zip`);
    } catch (error) {
      console.error('[ZipHandler] 创建 ZIP 失败:', error);
      if (!res.headersSent) {
        res.writeHead(500, { 'Content-Type': 'application/json' });
        res.end(JSON.stringify({ error: 'Failed to create zip', details: error.message }));
      }
    }
  }
}

module.exports = ZipHandler;