@jmylchreest/aide-plugin 0.1.1 → 0.1.2

package/src/hooks/session-end.ts

@@ -26,8 +26,16 @@ const T0 = performance.now();
 // Output continue IMMEDIATELY — before require(), before anything.
 console.log(JSON.stringify({ continue: true }));
 
-const { spawn, execFileSync } = require("child_process") as typeof import("child_process");
-const { existsSync, realpathSync, appendFileSync, mkdirSync, readFileSync, statSync } = require("fs") as typeof import("fs");
+const { spawn, execFileSync } =
+  require("child_process") as typeof import("child_process");
+const {
+  existsSync,
+  realpathSync,
+  appendFileSync,
+  mkdirSync,
+  readFileSync,
+  statSync,
+} = require("fs") as typeof import("fs");
 const { join, dirname } = require("path") as typeof import("path");
 const whichSync = (require("which") as typeof import("which")).sync;
 
@@ -41,7 +49,9 @@ function resolveRoot(cwd: string): string {
   if (override) {
     try {
       if (statSync(override).isDirectory()) return override;
-    } catch { /* fall through */ }
+    } catch {
+      /* fall through */
+    }
   }
   const candidates: { dir: string; hasAide: boolean; hasVCS: boolean }[] = [];
   let dir = cwd;
