tri-cli 0.0.53 → 0.0.55

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;
   }
@@ -279,13 +265,18 @@ const TreeRow = /*#__PURE__*/React.memo(({
     color: oneHunter.purple
   }, " [", formatCount(fileCount), " files]"), isLoading && /*#__PURE__*/React.createElement(Text, {
     color: oneHunter.yellow
-  }, " ", SPINNER_FRAMES[spinnerFrame])), isLoading ? /*#__PURE__*/React.createElement(Text, {
+  }, " ", SPINNER_FRAMES[spinnerFrame])), isLoading && fileCount === undefined ? /*#__PURE__*/React.createElement(Text, {
     color: oneHunter.yellow
   }, " ", SPINNER_FRAMES[spinnerFrame]) : displaySize > 0 ? /*#__PURE__*/React.createElement(Text, {
     color: oneHunter.cyan
   }, " ", formatBytes(displaySize)) : /*#__PURE__*/React.createElement(Text, null, " "), renderShortcut(), lastModified && /*#__PURE__*/React.createElement(Text, {
     color: oneHunter.text2
   }, " ", formatDate(lastModified))));
+},
+// Custom comparison function to prevent unnecessary re-renders
+(prevProps, nextProps) => {
+  // Only re-render if these specific props change
+  return prevProps.node.data.path === nextProps.node.data.path && prevProps.isSelected === nextProps.isSelected && prevProps.collapsed.has(prevProps.node.data.path) === nextProps.collapsed.has(nextProps.node.data.path) && prevProps.shortcutInput === nextProps.shortcutInput && prevProps.calculatedSizes.get(prevProps.node.data.path) === nextProps.calculatedSizes.get(nextProps.node.data.path) && prevProps.calculatedCounts.get(prevProps.node.data.path) === nextProps.calculatedCounts.get(nextProps.node.data.path) && prevProps.loadingPaths.has(prevProps.node.data.path) === nextProps.loadingPaths.has(nextProps.node.data.path) && prevProps.lastModifiedDates.get(prevProps.node.data.path) === nextProps.lastModifiedDates.get(nextProps.node.data.path) && prevProps.spinnerFrame === nextProps.spinnerFrame;
 });
 const TreeVisualization = ({
   dirPath = '.',
@@ -451,64 +442,39 @@ 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 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);
-      }
+    setLoadingPaths(new Set(nodesToCount.map(n => n.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;
-        });
+    // Track completed counts
+    const completedCounts = new Map();
+    const completedPaths = new Set();
+
+    // Process all nodes in parallel
+    const promises = nodesToCount.map(async targetNode => {
+      const count = await countFilesAsync(targetNode.data.path);
+
+      // Store result
+      completedCounts.set(targetNode.data.path, count);
+      completedPaths.add(targetNode.data.path);
+
+      // Batch update state (React will batch these automatically)
+      setCalculatedCounts(prev => {
+        const next = new Map(prev);
+        next.set(targetNode.data.path, count);
+        return next;
       });
-      await new Promise(r => setTimeout(r, 0));
-    }
+      setLoadingPaths(prev => {
+        const next = new Set(prev);
+        next.delete(targetNode.data.path);
+        return next;
+      });
+    });
+
+    // Wait for all counts to complete
+    await Promise.all(promises);
   };
   const getLastModifiedAsync = async node => {
     // Collect node and all its descendants
@@ -834,7 +800,7 @@ const TreeVisualization = ({
   }, "\u2191/\u2193: Navigate | Enter: Expand/Traverse | c: Count | d: Size | l: Date | m: Map | Shift+C: CountMap | q: Quit"), loadingPaths.size > 0 && /*#__PURE__*/React.createElement(Text, {
     color: oneHunter.yellow
   }, ' ', SPINNER_FRAMES[spinnerFrame], " Loading ", loadingPaths.size, "...")), viewportNodes.map((node, i) => /*#__PURE__*/React.createElement(TreeRow, {
-    key: start + i,
+    key: node.data.path,
     node: node,
     isSelected: start + i === selectedIndex,
     collapsed: collapsed,

package/package.json

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