@tekmidian/pai 0.5.3 → 0.5.5

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

@@ -150,6 +150,32 @@ async function sendNtfyNotification(message, retries = 2) {
   console.error("ntfy.sh notification failed after all retries");
   return false;
 }
+function getMonthDir(notesDir) {
+  const now = /* @__PURE__ */ new Date();
+  const year = String(now.getFullYear());
+  const month = String(now.getMonth() + 1).padStart(2, "0");
+  const monthDir = join2(notesDir, year, month);
+  if (!existsSync2(monthDir)) {
+    mkdirSync(monthDir, { recursive: true });
+  }
+  return monthDir;
+}
+function getNextNoteNumber(notesDir) {
+  const monthDir = getMonthDir(notesDir);
+  const files = readdirSync(monthDir).filter((f) => f.match(/^\d{3,4}[\s_-]/)).filter((f) => f.endsWith(".md")).sort();
+  if (files.length === 0) {
+    return "0001";
+  }
+  let maxNumber = 0;
+  for (const file of files) {
+    const digitMatch = file.match(/^(\d+)/);
+    if (digitMatch) {
+      const num = parseInt(digitMatch[1], 10);
+      if (num > maxNumber) maxNumber = num;
+    }
+  }
+  return String(maxNumber + 1).padStart(4, "0");
+}
 function getCurrentNotePath(notesDir) {
   if (!existsSync2(notesDir)) {
     return null;
@@ -178,10 +204,77 @@ function getCurrentNotePath(notesDir) {
   if (prevFound) return prevFound;
   return findLatestIn(notesDir);
 }
+function createSessionNote(notesDir, description) {
+  const noteNumber = getNextNoteNumber(notesDir);
+  const date = (/* @__PURE__ */ new Date()).toISOString().split("T")[0];
+  const safeDescription = "New Session";
+  const monthDir = getMonthDir(notesDir);
+  const filename = `${noteNumber} - ${date} - ${safeDescription}.md`;
+  const filepath = join2(monthDir, filename);
+  const content = `# Session ${noteNumber}: ${description}
+
+**Date:** ${date}
+**Status:** In Progress
+
+---
+
+## Work Done
+
+<!-- PAI will add completed work here during session -->
+
+---
+
+## Next Steps
+
+<!-- To be filled at session end -->
+
+---
+
+**Tags:** #Session
+`;
+  writeFileSync(filepath, content);
+  console.error(`Created session note: ${filename}`);
+  return filepath;
+}
 function appendCheckpoint(notePath, checkpoint) {
   if (!existsSync2(notePath)) {
-    console.error(`Note file not found: ${notePath}`);
-    return;
+    console.error(`Note file not found, recreating: ${notePath}`);
+    try {
+      const parentDir = join2(notePath, "..");
+      if (!existsSync2(parentDir)) {
+        mkdirSync(parentDir, { recursive: true });
+      }
+      const noteFilename = basename(notePath);
+      const numberMatch = noteFilename.match(/^(\d+)/);
+      const noteNumber = numberMatch ? numberMatch[1] : "0000";
+      const date = (/* @__PURE__ */ new Date()).toISOString().split("T")[0];
+      const content2 = `# Session ${noteNumber}: Recovered
+
+**Date:** ${date}
+**Status:** In Progress
+
+---
+
+## Work Done
+
+<!-- PAI will add completed work here during session -->
+
+---
+
+## Next Steps
+
+<!-- To be filled at session end -->
+
+---
+
+**Tags:** #Session
+`;
+      writeFileSync(notePath, content2);
+      console.error(`Recreated session note: ${noteFilename}`);
+    } catch (err) {
+      console.error(`Failed to recreate note: ${err}`);
+      return;
+    }
   }
   const content = readFileSync2(notePath, "utf-8");
   const timestamp = (/* @__PURE__ */ new Date()).toISOString();
@@ -237,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;
@@ -304,26 +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 backlogIndex = content.indexOf("## Backlog");
-  if (backlogIndex !== -1) {
-    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}
+
+---
 
 `;
-    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 {
+    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
@@ -360,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;
@@ -379,124 +517,117 @@ 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)}`);
-      }
-    }
-    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}`);
-      }
-    }
-    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;
+    console.error(`parseTranscript error: ${err}`);
   }
+  return data;
 }
-function extractWorkFromTranscript(transcriptPath) {
-  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 });
-          }
-        }
+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 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;
       }
     }
-    return workItems;
-  } catch {
-    return [];
   }
+  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);
 }
 async function main() {
   let hookInput = null;
@@ -523,30 +654,59 @@ 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);
+    const state = formatSessionState(data, hookInput.cwd);
+    let notePath = null;
     try {
       const notesInfo = hookInput.cwd ? findNotesDir(hookInput.cwd) : { path: join3(dirname(hookInput.transcript_path), "Notes"), isLocal: false };
-      const currentNotePath = getCurrentNotePath(notesInfo.path);
-      if (currentNotePath) {
-        const checkpointBody = state ? `Context compression triggered at ~${tokenDisplay} tokens with ${stats.messageCount} messages.
+      notePath = getCurrentNotePath(notesInfo.path);
+      if (!notePath) {
+        console.error("No session note found \u2014 creating one for checkpoint");
+        notePath = createSessionNote(notesInfo.path, "Recovered Session");
+      } else {
+        try {
+          const noteContent = readFileSync3(notePath, "utf-8");
+          if (noteContent.includes("**Status:** Completed") || noteContent.includes("**Completed:**")) {
+            console.error(`Latest note is completed (${basename2(notePath)}) \u2014 creating new one`);
+            notePath = createSessionNote(notesInfo.path, "Continued Session");
+          }
+        } catch {
+        }
+      }
+      const checkpointBody = state ? `Context compression triggered at ~${tokenDisplay} tokens with ${stats.messageCount} messages.
 
 ${state}` : `Context compression triggered at ~${tokenDisplay} tokens with ${stats.messageCount} messages.`;
-        appendCheckpoint(currentNotePath, checkpointBody);
-        const workItems = extractWorkFromTranscript(hookInput.transcript_path);
-        if (workItems.length > 0) {
-          addWorkToSessionNote(currentNotePath, workItems, `Pre-Compact (~${tokenDisplay} tokens)`);
-          console.error(`Added ${workItems.length} work item(s) to session note`);
+      appendCheckpoint(notePath, checkpointBody);
+      if (data.workItems.length > 0) {
+        addWorkToSessionNote(notePath, data.workItems, `Pre-Compact (~${tokenDisplay} tokens)`);
+        console.error(`Added ${data.workItems.length} work item(s) to session note`);
+      }
+      const title = deriveTitle(data);
+      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(currentNotePath)}`);
       }
+      console.error(`Rich checkpoint saved: ${basename2(notePath)}`);
     } catch (noteError) {
       console.error(`Could not save checkpoint: ${noteError}`);
     }
-    if (hookInput.cwd && state) {
+    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}`);
       }