@tekmidian/pai 0.7.0 → 0.7.2

package/dist/hooks/stop-hook.mjs

@@ -3,6 +3,8 @@
 // src/hooks/ts/stop/stop-hook.ts
 import { readFileSync as readFileSync5 } from "fs";
 import { basename as basename3, dirname } from "path";
+import { connect } from "net";
+import { randomUUID } from "crypto";
 
 // src/hooks/ts/lib/project-utils/paths.ts
 import { existsSync as existsSync2, mkdirSync, readdirSync, renameSync } from "fs";
@@ -266,6 +268,30 @@ function addWorkToSessionNote(notePath, workItems, sectionTitle) {
 function sanitizeForFilename(str) {
   return str.toLowerCase().replace(/[^a-z0-9\s-]/g, "").replace(/\s+/g, "-").replace(/-+/g, "-").replace(/^-|-$/g, "").substring(0, 50);
 }
+function isMeaninglessCandidate(text) {
+  const t = text.trim();
+  if (!t) return true;
+  if (t.length < 5) return true;
+  if (t.startsWith("/") || t.startsWith("~")) return true;
+  if (t.startsWith("#!")) return true;
+  if (t.includes("[object Object]")) return true;
+  if (/^\d{4}-\d{2}-\d{2}(T[\d:.Z+-]+)?$/.test(t)) return true;
+  if (/^\d{1,2}:\d{2}(:\d{2})?(\s*(AM|PM))?$/i.test(t)) return true;
+  if (/^<[a-z-]+[\s/>]/i.test(t)) return true;
+  if (/^[0-9a-f]{10,}$/i.test(t)) return true;
+  if (/^Exit code \d+/i.test(t)) return true;
+  if (/^Error:/i.test(t)) return true;
+  if (/^This session is being continued/i.test(t)) return true;
+  if (/^\(Bash completed/i.test(t)) return true;
+  if (/^Task Notification$/i.test(t)) return true;
+  if (/^New Session$/i.test(t)) return true;
+  if (/^Recovered Session$/i.test(t)) return true;
+  if (/^Continued Session$/i.test(t)) return true;
+  if (/^Untitled Session$/i.test(t)) return true;
+  if (/^Context Compression$/i.test(t)) return true;
+  if (/^[A-Fa-f0-9]{8,}\s+Output$/i.test(t)) return true;
+  return false;
+}
 function extractMeaningfulName(noteContent, summary) {
   const workDoneMatch = noteContent.match(/## Work Done\n\n([\s\S]*?)(?=\n---|\n## Next)/);
   if (workDoneMatch) {
@@ -273,23 +299,27 @@ function extractMeaningfulName(noteContent, summary) {
     const subheadings = workDoneSection.match(/### ([^\n]+)/g);
     if (subheadings && subheadings.length > 0) {
       const firstHeading = subheadings[0].replace("### ", "").trim();
-      if (firstHeading.length > 5 && firstHeading.length < 60) {
+      if (!isMeaninglessCandidate(firstHeading) && firstHeading.length > 5 && firstHeading.length < 60) {
         return sanitizeForFilename(firstHeading);
       }
     }
     const boldMatches = workDoneSection.match(/\*\*([^*]+)\*\*/g);
     if (boldMatches && boldMatches.length > 0) {
       const firstBold = boldMatches[0].replace(/\*\*/g, "").trim();
-      if (firstBold.length > 3 && firstBold.length < 50) {
+      if (!isMeaninglessCandidate(firstBold) && firstBold.length > 3 && firstBold.length < 50) {
         return sanitizeForFilename(firstBold);
       }
     }
     const numberedItems = workDoneSection.match(/^\d+\.\s+\*\*([^*]+)\*\*/m);
-    if (numberedItems) return sanitizeForFilename(numberedItems[1]);
+    if (numberedItems && !isMeaninglessCandidate(numberedItems[1])) {
+      return sanitizeForFilename(numberedItems[1]);
+    }
   }
-  if (summary && summary.length > 5 && summary !== "Session completed.") {
+  if (summary && summary.length > 5 && summary !== "Session completed." && !isMeaninglessCandidate(summary)) {
     const cleanSummary = summary.replace(/[^\w\s-]/g, " ").trim().split(/\s+/).slice(0, 5).join(" ");
-    if (cleanSummary.length > 3) return sanitizeForFilename(cleanSummary);
+    if (cleanSummary.length > 3 && !isMeaninglessCandidate(cleanSummary)) {
+      return sanitizeForFilename(cleanSummary);
+    }
   }
   return "";
 }
@@ -418,6 +448,105 @@ ${stateLines}
 }
 
 // src/hooks/ts/stop/stop-hook.ts
+var DAEMON_SOCKET = process.env.PAI_SOCKET ?? "/tmp/pai.sock";
+var DAEMON_TIMEOUT_MS = 3e3;
+function contentToText(content) {
+  if (typeof content === "string") return content;
+  if (Array.isArray(content)) {
+    return content.map((c) => {
+      if (typeof c === "string") return c;
+      if (c?.text) return c.text;
+      if (c?.content) return String(c.content);
+      return "";
+    }).join(" ").trim();
+  }
+  return "";
+}
+function extractCompletedMessage(lines) {
+  for (let i = lines.length - 1; i >= 0; i--) {
+    try {
+      const entry = JSON.parse(lines[i]);
+      if (entry.type === "assistant" && entry.message?.content) {
+        const content = contentToText(entry.message.content);
+        const m = content.match(/COMPLETED:\s*(.+?)(?:\n|$)/i);
+        if (m) {
+          return m[1].trim().replace(/\*+/g, "").replace(/\[.*?\]/g, "").trim();
+        }
+      }
+    } catch {
+    }
+  }
+  return "";
+}
+function enqueueWithDaemon(payload) {
+  return new Promise((resolve2) => {
+    let done = false;
+    let buffer = "";
+    let timer = null;
+    function finish(ok) {
+      if (done) return;
+      done = true;
+      if (timer !== null) {
+        clearTimeout(timer);
+        timer = null;
+      }
+      try {
+        client.destroy();
+      } catch {
+      }
+      resolve2(ok);
+    }
+    const client = connect(DAEMON_SOCKET, () => {
+      const msg = JSON.stringify({
+        id: randomUUID(),
+        method: "work_queue_enqueue",
+        params: {
+          type: "session-end",
+          priority: 2,
+          payload: {
+            transcriptPath: payload.transcriptPath,
+            cwd: payload.cwd,
+            message: payload.message
+          }
+        }
+      }) + "\n";
+      client.write(msg);
+    });
+    client.on("data", (chunk) => {
+      buffer += chunk.toString();
+      const nl = buffer.indexOf("\n");
+      if (nl === -1) return;
+      const line = buffer.slice(0, nl);
+      try {
+        const response = JSON.parse(line);
+        if (response.ok) {
+          console.error(`STOP-HOOK: Work enqueued with daemon (id=${response.result?.id}).`);
+          finish(true);
+        } else {
+          console.error(`STOP-HOOK: Daemon rejected enqueue: ${response.error}`);
+          finish(false);
+        }
+      } catch {
+        finish(false);
+      }
+    });
+    client.on("error", (e) => {
+      if (e.code === "ENOENT" || e.code === "ECONNREFUSED") {
+        console.error("STOP-HOOK: Daemon not running \u2014 falling back to direct execution.");
+      } else {
+        console.error(`STOP-HOOK: Daemon socket error: ${e.message}`);
+      }
+      finish(false);
+    });
+    client.on("end", () => {
+      if (!done) finish(false);
+    });
+    timer = setTimeout(() => {
+      console.error(`STOP-HOOK: Daemon timeout after ${DAEMON_TIMEOUT_MS}ms \u2014 falling back.`);
+      finish(false);
+    }, DAEMON_TIMEOUT_MS);
+  });
+}
 function extractWorkFromTranscript(lines) {
   const workItems = [];
   const seenSummaries = /* @__PURE__ */ new Set();
@@ -446,13 +575,10 @@ function extractWorkFromTranscript(lines) {
         }
         const completedMatch = content.match(/COMPLETED:\s*(.+?)(?:\n|$)/i);
         if (completedMatch && workItems.length === 0) {
-          const completed = completedMatch[1].trim().replace(/\*+/g, "");
+          const completed = completedMatch[1].trim().replace(/\*+/g, "").replace(/\[.*?\]/g, "");
           if (completed && !seenSummaries.has(completed) && completed.length > 5) {
             seenSummaries.add(completed);
-            workItems.push({
-              title: completed,
-              completed: true
-            });
+            workItems.push({ title: completed, completed: true });
           }
         }
       }
@@ -467,9 +593,7 @@ function generateTabTitle(prompt, completedLine) {
     const completedWords = cleanCompleted.split(/\s+/).filter((word) => word.length > 2 && !["the", "and", "but", "for", "are", "with", "his", "her", "this", "that", "you", "can", "will", "have", "been", "your", "from", "they", "were", "said", "what", "them", "just", "told", "how", "does", "into", "about", "completed"].includes(word.toLowerCase())).map((word) => word.charAt(0).toUpperCase() + word.slice(1).toLowerCase());
     if (completedWords.length >= 2) {
       const summary = completedWords.slice(0, 4);
-      while (summary.length < 4) {
-        summary.push("Done");
-      }
+      while (summary.length < 4) summary.push("Done");
       return summary.slice(0, 4).join(" ");
     }
   }
@@ -494,37 +618,80 @@ function generateTabTitle(prompt, completedLine) {
   }
   const remainingWords = words.filter((word) => !actionVerbs.includes(word.toLowerCase())).map((word) => word.charAt(0).toUpperCase() + word.slice(1).toLowerCase());
   for (const word of remainingWords) {
-    if (titleWords.length < 4) {
-      titleWords.push(word);
-    } else {
-      break;
-    }
+    if (titleWords.length < 4) titleWords.push(word);
+    else break;
   }
-  if (titleWords.length === 0) {
-    titleWords.push("Completed");
+  if (titleWords.length === 0) titleWords.push("Completed");
+  if (titleWords.length === 1) titleWords.push("Task");
+  if (titleWords.length === 2) titleWords.push("Successfully");
+  if (titleWords.length === 3) titleWords.push("Done");
+  return titleWords.slice(0, 4).join(" ");
+}
+async function executeDirectly(lines, transcriptPath, cwd, message, lastUserQuery) {
+  let tabTitle = message || "";
+  if (!tabTitle && lastUserQuery) {
+    tabTitle = generateTabTitle(lastUserQuery, "");
   }
-  if (titleWords.length === 1) {
-    titleWords.push("Task");
+  if (tabTitle) {
+    try {
+      const escapedTitle = tabTitle.replace(/'/g, "'\\''");
+      const { execSync } = await import("child_process");
+      execSync(`printf '\\033]0;${escapedTitle}\\007' >&2`);
+      execSync(`printf '\\033]2;${escapedTitle}\\007' >&2`);
+      execSync(`printf '\\033]30;${escapedTitle}\\007' >&2`);
+      console.error(`Tab title set to: "${tabTitle}"`);
+    } catch (e) {
+      console.error(`Failed to set tab title: ${e}`);
+    }
   }
-  if (titleWords.length === 2) {
-    titleWords.push("Successfully");
+  if (message) {
+    const finalTabTitle = message.slice(0, 50);
+    process.stderr.write(`\x1B]2;${finalTabTitle}\x07`);
   }
-  if (titleWords.length === 3) {
-    titleWords.push("Done");
+  try {
+    const notesInfo = findNotesDir(cwd);
+    const currentNotePath = getCurrentNotePath(notesInfo.path);
+    if (currentNotePath) {
+      const workItems = extractWorkFromTranscript(lines);
+      if (workItems.length > 0) {
+        addWorkToSessionNote(currentNotePath, workItems);
+        console.error(`Added ${workItems.length} work item(s) to session note`);
+      } else if (message) {
+        addWorkToSessionNote(currentNotePath, [{ title: message, completed: true }]);
+        console.error(`Added completion message to session note`);
+      }
+      const summary = message || "Session completed.";
+      finalizeSessionNote(currentNotePath, summary);
+      console.error(`Session note finalized: ${basename3(currentNotePath)}`);
+      try {
+        const stateLines = [];
+        stateLines.push(`Working directory: ${cwd}`);
+        if (workItems.length > 0) {
+          stateLines.push("", "Work completed:");
+          for (const item of workItems.slice(0, 5)) {
+            stateLines.push(`- ${item.title}`);
+          }
+        }
+        if (message) {
+          stateLines.push("", `Last completed: ${message}`);
+        }
+        updateTodoContinue(cwd, basename3(currentNotePath), stateLines.join("\n"), "session-end");
+      } catch (todoError) {
+        console.error(`Could not update TODO.md: ${todoError}`);
+      }
+    }
+  } catch (noteError) {
+    console.error(`Could not finalize session note: ${noteError}`);
   }
-  return titleWords.slice(0, 4).join(" ");
-}
-function contentToText(content) {
-  if (typeof content === "string") return content;
-  if (Array.isArray(content)) {
-    return content.map((c) => {
-      if (typeof c === "string") return c;
-      if (c?.text) return c.text;
-      if (c?.content) return String(c.content);
-      return "";
-    }).join(" ").trim();
+  try {
+    const transcriptDir = dirname(transcriptPath);
+    const movedCount = moveSessionFilesToSessionsDir(transcriptDir);
+    if (movedCount > 0) {
+      console.error(`Moved ${movedCount} session file(s) to sessions/`);
+    }
+  } catch (moveError) {
+    console.error(`Could not move session files: ${moveError}`);
   }
-  return "";
 }
 async function main() {
   if (isProbeSession()) {
@@ -590,45 +757,20 @@ STOP-HOOK TRIGGERED AT ${timestamp}`);
         }
         if (lastUserQuery) break;
       }
-    } catch (e) {
-    }
-  }
-  let message = "";
-  const lastResponse = lines[lines.length - 1];
-  try {
-    const entry = JSON.parse(lastResponse);
-    if (entry.type === "assistant" && entry.message?.content) {
-      const content = contentToText(entry.message.content);
-      const completedMatch = content.match(/COMPLETED:\s*(.+?)(?:\n|$)/i);
-      if (completedMatch) {
-        message = completedMatch[1].trim().replace(/\*+/g, "").replace(/\[.*?\]/g, "").trim();
-        console.error(`COMPLETION: ${message}`);
-      }
+    } catch {
     }
-  } catch (e) {
-    console.error("Error parsing assistant response:", e);
   }
+  const message = extractCompletedMessage(lines);
+  console.error(`User query: ${lastUserQuery || "No query found"}`);
+  console.error(`Message: ${message || "No completion message"}`);
   let tabTitle = message || "";
   if (!tabTitle && lastUserQuery) {
-    try {
-      const entry = JSON.parse(lastResponse);
-      if (entry.type === "assistant" && entry.message?.content) {
-        const content = contentToText(entry.message.content);
-        const completedMatch = content.match(/COMPLETED:\s*(.+?)(?:\n|$)/im);
-        if (completedMatch) {
-          tabTitle = completedMatch[1].trim().replace(/\*+/g, "").replace(/\[.*?\]/g, "").trim();
-        }
-      }
-    } catch (e) {
-    }
-    if (!tabTitle) {
-      tabTitle = generateTabTitle(lastUserQuery, "");
-    }
+    tabTitle = generateTabTitle(lastUserQuery, "");
   }
   if (tabTitle) {
     try {
-      const escapedTitle = tabTitle.replace(/'/g, "'\\''");
       const { execSync } = await import("child_process");
+      const escapedTitle = tabTitle.replace(/'/g, "'\\''");
       execSync(`printf '\\033]0;${escapedTitle}\\007' >&2`);
       execSync(`printf '\\033]2;${escapedTitle}\\007' >&2`);
       execSync(`printf '\\033]30;${escapedTitle}\\007' >&2`);
@@ -637,69 +779,22 @@ STOP-HOOK TRIGGERED AT ${timestamp}`);
       console.error(`Failed to set tab title: ${e}`);
     }
   }
-  console.error(`User query: ${lastUserQuery || "No query found"}`);
-  console.error(`Message: ${message || "No completion message"}`);
   if (message) {
-    const finalTabTitle = message.slice(0, 50);
-    process.stderr.write(`\x1B]2;${finalTabTitle}\x07`);
+    process.stderr.write(`\x1B]2;${message.slice(0, 50)}\x07`);
   }
   if (message) {
     await sendNtfyNotification(message);
   } else {
     await sendNtfyNotification("Session ended");
   }
-  try {
-    const notesInfo = findNotesDir(cwd);
-    console.error(`Notes directory: ${notesInfo.path} (${notesInfo.isLocal ? "local" : "central"})`);
-    const currentNotePath = getCurrentNotePath(notesInfo.path);
-    if (currentNotePath) {
-      const workItems = extractWorkFromTranscript(lines);
-      if (workItems.length > 0) {
-        addWorkToSessionNote(currentNotePath, workItems);
-        console.error(`Added ${workItems.length} work item(s) to session note`);
-      } else {
-        if (message) {
-          addWorkToSessionNote(currentNotePath, [{
-            title: message,
-            completed: true
-          }]);
-          console.error(`Added completion message to session note`);
-        }
-      }
-      const summary = message || "Session completed.";
-      finalizeSessionNote(currentNotePath, summary);
-      console.error(`Session note finalized: ${basename3(currentNotePath)}`);
-      try {
-        const stateLines = [];
-        stateLines.push(`Working directory: ${cwd}`);
-        if (workItems.length > 0) {
-          stateLines.push("");
-          stateLines.push("Work completed:");
-          for (const item of workItems.slice(0, 5)) {
-            stateLines.push(`- ${item.title}`);
-          }
-        }
-        if (message) {
-          stateLines.push("");
-          stateLines.push(`Last completed: ${message}`);
-        }
-        const state = stateLines.join("\n");
-        updateTodoContinue(cwd, basename3(currentNotePath), state, "session-end");
-      } catch (todoError) {
-        console.error(`Could not update TODO.md: ${todoError}`);
-      }
-    }
-  } catch (noteError) {
-    console.error(`Could not finalize session note: ${noteError}`);
-  }
-  try {
-    const transcriptDir = dirname(transcriptPath);
-    const movedCount = moveSessionFilesToSessionsDir(transcriptDir);
-    if (movedCount > 0) {
-      console.error(`Moved ${movedCount} session file(s) to sessions/`);
-    }
-  } catch (moveError) {
-    console.error(`Could not move session files: ${moveError}`);
+  const relayed = await enqueueWithDaemon({
+    transcriptPath,
+    cwd,
+    message
+  });
+  if (!relayed) {
+    console.error("STOP-HOOK: Using direct execution fallback.");
+    await executeDirectly(lines, transcriptPath, cwd, message, lastUserQuery);
   }
   console.error(`STOP-HOOK COMPLETED SUCCESSFULLY at ${(/* @__PURE__ */ new Date()).toISOString()}
 `);