@tekmidian/pai 0.5.3 → 0.5.4

package/src/hooks/ts/lib/project-utils.ts

@@ -437,8 +437,24 @@ export function createSessionNote(notesDir: string, description: string): string
  */
 export function appendCheckpoint(notePath: string, checkpoint: string): void {
   if (!existsSync(notePath)) {
-    console.error(`Note file not found: ${notePath}`);
-    return;
+    // Note vanished (cloud sync, cleanup, etc.) — recreate it
+    console.error(`Note file not found, recreating: ${notePath}`);
+    try {
+      const parentDir = join(notePath, '..');
+      if (!existsSync(parentDir)) {
+        mkdirSync(parentDir, { recursive: true });
+      }
+      const noteFilename = basename(notePath);
+      const numberMatch = noteFilename.match(/^(\d+)/);
+      const noteNumber = numberMatch ? numberMatch[1] : '0000';
+      const date = new Date().toISOString().split('T')[0];
+      const content = `# Session ${noteNumber}: Recovered\n\n**Date:** ${date}\n**Status:** In Progress\n\n---\n\n## Work Done\n\n<!-- PAI will add completed work here during session -->\n\n---\n\n## Next Steps\n\n<!-- To be filled at session end -->\n\n---\n\n**Tags:** #Session\n`;
+      writeFileSync(notePath, content);
+      console.error(`Recreated session note: ${noteFilename}`);
+    } catch (err) {
+      console.error(`Failed to recreate note: ${err}`);
+      return;
+    }
   }
 
   const content = readFileSync(notePath, 'utf-8');
@@ -891,6 +907,7 @@ ${sessionSummary ? `**Session Summary:** ${sessionSummary}\n\n` : ''}${backlogSe
 /**
  * Add a checkpoint entry to TODO.md (without replacing tasks)
  * Ensures only ONE timestamp line at the end
+ * Works regardless of TODO.md structure — appends if no known section found
  */
 export function addTodoCheckpoint(cwd: string, checkpoint: string): void {
   const todoPath = ensureTodoMd(cwd);
@@ -899,11 +916,28 @@ export function addTodoCheckpoint(cwd: string, checkpoint: string): void {
   // Remove ALL existing timestamp lines and trailing separators
   content = content.replace(/(\n---\s*)*(\n\*Last updated:.*\*\s*)+$/g, '');
 
-  // Add checkpoint before Backlog section
+  const checkpointText = `\n**Checkpoint (${new Date().toISOString()}):** ${checkpoint}\n\n`;
+
+  // Try to insert before Backlog section
   const backlogIndex = content.indexOf('## Backlog');
   if (backlogIndex !== -1) {
-    const checkpointText = `\n**Checkpoint (${new Date().toISOString()}):** ${checkpoint}\n\n`;
     content = content.substring(0, backlogIndex) + checkpointText + content.substring(backlogIndex);
+  } else {
+    // No Backlog section — try before Continue section, or just append
+    const continueIndex = content.indexOf('## Continue');
+    if (continueIndex !== -1) {
+      // Insert after the Continue section (find the next ## or ---)
+      const afterContinue = content.indexOf('\n---', continueIndex);
+      if (afterContinue !== -1) {
+        const insertAt = afterContinue + 4; // after \n---
+        content = content.substring(0, insertAt) + '\n' + checkpointText + content.substring(insertAt);
+      } else {
+        content = content.trimEnd() + '\n' + checkpointText;
+      }
+    } else {
+      // No known section — just append before the end
+      content = content.trimEnd() + '\n' + checkpointText;
+    }
   }
 
   // Add exactly ONE timestamp at the end

package/src/hooks/ts/pre-compact/context-compression-hook.ts

@@ -18,6 +18,7 @@ import { tmpdir } from 'os';
 import {
   sendNtfyNotification,
   getCurrentNotePath,
+  createSessionNote,
   appendCheckpoint,
   addWorkToSessionNote,
   findNotesDir,
@@ -297,24 +298,37 @@ async function main() {
       const notesInfo = hookInput.cwd
         ? findNotesDir(hookInput.cwd)
         : { path: join(dirname(hookInput.transcript_path), 'Notes'), isLocal: false };
-      const currentNotePath = getCurrentNotePath(notesInfo.path);
-
-      if (currentNotePath) {
-        // 1. Write rich checkpoint with full session state
-        const checkpointBody = state
-          ? `Context compression triggered at ~${tokenDisplay} tokens with ${stats.messageCount} messages.\n\n${state}`
-          : `Context compression triggered at ~${tokenDisplay} tokens with ${stats.messageCount} messages.`;
-        appendCheckpoint(currentNotePath, checkpointBody);
-
-        // 2. Write work items to "Work Done" section (same as stop-hook)
-        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`);
-        }
+      let notePath = getCurrentNotePath(notesInfo.path);
+
+      // If no note found, or the latest note is completed, create a new one
+      if (!notePath) {
+        console.error('No session note found — creating one for checkpoint');
+        notePath = createSessionNote(notesInfo.path, 'Recovered Session');
+      } else {
+        // Check if the found note is already completed — don't write to completed notes
+        try {
+          const noteContent = readFileSync(notePath, 'utf-8');
+          if (noteContent.includes('**Status:** Completed') || noteContent.includes('**Completed:**')) {
+            console.error(`Latest note is completed (${basename(notePath)}) — creating new one`);
+            notePath = createSessionNote(notesInfo.path, 'Continued Session');
+          }
+        } catch { /* proceed with existing note */ }
+      }
 
-        console.error(`Rich checkpoint saved: ${basename(currentNotePath)}`);
+      // 1. Write rich checkpoint with full session state
+      const checkpointBody = state
+        ? `Context compression triggered at ~${tokenDisplay} tokens with ${stats.messageCount} messages.\n\n${state}`
+        : `Context compression triggered at ~${tokenDisplay} tokens with ${stats.messageCount} messages.`;
+      appendCheckpoint(notePath, checkpointBody);
+
+      // 2. Write work items to "Work Done" section (same as stop-hook)
+      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`);
       }
+
+      console.error(`Rich checkpoint saved: ${basename(notePath)}`);
     } catch (noteError) {
       console.error(`Could not save checkpoint: ${noteError}`);
     }