tri-cli 0.0.54 → 0.0.55

package/dist/cli.js

@@ -265,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 = '.',
@@ -441,21 +446,31 @@ const TreeVisualization = ({
     nodesToCount.sort((a, b) => a.depth - b.depth);
 
     // Mark all nodes as loading
-    const loadingSet = new Set(nodesToCount.map(n => n.data.path));
-    setLoadingPaths(loadingSet);
-    const newMap = new Map(calculatedCounts);
+    setLoadingPaths(new Set(nodesToCount.map(n => n.data.path)));
 
-    // Process all nodes in parallel, but update UI as each completes
+    // 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);
 
-      // Update immediately when this specific node completes
-      newMap.set(targetNode.data.path, count);
-      loadingSet.delete(targetNode.data.path);
+      // Store result
+      completedCounts.set(targetNode.data.path, count);
+      completedPaths.add(targetNode.data.path);
 
-      // Trigger UI update
-      setCalculatedCounts(new Map(newMap));
-      setLoadingPaths(new Set(loadingSet));
+      // Batch update state (React will batch these automatically)
+      setCalculatedCounts(prev => {
+        const next = new Map(prev);
+        next.set(targetNode.data.path, count);
+        return next;
+      });
+      setLoadingPaths(prev => {
+        const next = new Set(prev);
+        next.delete(targetNode.data.path);
+        return next;
+      });
     });
 
     // Wait for all counts to complete
@@ -785,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.54",
+	"version": "0.0.55",
 	"description": "Interactive CLI directory tree visualizer with treemap view",
 	"author": "Jared Wilber",
 	"license": "MIT",