snippets - specify the depth,suffix file path traverser

TypeScript
import fs from 'fs/promises';
import path from 'path';

export interface WalkDirOptions {
  extensions: string[]; //文件后缀
  maxDepth: number; // 最大深度
}

export async function walkDir(
  dirPath: string,
  options: WalkDirOptions,
  currentDepth = 0
): Promise<string[]> {
  if (currentDepth > options.maxDepth) {
    return [];
  }
  const files = await fs.readdir(dirPath);
  const filePaths: string[] = [];
  for (const file of files) {
    const filePath = path.join(dirPath, file);
    const stat = await fs.stat(filePath);
    if (stat.isDirectory()) {
      const subDirFilePaths = await walkDir(
        filePath,
        options,
        currentDepth + 1
      );
      filePaths.push(...subDirFilePaths);
    } else {
      const ext = path.extname(filePath);
      if (options.extensions.includes(ext)) {
        filePaths.push(filePath);
      }
    }
  }
  return filePaths;
}

TypeScript
import fs from 'fs';
import path from 'path';
import readline from 'readline';

interface WalkDirOptions {
  extensions: string[];
  maxDepth: number;
}

async function walkDir(
  dirPath: string,
  options: WalkDirOptions,
  currentDepth = 0
): Promise<string[]> {
  if (currentDepth > options.maxDepth) {
    return [];
  }
  const files = await fs.promises.readdir(dirPath);
  const filePaths: string[] = [];
  for (const file of files) {
    const filePath = path.join(dirPath, file);
    const stat = await fs.promises.stat(filePath);
    if (stat.isDirectory()) {
      const subDirFilePaths = await walkDir(
        filePath,
        options,
        currentDepth + 1
      );
      filePaths.push(...subDirFilePaths);
    } else {
      const ext = path.extname(filePath);
      if (options.extensions.includes(ext)) {
        const stream = fs.createReadStream(filePath);
        const rl = readline.createInterface({
          input: stream,
          crlfDelay: Infinity,
        });
        for await (const line of rl) {
          // Do something with the line
          // console.log(line);
        }
        filePaths.push(filePath);
      }
    }
  }
  return filePaths;
}