@membank/cli 0.12.0 → 0.13.1

package/dist/index.mjs

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 import { cancel, confirm, intro, isCancel, multiselect, note, outro } from "@clack/prompts";
 import { DatabaseManager, EmbeddingService, MIGRATIONS, MODEL_NAME, MemoryTypeSchema, MemoryTypeSchema as MemoryTypeSchema$1, ModelDownloader, PIN_BUDGET_THRESHOLD, QueryEngine, SessionContextBuilder, createMemoryRepository, createProjectRepository, createSynthesisRepository, resolveProject, runScopeToProjectsMigration, saveMemory } from "@membank/core";
-import { runSynthesis, startServer } from "@membank/mcp";
+import { runExtraction, runSynthesis, startServer } from "@membank/mcp";
 import chalk from "chalk";
 import { Command } from "commander";
 import ora from "ora";
@@ -192,6 +192,95 @@ function exportCommand(db, formatter, opts) {
 	else process.stdout.write(`Exported ${memories.length} memories to ${outputPath}\n`);
 }
 //#endregion
+//#region src/commands/extract.ts
+const ExtractionHarnessSchema = z.enum(["claude-code"]);
+const ClaudeCodeStopInputSchema = z.object({
+	session_id: z.string(),
+	transcript_path: z.string(),
+	cwd: z.string().optional(),
+	hook_event_name: z.string().optional(),
+	stop_hook_active: z.boolean().optional()
+});
+function parseHookPayload(harness, raw) {
+	let parsedJson;
+	try {
+		parsedJson = JSON.parse(raw);
+	} catch {
+		return {
+			ok: false,
+			reason: "stdin is not valid JSON"
+		};
+	}
+	if (harness === "claude-code") {
+		const parsed = ClaudeCodeStopInputSchema.safeParse(parsedJson);
+		if (!parsed.success) return {
+			ok: false,
+			reason: `invalid hook payload: ${parsed.error.message}`
+		};
+		return {
+			ok: true,
+			value: {
+				sessionId: parsed.data.session_id,
+				transcriptPath: parsed.data.transcript_path,
+				stopHookActive: parsed.data.stop_hook_active === true
+			}
+		};
+	}
+	return {
+		ok: false,
+		reason: `unsupported harness: ${harness}`
+	};
+}
+async function readStdin() {
+	if (process.stdin.isTTY) return "";
+	const chunks = [];
+	for await (const chunk of process.stdin) chunks.push(typeof chunk === "string" ? Buffer.from(chunk) : chunk);
+	return Buffer.concat(chunks).toString("utf8");
+}
+async function extractCommand(opts) {
+	const harnessResult = ExtractionHarnessSchema.safeParse(opts.harness ?? "claude-code");
+	if (!harnessResult.success) {
+		process.stderr.write(`membank extract: unsupported harness "${opts.harness}". Supported: ${ExtractionHarnessSchema.options.join(", ")}\n`);
+		return;
+	}
+	const harness = harnessResult.data;
+	let sessionId = opts.sessionId;
+	let transcriptPath = opts.transcript;
+	let stopHookActive = false;
+	if (sessionId === void 0 || transcriptPath === void 0) {
+		const raw = await readStdin();
+		if (raw.trim().length > 0) {
+			const parsed = parseHookPayload(harness, raw);
+			if (!parsed.ok) {
+				process.stderr.write(`membank extract: ${parsed.reason}; skipping.\n`);
+				return;
+			}
+			sessionId = sessionId ?? parsed.value.sessionId;
+			transcriptPath = transcriptPath ?? parsed.value.transcriptPath;
+			stopHookActive = parsed.value.stopHookActive;
+		}
+	}
+	if (sessionId === void 0 || transcriptPath === void 0) {
+		process.stderr.write("membank extract: missing session_id or transcript_path (provide via stdin or --session/--transcript).\n");
+		return;
+	}
+	if (stopHookActive) {
+		process.stderr.write("membank extract: stop_hook_active=true; skipping to avoid recursion.\n");
+		return;
+	}
+	try {
+		const result = await runExtraction({
+			sessionId,
+			transcriptPath
+		});
+		if (result.status === "skipped") process.stderr.write(`membank extract: skipped (${result.reason})\n`);
+		else if (result.status === "failed") process.stderr.write(`membank extract: failed: ${result.error}\n`);
+	} catch (err) {
+		const msg = err instanceof Error ? err.message : String(err);
+		process.stderr.write(`membank extract: ${msg}\n`);
+	}
+}
+//#endregion
 //#region src/commands/import.ts
 async function importCommand(filePath, db, formatter, prompt) {
 	let raw;
@@ -290,10 +379,6 @@ async function buildText() {
 	}
 }
 async function handleEvent(harness, eventName) {
-	if (harness === "claude-code" && eventName === "Stop") {
-		process.stdout.write("{}");
-		return;
-	}
 	const text = await buildText().catch((err) => {
 		const msg = err instanceof Error ? err.message : String(err);
 		process.stderr.write(`membank inject: ${msg}\n`);
@@ -313,10 +398,6 @@ async function injectCommand(opts) {
 		await handleEvent(harness, "UserPromptSubmit");
 		return;
 	}
-	if (opts.event === "session-stop" || opts.event === "stop") {
-		await handleEvent(harness, "Stop");
-		return;
-	}
 	process.exit(0);
 }
 //#endregion
@@ -1017,7 +1098,7 @@ const writers = {
 					},
 					{
 						event: "Stop",
-						command: "npx -y @membank/cli inject --harness claude-code --event session-stop",
+						command: "npx -y @membank/cli extract --harness claude-code",
 						existingCommand: extractInjectCommand(stopInner) || null
 					}
 				]
@@ -1029,6 +1110,7 @@ const writers = {
 			const hooks = MaybeJsonObjectSchema.parse(cfg.hooks) ?? {};
 			const newHooks = { ...hooks };
 			pruneNestedEvent(newHooks, "PostToolUseFailure");
+			pruneNestedEvent(newHooks, "Stop");
 			if (events.includes("SessionStart")) newHooks.SessionStart = [...filterOutMembank(Array.isArray(hooks.SessionStart) ? hooks.SessionStart : []), {
 				matcher: "",
 				hooks: [{
@@ -1047,7 +1129,9 @@ const writers = {
 				matcher: "",
 				hooks: [{
 					type: "command",
-					command: "npx -y @membank/cli inject --harness claude-code --event session-stop"
+					command: "npx -y @membank/cli extract --harness claude-code",
+					async: true,
+					timeout: 600
 				}]
 			}];
 			writeJsonAtomic(cfgPath, {
@@ -1064,27 +1148,18 @@ const writers = {
 			const hooks = MaybeJsonObjectSchema.parse(cfg.hooks) ?? {};
 			const sessionStart = Array.isArray(hooks.sessionStart) ? hooks.sessionStart : [];
 			const userPromptSubmitted = Array.isArray(hooks.userPromptSubmitted) ? hooks.userPromptSubmitted : [];
-			const sessionEnd = Array.isArray(hooks.sessionEnd) ? hooks.sessionEnd : [];
 			return {
 				status: "ready",
 				configPath: cfgPath,
-				hooks: [
-					{
-						event: "sessionStart",
-						command: "npx -y @membank/cli inject --harness copilot-cli",
-						existingCommand: extractInjectCommand(sessionStart) || null
-					},
-					{
-						event: "userPromptSubmitted",
-						command: "npx -y @membank/cli inject --harness copilot-cli --event user-prompt-submit",
-						existingCommand: extractInjectCommand(userPromptSubmitted) || null
-					},
-					{
-						event: "sessionEnd",
-						command: "npx -y @membank/cli inject --harness copilot-cli --event session-stop",
-						existingCommand: extractInjectCommand(sessionEnd) || null
-					}
-				]
+				hooks: [{
+					event: "sessionStart",
+					command: "npx -y @membank/cli inject --harness copilot-cli",
+					existingCommand: extractInjectCommand(sessionStart) || null
+				}, {
+					event: "userPromptSubmitted",
+					command: "npx -y @membank/cli inject --harness copilot-cli --event user-prompt-submit",
+					existingCommand: extractInjectCommand(userPromptSubmitted) || null
+				}]
 			};
 		},
 		write(resolver, events) {
@@ -1093,6 +1168,7 @@ const writers = {
 			const hooks = MaybeJsonObjectSchema.parse(cfg.hooks) ?? {};
 			const newHooks = { ...hooks };
 			pruneFlatEvent(newHooks, "postToolUseFailure");
+			pruneFlatEvent(newHooks, "sessionEnd");
 			if (events.includes("sessionStart")) newHooks.sessionStart = [...filterOutMembankFlat(Array.isArray(hooks.sessionStart) ? hooks.sessionStart : []), {
 				type: "command",
 				bash: "npx -y @membank/cli inject --harness copilot-cli",
@@ -1103,11 +1179,6 @@ const writers = {
 				bash: "npx -y @membank/cli inject --harness copilot-cli --event user-prompt-submit",
 				timeoutSec: 30
 			}];
-			if (events.includes("sessionEnd")) newHooks.sessionEnd = [...filterOutMembankFlat(Array.isArray(hooks.sessionEnd) ? hooks.sessionEnd : []), {
-				type: "command",
-				bash: "npx -y @membank/cli inject --harness copilot-cli --event session-stop",
-				timeoutSec: 30
-			}];
 			writeJsonAtomic(cfgPath, {
 				version: OptionalNumberSchema.parse(cfg.version) ?? 1,
 				...cfg,
@@ -1123,27 +1194,18 @@ const writers = {
 			const hooks = MaybeJsonObjectSchema.parse(cfg.hooks) ?? {};
 			const sessionStartInner = (Array.isArray(hooks.SessionStart) ? hooks.SessionStart : []).flatMap(getHooksArray);
 			const userPromptSubmitInner = (Array.isArray(hooks.UserPromptSubmit) ? hooks.UserPromptSubmit : []).flatMap(getHooksArray);
-			const stopInner = (Array.isArray(hooks.Stop) ? hooks.Stop : []).flatMap(getHooksArray);
 			return {
 				status: "ready",
 				configPath: cfgPath,
-				hooks: [
-					{
-						event: "SessionStart",
-						command: "npx -y @membank/cli inject --harness codex",
-						existingCommand: extractInjectCommand(sessionStartInner) || null
-					},
-					{
-						event: "UserPromptSubmit",
-						command: "npx -y @membank/cli inject --harness codex --event user-prompt-submit",
-						existingCommand: extractInjectCommand(userPromptSubmitInner) || null
-					},
-					{
-						event: "Stop",
-						command: "npx -y @membank/cli inject --harness codex --event session-stop",
-						existingCommand: extractInjectCommand(stopInner) || null
-					}
-				]
+				hooks: [{
+					event: "SessionStart",
+					command: "npx -y @membank/cli inject --harness codex",
+					existingCommand: extractInjectCommand(sessionStartInner) || null
+				}, {
+					event: "UserPromptSubmit",
+					command: "npx -y @membank/cli inject --harness codex --event user-prompt-submit",
+					existingCommand: extractInjectCommand(userPromptSubmitInner) || null
+				}]
 			};
 		},
 		write(resolver, events) {
@@ -1152,6 +1214,7 @@ const writers = {
 			const hooks = MaybeJsonObjectSchema.parse(cfg.hooks) ?? {};
 			const newHooks = { ...hooks };
 			pruneNestedEvent(newHooks, "PostToolUse");
+			pruneNestedEvent(newHooks, "Stop");
 			if (events.includes("SessionStart")) newHooks.SessionStart = [...filterOutMembank(Array.isArray(hooks.SessionStart) ? hooks.SessionStart : []), {
 				matcher: "",
 				hooks: [{
@@ -1168,14 +1231,6 @@ const writers = {
 					timeout: 30
 				}]
 			}];
-			if (events.includes("Stop")) newHooks.Stop = [...filterOutMembank(Array.isArray(hooks.Stop) ? hooks.Stop : []), {
-				matcher: "",
-				hooks: [{
-					type: "command",
-					command: "npx -y @membank/cli inject --harness codex --event session-stop",
-					timeout: 30
-				}]
-			}];
 			writeJsonAtomic(cfgPath, {
 				...cfg,
 				hooks: newHooks
@@ -1639,6 +1694,18 @@ program.command("inject").description("output session context for harness inject
 		process.exit(2);
 	}
 });
+program.command("extract").description("(internal) run session-end memory extraction; reads the harness's Stop hook payload from stdin").option("--harness <name>", "harness whose stop-hook payload is on stdin (only claude-code is supported today)", "claude-code").option("--session <id>", "session id (otherwise read from stdin)").option("--transcript <path>", "transcript JSONL path (otherwise read from stdin)").action(async (cmdOptions) => {
+	try {
+		await extractCommand({
+			harness: cmdOptions.harness,
+			sessionId: cmdOptions.session,
+			transcript: cmdOptions.transcript
+		});
+	} catch (err) {
+		process.stderr.write(`${err instanceof Error ? err.message : String(err)}\n`);
+	}
+	process.exit(0);
+});
 const setupCmd = program.command("setup").description("detect installed harnesses and write MCP config for each").option("--yes", "skip all confirmation prompts").option("--dry-run", "print planned changes without writing any file").option("--harness <name>", "target only the named harness (skip detection)");
 setupCmd.command("upgrade").description("upgrade harness configs from membank --mcp to standalone membank-mcp").action(async () => {
 	const isJson = program.opts().json === true;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@membank/cli",
-  "version": "0.12.0",
+  "version": "0.13.1",
   "type": "module",
   "repository": {
     "type": "git",
@@ -21,8 +21,8 @@
     "commander": "^14.0.3",
     "ora": "^9.4.0",
     "zod": "^4.4.3",
-    "@membank/core": "0.10.0",
-    "@membank/mcp": "0.13.0"
+    "@membank/mcp": "0.14.1",
+    "@membank/core": "0.11.1"
   },
   "devDependencies": {
     "@types/node": "^25.6.0",