import { promises as fs } from 'node:fs';
import { join, normalize } from 'node:path';

const rootDir = process.cwd();

/**
 * 递归查找并删除目标目录
 * @param {string} currentDir - 当前遍历的目录路径
 * @param {string[]} targets - 要删除的目标列表
 */
async function cleanTargetsRecursively(currentDir, targets) {
  const items = await fs.readdir(currentDir);

  for (const item of items) {
    try {
      const itemPath = normalize(join(currentDir, item));
      const stat = await fs.lstat(itemPath);

      if (targets.includes(item)) {
        // 匹配到目标目录或文件时直接删除
        await fs.rm(itemPath, { force: true, recursive: true });
        console.log(`Deleted: ${itemPath}`);
      } else if (stat.isDirectory()) {
        // 只对目录进行递归处理
        await cleanTargetsRecursively(itemPath, targets);
      }
    } catch (error) {
      console.error(
        `Error handling item ${item} in ${currentDir}: ${error.message}`,
      );
    }
  }
}

(async function startCleanup() {
  // 要删除的目录及文件名称
  const targets = ['node_modules', 'dist', '.turbo', 'dist.zip'];
  const deleteLockFile = process.argv.includes('--del-lock');
  const cleanupTargets = [...targets];

  if (deleteLockFile) {
    cleanupTargets.push('pnpm-lock.yaml');
  }

  console.log(
    `Starting cleanup of targets: ${cleanupTargets.join(', ')} from root: ${rootDir}`,
  );

  try {
    await cleanTargetsRecursively(rootDir, cleanupTargets);
    console.log('Cleanup process completed successfully.');
  } catch (error) {
    console.error(`Unexpected error during cleanup: ${error.message}`);
    process.exit(1);
  }
})();