递归删除子目录中所有dist和node_modules

javascript

const fs = require("fs")
const path = require("path")
const delDirs = ["node_modules", "dist"]
var shell = require("shelljs")

/** 递归删除当前目录下所有 dist / node_modules */
function del(filePath, root = "") {
  fs.stat(filePath, function (err, stat) {
    if (err) return console.log(err)

    if (stat.isFile()) {
      // do nothing
    } else {
      fs.readdir(filePath, function (err, data) {
        if (err) return console.log(err)

        // filter 需要继续递归的文件
        data = data.filter((dir) => {
          // 符合删除条件的文件名删除
          if (delDirs.includes(dir)) {
            console.log(path.join(filePath, dir))
            // root 目录不删除
            if (filePath != root) shell.rm("-rf", path.join(filePath, dir))
            return false
          }

          return true
        })

        data = data.map((dir) => path.join(filePath, dir))
        data.forEach((fullPath) => {
          del(fullPath)
        })
      })
    }
  })
}

del(__dirname, __dirname)

rust

use std::env;
use std::fs::{self};
use std::io;
use std::path::Path;

fn main() {
    
    let args: Vec<String> = env::args().collect();
    if args.len() < 3 {
        panic!("not enough args");
    }

    let root_path = Path::new(&args[1]);
    let delete_path = &args[2];

    visit_dirs(root_path, delete_path).unwrap_or_default();
}

fn visit_dirs(dir: &Path, delete_path: &String) -> io::Result<()> {
    if dir.is_dir() {
        for entry in fs::read_dir(dir)? {
            let entry = entry?;
            let path = entry.path();

            if path.is_dir() {
                let path_str = path.to_string_lossy();
                if path_str.contains(delete_path) {
                    println!("{} has been deleted", path_str);
                    fs::remove_dir_all(String::from(path_str))?;
                } else {
                    visit_dirs(&path, delete_path)?;
                }
            }
        }
    }
    Ok(())
}