用Node.js遍历多级目录处理文件的代码详解

const fs = require('fs').promises;
const path = require('path');

// 定义支持的图片扩展名
const imageExtensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.svg'];

// 遍历文件夹的异步函数
async function traverseDirectory(inputDir, outputDir, callback) {
  try {
    const files = await fs.readdir(inputDir);
    
    for (const file of files) {
      const inputFilePath = path.join(inputDir, file);
      const stats = await fs.stat(inputFilePath);

      if (stats.isDirectory()) {
        // 如果是文件夹，递归遍历
        const newOutputDir = path.join(outputDir, file);
        await fs.mkdir(newOutputDir, { recursive: true });
        await traverseDirectory(inputFilePath, newOutputDir, callback);
      } else if (stats.isFile()) {
        // 如果是文件，检查是否是图片
        const ext = path.extname(file).toLowerCase();
        if (imageExtensions.includes(ext)) {
          const outputFilePath = path.join(outputDir, file);
          await callback(inputFilePath, outputFilePath);
        }
      }
    }
  } catch (err) {
    console.error(`Error processing directory ${inputDir}:`, err);
  }
}

// 使用示例
const inputDirectory = '/path/to/your/input/directory'; // 修改为你的输入文件夹路径
const outputDirectory = '/path/to/your/output/directory'; // 修改为你的输出文件夹路径

// 主函数，处理文件遍历和复制
async function main() {
  await traverseDirectory(inputDirectory, outputDirectory, async (inputFilePath, outputFilePath) => {
    // 替换成真实业务图片操作，这里简单拷贝
    try {
      await fs.copyFile(inputFilePath, outputFilePath);
      console.log(`Copied ${inputFilePath} to ${outputFilePath}`);
    } catch (err) {
      console.error(`Error copying ${inputFilePath} to ${outputFilePath}:`, err);
    }
  });
}

// 执行主函数
main();