clear_node_modules 1.1.6 → 1.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "clear_node_modules",
-  "version": "1.1.6",
+  "version": "1.2.0",
   "description": "help you remove all large file of node_modules in dist dir!",
   "main": "index.js",
   "directories": {

package/src/lib.js

@@ -1,36 +1,59 @@
-const fs = require('fs')
-const path = require('path')
-const rimraf = require('rimraf')
-const ora = require('ora')
+const fs = require("fs");
+const path = require("path");
+const rimraf = require("rimraf");
+const ora = require("ora");
 
-const noop = () => {}
+function getDirSize(dir) {
+  let total = 0;
+  const entries = fs.readdirSync(dir, { withFileTypes: true });
+  for (const entry of entries) {
+    const fullPath = path.join(dir, entry.name);
+    if (entry.isDirectory()) {
+      total += getDirSize(fullPath);
+    } else {
+      total += fs.statSync(fullPath).size;
+    }
+  }
+  return total;
+}
 
 function wrap(LIMIT_SIZE, NODE_MODULES) {
-  return function clearDir(disDir) {
-    const p = path.resolve(disDir)
-    if (!fs.statSync(p).isDirectory()) return
-    if (!fs.readdirSync(p).length) return
-    const { size } = fs.statSync(p)
-    if (size / 1024 < LIMIT_SIZE) return
-    const dirList = fs.readdirSync(p)
-    dirList.forEach(dir => {
-      const subP = path.resolve(p, dir)
-      if (dir === NODE_MODULES) {
-        const dirPath = path.resolve(__filename, subP)
-        const spinner = ora(`Removing ${dirPath}`).start()
-        rimraf(subP, noop)
-        spinner.succeed(`Done ${dirPath}`)
-        return
+  return function clearDir(dirPath) {
+    if (!fs.existsSync(dirPath) || !fs.statSync(dirPath).isDirectory()) return;
+
+    const entries = fs.readdirSync(dirPath);
+    if (entries.length === 0) return;
+
+    const dirSize = getDirSize(dirPath);
+    if (dirSize / 1024 < LIMIT_SIZE) return;
+
+    for (const entry of entries) {
+      const subPath = path.join(dirPath, entry);
+      if (entry === NODE_MODULES && fs.statSync(subPath).isDirectory()) {
+        const spinner = ora(`Removing ${subPath}`).start();
+        rimraf(subPath, (err) => {
+          if (err) {
+            spinner.fail(`Failed to remove ${subPath}: ${err.message}`);
+          } else {
+            spinner.succeed(`Done ${subPath}`);
+          }
+        });
+      } else if (fs.statSync(subPath).isDirectory()) {
+        clearDir(subPath);
       }
-      clearDir(subP)
-    })
-  }
+    }
+  };
 }
 
-function clearFunc(disDir = './', LIMIT_SIZE = 0, NODE_MODULES = 'node_modules') {
-  console.log('')
-  wrap(LIMIT_SIZE, NODE_MODULES)(disDir)
-  console.log('')
+function clearFunc(
+  disDir = "./",
+  LIMIT_SIZE = 0,
+  NODE_MODULES = "node_modules"
+) {
+  console.log("");
+  const clearDir = wrap(LIMIT_SIZE, NODE_MODULES);
+  clearDir(path.resolve(disDir));
+  console.log("");
 }
 
-module.exports = clearFunc
+module.exports = clearFunc;