@tmustier/pi-files-widget 0.1.0 → 0.1.7

package/CHANGELOG.md

@@ -2,6 +2,13 @@
 
 All notable changes to this extension will be documented in this file.
 
+## [0.1.7] - 2026-01-26
+
+### Changed
+- Cache line counts and skip large files to avoid freezes in big folders
+- Avoid recomputing tree stats on every render
+- Preserve line counts for open files across refreshes
+
 ## [0.1.6] - 2026-01-24
 
 ### Added

package/browser.ts

@@ -21,10 +21,17 @@ export interface BrowserController {
   invalidate(): void;
 }
 
+interface BrowserStats {
+  totalLines?: number;
+  additions: number;
+  deletions: number;
+}
 
 interface BrowserState {
   root: FileNode | null;
   flatList: FlatNode[];
+  fullList: FlatNode[];
+  stats: BrowserStats;
   selectedIndex: number;
   searchQuery: string;
   searchMode: boolean;
@@ -100,28 +107,16 @@ function collectChangedFiles(node: FileNode, ancestors: FileNode[] = []): Change
   return results;
 }
 
-function computeTotalStats(root: FileNode | null): { totalLines: number; additions: number; deletions: number } {
-  let totalLines = 0;
-  let additions = 0;
-  let deletions = 0;
-
-  function traverse(node: FileNode): void {
-    if (!node.isDirectory) {
-      totalLines += node.lineCount || 0;
-      if (node.diffStats) {
-        additions += node.diffStats.additions;
-        deletions += node.diffStats.deletions;
-      }
-    }
-    if (node.children) {
-      for (const child of node.children) {
-        traverse(child);
-      }
-    }
+function getTreeStats(root: FileNode | null): BrowserStats {
+  if (!root) {
+    return { totalLines: undefined, additions: 0, deletions: 0 };
   }
 
-  if (root) traverse(root);
-  return { totalLines, additions, deletions };
+  return {
+    totalLines: root.lineCountComplete ? root.totalLines ?? 0 : undefined,
+    additions: root.totalAdditions ?? 0,
+    deletions: root.totalDeletions ?? 0,
+  };
 }
 
 function formatNodeStatus(node: FileNode, theme: Theme): string {
@@ -146,7 +141,7 @@ function formatNodeMeta(node: FileNode, theme: Theme): string {
     if (node.totalDeletions && node.totalDeletions > 0) {
       parts.push(theme.fg("error", `-${node.totalDeletions}`));
     }
-    if (node.totalLines) {
+    if (node.totalLines && node.lineCountComplete !== false) {
       parts.push(theme.fg("dim", `${node.totalLines}L`));
     }
   } else if (!node.isDirectory) {
@@ -157,10 +152,10 @@ function formatNodeMeta(node: FileNode, theme: Theme): string {
       if (node.diffStats.deletions > 0) {
         parts.push(theme.fg("error", `-${node.diffStats.deletions}`));
       }
-    } else if (isUntrackedStatus(node.gitStatus) && node.lineCount) {
+    } else if (isUntrackedStatus(node.gitStatus) && node.lineCount !== undefined) {
       parts.push(theme.fg("success", `+${node.lineCount}`));
     }
-    if (node.lineCount) {
+    if (node.lineCount !== undefined) {
       parts.push(theme.fg("dim", `${node.lineCount}L`));
     }
   }
@@ -202,9 +197,13 @@ export function createFileBrowser(
 
   const viewer = createViewer(cwd, theme, requestComment);
 
+  const root = buildFileTree(cwd, cwd, gitStatus, diffStats, ignored, agentModifiedFiles);
+
   const browser: BrowserState = {
-    root: buildFileTree(cwd, cwd, gitStatus, diffStats, ignored, agentModifiedFiles),
+    root,
     flatList: [],
+    fullList: [],
+    stats: getTreeStats(root),
     selectedIndex: 0,
     searchQuery: "",
     searchMode: false,
@@ -214,40 +213,49 @@ export function createFileBrowser(
   };
 
   browser.flatList = browser.root ? flattenTree(browser.root) : [];
+  browser.fullList = browser.root ? flattenTree(browser.root, 0, true, true) : [];
 
   function refreshGitStatus(): void {
-    const currentPath = browser.flatList[browser.selectedIndex]?.node.path;
-    const viewingFilePath = viewer.getFile()?.path;
+    const previousDisplayList = getDisplayList();
+    const currentPath = previousDisplayList[browser.selectedIndex]?.node.path;
+    const viewingFile = viewer.getFile();
+    const viewingFilePath = viewingFile?.path;
 
     const expandedPaths = captureExpandedPaths(browser.root);
 
     gitStatus = getGitStatus(cwd);
     diffStats = getGitDiffStats(cwd);
     browser.root = buildFileTree(cwd, cwd, gitStatus, diffStats, ignored, agentModifiedFiles);
+    browser.stats = getTreeStats(browser.root);
 
     restoreExpandedPaths(browser.root, expandedPaths);
 
     browser.flatList = browser.root ? flattenTree(browser.root) : [];
+    browser.fullList = browser.root ? flattenTree(browser.root, 0, true, true) : [];
 
+    const updatedDisplayList = getDisplayList();
     if (currentPath) {
-      const newIdx = browser.flatList.findIndex(f => f.node.path === currentPath);
+      const newIdx = updatedDisplayList.findIndex(f => f.node.path === currentPath);
       if (newIdx !== -1) {
         browser.selectedIndex = newIdx;
       }
     }
 
-    browser.selectedIndex = Math.min(browser.selectedIndex, Math.max(0, browser.flatList.length - 1));
+    browser.selectedIndex = Math.min(browser.selectedIndex, Math.max(0, updatedDisplayList.length - 1));
 
     if (viewingFilePath && browser.root) {
       const newNode = findNodeByPath(browser.root, viewingFilePath);
       if (newNode) {
+        if (newNode.lineCount === undefined && viewingFile?.lineCount !== undefined) {
+          newNode.lineCount = viewingFile.lineCount;
+        }
         viewer.updateFileRef(newNode);
       }
     }
   }
 
   function getDisplayList(): FlatNode[] {
-    let list = browser.flatList;
+    let list = browser.searchQuery ? browser.fullList : browser.flatList;
 
     if (browser.showOnlyChanged) {
       list = list.filter(f =>
@@ -321,13 +329,14 @@ export function createFileBrowser(
     const lines: string[] = [];
     const pathDisplay = basename(cwd);
     const branchDisplay = gitBranch ? theme.fg("accent", ` (${gitBranch})`) : "";
-    const stats = computeTotalStats(browser.root);
+    const stats = browser.stats;
 
-    let statsDisplay = theme.fg("dim", ` ${stats.totalLines}L`);
-    if (stats.additions > 0 || stats.deletions > 0) {
-      if (stats.additions > 0) statsDisplay += theme.fg("success", ` +${stats.additions}`);
-      if (stats.deletions > 0) statsDisplay += theme.fg("error", ` -${stats.deletions}`);
+    let statsDisplay = "";
+    if (stats.totalLines !== undefined) {
+      statsDisplay += theme.fg("dim", ` ${stats.totalLines}L`);
     }
+    if (stats.additions > 0) statsDisplay += theme.fg("success", ` +${stats.additions}`);
+    if (stats.deletions > 0) statsDisplay += theme.fg("error", ` -${stats.deletions}`);
 
     const searchIndicator = browser.searchMode
       ? theme.fg("accent", `  /${browser.searchQuery}█`)

package/constants.ts

@@ -1,5 +1,6 @@
 export const MAX_TREE_DEPTH = 6;
 export const POLL_INTERVAL_MS = 3000;
+export const MAX_LINE_COUNT_BYTES = 256 * 1024;
 
 export const DEFAULT_VIEWER_HEIGHT = 18;
 export const DEFAULT_BROWSER_HEIGHT = 15;

package/file-tree.ts

@@ -1,15 +1,28 @@
 import { readdirSync, readFileSync, statSync } from "node:fs";
 import { basename, join, relative } from "node:path";
 
-import { MAX_TREE_DEPTH } from "./constants";
+import { MAX_LINE_COUNT_BYTES, MAX_TREE_DEPTH } from "./constants";
 import type { DiffStats, FileNode, FlatNode } from "./types";
 
-export function getFileLineCount(filePath: string): number {
+const lineCountCache = new Map<string, { mtimeMs: number; size: number; count: number }>();
+
+export function getFileLineCount(filePath: string, size: number, mtimeMs: number): number | undefined {
+  const cached = lineCountCache.get(filePath);
+  if (cached && cached.size === size && cached.mtimeMs === mtimeMs) {
+    return cached.count;
+  }
+
+  if (size > MAX_LINE_COUNT_BYTES) {
+    return undefined;
+  }
+
   try {
     const content = readFileSync(filePath, "utf-8");
-    return content.split("\n").length;
+    const count = content.split("\n").length;
+    lineCountCache.set(filePath, { mtimeMs, size, count });
+    return count;
   } catch {
-    return 0;
+    return undefined;
   }
 }
 
@@ -98,7 +111,7 @@ export function buildFileTree(
           isDirectory: false,
           gitStatus: fileGitStatus,
           agentModified: agentModified.has(fullPath),
-          lineCount: getFileLineCount(fullPath),
+          lineCount: getFileLineCount(fullPath, stat.size, stat.mtimeMs),
           diffStats: fileDiffStats,
         });
       }
@@ -110,14 +123,22 @@ export function buildFileTree(
     let totalLines = 0;
     let totalAdditions = 0;
     let totalDeletions = 0;
+    let lineCountComplete = true;
 
     for (const child of node.children) {
       if (child.isDirectory) {
-        totalLines += child.totalLines || 0;
-        totalAdditions += child.totalAdditions || 0;
-        totalDeletions += child.totalDeletions || 0;
+        totalLines += child.totalLines ?? 0;
+        totalAdditions += child.totalAdditions ?? 0;
+        totalDeletions += child.totalDeletions ?? 0;
+        if (child.lineCountComplete === false) {
+          lineCountComplete = false;
+        }
       } else {
-        totalLines += child.lineCount || 0;
+        if (child.lineCount === undefined) {
+          lineCountComplete = false;
+        } else {
+          totalLines += child.lineCount;
+        }
         if (child.diffStats) {
           totalAdditions += child.diffStats.additions;
           totalDeletions += child.diffStats.deletions;
@@ -128,27 +149,33 @@ export function buildFileTree(
     node.totalLines = totalLines;
     node.totalAdditions = totalAdditions;
     node.totalDeletions = totalDeletions;
+    node.lineCountComplete = lineCountComplete;
   } catch {}
 
   return node;
 }
 
-export function flattenTree(node: FileNode, depth = 0, isRoot = true): FlatNode[] {
+export function flattenTree(
+  node: FileNode,
+  depth = 0,
+  isRoot = true,
+  includeCollapsed = false
+): FlatNode[] {
   const result: FlatNode[] = [];
 
   // Skip the root "." node itself, just process its children
   if (isRoot && node.name === ".") {
     for (const child of node.children || []) {
-      result.push(...flattenTree(child, 0, false));
+      result.push(...flattenTree(child, 0, false, includeCollapsed));
     }
     return result;
   }
 
   result.push({ node, depth });
 
-  if (node.isDirectory && node.expanded && node.children) {
+  if (node.isDirectory && node.children && (includeCollapsed || node.expanded)) {
     for (const child of node.children) {
-      result.push(...flattenTree(child, depth + 1, false));
+      result.push(...flattenTree(child, depth + 1, false, includeCollapsed));
     }
   }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tmustier/pi-files-widget",
-  "version": "0.1.0",
+  "version": "0.1.7",
   "description": "In-terminal file browser and viewer for Pi.",
   "license": "MIT",
   "author": "Thomas Mustier",

package/types.ts

@@ -18,6 +18,7 @@ export interface FileNode {
   totalLines?: number;
   totalAdditions?: number;
   totalDeletions?: number;
+  lineCountComplete?: boolean;
 }
 
 export interface FlatNode {

package/viewer.ts

@@ -178,11 +178,11 @@ export function createViewer(
       if (state.file.diffStats.deletions > 0) {
         header += theme.fg("error", ` -${state.file.diffStats.deletions}`);
       }
-    } else if (isUntracked && state.file.lineCount) {
+    } else if (isUntracked && state.file.lineCount !== undefined) {
       header += theme.fg("success", ` +${state.file.lineCount}`);
     }
 
-    if (state.file.lineCount) {
+    if (state.file.lineCount !== undefined) {
       header += theme.fg("dim", ` ${state.file.lineCount}L`);
     }
 
@@ -245,8 +245,10 @@ export function createViewer(
 
       try {
         state.rawContent = readFileSync(file.path, "utf-8");
+        file.lineCount = state.rawContent.split("\n").length;
       } catch {
         state.rawContent = "";
+        file.lineCount = undefined;
       }
     },