clear_node_modules 1.4.2 → 1.4.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "clear_node_modules",
-  "version": "1.4.2",
+  "version": "1.4.3",
   "description": "Recursively remove node_modules folders from current directory and subdirectories",
   "main": "index.js",
   "bin": {

package/src/lib.js

@@ -6,16 +6,34 @@ const { promisify } = require("util");
 
 const rimrafAsync = promisify(rimraf);
 
+// 格式化文件大小
+function formatSize(bytes) {
+  if (bytes < 1024) return bytes + " B";
+  if (bytes < 1024 * 1024) return (bytes / 1024).toFixed(1) + " KB";
+  if (bytes < 1024 * 1024 * 1024) return (bytes / 1024 / 1024).toFixed(1) + " MB";
+  return (bytes / 1024 / 1024 / 1024).toFixed(2) + " GB";
+}
+
 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;
+  try {
+    const entries = fs.readdirSync(dir, { withFileTypes: true });
+    for (const entry of entries) {
+      const fullPath = path.join(dir, entry.name);
+      try {
+        if (entry.isSymbolicLink()) {
+          continue;
+        } else if (entry.isDirectory()) {
+          total += getDirSize(fullPath);
+        } else if (entry.isFile()) {
+          total += fs.statSync(fullPath).size;
+        }
+      } catch (e) {
+        continue;
+      }
     }
+  } catch (e) {
+    // 目录不可读
   }
   return total;
 }
@@ -25,61 +43,111 @@ function wrap(LIMIT_SIZE, NODE_MODULES) {
     const tasks = [];
     let totalCount = 0;
     let doneCount = 0;
+    let totalFreed = 0;
+    let failedCount = 0;
 
-    // 先扫描一次，预估总数
+    // 扫描阶段
+    const scanSpinner = ora("🔍 Scanning for node_modules...").start();
+    
     function scan(pathToScan) {
-      if (!fs.existsSync(pathToScan) || !fs.statSync(pathToScan).isDirectory())
-        return;
-      const entries = fs.readdirSync(pathToScan);
-      for (const entry of entries) {
-        const subPath = path.join(pathToScan, entry);
-        if (entry === NODE_MODULES && fs.existsSync(subPath)) {
-          totalCount++;
-        } else if (fs.statSync(subPath).isDirectory()) {
-          scan(subPath);
+      try {
+        if (!fs.existsSync(pathToScan)) return;
+        const stat = fs.lstatSync(pathToScan);
+        if (!stat.isDirectory() || stat.isSymbolicLink()) return;
+        
+        const entries = fs.readdirSync(pathToScan, { withFileTypes: true });
+        for (const entry of entries) {
+          const subPath = path.join(pathToScan, entry.name);
+          try {
+            if (entry.isSymbolicLink()) continue;
+            if (entry.name === NODE_MODULES && entry.isDirectory()) {
+              totalCount++;
+            } else if (entry.isDirectory()) {
+              scan(subPath);
+            }
+          } catch (e) {
+            continue;
+          }
         }
+      } catch (e) {
+        // 忽略
       }
     }
 
     scan(dirPath);
+    
+    if (totalCount === 0) {
+      scanSpinner.info("No node_modules found.");
+      return;
+    }
+    
+    scanSpinner.succeed(`Found ${totalCount} node_modules folder(s)`);
+    console.log("");
 
     async function recurse(currentPath) {
-      if (
-        !fs.existsSync(currentPath) ||
-        !fs.statSync(currentPath).isDirectory()
-      )
-        return;
+      try {
+        if (!fs.existsSync(currentPath)) return;
+        const stat = fs.lstatSync(currentPath);
+        if (!stat.isDirectory() || stat.isSymbolicLink()) return;
 
-      const entries = fs.readdirSync(currentPath);
-      if (entries.length === 0) return;
+        const entries = fs.readdirSync(currentPath, { withFileTypes: true });
+        if (entries.length === 0) return;
 
-      for (const entry of entries) {
-        const subPath = path.join(currentPath, entry);
-        if (entry === NODE_MODULES && fs.statSync(subPath).isDirectory()) {
-          const dirSize = getDirSize(subPath);
-          if (dirSize / 1024 / 1024 < LIMIT_SIZE) continue;
+        for (const entry of entries) {
+          const subPath = path.join(currentPath, entry.name);
+          try {
+            if (entry.isSymbolicLink()) continue;
+            
+            if (entry.name === NODE_MODULES && entry.isDirectory()) {
+              const dirSize = getDirSize(subPath);
+              const sizeMB = dirSize / 1024 / 1024;
+              
+              if (sizeMB < LIMIT_SIZE) {
+                doneCount++;
+                console.log(`  ⏭️  [${doneCount}/${totalCount}] Skipped (${formatSize(dirSize)}) ${subPath}`);
+                continue;
+              }
 
-          const spinner = ora(
-            `(${++doneCount}/${totalCount}) Removing ${subPath}`
-          ).start();
-          const task = rimrafAsync(subPath)
-            .then(() =>
-              spinner.succeed(`(${doneCount}/${totalCount}) Done ${subPath}`)
-            )
-            .catch((err) =>
-              spinner.fail(
-                `(${doneCount}/${totalCount}) Failed ${subPath}: ${err.message}`
-              )
-            );
-          tasks.push(task);
-        } else if (fs.statSync(subPath).isDirectory()) {
-          await recurse(subPath);
+              const spinner = ora({
+                text: `[${++doneCount}/${totalCount}] Removing ${formatSize(dirSize)} - ${subPath}`,
+                prefixText: "  "
+              }).start();
+              
+              const task = rimrafAsync(subPath)
+                .then(() => {
+                  totalFreed += dirSize;
+                  spinner.succeed(`[${doneCount}/${totalCount}] ✓ Freed ${formatSize(dirSize)} - ${subPath}`);
+                })
+                .catch((err) => {
+                  failedCount++;
+                  spinner.fail(`[${doneCount}/${totalCount}] ✗ Failed - ${subPath}: ${err.message}`);
+                });
+              tasks.push(task);
+            } else if (entry.isDirectory()) {
+              await recurse(subPath);
+            }
+          } catch (e) {
+            continue;
+          }
         }
+      } catch (e) {
+        // 忽略
       }
     }
 
     await recurse(dirPath);
     await Promise.all(tasks);
+    
+    // 汇总信息
+    console.log("");
+    console.log("─".repeat(50));
+    console.log(`  ✨ Completed!`);
+    console.log(`     📁 Processed: ${totalCount} folder(s)`);
+    console.log(`     💾 Freed: ${formatSize(totalFreed)}`);
+    if (failedCount > 0) {
+      console.log(`     ⚠️  Failed: ${failedCount}`);
+    }
+    console.log("─".repeat(50));
   };
 }
 
@@ -89,9 +157,18 @@ async function clearFunc(
   NODE_MODULES = "node_modules"
 ) {
   console.log("");
+  console.log("🧹 Clear Node Modules");
+  console.log("─".repeat(50));
+  console.log(`  📂 Target: ${path.resolve(disDir)}`);
+  if (LIMIT_SIZE > 0) {
+    console.log(`  📏 Min size: ${LIMIT_SIZE} MB`);
+  }
+  console.log("─".repeat(50));
+  console.log("");
+  
   const clearDir = wrap(LIMIT_SIZE, NODE_MODULES);
   await clearDir(path.resolve(disDir));
-  console.log("\nAll done!\n");
+  console.log("");
 }
 
 module.exports = clearFunc;