eSIM-Tools/scripts/compress.js

const fs = require('fs');
const path = require('path');
const zlib = require('zlib');
const { promisify } = require('util');

const gzip = promisify(zlib.gzip);
const brotliCompress = promisify(zlib.brotliCompress);

// 压缩配置
const compressionOptions = {
  gzip: {
    level: 9,
    memLevel: 9
  },
  brotli: {
    params: {
      [zlib.constants.BROTLI_PARAM_QUALITY]: 11,
      [zlib.constants.BROTLI_PARAM_MODE]: zlib.constants.BROTLI_MODE_GENERIC,
      [zlib.constants.BROTLI_PARAM_SIZE_HINT]: 0
    }
  }
};

// 需要压缩的文件类型
const compressibleExtensions = ['.js', '.css', '.html', '.json', '.xml', '.svg'];

// 压缩单个文件
async function compressFile(filePath) {
  try {
    const content = fs.readFileSync(filePath);
    const ext = path.extname(filePath);
    
    // 只压缩特定类型的文件
    if (!compressibleExtensions.includes(ext)) {
      return null;
    }
    
    const fileName = path.basename(filePath);
    const dir = path.dirname(filePath);
    
    // Gzip压缩
    const gzipped = await gzip(content, compressionOptions.gzip);
    const gzipPath = path.join(dir, `${fileName}.gz`);
    fs.writeFileSync(gzipPath, gzipped);
    
    // Brotli压缩（如果支持）
    try {
      const brotlied = await brotliCompress(content, compressionOptions.brotli);
      const brotliPath = path.join(dir, `${fileName}.br`);
      fs.writeFileSync(brotliPath, brotlied);
    } catch (error) {
      console.log(`Brotli压缩失败 ${fileName}:`, error.message);
    }
    
    const originalSize = content.length;
    const gzipSize = gzipped.length;
    const compressionRatio = ((originalSize - gzipSize) / originalSize * 100).toFixed(1);
    
    return {
      file: fileName,
      original: originalSize,
      gzip: gzipSize,
      ratio: compressionRatio
    };
  } catch (error) {
    console.error(`压缩文件失败 ${filePath}:`, error.message);
    return null;
  }
}

// 递归压缩目录
async function compressDirectory(dirPath) {
  const results = [];
  
  async function scanDirectory(dir) {
    const files = fs.readdirSync(dir);
    
    for (const file of files) {
      const fullPath = path.join(dir, file);
      const stat = fs.statSync(fullPath);
      
      if (stat.isDirectory()) {
        await scanDirectory(fullPath);
      } else if (stat.isFile()) {
        const result = await compressFile(fullPath);
        if (result) {
          results.push(result);
        }
      }
    }
  }
  
  await scanDirectory(dirPath);
  return results;
}

// 主压缩函数
async function compressBuild() {
  const distDir = path.join(__dirname, '../dist');
  
  if (!fs.existsSync(distDir)) {
    console.error('dist目录不存在，请先运行构建命令');
    return;
  }
  
  console.log('开始压缩构建文件...');
  
  try {
    const results = await compressDirectory(distDir);
    
    if (results.length === 0) {
      console.log('没有找到需要压缩的文件');
      return;
    }
    
    // 显示压缩结果
    console.log('\n压缩结果:');
    console.log('文件名'.padEnd(30) + '原始大小'.padEnd(12) + 'Gzip大小'.padEnd(12) + '压缩率');
    console.log('-'.repeat(70));
    
    let totalOriginal = 0;
    let totalGzip = 0;
    
    results.forEach(result => {
      totalOriginal += result.original;
      totalGzip += result.gzip;
      
      const originalKB = (result.original / 1024).toFixed(1);
      const gzipKB = (result.gzip / 1024).toFixed(1);
      
      console.log(
        result.file.padEnd(30) +
        `${originalKB}KB`.padEnd(12) +
        `${gzipKB}KB`.padEnd(12) +
        `${result.ratio}%`
      );
    });
    
    const totalRatio = ((totalOriginal - totalGzip) / totalOriginal * 100).toFixed(1);
    console.log('-'.repeat(70));
    console.log(
      '总计'.padEnd(30) +
      `${(totalOriginal / 1024).toFixed(1)}KB`.padEnd(12) +
      `${(totalGzip / 1024).toFixed(1)}KB`.padEnd(12) +
      `${totalRatio}%`
    );
    
    console.log('\n压缩完成！');
  } catch (error) {
    console.error('压缩失败:', error);
  }
}

// 如果直接运行此脚本
if (require.main === module) {
  compressBuild();
}

module.exports = { compressBuild, compressFile };