zcode-acp-server 0.14.3 → 0.17.2

package/dist/handlers/session.js

@@ -11,19 +11,24 @@
  * notifications.
  */
 import process from "node:process";
+import path from "node:path";
 import { randomUUID } from "node:crypto";
+import { realpathSync } from "node:fs";
 import { RequestError } from "@agentclientprotocol/sdk";
 import { EventStreamListener, TurnMonitor } from "../backend/listener.js";
+import { resolveReal } from "../backend/sandbox.js";
 import { buildModes, buildConfigOptions, formatModelValue, loadAllModels, } from "../config/options.js";
 import { emitInitialUsage } from "../config/model-cache.js";
 import { buildProviderRegistry } from "../config/provider-registry.js";
 import { buildResumeRuntimeModel } from "../config/runtime-model.js";
+import { messages } from "../i18n.js";
 import { lookupLazySession, recordMaterializedSession, rememberLazySession, } from "../lazy-sessions.js";
 import { buildDiffContent, EventTranslator, extractLocations, formatTurnError, isTransientTurnError, ProjectionDiffer, } from "../translators/index.js";
 import { log, warn } from "../utils.js";
 import { dispatchEvent } from "./dispatch.js";
 import { sendSessionUpdate, sendTextChunk, withReplayBatch } from "./io.js";
 import { fetchMessages, fullSlice, readTailLimit, replayMessages, sliceTail } from "./replay.js";
