tri-cli 0.0.53 → 0.0.54

package/dist/cli.js

@@ -73,33 +73,19 @@ const getDirSizeAsync = async dirPath => {
   }
 };
 
-// Fast file counting - much faster than size calculation
+// Fast file counting using shell command
 const countFilesAsync = async dirPath => {
   try {
     const stats = fs.statSync(dirPath);
     if (!stats.isDirectory()) return 1;
-    let count = 0;
-    const countRecursive = async dir => {
-      try {
-        const entries = fs.readdirSync(dir, {
-          withFileTypes: true
-        });
-        for (const entry of entries) {
-          // Skip hidden files
-          if (entry.name.startsWith('.')) continue;
-          const fullPath = path.join(dir, entry.name);
-          if (entry.isFile()) {
-            count++;
-          } else if (entry.isDirectory()) {
-            await countRecursive(fullPath);
-          }
-        }
-      } catch {
-        // Skip directories we can't read
-      }
-    };
-    await countRecursive(dirPath);
-    return count;
+
+    // Use find command for fast counting, excluding hidden files
+    const escapedPath = dirPath.replace(/"/g, '\\"');
+    const {
+      stdout
+    } = await execAsync(`find "${escapedPath}" -type f ! -path '*/.*' 2>/dev/null | wc -l`);
+    const count = parseInt(stdout.trim(), 10);
+    return isNaN(count) ? 0 : count;
   } catch {
     return 0;
   }
@@ -451,64 +437,29 @@ const TreeVisualization = ({
     collectNodes(node);
     if (nodesToCount.length === 0) return; // Nothing to count
 
+    // Sort nodes by depth (shallowest first) for top-down display
+    nodesToCount.sort((a, b) => a.depth - b.depth);
+
     // Mark all nodes as loading
-    const pathsToLoad = nodesToCount.map(n => n.data.path);
-    setLoadingPaths(prev => new Set([...prev, ...pathsToLoad]));
+    const loadingSet = new Set(nodesToCount.map(n => n.data.path));
+    setLoadingPaths(loadingSet);
     const newMap = new Map(calculatedCounts);
 
-    // Count and stream updates for each node individually
-    for (const targetNode of nodesToCount) {
-      // For directories, stream partial counts as we count
-      if (!targetNode.data.isFile) {
-        let runningCount = 0;
-        let updateCounter = 0;
-        const countRecursive = async dir => {
-          try {
-            const entries = fs.readdirSync(dir, {
-              withFileTypes: true
-            });
-            for (const entry of entries) {
-              // Skip hidden files
-              if (entry.name.startsWith('.')) continue;
-              const fullPath = path.join(dir, entry.name);
-              if (entry.isFile()) {
-                runningCount++;
-                updateCounter++;
-                // Stream update every 100 files for better performance
-                if (updateCounter >= 100) {
-                  updateCounter = 0;
-                  newMap.set(targetNode.data.path, runningCount);
-                  React.startTransition(() => {
-                    setCalculatedCounts(new Map(newMap));
-                  });
-                  await new Promise(r => setTimeout(r, 0));
-                }
-              } else if (entry.isDirectory()) {
-                await countRecursive(fullPath);
-              }
-            }
-          } catch {
-            // Skip directories we can't read
-          }
-        };
-        await countRecursive(targetNode.data.path);
-        newMap.set(targetNode.data.path, runningCount);
-      } else {
-        // Files are always 1
-        newMap.set(targetNode.data.path, 1);
-      }
+    // Process all nodes in parallel, but update UI as each completes
+    const promises = nodesToCount.map(async targetNode => {
+      const count = await countFilesAsync(targetNode.data.path);
 
-      // Stream the final update and remove from loading
-      React.startTransition(() => {
-        setCalculatedCounts(new Map(newMap));
-        setLoadingPaths(prev => {
-          const next = new Set(prev);
-          next.delete(targetNode.data.path);
-          return next;
-        });
-      });
-      await new Promise(r => setTimeout(r, 0));
-    }
+      // Update immediately when this specific node completes
+      newMap.set(targetNode.data.path, count);
+      loadingSet.delete(targetNode.data.path);
+
+      // Trigger UI update
+      setCalculatedCounts(new Map(newMap));
+      setLoadingPaths(new Set(loadingSet));
+    });
+
+    // Wait for all counts to complete
+    await Promise.all(promises);
   };
   const getLastModifiedAsync = async node => {
     // Collect node and all its descendants

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "tri-cli",
-	"version": "0.0.53",
+	"version": "0.0.54",
 	"description": "Interactive CLI directory tree visualizer with treemap view",
 	"author": "Jared Wilber",
 	"license": "MIT",