@tekmidian/pai 0.14.2 → 0.16.0

package/dist/hooks/stop-hook.mjs

@@ -1,8 +1,8 @@
 #!/usr/bin/env node
 
 // src/hooks/ts/stop/stop-hook.ts
-import { readFileSync as readFileSync5, writeFileSync as writeFileSync3, mkdirSync as mkdirSync4, existsSync as existsSync6, unlinkSync } from "fs";
-import { join as join6, basename as basename3, dirname } from "path";
+import { readFileSync as readFileSync6, writeFileSync as writeFileSync4, mkdirSync as mkdirSync5, existsSync as existsSync7, unlinkSync } from "fs";
+import { join as join7, basename as basename3, dirname } from "path";
 import { connect } from "net";
 import { randomUUID } from "crypto";
 
@@ -387,13 +387,260 @@ ${summary || "Session completed."}`
 }
 
 // src/hooks/ts/lib/project-utils/todo.ts
-import { existsSync as existsSync5, mkdirSync as mkdirSync3, readFileSync as readFileSync4, writeFileSync as writeFileSync2 } from "fs";
-import { join as join5 } from "path";
+import { existsSync as existsSync6, mkdirSync as mkdirSync4, readFileSync as readFileSync5, writeFileSync as writeFileSync3 } from "fs";
+import { join as join6 } from "path";
+
+// src/session/checkpoint-block.ts
+import {
+  existsSync as existsSync5,
+  readFileSync as readFileSync4,
+  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 (existsSync5(full)) {
+      try {
+        return { path: full, content: readFileSync4(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 (!existsSync5(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 (existsSync5(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 (!existsSync5(todoPath)) {
-    const parentDir = join5(todoPath, "..");
-    if (!existsSync5(parentDir)) mkdirSync3(parentDir, { recursive: true });
+  if (!existsSync6(todoPath)) {
+    const parentDir = join6(todoPath, "..");
+    if (!existsSync6(parentDir)) mkdirSync4(parentDir, { recursive: true });
     const content = `# TODO
 
 ## Current Session
@@ -408,43 +655,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 = readFileSync4(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;
+  }
+  if (result.action === "failed") {
+    console.error(`TODO.md ## Continue update failed: ${result.error}`);
+    return;
   }
-  content = content.replace(/(\n---\s*)*(\n\*Last updated:.*\*\s*)+$/g, "");
-  content = content.trimEnd() + `
+  try {
+    const todoPath = result.path;
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    let content = readFileSync5(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/stop/stop-hook.ts
@@ -458,7 +714,7 @@ var AUTO_SAVE_INTERVAL = (() => {
   }
   return 15;
 })();
-var SESSION_STATE_DIR = join6(
+var SESSION_STATE_DIR = join7(
   process.env.HOME ?? "/tmp",
   ".config",
   "pai",
@@ -466,9 +722,9 @@ var SESSION_STATE_DIR = join6(
 );
 function readSessionState(sessionId) {
   try {
-    const stateFile = join6(SESSION_STATE_DIR, `${sessionId}.json`);
-    if (!existsSync6(stateFile)) return { humanMessageCount: 0 };
-    const raw = readFileSync5(stateFile, "utf-8");
+    const stateFile = join7(SESSION_STATE_DIR, `${sessionId}.json`);
+    if (!existsSync7(stateFile)) return { humanMessageCount: 0 };
+    const raw = readFileSync6(stateFile, "utf-8");
     const parsed = JSON.parse(raw);
     return {
       humanMessageCount: typeof parsed.humanMessageCount === "number" ? parsed.humanMessageCount : 0
@@ -479,17 +735,17 @@ function readSessionState(sessionId) {
 }
 function writeSessionState(sessionId, state) {
   try {
-    mkdirSync4(SESSION_STATE_DIR, { recursive: true });
-    const stateFile = join6(SESSION_STATE_DIR, `${sessionId}.json`);
-    writeFileSync3(stateFile, JSON.stringify(state, null, 2), "utf-8");
+    mkdirSync5(SESSION_STATE_DIR, { recursive: true });
+    const stateFile = join7(SESSION_STATE_DIR, `${sessionId}.json`);
+    writeFileSync4(stateFile, JSON.stringify(state, null, 2), "utf-8");
   } catch (e) {
     console.error(`STOP-HOOK: Could not write session state: ${e}`);
   }
 }
 function deleteSessionState(sessionId) {
   try {
-    const stateFile = join6(SESSION_STATE_DIR, `${sessionId}.json`);
-    if (existsSync6(stateFile)) {
+    const stateFile = join7(SESSION_STATE_DIR, `${sessionId}.json`);
+    if (existsSync7(stateFile)) {
       unlinkSync(stateFile);
     }
   } catch {
@@ -996,7 +1252,7 @@ STOP-HOOK TRIGGERED AT ${timestamp}`);
   }
   let transcript;
   try {
-    transcript = readFileSync5(transcriptPath, "utf-8");
+    transcript = readFileSync6(transcriptPath, "utf-8");
     debug(`Transcript loaded: ${transcript.split("\n").length} lines`);
   } catch (e) {
     console.error(`Error reading transcript: ${e}`);