+import { extractPermDeniedPath, extractSandboxDenial, GENERIC_HINT_KEY, handleSandboxDenial, READ_ONLY_TOOLS, } from "./sandbox-allow.js";
 import { handleServerRequests } from "./server-requests.js";
 /** Workspace descriptor used in session create/resume calls. */
 function workspaceFor(cwd) {
@@ -39,6 +44,20 @@ function workspaceFor(cwd) {
 function sanitizeClientCwd(client) {
     return client && client !== "/" ? client : null;
 }
+/**
+ * Same-directory check tolerant of spelling: a workspace can be recorded or
+ * reported under a symlinked / non-canonical spelling while the serve bridge
+ * holds the resolved process cwd. Compare realpaths; a vanished path falls
+ * back to raw equality (both sides unchanged → still equal).
+ */
+function sameProjectDir(a, b) {
+    try {
+        return realpathSync(a) === realpathSync(b);
+    }
+    catch {
+        return a === b;
+    }
+}
 /**
  * The authoritative Session Root for an EXISTING session: what the bridge
  * already recorded (set at creation, or refreshed from the backend's resume
@@ -100,7 +119,10 @@ function toIso(ms) {
 export async function newSession(server, params) {
     // Creation is the one moment a client's cwd is trusted (the editor
     // declaring its worktree); "/" is still rejected as a degenerate root.
-    const cwd = sanitizeClientCwd(params.cwd) ?? process.cwd();
+    // Serve mode (ADR-0014) is the exception: a headless bridge exists for ONE
+    // hub-chosen project — the process cwd wins and client-supplied values are
+    // ignored, so the remote create-whitelist cannot be bypassed via session/new.
+    const cwd = server.serveMode ? process.cwd() : (sanitizeClientCwd(params.cwd) ?? process.cwd());
     // Placeholder id — the client addresses this session with it until the
     // backend session materializes; never shown in session/list.
     const acpSid = randomUUID();
@@ -165,15 +187,29 @@ export async function ensureRealSession(server, acpSid) {
         // (the backend session still exists — re-register the alias); one without
         // re-hydrates the pending entry so the create path below runs.
         const record = lookupLazySession(acpSid);
+        // Serve mode (ADR-0014) honors durable records for ITS OWN project only.
+        // Aliases are minted by editor bridges, which trust their local client's
+        // session/new cwd — a remote client can mint {sid → arbitrary cwd} there
+        // and then resume it here to drag this bridge into a foreign workspace.
+        // Records from another cwd read as unknown ids. The comparison tolerates
+        // spelling differences (symlinked record cwd vs the resolved process cwd)
+        // — same project under another spelling stays resumable.
+        if (server.serveMode && record && !sameProjectDir(record.cwd, process.cwd())) {
+            log(`ensureRealSession: serve mode ignores a foreign lazy record (${acpSid})`);
+            throw new Error(`session ${acpSid} not found`);
+        }
         if (record?.zcodeSid) {
             server.registerSession(acpSid, record.zcodeSid);
             return record.zcodeSid;
         }
         if (record) {
-            pending = { cwd: record.cwd };
+            // Serve mode pins to its process cwd even here (belt and suspenders —
+            // the guard above already proved the record's cwd matches).
+            const cwd = server.serveMode ? process.cwd() : record.cwd;
+            pending = { cwd };
             server.pendingSessions.set(acpSid, pending);
-            if (record.cwd !== "/")
-                server.sessionCwds.set(acpSid, record.cwd);
+            if (cwd !== "/")
+                server.sessionCwds.set(acpSid, cwd);
         }
     }
     if (!pending)
@@ -248,8 +284,12 @@ export async function ensureRealSession(server, acpSid) {
 export async function listSessions(server, params) {
     const backend = server.ensureBackend();
     const zcParams = {};
-    if (params.cwd) {
-        zcParams.workspace = workspaceFor(params.cwd);
+    // Serve mode (ADR-0014) pins the workspace: a remote client must not use a
+    // client-supplied cwd to enumerate the machine's sessions in OTHER projects
+    // (the backend scopes the listing to the workspace it is given).
+    const listCwd = server.serveMode ? process.cwd() : params.cwd;
+    if (listCwd) {
+        zcParams.workspace = workspaceFor(listCwd);
     }
     const resp = await backend.request(server.nextId(), "session/list", zcParams, 15000);
     if (resp.error)
@@ -342,8 +382,10 @@ export async function resumeSession(server, params, cx) {
     // The Session Root never comes from the client (params.cwd is ignored):
     // start from what the bridge recorded, then let the backend's own resume
     // result correct it below — a remote client must not move a session's
-    // file scope by sending its own cwd.
-    let cwd = authoritativeSessionCwd(server, acpSid);
+    // file scope by sending its own cwd. Serve mode (ADR-0014) skips both
+    // sources: the bridge exists for ONE hub-chosen project, so the root (and
+    // the workspace sent to the backend's resume) stays the process cwd.
+    let cwd = server.serveMode ? process.cwd() : authoritativeSessionCwd(server, acpSid);
     // Lazy placeholders (session/new) resolve to their real backend session
     // here; alreadyLive targets skip the resume RPC because the session is live
     // in this backend subprocess.
@@ -375,9 +417,16 @@ export async function resumeSession(server, params, cx) {
         await repairUnavailableModel(server, zcodeSid);
         // The backend's session record is the root authority: adopt its
         // workspace as the session root (heals any stale/polluted entry).
+        // Serve mode keeps its pinned cwd (see above) — and a session that
+        // genuinely lives in ANOTHER workspace is refused outright: a raw
+        // backend id from session/list elsewhere must not be replayed through
+        // a serve bridge pinned to one project.
         const backendWs = workspaceFromResumeResult(resumeResult);
-        if (backendWs)
+        if (backendWs && !server.serveMode)
             cwd = backendWs;
+        if (server.serveMode && backendWs && !sameProjectDir(backendWs, process.cwd())) {
+            throw new Error("session belongs to another workspace");
+        }
     }
     server.registerSession(acpSid, zcodeSid);
     // The session root for remote file access — backend-authoritative (see
@@ -407,8 +456,9 @@ export async function loadSession(server, params, cx) {
     // The Session Root never comes from the client (params.cwd is ignored):
     // start from what the bridge recorded, then let the backend's own resume
     // result correct it below — a remote client must not move a session's
-    // file scope by sending its own cwd.
-    let cwd = authoritativeSessionCwd(server, acpSid);
+    // file scope by sending its own cwd. Serve mode (ADR-0014) pins the root
+    // (and the workspace sent to resume) to the process cwd throughout.
+    let cwd = server.serveMode ? process.cwd() : authoritativeSessionCwd(server, acpSid);
     // Same placeholder resolution as resumeSession; alreadyLive targets skip the
     // backend resume RPC (the session is live in this subprocess).
     const { zcodeSid, alreadyLive } = await resolveResumeTarget(server, acpSid);
@@ -428,9 +478,14 @@ export async function loadSession(server, params, cx) {
         await repairUnavailableModel(server, zcodeSid);
         // The backend's session record is the root authority: adopt its
         // workspace as the session root (heals any stale/polluted entry).
+        // Serve mode keeps its pinned cwd (see above) — and refuses sessions
+        // from another workspace outright (same rule as resumeSession).
         const backendWs = workspaceFromResumeResult(resumeResult);
-        if (backendWs)
+        if (backendWs && !server.serveMode)
             cwd = backendWs;
+        if (server.serveMode && backendWs && !sameProjectDir(backendWs, process.cwd())) {
+            throw new Error("session belongs to another workspace");
+        }
     }
     server.registerSession(acpSid, zcodeSid);
     // Same as resumeSession: backend-authoritative session root for file access.
@@ -488,8 +543,56 @@ export async function loadSession(server, params, cx) {
     };
     return result;
 }
-/** `session/prompt` → subscribe-before-send, run the event-driven turn loop. */
+/**
+ * `session/prompt` → subscribe-before-send, run the event-driven turn loop.
+ *
+ * Sandbox allow-restart chaining (ADR-0011): when a turn unwinds as cancelled
+ * because the user approved a new writable root (the backend was killed to be
+ * respawned under the widened profile), the continuation prompt runs INSIDE
+ * this same request via runPrompt. The editor's spinner then spans the
+ * respawn+reload window and the resumed work renders as the same turn. A
+ * detached follow-up prompt (the previous design) has no pending editor
+ * request behind it, so the editor showed no running state for it at all —
+ * the window read as "it just stopped", any message typed there preempted
+ * the continuation for real, and only the session/load replay later surfaced
+ * the orphaned continuation bubble.
+ */
 export async function prompt(server, params, cx, requestId) {
+    const result = await runPrompt(server, params, cx, requestId);
+    const continuation = server.sandboxContinuations.get(params.sessionId);
+    if (continuation && result.stopReason === "cancelled") {
+        server.sandboxContinuations.delete(params.sessionId);
+        // Same request, fresh round: runPrompt re-enters the whole machinery
+        // (ensureBackend respawns; ensureRealSession/subscribe-recovery reloads
+        // the session into it). The user's preempt/ESC during the continuation
+        // still cancels it — the round registers itself in pendingTurns.
+        try {
+            return await runPrompt(server, { sessionId: params.sessionId, prompt: [{ type: "text", text: continuation }] }, cx, `sandbox-cont-${randomUUID()}`, true);
+        }
+        catch (e) {
+            // The chained round runs right after a sandbox-allow respawn, inside
+            // the fresh backend's warm-up window. A failure here used to propagate
+            // as an RPC error and leave the session dead-silent with the
+            // continuation lost — the user had to blindly resend to recover
+            // (observed in production logs: "历史任务使用的模型已不可用" rejected
+            // the automatic send, a manual resend a minute later went through).
+            // Tell the user what happened and how to resume instead.
+            const err = e instanceof Error ? e.message : String(e);
+            warn(`sandbox: continuation prompt failed: ${err}`);
+            await sendTextChunk(cx, params.sessionId, messages().sandboxContinuationFailed(err), randomUUID()).catch(() => undefined);
+            return { stopReason: "cancelled" };
+        }
+    }
+    return result;
+}
+/** One prompt round: subscribe-before-send, run the event-driven turn loop. */
+async function runPrompt(server, params, cx, requestId, continuationRound = false) {
+    // Project-level sandbox flip (ADR-0011): a .zcode/acp/sandbox.json created
+    // after the backend spawned unsandboxed must arm on THIS prompt, not on the
+    // next bridge restart — kill the old process; the ensureBackend() below
+    // respawns under the profile and the subscribe-recovery path reloads the
+    // session. No-op unless the config appeared mid-run.
+    await server.applySandboxFlip();
     const backend = server.ensureBackend();
     // Extract prompt text + image attachments from ACP ContentBlock[].
     const text = extractPromptText(params.prompt);
@@ -659,7 +762,7 @@ export async function prompt(server, params, cx, requestId) {
                     return { stopReason: "cancelled" };
                 }
                 differ.markSeen(await fetchMessages(server, zcodeSid));
-                await sendTextChunk(cx, params.sessionId, `[网络异常，正在重试 (${attempt - 1}/${MAX_TURN_ATTEMPTS - 1})…]`, randomUUID());
+                await sendTextChunk(cx, params.sessionId, messages().networkRetry(attempt - 1, MAX_TURN_ATTEMPTS - 1), randomUUID());
                 log(`  [retry] transient turn failed, re-sending (attempt ${attempt}/${MAX_TURN_ATTEMPTS})`);
                 await sleep(backoffMs(attempt - 1));
             }
@@ -691,6 +794,9 @@ export async function prompt(server, params, cx, requestId) {
             // in-flight one) and the stop-recovery window after a manual cancel.
             const SEND_RETRY_INTERVAL_MS = 500;
             const SEND_RETRY_TIMEOUT_MS = 30_000;
+            // Cold-start rejections recover in seconds (observed); a longer budget
+            // would spin pointlessly when the model is PERMANENTLY unavailable.
+            const WARMUP_RETRY_TIMEOUT_MS = 10_000;
             const sendParams = attachments.length > 0
                 ? { sessionId: zcodeSid, content: sendText, attachments }
                 : { sessionId: zcodeSid, content: sendText };
@@ -728,14 +834,27 @@ export async function prompt(server, params, cx, requestId) {
                 const isBusy = sendErrCode === 1308 ||
                     sendErrMsg.includes("prompt is running") ||
                     sendErrMsg.includes("already running");
-                if (!isBusy) {
-                    // Non-busy error (auth, malformed, etc.) — don't retry, surface it.
+                const isWarming = isTransientSendError(sendResp.error.message ?? "");
+                if (!isBusy && !isWarming) {
+                    // Permanent error (auth, malformed, a truly removed model, …) —
+                    // don't retry, surface it. The stale-history-model case is
+                    // repaired before the send (repairUnavailableModel on every
+                    // resume path); the cold-start form is transient and retries below.
                     throw new Error(`zcode send failed: ${sendResp.error.message ?? ""}`);
                 }
-                if (Date.now() - sendT0 > SEND_RETRY_TIMEOUT_MS) {
-                    throw new Error(`zcode send failed: backend still busy after ${Math.round(SEND_RETRY_TIMEOUT_MS / 1000)}s (${sendResp.error.message ?? ""})`);
+                const budget = isBusy ? SEND_RETRY_TIMEOUT_MS : WARMUP_RETRY_TIMEOUT_MS;
+                if (Date.now() - sendT0 > budget) {
+                    throw new Error(`zcode send failed: ${isBusy ? "backend still busy" : "send keeps being rejected"} after ${Math.round(budget / 1000)}s (${sendResp.error.message ?? ""})`);
                 }
-                log(`  [send] backend busy (${sendResp.error.message ?? ""}), retrying in ${SEND_RETRY_INTERVAL_MS}ms`);
+                log(`  [send] backend ${isBusy ? "busy" : "cold-start reject"} (${sendResp.error.message ?? ""}), retrying in ${SEND_RETRY_INTERVAL_MS}ms`);
+            }
+            // Continuation round status line: the respawn+reload window (~seconds)
+            // shows only the editor spinner before the model's first output — a
+            // bare thinking block with no context. Announce the resumed turn at
+            // send-accept so every phase of the allow→restart→continue flow is
+            // visibly accounted for (the allow-time hint covers the restart start).
+            if (continuationRound) {
+                await sendTextChunk(cx, params.sessionId, messages().sandboxResumedStatus, randomUUID());
             }
             try {
                 // Event-driven turn loop: translate events via EventTranslator + dispatch.
@@ -760,6 +879,9 @@ export async function prompt(server, params, cx, requestId) {
                     const { maybeAutoCompact } = await import("../config/auto-compact.js");
                     await maybeAutoCompact(server, cx, params.sessionId, zcodeSid);
                 }
+                // (Sandbox allow-restart continuation: handled by the prompt() wrapper
+                // — the chained round keeps the editor's original request pending so
+                // the restart window stays visibly "running".)
                 return result;
             }
             catch (e) {
@@ -783,7 +905,7 @@ export async function prompt(server, params, cx, requestId) {
         // surfacing a hard error and stopping. Skip auto-compact here: compaction
         // after a failed turn is more likely to confuse state than help.
         const errMsg = formatTurnError(lastTurnError) || "turn failed after retries";
-        await sendTextChunk(cx, params.sessionId, `[请求失败：${errMsg}。会话仍可用，请重新发送消息重试。]`, randomUUID());
+        await sendTextChunk(cx, params.sessionId, messages().requestFailed(errMsg), randomUUID());
         return { stopReason: "end_turn" };
     }
     finally {
@@ -1064,7 +1186,7 @@ export async function drainBackendAfterCancel(server, deps) {
         }
         if (!noticed) {
             noticed = true;
-            await sendTextChunk(cx, acpSid, "[上一个回复仍在生成，等待结束后发送…]", randomUUID());
+            await sendTextChunk(cx, acpSid, messages().promptQueuedBehindTurn, randomUUID());
         }
         await sleep(DRAIN_POLL_MS);
     }
@@ -1299,12 +1421,22 @@ async function resumeBackendSession(server, zcParams) {
  * fails). Marks the session backend-loaded on success.
  */
 async function reloadBackendSession(server, acpSid, zcodeSid) {
+    const cwd = server.sessionCwds.get(acpSid) ?? process.cwd();
     const zcParams = {
         sessionId: zcodeSid,
-        workspace: workspaceFor(server.sessionCwds.get(acpSid) ?? process.cwd()),
+        workspace: workspaceFor(cwd),
     };
+    // Same pre-resume steps as the ACP resume/load handlers: register the
+    // provider registry (a resumed session's history references a model the
+    // fresh backend can't process until its provider is registered — sends
+    // then fail with the backend's stale-history-model error, persistent, not
+    // transient), then repair a model that is no longer enabled. Skipping
+    // these made every backend respawn (sandbox allow-restart, eviction
+    // recovery) deaf-fail its first sends.
+    await syncProviderRegistry(server, cwd);
     await resumePreservingModel(server, zcParams);
     server.markBackendLoaded(acpSid);
+    await repairUnavailableModel(server, zcodeSid);
 }
 /**
  * Resume WITHOUT pinning a model, so the session keeps its own selection (the
@@ -1358,6 +1490,19 @@ async function repairUnavailableModel(server, zcodeSid) {
         warn(`model repair check failed: ${e instanceof Error ? e.message : String(e)}`);
     }
 }
+/**
+ * Cold-start send rejection worth retrying: right after a sandbox-allow
+ * respawn, the reloaded session's recorded model briefly reads as
+ * unavailable while the fresh backend finishes loading its model list — the
+ * same send succeeds seconds later (verified in production: the automatic
+ * continuation was rejected with "历史任务使用的模型已不可用", a manual
+ * resend a minute later went through). Matches the backend's Chinese wording
+ * plus a generic English form.
+ */
+export function isTransientSendError(message) {
+    const m = message.toLowerCase();
+    return m.includes("模型已不可用") || /model .*(unavailable|no longer available)/.test(m);
+}
 /** Get or create the session-level ProjectionDiffer (persists across turns). */
 function getOrCreateDiffer(server, zcodeSid) {
     let d = server.differs.get(zcodeSid);
@@ -1524,7 +1669,7 @@ export async function runEventTurn(server, listener, monitor, differ, cx, acpSid
                 thinkingHintSent = true;
                 await sendSessionUpdate(cx, acpSid, {
                     sessionUpdate: "agent_thought_chunk",
-                    content: { type: "text", text: "正在思考…" },
+                    content: { type: "text", text: messages().thinkingPlaceholder },
                     messageId: `thinking_${chunkMsgId}`,
                 });
             }
@@ -1601,7 +1746,7 @@ export async function runEventTurn(server, listener, monitor, differ, cx, acpSid
         // with the message lost. turn.steerQueued is definitive proof of the
         // swallow: report it at once so the user can resend immediately.
         if (ev.type === "turn.steerQueued" && !translator.turnStarted && gateArmed) {
-            await sendTextChunk(cx, acpSid, "[消息被并入仍在生成的回合，将被丢弃，请重新发送]", chunkMsgId);
+            await sendTextChunk(cx, acpSid, messages().messageSwallowedByTurn, chunkMsgId);
             return { stopReason: "max_turn_requests" };
         }
         // Turn-attribution gate: before this turn's own turn.started arrives, any
@@ -1654,6 +1799,68 @@ export async function runEventTurn(server, listener, monitor, differ, cx, acpSid
                 differ.setLastUsage(iev.used);
             await dispatchEvent(server, cx, acpSid, iev, chunkMsgId);
         }
+        // Filesystem-permission failure surfacing (the tool output itself already
+        // reached the editor above). Two scans, both only on tool.updated so
+        // mid-stream deltas aren't scanned, and both skipping read-only tools —
+        // their output merely ECHOES text, and acting on it would raise a phantom
+        // ask for a path nothing tried to touch.
+        if (ev.type === "tool.updated") {
+            const outputText = JSON.stringify(ev.payload ?? {});
+            const toolCallId = String(ev.payload?.toolCallId ?? "");
+            const toolName = (translator.toolNames.get(toolCallId) ?? "").toLowerCase();
+            // Case-insensitive: zsh prints redirects as "operation not permitted".
+            const lower = outputText.toLowerCase();
+            const scannable = !READ_ONLY_TOOLS.has(toolName);
+            if (scannable && server.backendSandboxed && lower.includes("operation not permitted")) {
+                // Sandbox write-denial (ADR-0011): EPERM outside the Seatbelt
+                // whitelist → dynamic-allow flow (ask → persist → kill backend → the
+                // turn unwinds as cancelled; prompt() fires the continuation). Gated
+                // on the PROCESS fact backendSandboxed (not the config wish) — EPERM
+                // can only come from a sandboxed process; unsandboxed EPERM is
+                // ordinary filesystem permissions.
+                const denial = extractSandboxDenial(outputText);
+                if (denial) {
+                    await handleSandboxDenial(server, cx, acpSid, turn, denial, toolCallId);
+                    if (turn.cancelled)
+                        return { stopReason: "cancelled" };
+                }
+                else {
+                    // No path parsed — one generic hint per session, not one per retry.
+                    let asked = server.sandboxAskedPaths.get(acpSid);
+                    if (!asked) {
+                        asked = new Set();
+                        server.sandboxAskedPaths.set(acpSid, asked);
+                    }
+                    if (!asked.has(GENERIC_HINT_KEY)) {
+                        asked.add(GENERIC_HINT_KEY);
+                        await sendTextChunk(cx, acpSid, messages().sandboxGenericDenialHint, chunkMsgId);
+                    }
+                }
+            }
+            else if (scannable && lower.includes("permission denied")) {
+                // EACCES — ordinary filesystem permissions, sandbox or not. Nothing
+                // the bridge can "allow" (no popup fixes chmod/ownership), so surface
+                // it as a one-time hint per path per session; without it the model
+                // silently swallows the failure and reroutes, and the user never
+                // learns why the command died.
+                const deniedPath = extractPermDeniedPath(outputText);
+                if (deniedPath) {
+                    const cwd = server.sessionCwds.get(acpSid);
+                    const real = resolveReal(path.isAbsolute(deniedPath)
+                        ? deniedPath
+                        : path.resolve(cwd ?? process.cwd(), deniedPath));
+                    let hinted = server.fsDeniedPaths.get(acpSid);
+                    if (!hinted) {
+                        hinted = new Set();
+                        server.fsDeniedPaths.set(acpSid, hinted);
+                    }
+                    if (!hinted.has(real)) {
+                        hinted.add(real);
+                        await sendTextChunk(cx, acpSid, messages().fsPermDeniedHint(real), chunkMsgId);
+                    }
+                }
+            }
+        }
         // Edit/Write diff eager dispatch: on tool.updated result, grab the
         // structured patch from session/messages immediately (don't wait for turn
         // completion — model rate-limiting could delay it indefinitely).