@tekmidian/pai 0.5.4 → 0.5.6

package/dist/hooks/context-compression-hook.mjs

@@ -7,7 +7,7 @@ var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require
 });
 
 // src/hooks/ts/pre-compact/context-compression-hook.ts
-import { readFileSync as readFileSync3, writeFileSync as writeFileSync2 } from "fs";
+import { existsSync as existsSync3, readFileSync as readFileSync3, writeFileSync as writeFileSync2 } from "fs";
 import { basename as basename2, dirname, join as join3 } from "path";
 import { tmpdir } from "os";
 
@@ -330,6 +330,35 @@ function addWorkToSessionNote(notePath, workItems, sectionTitle) {
   writeFileSync(notePath, content);
   console.error(`Added ${workItems.length} work item(s) to: ${basename(notePath)}`);
 }
+function renameSessionNote(notePath, meaningfulName) {
+  if (!meaningfulName || !existsSync2(notePath)) {
+    return notePath;
+  }
+  const dir = join2(notePath, "..");
+  const oldFilename = basename(notePath);
+  const correctMatch = oldFilename.match(/^(\d{3,4}) - (\d{4}-\d{2}-\d{2}) - .*\.md$/);
+  const legacyMatch = oldFilename.match(/^(\d{3,4})_(\d{4}-\d{2}-\d{2})_.*\.md$/);
+  const match = correctMatch || legacyMatch;
+  if (!match) {
+    return notePath;
+  }
+  const [, noteNumber, date] = match;
+  const titleCaseName = meaningfulName.split(/[\s_-]+/).map((word) => word.charAt(0).toUpperCase() + word.slice(1).toLowerCase()).join(" ").trim();
+  const paddedNumber = noteNumber.padStart(4, "0");
+  const newFilename = `${paddedNumber} - ${date} - ${titleCaseName}.md`;
+  const newPath = join2(dir, newFilename);
+  if (newFilename === oldFilename) {
+    return notePath;
+  }
+  try {
+    renameSync(notePath, newPath);
+    console.error(`Renamed note: ${oldFilename} \u2192 ${newFilename}`);
+    return newPath;
+  } catch (error) {
+    console.error(`Could not rename note: ${error}`);
+    return notePath;
+  }
+}
 function calculateSessionTokens(jsonlPath) {
   if (!existsSync2(jsonlPath)) {
     return 0;
@@ -397,39 +426,38 @@ function ensureTodoMd(cwd) {
   }
   return todoPath;
 }
-function addTodoCheckpoint(cwd, checkpoint) {
+function updateTodoContinue(cwd, noteFilename, state, tokenDisplay) {
   const todoPath = ensureTodoMd(cwd);
   let content = readFileSync2(todoPath, "utf-8");
-  content = content.replace(/(\n---\s*)*(\n\*Last updated:.*\*\s*)+$/g, "");
-  const checkpointText = `
-**Checkpoint (${(/* @__PURE__ */ new Date()).toISOString()}):** ${checkpoint}
+  content = content.replace(/## Continue\n[\s\S]*?\n---\n+/, "");
+  const now = (/* @__PURE__ */ new Date()).toISOString();
+  const stateLines = state ? state.split("\n").filter((l) => l.trim()).slice(0, 10).map((l) => `> ${l}`).join("\n") : `> Check the latest session note for details.`;
+  const continueSection = `## Continue
+
+> **Last session:** ${noteFilename.replace(".md", "")}
+> **Paused at:** ${now}
+>
+${stateLines}
+
+---
 
 `;
-  const backlogIndex = content.indexOf("## Backlog");
-  if (backlogIndex !== -1) {
-    content = content.substring(0, backlogIndex) + checkpointText + content.substring(backlogIndex);
+  content = content.replace(/^\s+/, "");
+  const titleMatch = content.match(/^(# [^\n]+\n+)/);
+  if (titleMatch) {
+    content = titleMatch[1] + continueSection + content.substring(titleMatch[0].length);
   } else {
-    const continueIndex = content.indexOf("## Continue");
-    if (continueIndex !== -1) {
-      const afterContinue = content.indexOf("\n---", continueIndex);
-      if (afterContinue !== -1) {
-        const insertAt = afterContinue + 4;
-        content = content.substring(0, insertAt) + "\n" + checkpointText + content.substring(insertAt);
-      } else {
-        content = content.trimEnd() + "\n" + checkpointText;
-      }
-    } else {
-      content = content.trimEnd() + "\n" + checkpointText;
-    }
+    content = continueSection + content;
   }
+  content = content.replace(/(\n---\s*)*(\n\*Last updated:.*\*\s*)+$/g, "");
   content = content.trimEnd() + `
 
 ---
 
-*Last updated: ${(/* @__PURE__ */ new Date()).toISOString()}*
+*Last updated: ${now}*
 `;
   writeFileSync(todoPath, content);
-  console.error(`Checkpoint added to TODO.md`);
+  console.error("TODO.md ## Continue section updated");
 }
 
 // src/hooks/ts/pre-compact/context-compression-hook.ts
@@ -466,15 +494,19 @@ function getTranscriptStats(transcriptPath) {
     return { messageCount: 0, isLarge: false };
   }
 }
-function extractSessionState(transcriptPath, cwd) {
+function parseTranscript(transcriptPath) {
+  const data = {
+    userMessages: [],
+    summaries: [],
+    captures: [],
+    lastCompleted: "",
+    filesModified: [],
+    workItems: []
+  };
   try {
     const raw = readFileSync3(transcriptPath, "utf-8");
     const lines = raw.trim().split("\n");
-    const userMessages = [];
-    const summaries = [];
-    const captures = [];
-    let lastCompleted = "";
-    const filesModified = /* @__PURE__ */ new Set();
+    const seenSummaries = /* @__PURE__ */ new Set();
     for (const line of lines) {
       if (!line.trim()) continue;
       let entry;
@@ -485,123 +517,164 @@ function extractSessionState(transcriptPath, cwd) {
       }
       if (entry.type === "user" && entry.message?.content) {
         const text = contentToText(entry.message.content).slice(0, 300);
-        if (text) userMessages.push(text);
+        if (text) data.userMessages.push(text);
       }
       if (entry.type === "assistant" && entry.message?.content) {
         const text = contentToText(entry.message.content);
         const summaryMatch = text.match(/SUMMARY:\s*(.+?)(?:\n|$)/i);
         if (summaryMatch) {
           const s = summaryMatch[1].trim();
-          if (s.length > 5 && !summaries.includes(s)) summaries.push(s);
+          if (s.length > 5 && !data.summaries.includes(s)) {
+            data.summaries.push(s);
+            if (!seenSummaries.has(s)) {
+              seenSummaries.add(s);
+              const details = [];
+              const actionsMatch = text.match(/ACTIONS:\s*(.+?)(?=\n[A-Z]+:|$)/is);
+              if (actionsMatch) {
+                const actionLines = actionsMatch[1].split("\n").map((l) => l.replace(/^[-*•]\s*/, "").replace(/^\d+\.\s*/, "").trim()).filter((l) => l.length > 3 && l.length < 100);
+                details.push(...actionLines.slice(0, 3));
+              }
+              data.workItems.push({ title: s, details: details.length > 0 ? details : void 0, completed: true });
+            }
+          }
         }
         const captureMatch = text.match(/CAPTURE:\s*(.+?)(?:\n|$)/i);
         if (captureMatch) {
           const c = captureMatch[1].trim();
-          if (c.length > 5 && !captures.includes(c)) captures.push(c);
+          if (c.length > 5 && !data.captures.includes(c)) data.captures.push(c);
         }
         const completedMatch = text.match(/COMPLETED:\s*(.+?)(?:\n|$)/i);
         if (completedMatch) {
-          lastCompleted = completedMatch[1].trim().replace(/\*+/g, "");
+          data.lastCompleted = completedMatch[1].trim().replace(/\*+/g, "");
+          if (data.workItems.length === 0 && !seenSummaries.has(data.lastCompleted) && data.lastCompleted.length > 5) {
+            seenSummaries.add(data.lastCompleted);
+            data.workItems.push({ title: data.lastCompleted, completed: true });
+          }
         }
-      }
-      if (entry.type === "assistant" && entry.message?.content && Array.isArray(entry.message.content)) {
-        for (const block of entry.message.content) {
-          if (block.type === "tool_use") {
-            const tool = block.name;
-            if ((tool === "Edit" || tool === "Write") && block.input?.file_path) {
-              filesModified.add(block.input.file_path);
+        if (Array.isArray(entry.message.content)) {
+          for (const block of entry.message.content) {
+            if (block.type === "tool_use") {
+              const tool = block.name;
+              if ((tool === "Edit" || tool === "Write") && block.input?.file_path) {
+                if (!data.filesModified.includes(block.input.file_path)) {
+                  data.filesModified.push(block.input.file_path);
+                }
+              }
             }
           }
         }
       }
     }
-    const parts = [];
-    if (cwd) {
-      parts.push(`Working directory: ${cwd}`);
-    }
-    const recentUser = userMessages.slice(-3);
-    if (recentUser.length > 0) {
-      parts.push("\nRecent user requests:");
-      for (const msg of recentUser) {
-        const firstLine = msg.split("\n")[0].slice(0, 200);
-        parts.push(`- ${firstLine}`);
-      }
-    }
-    const recentSummaries = summaries.slice(-3);
-    if (recentSummaries.length > 0) {
-      parts.push("\nWork summaries:");
-      for (const s of recentSummaries) {
-        parts.push(`- ${s.slice(0, 150)}`);
-      }
-    }
-    const recentCaptures = captures.slice(-5);
-    if (recentCaptures.length > 0) {
-      parts.push("\nCaptured context:");
-      for (const c of recentCaptures) {
-        parts.push(`- ${c.slice(0, 150)}`);
-      }
+  } catch (err) {
+    console.error(`parseTranscript error: ${err}`);
+  }
+  return data;
+}
+function formatSessionState(data, cwd) {
+  const parts = [];
+  if (cwd) parts.push(`Working directory: ${cwd}`);
+  const recentUser = data.userMessages.slice(-3);
+  if (recentUser.length > 0) {
+    parts.push("\nRecent user requests:");
+    for (const msg of recentUser) {
+      parts.push(`- ${msg.split("\n")[0].slice(0, 200)}`);
     }
-    const files = Array.from(filesModified).slice(-10);
-    if (files.length > 0) {
-      parts.push("\nFiles modified this session:");
-      for (const f of files) {
-        parts.push(`- ${f}`);
+  }
+  const recentSummaries = data.summaries.slice(-3);
+  if (recentSummaries.length > 0) {
+    parts.push("\nWork summaries:");
+    for (const s of recentSummaries) parts.push(`- ${s.slice(0, 150)}`);
+  }
+  const recentCaptures = data.captures.slice(-5);
+  if (recentCaptures.length > 0) {
+    parts.push("\nCaptured context:");
+    for (const c of recentCaptures) parts.push(`- ${c.slice(0, 150)}`);
+  }
+  const files = data.filesModified.slice(-10);
+  if (files.length > 0) {
+    parts.push("\nFiles modified this session:");
+    for (const f of files) parts.push(`- ${f}`);
+  }
+  if (data.lastCompleted) {
+    parts.push(`
+Last completed: ${data.lastCompleted.slice(0, 150)}`);
+  }
+  const result = parts.join("\n");
+  return result.length > 50 ? result : null;
+}
+function deriveTitle(data) {
+  let title = "";
+  if (data.workItems.length > 0) {
+    title = data.workItems[data.workItems.length - 1].title;
+  } else if (data.summaries.length > 0) {
+    title = data.summaries[data.summaries.length - 1];
+  } else if (data.lastCompleted && data.lastCompleted.length > 5) {
+    title = data.lastCompleted;
+  } else if (data.userMessages.length > 0) {
+    for (let i = data.userMessages.length - 1; i >= 0; i--) {
+      const msg = data.userMessages[i].split("\n")[0].trim();
+      if (msg.length > 10 && msg.length < 80 && !msg.toLowerCase().startsWith("yes") && !msg.toLowerCase().startsWith("ok")) {
+        title = msg;
+        break;
       }
     }
-    if (lastCompleted) {
-      parts.push(`
-Last completed: ${lastCompleted.slice(0, 150)}`);
-    }
-    const result = parts.join("\n");
-    return result.length > 50 ? result : null;
-  } catch (err) {
-    console.error(`extractSessionState error: ${err}`);
-    return null;
   }
+  if (!title && data.filesModified.length > 0) {
+    const basenames = data.filesModified.slice(-5).map((f) => {
+      const b = basename2(f);
+      return b.replace(/\.[^.]+$/, "");
+    });
+    const unique = [...new Set(basenames)];
+    title = unique.length <= 3 ? `Updated ${unique.join(", ")}` : `Modified ${data.filesModified.length} files`;
+  }
+  return title.replace(/[^\w\s-]/g, " ").replace(/\s+/g, " ").trim().substring(0, 60);
 }
-function extractWorkFromTranscript(transcriptPath) {
+var CUMULATIVE_STATE_FILE = ".compact-state.json";
+function loadCumulativeState(notesDir) {
   try {
-    const raw = readFileSync3(transcriptPath, "utf-8");
-    const lines = raw.trim().split("\n");
-    const workItems = [];
-    const seenSummaries = /* @__PURE__ */ new Set();
-    for (const line of lines) {
-      if (!line.trim()) continue;
-      let entry;
-      try {
-        entry = JSON.parse(line);
-      } catch {
-        continue;
-      }
-      if (entry.type === "assistant" && entry.message?.content) {
-        const text = contentToText(entry.message.content);
-        const summaryMatch = text.match(/SUMMARY:\s*(.+?)(?:\n|$)/i);
-        if (summaryMatch) {
-          const summary = summaryMatch[1].trim();
-          if (summary && !seenSummaries.has(summary) && summary.length > 5) {
-            seenSummaries.add(summary);
-            const details = [];
-            const actionsMatch = text.match(/ACTIONS:\s*(.+?)(?=\n[A-Z]+:|$)/is);
-            if (actionsMatch) {
-              const actionLines = actionsMatch[1].split("\n").map((l) => l.replace(/^[-*•]\s*/, "").replace(/^\d+\.\s*/, "").trim()).filter((l) => l.length > 3 && l.length < 100);
-              details.push(...actionLines.slice(0, 3));
-            }
-            workItems.push({ title: summary, details: details.length > 0 ? details : void 0, completed: true });
-          }
-        }
-        const completedMatch = text.match(/COMPLETED:\s*(.+?)(?:\n|$)/i);
-        if (completedMatch && workItems.length === 0) {
-          const completed = completedMatch[1].trim().replace(/\*+/g, "");
-          if (completed && !seenSummaries.has(completed) && completed.length > 5) {
-            seenSummaries.add(completed);
-            workItems.push({ title: completed, completed: true });
-          }
-        }
-      }
-    }
-    return workItems;
+    const filePath = join3(notesDir, CUMULATIVE_STATE_FILE);
+    if (!existsSync3(filePath)) return null;
+    const raw = JSON.parse(readFileSync3(filePath, "utf-8"));
+    return {
+      userMessages: raw.userMessages || [],
+      summaries: raw.summaries || [],
+      captures: raw.captures || [],
+      lastCompleted: raw.lastCompleted || "",
+      filesModified: raw.filesModified || [],
+      workItems: raw.workItems || []
+    };
   } catch {
-    return [];
+    return null;
+  }
+}
+function mergeTranscriptData(accumulated, current) {
+  if (!accumulated) return current;
+  const mergeArrays = (a, b) => {
+    const seen = new Set(a);
+    return [...a, ...b.filter((x) => !seen.has(x))];
+  };
+  const seenTitles = new Set(accumulated.workItems.map((w) => w.title));
+  const newWorkItems = current.workItems.filter((w) => !seenTitles.has(w.title));
+  return {
+    userMessages: mergeArrays(accumulated.userMessages, current.userMessages).slice(-20),
+    summaries: mergeArrays(accumulated.summaries, current.summaries),
+    captures: mergeArrays(accumulated.captures, current.captures),
+    lastCompleted: current.lastCompleted || accumulated.lastCompleted,
+    filesModified: mergeArrays(accumulated.filesModified, current.filesModified),
+    workItems: [...accumulated.workItems, ...newWorkItems]
+  };
+}
+function saveCumulativeState(notesDir, data, notePath) {
+  try {
+    const filePath = join3(notesDir, CUMULATIVE_STATE_FILE);
+    writeFileSync2(filePath, JSON.stringify({
+      ...data,
+      notePath,
+      lastUpdated: (/* @__PURE__ */ new Date()).toISOString()
+    }, null, 2));
+    console.error(`Cumulative state saved (${data.workItems.length} work items, ${data.filesModified.length} files)`);
+  } catch (err) {
+    console.error(`Failed to save cumulative state: ${err}`);
   }
 }
 async function main() {
@@ -629,10 +702,22 @@ async function main() {
     const stats = getTranscriptStats(hookInput.transcript_path);
     tokenCount = calculateSessionTokens(hookInput.transcript_path);
     const tokenDisplay = tokenCount > 1e3 ? `${Math.round(tokenCount / 1e3)}k` : String(tokenCount);
-    const state = extractSessionState(hookInput.transcript_path, hookInput.cwd);
+    const data = parseTranscript(hookInput.transcript_path);
+    let notesInfo;
+    try {
+      notesInfo = hookInput.cwd ? findNotesDir(hookInput.cwd) : { path: join3(dirname(hookInput.transcript_path), "Notes"), isLocal: false };
+    } catch {
+      notesInfo = { path: join3(dirname(hookInput.transcript_path), "Notes"), isLocal: false };
+    }
+    const accumulated = loadCumulativeState(notesInfo.path);
+    const merged = mergeTranscriptData(accumulated, data);
+    const state = formatSessionState(merged, hookInput.cwd);
+    if (accumulated) {
+      console.error(`Loaded cumulative state: ${accumulated.workItems.length} work items, ${accumulated.filesModified.length} files from previous compaction(s)`);
+    }
+    let notePath = null;
     try {
-      const notesInfo = hookInput.cwd ? findNotesDir(hookInput.cwd) : { path: join3(dirname(hookInput.transcript_path), "Notes"), isLocal: false };
-      let notePath = getCurrentNotePath(notesInfo.path);
+      notePath = getCurrentNotePath(notesInfo.path);
       if (!notePath) {
         console.error("No session note found \u2014 creating one for checkpoint");
         notePath = createSessionNote(notesInfo.path, "Recovered Session");
@@ -650,30 +735,53 @@ async function main() {
 
 ${state}` : `Context compression triggered at ~${tokenDisplay} tokens with ${stats.messageCount} messages.`;
       appendCheckpoint(notePath, checkpointBody);
-      const workItems = extractWorkFromTranscript(hookInput.transcript_path);
-      if (workItems.length > 0) {
-        addWorkToSessionNote(notePath, workItems, `Pre-Compact (~${tokenDisplay} tokens)`);
-        console.error(`Added ${workItems.length} work item(s) to session note`);
+      if (merged.workItems.length > 0) {
+        addWorkToSessionNote(notePath, merged.workItems, `Pre-Compact (~${tokenDisplay} tokens)`);
+        console.error(`Added ${merged.workItems.length} work item(s) to session note`);
+      }
+      const title = deriveTitle(merged);
+      if (title) {
+        const newPath = renameSessionNote(notePath, title);
+        if (newPath !== notePath) {
+          try {
+            let noteContent = readFileSync3(newPath, "utf-8");
+            noteContent = noteContent.replace(
+              /^(# Session \d+:)\s*.*$/m,
+              `$1 ${title}`
+            );
+            writeFileSync2(newPath, noteContent);
+            console.error(`Updated note H1 to match rename`);
+          } catch {
+          }
+          notePath = newPath;
+        }
       }
       console.error(`Rich checkpoint saved: ${basename2(notePath)}`);
     } catch (noteError) {
       console.error(`Could not save checkpoint: ${noteError}`);
     }
-    if (hookInput.cwd && state) {
+    saveCumulativeState(notesInfo.path, merged, notePath);
+    if (hookInput.cwd && notePath) {
       try {
-        addTodoCheckpoint(hookInput.cwd, `Pre-compact checkpoint (~${tokenDisplay} tokens):
-${state}`);
-        console.error("TODO.md checkpoint added");
+        const noteFilename = basename2(notePath);
+        updateTodoContinue(hookInput.cwd, noteFilename, state, tokenDisplay);
+        console.error("TODO.md ## Continue section updated");
       } catch (todoError) {
         console.error(`Could not update TODO.md: ${todoError}`);
       }
     }
-    if (state && hookInput.session_id) {
+    if (hookInput.session_id) {
+      const stateText = state || `Working directory: ${hookInput.cwd || "unknown"}`;
+      const noteInfo = notePath ? `
+SESSION NOTE: ${notePath}
+If this note still has a generic title (e.g. "New Session", "Context Compression"),
+rename it based on actual work done and add a rich summary.` : "";
       const injection = [
         "<system-reminder>",
         `SESSION STATE RECOVERED AFTER COMPACTION (${compactType}, ~${tokenDisplay} tokens)`,
         "",
-        state,
+        stateText,
+        noteInfo,
         "",
         "IMPORTANT: This session state was captured before context compaction.",
         "Use it to maintain continuity. Continue the conversation from where",