mainflow 0.27.0 → 0.29.0-alpha.1

package/dist/hooks.js

@@ -1,5 +1,6 @@
-import { B as setSessionStatus, G as grantScope, I as loadSessions, J as endsWithMarker, K as matchGrants, L as readSession, M as turnReminder, P as ensureSession, V as settleSession, X as mayDecorate, Y as headerLine, Z as appendOps, _t as DATA_DIR, bt as worktreesOf, gt as readYaml, i as hostName, j as sessionBriefing, ot as HubError, q as decorateReply, st as loadHub, vt as currentBranch, yt as findHubRoot, z as setSessionInfo } from "./host-sSWrUJeC.js";
-import { Z as describeRequest, j as configSchema } from "./dist-CBGKXsoi.js";
+import { D as endsWithMarker, E as decorateReply, G as loadHub, O as headerLine, S as readSession, T as matchGrants, W as HubError, b as turnReminder, et as readYaml, i as hostName, it as worktreesOf, k as mayDecorate, nt as currentBranch, ot as DATA_DIR, rt as findHubRoot, w as grantScope, x as loadSessions, y as sessionBriefing } from "./host-BJnrv8xF.js";
+import { Z as describeRequest, j as configSchema } from "./dist-BbamrIPW.js";
+import { n as hubWriter, r as writtenLines } from "./operation-Bp4U2bx3.js";
 import { resolveFixed, scanCommand, scopedCwd } from "./grant-commands.js";
 import { appendFileSync, existsSync, readFileSync } from "node:fs";
 import { basename, dirname, isAbsolute, join, relative, resolve } from "node:path";
@@ -74,24 +75,27 @@ const claudeCodeSessionId = (raw) => `claude-code-${shortId(raw)}`;
 * bring an ended or archived one back, since the viewer is working in it again. Its environment facts are
 * written when it is created and whenever `refresh` asks, such as on SessionStart after a resume.
 */
