engrm 0.1.0

package/hooks/stop.ts

@@ -0,0 +1,131 @@
+#!/usr/bin/env bun
+/**
+ * Stop hook for Claude Code.
+ *
+ * Fires when Claude finishes responding. We:
+ *   1. Check stop_hook_active to prevent infinite loops
+ *   2. Complete the session in the database
+ *   3. Generate a retrospective summary from session observations
+ *   4. Exit 0 to allow Claude to stop
+ */
+
+import { loadConfig, configExists, getDbPath } from "../src/config.js";
+import { MemDatabase } from "../src/storage/sqlite.js";
+import { extractRetrospective } from "../src/capture/retrospective.js";
+
+import type { InsertSessionSummary } from "../src/storage/sqlite.js";
+
+function printRetrospective(summary: InsertSessionSummary): void {
+  const lines: string[] = [];
+  lines.push("");
+  lines.push("━━━ Engrm Session Summary ━━━");
+  lines.push("");
+
+  if (summary.request) {
+    lines.push(`📋 Request: ${summary.request}`);
+    lines.push("");
+  }
+  if (summary.investigated) {
+    lines.push("🔍 Investigated:");
+    lines.push(summary.investigated);
+    lines.push("");
+  }
+  if (summary.learned) {
+    lines.push("💡 Learned:");
+    lines.push(summary.learned);
+    lines.push("");
+  }
+  if (summary.completed) {
+    lines.push("✅ Completed:");
+    lines.push(summary.completed);
+    lines.push("");
+  }
+  if (summary.next_steps) {
+    lines.push("➡️  Next Steps:");
+    lines.push(summary.next_steps);
+    lines.push("");
+  }
+
+  lines.push("━━━━━━━━━━━━━━━━━━━━━━━━━━━━━");
+  console.log(lines.join("\n"));
+}
+
+interface StopEvent {
+  session_id: string;
+  hook_event_name: string;
+  stop_hook_active: boolean;
+  last_assistant_message: string;
+  cwd: string;
+}
+
+async function main(): Promise<void> {
+  // Read stdin
+  const chunks: string[] = [];
+  for await (const chunk of Bun.stdin.stream()) {
+    chunks.push(new TextDecoder().decode(chunk));
+  }
+  const raw = chunks.join("");
+
+  if (!raw.trim()) process.exit(0);
+
+  let event: StopEvent;
+  try {
+    event = JSON.parse(raw) as StopEvent;
+  } catch {
+    process.exit(0);
+  }
+
+  // Prevent infinite loops — if we're already in a stop hook, just exit
+  if (event.stop_hook_active) process.exit(0);
+
+  // Bail if not configured
+  if (!configExists()) process.exit(0);
+
+  let config;
+  let db;
+  try {
+    config = loadConfig();
+    db = new MemDatabase(getDbPath());
+  } catch {
+    process.exit(0);
+  }
+
+  try {
+    // Complete the session
+    if (event.session_id) {
+      db.completeSession(event.session_id);
+
+      // Generate retrospective — only if we haven't already for this session
+      const existing = db.getSessionSummary(event.session_id);
+      if (!existing) {
+        const observations = db.getObservationsBySession(event.session_id);
+        if (observations.length > 0) {
+          const session = db.getSessionMetrics(event.session_id);
+          const summary = extractRetrospective(
+            observations,
+            event.session_id,
+            session?.project_id ?? null,
+            config.user_id
+          );
+
+          if (summary) {
+            const row = db.insertSessionSummary(summary);
+            db.addToOutbox("summary", row.id);
+
+            // Display session retrospective to the user
+            printRetrospective(summary);
+          }
+        }
+      }
+    }
+  } finally {
+    db.close();
+  }
+
+  // Exit 0 — allow Claude to stop
+  process.exit(0);
+}
+
+main().catch(() => {
+  process.exit(0);
+});