pi-goosedump 0.1.3 → 0.1.4

package/index.ts

@@ -17,20 +17,17 @@ import { join } from 'node:path';
 import { Key, matchesKey, truncateToWidth, visibleWidth } from '@earendil-works/pi-tui';
 import { Type } from '@sinclair/typebox';
 
-const GOOSEDUMP_VERSION = [0, 1, 0] as const;
+const GOOSEDUMP_VERSION = [0, 1, 4] as const;
 
 interface GoosedumpListing {
   id: string;
-  created: string;
-  modified: string;
-  entryCount: number;
 }
 
 interface GoosedumpMessage {
   id: string;
   role: string;
   content: string;
-  timestamp: string;
+  score?: number;
 }
 
 interface GoosedumpContextResult {
@@ -91,17 +88,14 @@ function hasMinimumVersion(version: string, minimum: readonly [number, number, n
 
 function parseListings(output: string): GoosedumpListing[] {
   const listings: GoosedumpListing[] = [];
-  // Format: <id> <created> <entryCount> entries <modified>
-  const regex = /^(\S+)\s+(\S+\s+\S+)\s+(\d+)\s+entries?\s+(.+)$/gm;
-  let match: RegExpExecArray | null;
-
-  while ((match = regex.exec(output)) !== null) {
-    listings.push({
-      id: match[1],
-      created: match[2],
-      entryCount: parseInt(match[3], 10),
-      modified: match[4],
-    });
+  const lines = output.split('\n');
+
+  for (const line of lines) {
+    if (line === 'Sessions:' || line.trim() === '') continue;
+    const match = line.match(/^  (\S+)/);
+    if (match) {
+      listings.push({ id: match[1] });
+    }
   }
 
   return listings;
@@ -109,41 +103,46 @@ function parseListings(output: string): GoosedumpListing[] {
 
 function parseMessages(output: string): GoosedumpMessage[] {
   const messages: GoosedumpMessage[] = [];
-  // Format from goosedump plain output: [id] role: content
   const lines = output.split('\n');
-  let currentId = '';
-  let currentRole = '';
+  let currentMsg: GoosedumpMessage | null = null;
   let contentLines: string[] = [];
-  let currentTimestamp = '';
 
   for (const line of lines) {
-    const headerMatch = line.match(/^\[([^\]]+)\]\s+(user|assistant|tool|system)\s*(\([^)]+\))?:/);
-    if (headerMatch) {
-      if (currentId) {
-        messages.push({
-          id: currentId,
-          role: currentRole,
-          content: contentLines.join('\n').trim(),
-          timestamp: currentTimestamp,
-        });
+    if (line.startsWith('results for ')) continue;
+
+    const entryMatch = line.match(/^entryId:\s+(\S+)\s+\((.+)\)$/);
+    if (entryMatch) {
+      if (currentMsg) {
+        currentMsg.content = contentLines.join('\n').trim();
+        messages.push(currentMsg);
+      }
+      currentMsg = {
+        id: entryMatch[1],
+        role: entryMatch[2],
+        content: '',
+      };
+      contentLines = [];
+      continue;
+    }
+
+    if (currentMsg) {
+      const indentedMatch = line.match(/^  (.+)$/);
+      if (indentedMatch) {
+        const content = indentedMatch[1];
+        if (content.startsWith('score: ')) {
+          const scoreVal = parseFloat(content.slice(7));
+          if (!isNaN(scoreVal)) currentMsg.score = scoreVal;
+          continue;
+        }
+        if (/^\.\.\.\(\d+ lines (?:above|below)\)$/.test(content)) continue;
+        contentLines.push(content);
       }
-      currentId = headerMatch[1];
-      currentRole = headerMatch[2];
-      currentTimestamp = headerMatch[3]?.replace(/[()]/g, '') ?? '';
-      contentLines = [line.slice(headerMatch[0].length).trim()];
-    } else if (currentId) {
-      if (line.startsWith('---') && messages.length > 0) continue;
-      contentLines.push(line);
     }
   }
 
-  if (currentId) {
-    messages.push({
-      id: currentId,
-      role: currentRole,
-      content: contentLines.join('\n').trim(),
-      timestamp: currentTimestamp,
-    });
+  if (currentMsg) {
+    currentMsg.content = contentLines.join('\n').trim();
+    messages.push(currentMsg);
   }
 
   return messages;
@@ -215,7 +214,7 @@ function goosedumpContext(
 
 function goosedumpExpand(
   contextId: string,
-  entryIds: number[],
+  entryIds: string[],
   scope: string = 'lineage',
 ): GoosedumpMessage[] {
   const ids = entryIds.join(',');
@@ -472,7 +471,7 @@ export function createGoosedumpIntegration(
               }),
             ),
             ids: Type.Optional(
-              Type.Array(Type.Integer(), {
+              Type.Array(Type.String(), {
                 description: 'Entry IDs to expand (show full content)',
               }),
             ),
@@ -510,9 +509,7 @@ export function createGoosedumpIntegration(
 
                   const lines = ['Available sessions:', ''];
                   for (const listing of listings) {
-                    lines.push(
-                      `  ${listing.id} — ${listing.entryCount} entries (modified ${listing.modified})`,
-                    );
+                    lines.push(`  ${listing.id}`);
                   }
 
                   return {
@@ -755,9 +752,8 @@ export function createGoosedumpIntegration(
                     const listing = listings[i];
                     const isSelected = i === selectedIndex;
                     const cursor = isSelected ? accent('▶') : ' ';
-                    const entryInfo = dim(`${listing.entryCount} entries`);
-                    const idText = isSelected ? accent(`[${listing.id}]`) : text(`[${listing.id}]`);
-                    const line = `  ${cursor} ${idText} ${entryInfo}`;
+                    const idText = isSelected ? accent(listing.id) : text(listing.id);
+                    const line = `  ${cursor} ${idText}`;
                     lines.push(makeRow(truncateToWidth(line, innerW), innerW, border));
                   }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-goosedump",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "Coding agent context data browser plugin for pi",
   "keywords": [
     "goosedump",
@@ -29,7 +29,7 @@
   },
   "dependencies": {
     "@earendil-works/pi-tui": "^0.78.0",
-    "@jarkkojs/goosedump": "^0.1.2",
+    "@jarkkojs/goosedump": "^0.1.4",
     "@sinclair/typebox": "^0.34.49"
   },
   "devDependencies": {