pi-goosedump 0.6.4 → 0.7.1

package/index.ts

@@ -49,6 +49,7 @@ interface GoosedumpListing {
   provider: string;
   nativeId: string;
   cwd: string;
+  parent: string | null;
   label: string | null;
   path: string | null;
 }
@@ -273,6 +274,7 @@ function goosedumpList(provider = 'pi'): GoosedumpListing[] {
       provider: entry.provider,
       nativeId: entry.native_id,
       cwd: entry.cwd,
+      parent: entry.parent,
       label: entry.label ?? null,
       // Only pi sessions map back to a local file we can resume.
       path: provider === 'pi' ? piSessionFilePath(entry.native_id) : null,
@@ -662,8 +664,46 @@ async function runSessionBrowser(
 
   const OVERLAY_WIDTH = 72;
 
+  // Build tree from flat listings using parent relationships.
+  const listingIds = new Set(listings.map((l) => l.id));
+  const roots = listings.filter((l) => l.parent === null || !listingIds.has(l.parent));
+  const childrenMap = new Map<string, GoosedumpListing[]>();
+  for (const l of listings) {
+    if (l.parent !== null) {
+      const siblings = childrenMap.get(l.parent) ?? [];
+      siblings.push(l);
+      childrenMap.set(l.parent, siblings);
+    }
+  }
+
+  function truncateCwd(cwd: string, maxLen: number): string {
+    if (cwd.length <= maxLen) return cwd;
+    const head = Math.floor(maxLen / 2) + 2;
+    const tail = maxLen - head - 3;
+    if (tail < 4) return cwd.slice(0, maxLen - 3) + '...';
+    return cwd.slice(0, head) + '...' + cwd.slice(cwd.length - tail);
+  }
+
+  function flattenTree(
+    nodes: GoosedumpListing[],
+    expanded: Set<string>,
+    depth: number,
+  ): { listing: GoosedumpListing; depth: number; hasChildren: boolean }[] {
+    const result: { listing: GoosedumpListing; depth: number; hasChildren: boolean }[] = [];
+    for (const node of nodes) {
+      const children = childrenMap.get(node.id) ?? [];
+      result.push({ listing: node, depth, hasChildren: children.length > 0 });
+      if (children.length > 0 && expanded.has(node.id)) {
+        result.push(...flattenTree(children, expanded, depth + 1));
+      }
+    }
+    return result;
+  }
+
   await ctx.ui.custom<void>(
     (tui, theme, _kb, done) => {
+      let expandedNodes = new Set<string>();
+      let displayList = flattenTree(roots, expandedNodes, 0);
       let selectedIndex = 0;
       let listScroll = 0;
       let mode: 'list' | 'compact' = 'list';
@@ -675,6 +715,10 @@ async function runSessionBrowser(
       const LIST_VIEWPORT = 20;
       const wrapWidth = OVERLAY_WIDTH - 4;
 
+      function rebuildDisplayList(): void {
+        displayList = flattenTree(roots, expandedNodes, 0);
+      }
+
       return {
         render(width: number): string[] {
           const innerW = width - 4;
@@ -723,15 +767,29 @@ async function runSessionBrowser(
           lines.push(makeRow('', innerW, border));
 
           // Session list
-          const total = listings.length;
-          const window = listings.slice(listScroll, listScroll + LIST_VIEWPORT);
+          const total = displayList.length;
+          const window = displayList.slice(listScroll, listScroll + LIST_VIEWPORT);
           for (let i = 0; i < window.length; i++) {
             const absolute = listScroll + i;
-            const listing = window[i];
+            const entry = window[i];
             const isSelected = absolute === selectedIndex;
+
             const cursor = isSelected ? accent('▶') : ' ';
-            const idText = isSelected ? accent(listing.id) : text(listing.id);
-            const line = `  ${cursor} ${idText}`;
+            const indent = ' '.repeat(entry.depth * 2);
+            const toggle = entry.hasChildren
+              ? expandedNodes.has(entry.listing.id)
+                ? muted('−')
+                : muted('+')
+              : ' ';
+            const idText = isSelected ? accent(entry.listing.id) : text(entry.listing.id);
+
+            // Prefix: "  ▶" (3) + indent + toggle + " " + id (8) + " "
+            const prefixLen = 3 + entry.depth * 2 + 1 + 1 + 8 + 1;
+            const cwdMax = Math.max(8, innerW - prefixLen);
+            const cwdTruncated = truncateCwd(entry.listing.cwd, cwdMax);
+            const cwdText = isSelected ? accent(cwdTruncated) : muted(cwdTruncated);
+
+            const line = `  ${cursor}${indent}${toggle} ${idText} ${cwdText}`;
             lines.push(makeRow(line, innerW, border));
           }
 
@@ -744,7 +802,7 @@ async function runSessionBrowser(
           lines.push(makeRow('', innerW, border));
           lines.push(
             makeRow(
-              dim('↑↓ navigate  enter compact  shift+enter resume  esc close'),
+              dim('↑↓ navigate  space/+/− fold  enter compact  shift+enter resume  esc close'),
               innerW,
               border,
             ),
@@ -800,7 +858,7 @@ async function runSessionBrowser(
           }
 
           if (matchesKey(data, Key.down)) {
-            selectedIndex = Math.min(listings.length - 1, selectedIndex + 1);
+            selectedIndex = Math.min(displayList.length - 1, selectedIndex + 1);
             if (selectedIndex >= listScroll + LIST_VIEWPORT) {
               listScroll = selectedIndex - LIST_VIEWPORT + 1;
             }
@@ -808,14 +866,39 @@ async function runSessionBrowser(
             return;
           }
 
+          // Expand/collapse: space, +, or - key
+          const entry = displayList[selectedIndex];
+          if (entry && entry.hasChildren && (data === ' ' || data === '+' || data === '-')) {
+            const id = entry.listing.id;
+            if (data === '+') {
+              expandedNodes.add(id);
+            } else if (data === '-') {
+              expandedNodes.delete(id);
+            } else {
+              // space = toggle
+              if (expandedNodes.has(id)) {
+                expandedNodes.delete(id);
+              } else {
+                expandedNodes.add(id);
+              }
+            }
+            rebuildDisplayList();
+            // Keep selection on the same entry
+            const newIndex = displayList.findIndex((d) => d.listing.id === id);
+            if (newIndex >= 0) selectedIndex = newIndex;
+            if (selectedIndex < listScroll) listScroll = selectedIndex;
+            tui.requestRender();
+            return;
+          }
+
           if (matchesKey(data, Key.shift(Key.enter))) {
-            resumePath = listings[selectedIndex]?.path;
+            resumePath = displayList[selectedIndex]?.listing.path;
             done(undefined);
             return;
           }
 
           if (matchesKey(data, Key.enter) || matchesKey(data, Key.return)) {
-            const listing = listings[selectedIndex];
+            const listing = displayList[selectedIndex]?.listing;
             if (!listing) return;
             compactTitle = listing.id;
             compactScroll = 0;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-goosedump",
-  "version": "0.6.4",
+  "version": "0.7.1",
   "description": "Coding agent context data browser plugin for pi",
   "keywords": [
     "goosedump",