@tekmidian/pai 0.14.2 → 0.16.0

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

@@ -1,8 +1,8 @@
 #!/usr/bin/env node
 
 // src/hooks/ts/pre-compact/context-compression-hook.ts
-import { existsSync as existsSync7, readFileSync as readFileSync6, writeFileSync as writeFileSync3 } from "fs";
-import { basename as basename3, dirname, join as join6 } from "path";
+import { existsSync as existsSync8, readFileSync as readFileSync7, writeFileSync as writeFileSync4 } from "fs";
+import { basename as basename3, dirname, join as join7 } from "path";
 import { tmpdir } from "os";
 import { connect } from "net";
 import { randomUUID } from "crypto";
@@ -420,13 +420,260 @@ function calculateSessionTokens(jsonlPath) {
 }
 
 // src/hooks/ts/lib/project-utils/todo.ts
-import { existsSync as existsSync6, mkdirSync as mkdirSync3, readFileSync as readFileSync5, writeFileSync as writeFileSync2 } from "fs";
-import { join as join5 } from "path";
+import { existsSync as existsSync7, mkdirSync as mkdirSync4, readFileSync as readFileSync6, writeFileSync as writeFileSync3 } from "fs";
+import { join as join6 } from "path";
+
+// src/session/checkpoint-block.ts
+import {
+  existsSync as existsSync6,
+  readFileSync as readFileSync5,
+  writeFileSync as writeFileSync2,
+  readdirSync as readdirSync3,
+  renameSync as renameSync3,
+  mkdirSync as mkdirSync3,
+  realpathSync
+} from "node:fs";
+import { join as join5 } from "node:path";
+var TODO_LOCATIONS = [
+  "Notes/TODO.md",
+  ".claude/Notes/TODO.md",
+  "tasks/todo.md",
+  "TODO.md"
+];
+var MARKER_OPEN = "<!-- pai:checkpoint";
+var MARKER_CLOSE = "<!-- /pai:checkpoint -->";
+var CONTINUE_HEADING = "## Continue";
+function findProjectTodo(rootPath) {
+  for (const rel of TODO_LOCATIONS) {
+    const full = join5(rootPath, rel);
+    if (existsSync6(full)) {
+      try {
+        return { path: full, content: readFileSync5(full, "utf8") };
+      } catch {
+      }
+    }
+  }
+  return null;
+}
+function resolveTodoTarget(rootPath, opts = {}) {
+  const found = findProjectTodo(rootPath);
+  if (found) return found;
+  const notesDir = join5(rootPath, "Notes");
+  if (opts.create !== false) {
+    try {
+      if (!existsSync6(notesDir)) mkdirSync3(notesDir, { recursive: true });
+    } catch {
+      return null;
+    }
+  }
+  return { path: join5(notesDir, "TODO.md"), content: "" };
+}
+function parseMarker(line) {
+  const trimmed = line.trim();
+  if (!trimmed.startsWith(MARKER_OPEN)) return null;
+  const attrs = {};
+  for (const m of trimmed.matchAll(/([a-zA-Z][\w-]*)="([^"]*)"/g)) {
+    attrs[m[1]] = m[2];
+  }
+  return {
+    authored: attrs.authored === "model" ? "model" : "auto",
+    session: attrs.session || void 0,
+    sessionId: attrs["session-id"] || void 0,
+    ts: attrs.ts || void 0
+  };
+}
+var BOILERPLATE_PATTERNS = [
+  /^##\s+Continue$/,
+  /^<!--\s*\/?pai:checkpoint/,
+  /^>\s*\*\*Last session:\*\*/,
+  /^>\s*\*\*Paused at:\*\*/,
+  /^>\s*Working directory:/,
+  /^>\s*Resume with:/,
+  /^>\s*_No checkpoint body was recorded/,
+  /^>?\s*$/,
+  /^-{3,}$/
+];
+function isBoilerplateOnly(lines) {
+  return lines.every((line) => {
+    const t = line.trim();
+    return BOILERPLATE_PATTERNS.some((re) => re.test(t));
+  });
+}
+function extractSectionContent(lines) {
+  const kept = [];
+  for (const line of lines) {
+    const t = line.trim();
+    if (BOILERPLATE_PATTERNS.some((re) => re.test(t))) continue;
+    kept.push(line);
+  }
+  return kept.join("\n").trim();
+}
+function locateContinue(content) {
+  const lines = content.split("\n");
+  const startIdx = lines.findIndex((l) => l.trim() === CONTINUE_HEADING);
+  if (startIdx === -1) return null;
+  let meta = null;
+  let markerIdx = -1;
+  for (let i = startIdx + 1; i < Math.min(startIdx + 6, lines.length); i++) {
+    const parsed = parseMarker(lines[i]);
+    if (parsed) {
+      meta = parsed;
+      markerIdx = i;
+      break;
+    }
+    if (lines[i].trim() !== "") break;
+  }
+  let endIdx = lines.length;
+  if (markerIdx !== -1) {
+    const closeIdx = lines.findIndex(
+      (l, i) => i > markerIdx && l.trim() === MARKER_CLOSE
+    );
+    if (closeIdx !== -1) {
+      endIdx = closeIdx + 1;
+    } else {
+      endIdx = heuristicEnd(lines, startIdx);
+    }
+  } else {
+    endIdx = heuristicEnd(lines, startIdx);
+  }
+  let trailingEnd = endIdx;
+  while (trailingEnd < lines.length && lines[trailingEnd].trim() === "") {
+    trailingEnd += 1;
+  }
+  if (trailingEnd < lines.length && lines[trailingEnd].trim() === "---") {
+    trailingEnd += 1;
+  } else {
+    trailingEnd = endIdx;
+  }
+  return {
+    startIdx,
+    endIdx: trailingEnd,
+    meta,
+    lines: lines.slice(startIdx, trailingEnd)
+  };
+}
+function heuristicEnd(lines, startIdx) {
+  for (let i = startIdx + 1; i < lines.length; i++) {
+    const trimmed = lines[i].trim();
+    if (trimmed === "---" || /^##(?!#)/.test(trimmed) && trimmed !== CONTINUE_HEADING) {
+      return i;
+    }
+  }
+  return lines.length;
+}
+function stripContinue(content) {
+  const found = locateContinue(content);
+  if (!found) return content;
+  const lines = content.split("\n");
+  const before = lines.slice(0, found.startIdx);
+  const after = lines.slice(found.endIdx);
+  while (after.length > 0 && after[0].trim() === "") after.shift();
+  return [...before, ...after].join("\n");
+}
+function escapeAttr(value) {
+  return value.replace(/"/g, "'");
+}
+function buildContinueBlock(opts) {
+  const ts = opts.timestamp ?? (/* @__PURE__ */ new Date()).toISOString();
+  const attrs = [
+    `authored="${opts.authored}"`,
+    `session="${escapeAttr(opts.sessionLine)}"`,
+    opts.sessionId ? `session-id="${escapeAttr(opts.sessionId)}"` : null,
+    `ts="${ts}"`
+  ].filter(Boolean).join(" ");
+  const header = [
+    `> **Last session:** ${opts.sessionLine}`,
+    `> **Paused at:** ${ts}`,
+    ">",
+    `> Working directory: ${opts.cwd}`
+  ];
+  if (opts.sessionId) {
+    header.push(">", `> Resume with: \`claude --resume ${opts.sessionId}\``);
+  }
+  const body = (opts.body ?? "").trim();
+  const parts = [
+    CONTINUE_HEADING,
+    "",
+    `${MARKER_OPEN} ${attrs} -->`,
+    "",
+    ...header
+  ];
+  if (body) {
+    parts.push("", body);
+  } else {
+    parts.push(
+      ">",
+      "> _No checkpoint body was recorded \u2014 see the latest session note._"
+    );
+  }
+  parts.push("", MARKER_CLOSE, "", "---", "");
+  return parts.join("\n");
+}
+function applyContinue(opts) {
+  const target = resolveTodoTarget(opts.rootPath, { create: !opts.dryRun });
+  if (!target) {
+    return {
+      action: "failed",
+      path: null,
+      block: buildContinueBlock(opts),
+      error: "Could not resolve or create a TODO.md target"
+    };
+  }
+  const existing = locateContinue(target.content);
+  let carriedForward = false;
+  let effectiveBody = opts.body;
+  if (opts.authored === "auto" && existing) {
+    if (existing.meta?.authored === "model" && existing.meta.session === opts.sessionLine) {
+      return {
+        action: "preserved",
+        path: target.path,
+        block: buildContinueBlock(opts),
+        preservedMeta: existing.meta
+      };
+    }
+    if (!existing.meta && !isBoilerplateOnly(existing.lines)) {
+      const salvaged = extractSectionContent(existing.lines);
+      if (salvaged) {
+        effectiveBody = [
+          "_Carried forward from the previous checkpoint (author unknown \u2014 this",
+          "block predates checkpoint authorship markers):_",
+          "",
+          salvaged
+        ].join("\n");
+        carriedForward = true;
+      }
+    }
+  }
+  const block = buildContinueBlock({ ...opts, body: effectiveBody });
+  if (opts.dryRun) {
+    return { action: "written", path: target.path, block, carriedForward };
+  }
+  const newContent = block + stripContinue(target.content).trimStart();
+  const tmpPath = `${target.path}.continue.tmp`;
+  try {
+    writeFileSync2(tmpPath, newContent, "utf8");
+    renameSync3(tmpPath, target.path);
+  } catch (err) {
+    try {
+      if (existsSync6(tmpPath)) renameSync3(tmpPath, `${tmpPath}.dead`);
+    } catch {
+    }
+    return {
+      action: "failed",
+      path: target.path,
+      block,
+      error: String(err)
+    };
+  }
+  return { action: "written", path: target.path, block, carriedForward };
+}
+
+// src/hooks/ts/lib/project-utils/todo.ts
 function ensureTodoMd(cwd) {
   const todoPath = findTodoPath(cwd);
-  if (!existsSync6(todoPath)) {
-    const parentDir = join5(todoPath, "..");
-    if (!existsSync6(parentDir)) mkdirSync3(parentDir, { recursive: true });
+  if (!existsSync7(todoPath)) {
+    const parentDir = join6(todoPath, "..");
+    if (!existsSync7(parentDir)) mkdirSync4(parentDir, { recursive: true });
     const content = `# TODO
 
 ## Current Session
@@ -441,43 +688,52 @@ function ensureTodoMd(cwd) {
 
 *Last updated: ${(/* @__PURE__ */ new Date()).toISOString()}*
 `;
-    writeFileSync2(todoPath, content);
+    writeFileSync3(todoPath, content);
     console.error(`Created TODO.md: ${todoPath}`);
   }
   return todoPath;
 }
 function updateTodoContinue(cwd, noteFilename, state, tokenDisplay) {
-  const todoPath = ensureTodoMd(cwd);
-  let content = readFileSync5(todoPath, "utf-8");
-  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") : `> Working directory: ${cwd}. Check the latest session note for details.`;
-  const continueSection = `## Continue
-
-> **Last session:** ${noteFilename.replace(".md", "")}
-> **Paused at:** ${now}
->
-${stateLines}
-
----
-
-`;
-  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;
+  ensureTodoMd(cwd);
+  const result = applyContinue({
+    rootPath: cwd,
+    authored: "auto",
+    // The hooks identify a session by its note filename; `pai pause` resolves
+    // the same string from the registry (and falls back to this filename when
+    // the registry has no session row). They must agree, because this string
+    // is the key that decides whether an authored checkpoint belongs to the
+    // current session and must be preserved.
+    sessionLine: noteFilename.replace(/\.md$/, ""),
+    cwd,
+    body: state?.trim() || void 0
+  });
+  if (result.action === "preserved") {
+    console.error(
+      "TODO.md ## Continue left intact \u2014 authored checkpoint for this session"
+    );
+    return;
   }
-  content = content.replace(/(\n---\s*)*(\n\*Last updated:.*\*\s*)+$/g, "");
-  content = content.trimEnd() + `
+  if (result.action === "failed") {
+    console.error(`TODO.md ## Continue update failed: ${result.error}`);
+    return;
+  }
+  try {
+    const todoPath = result.path;
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    let content = readFileSync6(todoPath, "utf-8");
+    content = content.replace(/(\n---\s*)*(\n\*Last updated:.*\*\s*)+$/g, "");
+    content = content.trimEnd() + `
 
 ---
 
 *Last updated: ${now}*
 `;
-  writeFileSync2(todoPath, content);
-  console.error("TODO.md ## Continue section updated");
+    writeFileSync3(todoPath, content);
+  } catch {
+  }
+  console.error(
+    result.carriedForward ? "TODO.md ## Continue section updated (previous content carried forward)" : "TODO.md ## Continue section updated"
+  );
 }
 
 // src/hooks/ts/pre-compact/context-compression-hook.ts
@@ -497,7 +753,7 @@ function contentToText(content) {
 }
 function getTranscriptStats(transcriptPath) {
   try {
-    const content = readFileSync6(transcriptPath, "utf-8");
+    const content = readFileSync7(transcriptPath, "utf-8");
     const lines = content.trim().split("\n");
     let userMessages = 0;
     let assistantMessages = 0;
@@ -526,7 +782,7 @@ function parseTranscript(transcriptPath) {
     workItems: []
   };
   try {
-    const raw = readFileSync6(transcriptPath, "utf-8");
+    const raw = readFileSync7(transcriptPath, "utf-8");
     const lines = raw.trim().split("\n");
     const seenSummaries = /* @__PURE__ */ new Set();
     for (const line of lines) {
@@ -662,9 +918,9 @@ function deriveTitle(data) {
 var CUMULATIVE_STATE_FILE = ".compact-state.json";
 function loadCumulativeState(notesDir) {
   try {
-    const filePath = join6(notesDir, CUMULATIVE_STATE_FILE);
-    if (!existsSync7(filePath)) return null;
-    const raw = JSON.parse(readFileSync6(filePath, "utf-8"));
+    const filePath = join7(notesDir, CUMULATIVE_STATE_FILE);
+    if (!existsSync8(filePath)) return null;
+    const raw = JSON.parse(readFileSync7(filePath, "utf-8"));
     return {
       userMessages: raw.userMessages || [],
       summaries: raw.summaries || [],
@@ -696,8 +952,8 @@ function mergeTranscriptData(accumulated, current) {
 }
 function saveCumulativeState(notesDir, data, notePath) {
   try {
-    const filePath = join6(notesDir, CUMULATIVE_STATE_FILE);
-    writeFileSync3(filePath, JSON.stringify({
+    const filePath = join7(notesDir, CUMULATIVE_STATE_FILE);
+    writeFileSync4(filePath, JSON.stringify({
       ...data,
       notePath,
       lastUpdated: (/* @__PURE__ */ new Date()).toISOString()
@@ -840,9 +1096,9 @@ async function main() {
     const data = parseTranscript(hookInput.transcript_path);
     let notesInfo;
     try {
-      notesInfo = hookInput.cwd ? findNotesDir(hookInput.cwd) : { path: join6(dirname(hookInput.transcript_path), "Notes"), isLocal: false };
+      notesInfo = hookInput.cwd ? findNotesDir(hookInput.cwd) : { path: join7(dirname(hookInput.transcript_path), "Notes"), isLocal: false };
     } catch {
-      notesInfo = { path: join6(dirname(hookInput.transcript_path), "Notes"), isLocal: false };
+      notesInfo = { path: join7(dirname(hookInput.transcript_path), "Notes"), isLocal: false };
     }
     const accumulated = loadCumulativeState(notesInfo.path);
     const merged = mergeTranscriptData(accumulated, data);
@@ -858,10 +1114,10 @@ async function main() {
         notePath = createSessionNote(notesInfo.path, "Untitled Session");
       } else {
         try {
-          let noteContent = readFileSync6(notePath, "utf-8");
+          let noteContent = readFileSync7(notePath, "utf-8");
           if (noteContent.includes("**Status:** Completed")) {
             noteContent = noteContent.replace("**Status:** Completed", "**Status:** In Progress");
-            writeFileSync3(notePath, noteContent);
+            writeFileSync4(notePath, noteContent);
             console.error(`Reopened completed note for continued session: ${basename3(notePath)}`);
           }
         } catch {
@@ -880,12 +1136,12 @@ ${state}` : `Context compression triggered at ~${tokenDisplay} tokens with ${sta
         const newPath = renameSessionNote(notePath, title);
         if (newPath !== notePath) {
           try {
-            let noteContent = readFileSync6(newPath, "utf-8");
+            let noteContent = readFileSync7(newPath, "utf-8");
             noteContent = noteContent.replace(
               /^(# Session \d+:)\s*.*$/m,
               `$1 ${title}`
             );
-            writeFileSync3(newPath, noteContent);
+            writeFileSync4(newPath, noteContent);
             console.error(`Updated note H1 to match rename`);
           } catch {
           }
@@ -926,8 +1182,8 @@ rename it based on actual work done and add a rich summary.` : "";
         "</system-reminder>"
       ].join("\n");
       try {
-        const stateFile = join6(tmpdir(), `pai-compact-state-${hookInput.session_id}.txt`);
-        writeFileSync3(stateFile, injection, "utf-8");
+        const stateFile = join7(tmpdir(), `pai-compact-state-${hookInput.session_id}.txt`);
+        writeFileSync4(stateFile, injection, "utf-8");
         console.error(`Session state saved to ${stateFile} (${injection.length} chars)`);
       } catch (err) {
         console.error(`Failed to save state file: ${err}`);