@@ -49,7 +59,10 @@ function resolveRoot(cwd: string): string {
     const hasAide = existsSync(join(dir, ".aide"));
     let hasVCS = false;
     for (const m of [".git", ".hg", ".svn", ".bzr", ".fossil"]) {
-      if (existsSync(join(dir, m))) { hasVCS = true; break; }
+      if (existsSync(join(dir, m))) {
+        hasVCS = true;
+        break;
+      }
     }
     if (hasAide || hasVCS) candidates.push({ dir, hasAide, hasVCS });
     const parent = dirname(dir);
@@ -79,15 +92,22 @@ function log(cwd: string, msg: string): void {
     if (!existsSync(logDir)) mkdirSync(logDir, { recursive: true });
     const line = `[${new Date().toISOString()}] [session-end] ${ms()} ${msg}\n`;
     appendFileSync(join(logDir, "session-end.log"), line);
-  } catch { /* best effort */ }
+  } catch {
+    /* best effort */
+  }
 }
 
 /** Find aide binary — inline, no external module imports. */
 function findBinary(cwd?: string): string | null {
-  const pluginRoot = process.env.AIDE_PLUGIN_ROOT || process.env.CLAUDE_PLUGIN_ROOT;
+  const pluginRoot =
+    process.env.AIDE_PLUGIN_ROOT || process.env.CLAUDE_PLUGIN_ROOT;
   let resolvedRoot = pluginRoot;
   if (resolvedRoot) {
-    try { resolvedRoot = realpathSync(resolvedRoot); } catch { /* symlink may not resolve */ }
+    try {
+      resolvedRoot = realpathSync(resolvedRoot);
+    } catch {
+      /* symlink may not resolve */
+    }
     const p = join(resolvedRoot, "bin", "aide");
     if (existsSync(p)) return p;
   }
@@ -97,7 +117,9 @@ function findBinary(cwd?: string): string | null {
   }
   try {
     return whichSync("aide", { nothrow: true });
-  } catch { return null; }
+  } catch {
+    return null;
+  }
 }
 
 function main(): void {
@@ -143,15 +165,40 @@ function main(): void {
       return;
     }
 
+    // Emit a lifecycle trigger so SessionEnd is traceable in the dashboard,
+    // symmetric with session-start and subagent-start/stop. Inlined (no ES
+    // import) to keep this hook's startup cheap. Fire-and-forget.
+    try {
+      execFileSync(
+        binary,
+        [
+          "observe",
+          "record",
+          "--kind=session",
+          "--name=session-end",
+          "--category=lifecycle",
+          `--session=${sessionId}`,
+        ],
+        { cwd, timeout: 3000, stdio: ["pipe", "pipe", "pipe"] },
+      );
+    } catch {
+      // Non-fatal telemetry — never block session end.
+    }
+
     // Mode guard: skip cleanup if autopilot is active (session is continuing,
     // not ending). This matters when Codex CLI invokes session-end from Stop
     // hook since there's no separate SessionEnd event.
     try {
       const mode = execFileSync(binary, ["state", "get", "mode"], {
-        cwd, timeout: 500, encoding: "utf-8",
+        cwd,
+        timeout: 500,
+        encoding: "utf-8",
       }).trim();
       if (mode === "autopilot") {
-        log(cwd, "autopilot mode active, skipping cleanup (session continuing)");
+        log(
+          cwd,
+          "autopilot mode active, skipping cleanup (session continuing)",
+        );
         return;
       }
     } catch {
@@ -160,7 +207,9 @@ function main(): void {
 
     log(cwd, `spawning cleanup: session end --session=${sessionId}`);
     const child = spawn(binary, ["session", "end", `--session=${sessionId}`], {
-      cwd, detached: true, stdio: "ignore",
+      cwd,
+      detached: true,
+      stdio: "ignore",
     });
     child.unref();
     log(cwd, "cleanup spawned (detached)");

package/src/hooks/session-start.ts

@@ -1,4 +1,4 @@
-#!/usr/bin/env node
+#!/usr/bin/env bun
 /**
  * Session Start Hook (SessionStart)
  *
@@ -24,10 +24,20 @@ import { join, dirname } from "path";
 import { fileURLToPath } from "url";
 import { homedir } from "os";
 import { Logger, debug, setDebugCwd } from "../lib/logger.js";
-import { readStdin, detectPlatform, isFalsy } from "../lib/hook-utils.js";
+import {
+  readStdin,
+  detectPlatform,
+  isFalsy,
+  emitHookResult,
+  installHookSafetyNet,
+} from "../lib/hook-utils.js";
 import { findAideBinary, ensureAideBinary } from "../lib/aide-downloader.js";
 import { findProjectRoot } from "../lib/project-root.js";
-import { emitInjectionEvent } from "../core/read-tracking.js";
+import {
+  emitInjectionEvent,
+  recordObserveEvent,
+} from "../core/read-tracking.js";
+import { buildResumeContext } from "../core/session-resume-logic.js";
 import {
   ensureDirectories as coreEnsureDirectories,
   loadConfig as coreLoadConfig,
@@ -58,10 +68,13 @@ interface HookInput {
   cwd: string;
   transcript_path?: string;
   permission_mode?: string;
+  /** How the session began: startup | resume | clear | compact */
+  source?: string;
 }
 
 interface HookOutput {
   continue: boolean;
+  suppressOutput?: boolean;
   hookSpecificOutput?: {
     hookEventName: string;
     additionalContext?: string;
@@ -189,7 +202,11 @@ function installHudWrapper(log: Logger): void {
 
     // Copy wrapper script
     copyFileSync(wrapperSrc, wrapperDest);
-    try { chmodSync(wrapperDest, 0o755); } catch { /* no-op on Windows */ }
+    try {
+      chmodSync(wrapperDest, 0o755);
+    } catch {
+      /* no-op on Windows */
+    }
 
     log.info(`Installed HUD wrapper to ${wrapperDest}`);
     log.end("installHudWrapper", { success: true, path: wrapperDest });
@@ -301,28 +318,8 @@ function debugLog(msg: string): void {
   debug(SOURCE, msg);
 }
 
-// Ensure we always output valid JSON, even on catastrophic errors
-function outputContinue(): void {
-  try {
-    console.log(JSON.stringify({ continue: true }));
-  } catch {
-    // Last resort - raw JSON string
-    console.log('{"continue":true}');
-  }
-}
-
 // Global error handlers to prevent hook crashes without JSON output
-process.on("uncaughtException", (err) => {
-  debugLog(`UNCAUGHT EXCEPTION: ${err}`);
-  outputContinue();
-  process.exit(0);
-});
-
-process.on("unhandledRejection", (reason) => {
-  debugLog(`UNHANDLED REJECTION: ${reason}`);
-  outputContinue();
-  process.exit(0);
-});
+installHookSafetyNet(SOURCE);
 
 async function main(): Promise<void> {
   let log: Logger | null = null;
@@ -337,7 +334,7 @@ async function main(): Promise<void> {
 
     if (!input.trim()) {
       debugLog("Empty input, exiting");
-      console.log(JSON.stringify({ continue: true }));
+      emitHookResult();
       return;
     }
 
@@ -356,7 +353,7 @@ async function main(): Promise<void> {
             `Set \`requireGit\`: false in ~/.aide/config/aide.json to allow ` +
             `init in arbitrary directories. Skipping AIDE bootstrap.\n`,
         );
-        console.log(JSON.stringify({ continue: true }));
+        emitHookResult();
         return;
       }
       process.stderr.write(
@@ -485,12 +482,52 @@ async function main(): Promise<void> {
     // Build welcome context with injected memories
     debugLog("buildWelcomeContext starting...");
     log.start("buildWelcomeContext");
-    const context = buildWelcomeContext(state, memories, notices);
+    let context = buildWelcomeContext(state, memories, notices);
     log.end("buildWelcomeContext");
     debugLog(`buildWelcomeContext complete (${Date.now() - hookStart}ms)`);
 
+    // On resume/compact, re-inject the last session checkpoint so the agent
+    // rebuilds working context instead of relearning it (MiMo-style context
+    // reconstruction, done at the SessionStart boundary).
+    try {
+      const resumeBinary = findAideBinary(cwd);
+      if (resumeBinary) {
+        const resume = buildResumeContext(
+          resumeBinary,
+          cwd,
+          sessionId,
+          data.source,
+        );
+        if (resume) {
+          context = `${context}\n\n${resume}`;
+          debugLog(`Injected resume checkpoint (source=${data.source})`);
+          recordObserveEvent(resumeBinary, cwd, {
+            kind: "injection",
+            name: "resume-checkpoint",
+            category: "resume",
+            subtype: data.source || "resume",
+            session: sessionId,
+          });
+        }
+      }
+    } catch (err) {
+      debugLog(`Resume checkpoint injection failed (non-fatal): ${err}`);
+    }
+
     try {
       const binary = findAideBinary(cwd);
+      if (binary) {
+        // Emit a lifecycle trigger so SessionStart is traceable in the
+        // dashboard, symmetric with subagent-start/stop. Fires regardless of
+        // whether any memories were injected.
+        recordObserveEvent(binary, cwd, {
+          kind: "session",
+          name: "session-start",
+          category: "lifecycle",
+          subtype: data.source || "startup",
+          session: sessionId,
+        });
+      }
       if (binary && context && memories.sources) {
         for (const src of memories.sources) {
           const attrs: Record<string, string> = { scope: src.scope };
@@ -523,13 +560,14 @@ async function main(): Promise<void> {
 
     const output: HookOutput = {
       continue: true,
+      ...(process.env.AIDE_DEBUG ? {} : { suppressOutput: true }),
       hookSpecificOutput: {
         hookEventName: "SessionStart",
         additionalContext: context,
       },
     };
 
-    console.log(JSON.stringify(output));
+    emitHookResult(output);
   } catch (error) {
     debugLog(`ERROR: ${error}`);
     // Log error if logger is available
@@ -538,7 +576,7 @@ async function main(): Promise<void> {
       log.flush();
     }
     // On error, allow continuation without context
-    console.log(JSON.stringify({ continue: true }));
+    emitHookResult();
   }
 }
 

package/src/hooks/session-summary.ts

@@ -10,8 +10,12 @@
  */
 
 import { debug, setDebugCwd } from "../lib/logger.js";
-import { readStdin } from "../lib/hook-utils.js";
-import { findAideBinary } from "../core/aide-client.js";
+import {
+  readStdin,
+  emitHookResult,
+  installHookSafetyNet,
+  findAideBinary,
+} from "../lib/hook-utils.js";
 import {
   buildSessionSummary,
   getSessionCommits,
@@ -45,10 +49,7 @@ function captureSessionSummary(
   sessionId: string,
   transcriptPath: string,
 ): boolean {
-  const binary = findAideBinary({
-    cwd,
-    pluginRoot: process.env.AIDE_PLUGIN_ROOT || process.env.CLAUDE_PLUGIN_ROOT,
-  });
+  const binary = findAideBinary(cwd);
   if (!binary) {
     debug(SOURCE, "aide binary not found, cannot capture session summary");
     return false;
@@ -98,7 +99,7 @@ async function main(): Promise<void> {
   try {
     const input = await readStdin();
     if (!input.trim()) {
-      console.log(JSON.stringify({ continue: true }));
+      emitHookResult({ continue: true });
       return;
     }
 
@@ -118,30 +119,13 @@ async function main(): Promise<void> {
       }
     }
 
-    console.log(JSON.stringify({ continue: true }));
+    emitHookResult({ continue: true });
   } catch (err) {
     debug(SOURCE, `Error: ${err}`);
-    console.log(JSON.stringify({ continue: true }));
+    emitHookResult({ continue: true });
   }
 }
 
-process.on("uncaughtException", (err) => {
-  debug(SOURCE, `UNCAUGHT EXCEPTION: ${err}`);
-  try {
-    console.log(JSON.stringify({ continue: true }));
-  } catch {
-    console.log('{"continue":true}');
-  }
-  process.exit(0);
-});
-process.on("unhandledRejection", (reason) => {
-  debug(SOURCE, `UNHANDLED REJECTION: ${reason}`);
-  try {
-    console.log(JSON.stringify({ continue: true }));
-  } catch {
-    console.log('{"continue":true}');
-  }
-  process.exit(0);
-});
+installHookSafetyNet(SOURCE);
 
 main();

package/src/hooks/skill-injector.ts

@@ -18,7 +18,13 @@
 import { existsSync, mkdirSync } from "fs";
 import { join } from "path";
 import { Logger, debug, setDebugCwd } from "../lib/logger.js";
-import { readStdin, detectPlatform } from "../lib/hook-utils.js";
+import {
+  readStdin,
+  detectPlatform,
+  emitHookResult,
+  installHookSafetyNet,
+  findAideBinary,
+} from "../lib/hook-utils.js";
 import { findProjectRoot } from "../lib/project-root.js";
 import {
   discoverSkills as coreDiscoverSkills,
@@ -26,7 +32,6 @@ import {
   formatSkillsContext as coreFormatSkillsContext,
 } from "../core/skill-matcher.js";
 import type { Skill } from "../core/types.js";
-import { findAideBinary } from "../core/aide-client.js";
 import {
   recordObserveEvent,
   previewContent,
@@ -124,28 +129,8 @@ function debugLog(msg: string): void {
   debug(SOURCE, msg);
 }
 
-// Ensure we always output valid JSON, even on catastrophic errors
-function outputContinue(): void {
-  try {
-    console.log(JSON.stringify({ continue: true }));
-  } catch {
-    // Last resort - raw JSON string
-    console.log('{"continue":true}');
-  }
-}
-
 // Global error handlers to prevent hook crashes without JSON output
-process.on("uncaughtException", (err) => {
-  debugLog(`UNCAUGHT EXCEPTION: ${err}`);
-  outputContinue();
-  process.exit(0);
-});
-
-process.on("unhandledRejection", (reason) => {
-  debugLog(`UNHANDLED REJECTION: ${reason}`);
-  outputContinue();
-  process.exit(0);
-});
+installHookSafetyNet(SOURCE);
 
 async function main(): Promise<void> {
   const hookStart = Date.now();
@@ -158,7 +143,7 @@ async function main(): Promise<void> {
 
     if (!input.trim()) {
       debugLog("Empty input, exiting");
-      console.log(JSON.stringify({ continue: true }));
+      emitHookResult({ continue: true });
       return;
     }
 
@@ -190,7 +175,7 @@ async function main(): Promise<void> {
     // sessions that aren't running reflect don't accrue extra events.
     if (prompt && reflectEnabled(cwd)) {
       try {
-        const binary = findAideBinary({ cwd });
+        const binary = findAideBinary(cwd);
         if (binary) {
           recordObserveEvent(binary, cwd, {
             kind: "hook",
@@ -211,7 +196,7 @@ async function main(): Promise<void> {
       log.info("No prompt provided");
       log.end("total");
       log.flush();
-      console.log(JSON.stringify({ continue: true }));
+      emitHookResult({ continue: true });
       return;
     }
 
@@ -242,7 +227,7 @@ async function main(): Promise<void> {
       // single "skill-injector" aggregate). Subtype="skill" keeps the
       // category roll-up; Name carries the per-skill identifier.
       try {
-        const binary = findAideBinary({ cwd });
+        const binary = findAideBinary(cwd);
         if (binary) {
           for (const skill of matched) {
             const text = `### ${skill.name}\n${skill.description ?? ""}\n${skill.content}`;
@@ -270,13 +255,13 @@ async function main(): Promise<void> {
           additionalContext: skillContext,
         },
       };
-      console.log(JSON.stringify(output));
+      emitHookResult(output);
     } else {
       log.info("No matching skills");
       debugLog(`Flushing logs...`);
       log.flush();
       debugLog(`Hook complete (${Date.now() - hookStart}ms total)`);
-      console.log(JSON.stringify({ continue: true }));
+      emitHookResult({ continue: true });
     }
   } catch (error) {
     debugLog(`ERROR: ${error}`);
@@ -286,7 +271,7 @@ async function main(): Promise<void> {
       log.flush();
     }
     // On error, allow continuation
-    console.log(JSON.stringify({ continue: true }));
+    emitHookResult({ continue: true });
   }
 }
 

package/src/hooks/subagent-tracker.ts

@@ -17,9 +17,18 @@
 import { execFileSync } from "child_process";
 import { basename } from "path";
 import { Logger } from "../lib/logger.js";
-import { readStdin, setMemoryState, isFalsy } from "../lib/hook-utils.js";
-import { findAideBinary } from "../core/aide-client.js";
-import { emitInjectionEvent } from "../core/read-tracking.js";
+import {
+  readStdin,
+  setMemoryState,
+  isFalsy,
+  emitHookResult,
+  installHookSafetyNet,
+  findAideBinary,
+} from "../lib/hook-utils.js";
+import {
+  emitInjectionEvent,
+  recordObserveEvent,
+} from "../core/read-tracking.js";
 import { refreshHud } from "../lib/hud.js";
 
 // Global logger instance
@@ -117,10 +126,7 @@ function fetchSubagentMemories(cwd: string): {
     return result;
   }
 
-  const binary = findAideBinary({
-    cwd,
-    pluginRoot: process.env.AIDE_PLUGIN_ROOT || process.env.CLAUDE_PLUGIN_ROOT,
-  });
+  const binary = findAideBinary(cwd);
   if (!binary) {
     return result;
   }
@@ -341,6 +347,20 @@ async function processSubagentStart(
   log?.info(
     `Injecting context for subagent: ${memories.global.length} preferences, ${memories.project.length} project, ${memories.decisions.length} decisions`,
   );
+
+  // Emit session observe event so SubagentStart is traceable in the dashboard
+  const binary = findAideBinary(cwd);
+  if (binary) {
+    recordObserveEvent(binary, cwd, {
+      kind: "session",
+      name: "subagent-start",
+      category: "lifecycle",
+      subtype: type,
+      session: session_id,
+      attrs: { agent_id, agent_type: type },
+    });
+  }
+
   return context;
 }
 
@@ -365,6 +385,18 @@ async function processSubagentStop(data: SubagentStopInput): Promise<void> {
   refreshHud(cwd, session_id);
   log?.end("refreshHud");
 
+  // Emit session observe event so SubagentStop is traceable in the dashboard
+  const binary = findAideBinary(cwd);
+  if (binary) {
+    recordObserveEvent(binary, cwd, {
+      kind: "session",
+      name: "subagent-stop",
+      category: "lifecycle",
+      session: session_id,
+      attrs: { agent_id, agent_type: data.agent_type ?? "" },
+    });
+  }
+
   log?.debug(`SubagentStop: agent ${agent_id} marked as completed`);
 }
 
@@ -372,7 +404,7 @@ async function main(): Promise<void> {
   try {
     const input = await readStdin();
     if (!input.trim()) {
-      console.log(JSON.stringify({ continue: true }));
+      emitHookResult({ continue: true });
       return;
     }
 
@@ -412,11 +444,7 @@ async function main(): Promise<void> {
         additionalContext,
       };
       try {
-        const binary = findAideBinary({
-          cwd,
-          pluginRoot:
-            process.env.AIDE_PLUGIN_ROOT || process.env.CLAUDE_PLUGIN_ROOT,
-        });
+        const binary = findAideBinary(cwd);
         if (binary) {
           const startData = data as SubagentStartInput;
           emitInjectionEvent(binary, cwd, {
@@ -438,40 +466,17 @@ async function main(): Promise<void> {
       }
     }
 
-    console.log(JSON.stringify(output));
+    emitHookResult(output);
   } catch (error) {
     // Log error but don't block
     if (log) {
       log.error("Subagent tracker failed", error);
       log.flush();
     }
-    console.log(JSON.stringify({ continue: true }));
+    emitHookResult({ continue: true });
   }
 }
 
-process.on("uncaughtException", (err) => {
-  if (log) {
-    log.error(`UNCAUGHT EXCEPTION: ${err}`);
-    log.flush();
-  }
-  try {
-    console.log(JSON.stringify({ continue: true }));
-  } catch {
-    console.log('{"continue":true}');
-  }
-  process.exit(0);
-});
-process.on("unhandledRejection", (reason) => {
-  if (log) {
-    log.error(`UNHANDLED REJECTION: ${reason}`);
-    log.flush();
-  }
-  try {
-    console.log(JSON.stringify({ continue: true }));
-  } catch {
-    console.log('{"continue":true}');
-  }
-  process.exit(0);
-});
+installHookSafetyNet("subagent-tracker");
 
 main();

package/src/hooks/task-completed.ts

@@ -26,7 +26,11 @@ import { existsSync, readFileSync } from "fs";
 import { join } from "path";
 import which from "which";
 import { debug, setDebugCwd } from "../lib/logger.js";
-import { readStdin } from "../lib/hook-utils.js";
+import {
+  readStdin,
+  emitHookResult,
+  installHookSafetyNet,
+} from "../lib/hook-utils.js";
 
 const SOURCE = "task-completed";
 
@@ -416,13 +420,6 @@ async function main(): Promise<void> {
   }
 }
 
-process.on("uncaughtException", (err) => {
-  debug(SOURCE, `UNCAUGHT EXCEPTION: ${err}`);
-  process.exit(0);
-});
-process.on("unhandledRejection", (reason) => {
-  debug(SOURCE, `UNHANDLED REJECTION: ${reason}`);
-  process.exit(0);
-});
+installHookSafetyNet(SOURCE);
 
 main();