-function recordSession(root, payload, id, refresh) {
+async function recordSession(write, root, payload, id, refresh) {
 	const cwd = payload.cwd ?? root;
-	const { session, created } = ensureSession(root, {
+	const { session, created } = (await write("session.ensure", {
 		id,
 		agent: "claude-code",
 		cwd
-	});
+	})).value;
 	if (!created && !refresh) return session;
-	return setSessionInfo(root, id, {
-		transcript: payload.transcript_path,
-		worktree: worktreesOf(cwd)?.current,
-		branch: currentBranch(cwd) ?? void 0,
-		effort: payload.effort?.level
-	});
+	return (await write("session.info", {
+		id,
+		info: {
+			transcript: payload.transcript_path,
+			worktree: worktreesOf(cwd)?.current,
+			branch: currentBranch(cwd) ?? void 0,
+			effort: payload.effort?.level
+		}
+	})).value;
 }
 /** Settle the main session by the rules settleSession states; prints nothing. */
-const settleOnEnd = (root, { payload }) => {
-	if (payload.session_id) settleSession(root, claudeCodeSessionId(payload.session_id));
+const settleOnEnd = async (_root, { payload, write }) => {
+	if (payload.session_id) await write("session.settle", { id: claudeCodeSessionId(payload.session_id) });
 	return null;
 };
 /** How a hub's interface is linked from a reply: through its /open route, which shows the address in the app. */
@@ -125,10 +129,55 @@ function openMainlinesIn(root, text) {
 	if (mentioned.size === 0) return [];
 	return loadHub(root, { remotes: false }).mainlines.filter((m) => mentioned.has(m.id) && !m.archivedAt && (m.status === "active" || m.status === "planning")).map((m) => m.id);
 }
+/** Tools whose `file_path` input writes a file. */
+const WRITE_TOOLS = /* @__PURE__ */ new Set([
+	"Edit",
+	"Write",
+	"MultiEdit",
+	"NotebookEdit"
+]);
+/** A Bash command that runs the mainflow CLI: by the form the hook texts give, or as a `mainflow` command word that starts a command, not one that is only an argument. */
+const isMainflowCommand = (command, cli) => command.includes(cli) || /(?:^|[;&|(])\s*mainflow\s+\S|(?:\.bin|mainflow\/dist)\/(?:cli\.js|mainflow)\b/.test(command);
+/** Whether a path lies under a `.mainflow/` directory. */
+const inMainflowData = (path) => /(?:^|\/)\.mainflow\//.test(path);
+/** A user record that holds what the user typed; tool results come back as user records too, but carry only tool_result blocks. */
+const isUserPrompt = (r) => r.type === "user" && !r.isSidechain && (typeof r.message?.content === "string" || Array.isArray(r.message?.content) && r.message.content.some((b) => b.type === "text"));
+/**
+* Whether the turn that is ending, everything after the last user message in the transcript, ran a mainflow
+* command or wrote a file under `.mainflow/`. False when the transcript is missing or unreadable, so a hook that
+* cannot tell never asks.
+*/
+function turnTouchedMainflow(transcript, cli) {
+	if (!transcript || !existsSync(transcript)) return false;
+	let lines;
+	try {
+		lines = readFileSync(transcript, "utf8").split("\n");
+	} catch {
+		return false;
+	}
+	for (let i = lines.length - 1; i >= 0; i--) {
+		if (!lines[i].trim()) continue;
+		let record;
+		try {
+			record = JSON.parse(lines[i]);
+		} catch {
+			continue;
+		}
+		if (isUserPrompt(record)) return false;
+		const content = Array.isArray(record.message?.content) ? record.message.content : [];
+		for (const block of content) {
+			if (block.type !== "tool_use") continue;
+			if (block.name === "Bash" && typeof block.input?.command === "string" && isMainflowCommand(block.input.command, cli)) return true;
+			if (block.name && WRITE_TOOLS.has(block.name) && typeof block.input?.file_path === "string" && inMainflowData(block.input.file_path)) return true;
+		}
+	}
+	return false;
+}
 /**
 * What to ask for when a turn ends without its header line: the header itself when an action is in progress, or
-* an action first when the queue is empty but the reply talks about an open mainline, since the header only names
-* the session's own action in progress. Null when there is nothing to ask for.
+* an action first when the queue is empty, the reply talks about an open mainline and the turn ran a mainflow
+* command or wrote under `.mainflow/`, since the header only names the session's own action in progress. A reply
+* that only mentions a mainline, in a draft or a question, asks for nothing. Null when there is nothing to ask for.
 */
 function missingHeaderRequest(root, { payload, cli }) {
 	if (!payload.session_id || payload.stop_hook_active) return null;
@@ -139,8 +188,8 @@ function missingHeaderRequest(root, { payload, cli }) {
 	if (header) return `[mainflow] 這一輪最後一則回覆的最後一行不是標頭。請只回一則內容為「${header}」的訊息，不加其他文字。`;
 	if (session && session.actions.length > 0) return null;
 	const mainlines = openMainlinesIn(root, message);
-	if (mainlines.length === 0) return null;
-	return `[mainflow] 這一輪的回覆提到 ${mainlines.join("、")}，但這場會話沒有行動，所以回覆結尾沒有標頭。這一輪若在推進它，先用 ${cli} action add "<做法>" --serves <m/t#c>（還沒拆任務時用解決條件 <m#r>）列出行動並用 ${cli} action set <a編號> --status doing 標為進行中，或用 ${cli} session handoff <會話> 接手上一場沒做完的行動；再只回一則內容為「▸ <m/t> · <a編號>」的訊息，不加其他文字。只是提到、沒有推進時，不必處理。`;
+	if (mainlines.length === 0 || !turnTouchedMainflow(payload.transcript_path, cli)) return null;
+	return `[mainflow] 這一輪的回覆提到 ${mainlines.join("、")}，也執行了 mainflow 指令或寫入 .mainflow/，但這場會話沒有行動，所以回覆結尾沒有標頭。這一輪若在推進它，先用 ${cli} action add "<做法>" --serves <m/t#c>（還沒拆任務時用解決條件 <m#r>）列出行動並用 ${cli} action set <a編號> --status doing 標為進行中，或用 ${cli} session handoff <會話> 接手上一場沒做完的行動；再只回一則內容為「▸ <m/t> · <a編號>」的訊息，不加其他文字。只是提到、沒有推進時，不必處理。`;
 }
 /**
 * Stop: ask for what the turn's header needs, then settle the session. Only the model knows which message closes a
@@ -148,20 +197,24 @@ function missingHeaderRequest(root, { payload, cli }) {
 * stops it from asking twice in a row. A session asked to list an action is not settled yet: settling removes a
 * session with no actions, and the action it is asked to add needs the session to exist.
 */
-const settleAndCheckMarker = (root, context) => {
+const settleAndCheckMarker = async (root, context) => {
 	const request = missingHeaderRequest(root, context);
 	const session = context.payload.session_id ? readSession(root, claudeCodeSessionId(context.payload.session_id)) : null;
-	if (!request || session && session.actions.length > 0) settleOnEnd(root, context);
+	if (!request || session && session.actions.length > 0) await settleOnEnd(root, context);
 	return request ? JSON.stringify({ hookSpecificOutput: {
 		hookEventName: "Stop",
 		additionalContext: request
 	} }) : null;
 };
 const CLAUDE_CODE_EVENTS = {
-	"session-start": (root, { payload, envFile, cli, startUi }) => {
+	"session-start": async (root, { payload, envFile, cli, startUi, write }) => {
 		if (!payload.session_id) return null;
 		const id = claudeCodeSessionId(payload.session_id);
-		const session = recordSession(root, payload, id, true);
+		const session = await recordSession(write, root, payload, id, true);
+		await write("session.reminder", {
+			id,
+			digest: null
+		});
 		if (envFile) appendFileSync(envFile, `export MAINFLOW_SESSION=${id}\n`);
 		startUi?.(root);
 		return sessionBriefing(root, session, {
@@ -169,16 +222,26 @@ const CLAUDE_CODE_EVENTS = {
 			envNote: envFile ? "已設定環境變數 MAINFLOW_SESSION，指令不必再帶 --session" : `執行指令時帶 --session ${id}`
 		});
 	},
-	"user-prompt-submit": (root, { payload, cli }) => {
+	"user-prompt-submit": async (root, { payload, cli, write }) => {
 		if (!payload.session_id) return null;
-		return turnReminder(root, recordSession(root, payload, claudeCodeSessionId(payload.session_id), false), { cli });
+		const session = await recordSession(write, root, payload, claudeCodeSessionId(payload.session_id), false);
+		const saved = [];
+		const text = turnReminder(root, session, {
+			cli,
+			saveDigest: (digest) => saved.push(write("session.reminder", {
+				id: session.id,
+				digest
+			}).catch(() => void 0))
+		});
+		await Promise.all(saved);
+		return text;
 	},
 	stop: settleAndCheckMarker,
 	"session-end": settleOnEnd,
-	"subagent-start": (root, { payload }) => {
+	"subagent-start": async (root, { payload, write }) => {
 		if (!payload.agent_id) return null;
 		const parent = payload.session_id ? claudeCodeSessionId(payload.session_id) : null;
-		ensureSession(root, {
+		await write("session.ensure", {
 			id: claudeCodeSessionId(payload.agent_id),
 			agent: payload.agent_type ? `claude-code/${payload.agent_type}` : "claude-code",
 			cwd: payload.cwd ?? root,
@@ -186,18 +249,24 @@ const CLAUDE_CODE_EVENTS = {
 		});
 		return null;
 	},
-	"subagent-stop": (root, { payload }) => {
+	"subagent-stop": async (root, { payload, write }) => {
 		const id = payload.agent_id ? claudeCodeSessionId(payload.agent_id) : null;
 		if (!id || !readSession(root, id)) return null;
 		const transcript = payload.transcript_path ? subagentTranscript(payload.transcript_path, payload.agent_id) : void 0;
 		const meta = transcript ? readMeta(transcript) : {};
-		setSessionInfo(root, id, {
-			transcript: transcript && existsSync(transcript) ? transcript : void 0,
-			description: meta.description,
-			depth: meta.spawnDepth,
-			summary: brief(payload.last_assistant_message) || void 0
+		await write("session.info", {
+			id,
+			info: {
+				transcript: transcript && existsSync(transcript) ? transcript : void 0,
+				description: meta.description,
+				depth: meta.spawnDepth,
+				summary: brief(payload.last_assistant_message) || void 0
+			}
+		});
+		await write("session.set", {
+			id,
+			status: "ended"
 		});
-		setSessionStatus(root, id, "ended");
 		return null;
 	},
 	"pre-tool-use": decideByGrants,
@@ -243,9 +312,9 @@ function askWhenGrantable({ payload }, failure) {
 /** The ops log's fields cannot hold its separator, so notes that contain one are written with a broken bar. */
 const opsText = (text) => text.replace(/\s\|\s/g, " ¦ ").replace(/\s+/g, " ").slice(0, 200);
 /** Log the allowed operation with the grant that covered it; a failure only warns, the call stays allowed. */
-function logGranted(root, scope, call, request, grant, warn) {
+async function logGranted(write, scope, call, request, grant, warn) {
 	try {
-		appendOps(root, {
+		await write("ops.add", {
 			target: request.action === "publish" ? `npm ${request.name}` : `github release ${request.repos.join("/")}`,
 			operation: opsText(call.text),
 			result: `授權放行 ${grant.ref}（${describeRequest(request)}）`,
@@ -263,7 +332,7 @@ function logGranted(root, scope, call, request, grant, warn) {
 * print nothing, so other hooks and the usual permission rules decide them. Any failure asks: the caller turns a
 * throw into the fallback above.
 */
-function decideByGrants(root, { payload, warn }) {
+async function decideByGrants(root, { payload, warn, write }) {
 	const scan = grantableCall(payload);
 	if (!scan) return null;
 	if (scan.kind === "refused") return askBecause(scan.reason);
@@ -283,7 +352,7 @@ function decideByGrants(root, { payload, warn }) {
 	const match = matchGrants(scoped.scope, [resolved.request]);
 	if (!match.ok) return askBecause(match.reason);
 	const { grant } = match.matches[0];
-	logGranted(root, scoped.scope, scan.call, resolved.request, grant, warn);
+	await logGranted(write, scoped.scope, scan.call, resolved.request, grant, warn);
 	return permission("allow", `主脈絡授權 ${grant.ref} 涵蓋 ${describeRequest(resolved.request)}`);
 }
 /** How many problems outside the written file the reminder lists before it only gives the count. */
@@ -371,7 +440,7 @@ function hookFallback(platform, event, ctx, failure) {
 * failure is reported on stderr while the caller still exits 0. Events listed in the fallbacks print their
 * fallback instead of staying silent.
 */
-function runHook(platform, event, ctx) {
+async function runHook(platform, event, ctx) {
 	const handler = HOOK_PLATFORMS[platform]?.[event];
 	if (!handler) return {
 		stdout: null,
@@ -385,10 +454,18 @@ function runHook(platform, event, ctx) {
 			stdout: hookFallback(platform, event, ctx, { kind: "outside-hub" }),
 			error: null
 		};
+		const warn = (message) => warnings.push(message);
+		const writer = ctx.write ?? hubWriter(root);
+		const write = async (op, input) => {
+			const written = await writer(op, input);
+			for (const line of writtenLines(written)) if (line.startsWith("!")) warn(`mainflow hook ${platform} ${event}: ${line.slice(2)}`);
+			return written;
+		};
 		return {
-			stdout: handler(root, {
+			stdout: await handler(root, {
 				...ctx,
-				warn: (message) => warnings.push(message)
+				warn,
+				write
 			}),
 			error: errorText()
 		};