mainflow 0.28.0 → 0.29.0

package/dist/hooks.js

@@ -1,5 +1,6 @@
-import { $ as appendOps, F as ensureSession, H as setSessionStatus, J as matchGrants, L as loadSessions, M as sessionBriefing, N as turnReminder, Q as mayDecorate, R as readSession, St as worktreesOf, U as settleSession, V as setSessionInfo, X as endsWithMarker, Y as decorateReply, Z as headerLine, bt as currentBranch, ct as HubError, i as hostName, lt as loadHub, q as grantScope, vt as readYaml, xt as findHubRoot, yt as DATA_DIR, z as setReminderDigest } from "./host-CeN3dvXg.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. */
@@ -193,21 +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);
-		setReminderDigest(root, id, null);
+		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, {
@@ -215,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,
@@ -232,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,
@@ -289,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)}）`,
@@ -309,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);
@@ -329,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. */
@@ -417,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,
@@ -431,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()
 		};