@hassangameryt/freeflow 0.7.3 → 0.8.0

package/pi-extension/dist/cognitive-routing/event-schema.js

@@ -0,0 +1,388 @@
+import { EFFORTS, PROFILES, DELEGATION_MODES, isObject, requireCondition as check } from "./types.js";
+const text = (x, max = 32768) => typeof x === "string" && x.trim().length > 0 && x.length <= max;
+const identity = (x) => text(x, 512);
+const nullableId = (x) => x === null || identity(x);
+const shape = (x, keys) => isObject(x) && Object.keys(x).every((k) => keys.includes(k));
+const uniqueStrings = (x) => Array.isArray(x) && x.every(identity) && new Set(x).size === x.length;
+const pair = (x) =>
+  shape(x, ["provider", "modelId", "thinking"]) &&
+  identity(x.provider) &&
+  identity(x.modelId) &&
+  EFFORTS.includes(x.thinking);
+const profileOverrides = (x) =>
+  isObject(x) &&
+  Object.keys(x).length > 0 &&
+  Object.keys(x).every((key) => PROFILES.includes(key)) &&
+  Object.values(x).every((value) => value === null || pair(value));
+const problemList = (xs) =>
+  Array.isArray(xs) &&
+  xs.every(
+    (x) =>
+      shape(x, ["ref", "code", "detail"]) && typeof x.ref === "string" && text(x.code, 256) && text(x.detail, 4096),
+  );
+const reservation = (r) =>
+  shape(r, [
+    "id",
+    "selectionRevision",
+    "receiver",
+    "target",
+    "qualification",
+    "sources",
+    "maximumInputTokens",
+    "outputReserve",
+    "estimateMethod",
+  ]) &&
+  identity(r.id) &&
+  Number.isSafeInteger(r.selectionRevision) &&
+  r.selectionRevision >= 0 &&
+  PROFILES.includes(r.receiver) &&
+  pair(r.target) &&
+  text(r.qualification) &&
+  Array.isArray(r.sources) &&
+  r.sources.every((s) => shape(s, ["ref", "bodyHash"]) && identity(s.ref) && /^[a-f0-9]{64}$/.test(s.bodyHash)) &&
+  new Set(r.sources.map((s) => s.ref)).size === r.sources.length &&
+  Number.isFinite(r.maximumInputTokens) &&
+  r.maximumInputTokens > 0 &&
+  Number.isFinite(r.outputReserve) &&
+  r.outputReserve >= 0 &&
+  text(r.estimateMethod);
+const recovery = (r) => {
+  const requestedResults = r?.requestedResults ?? [];
+  const results = r?.results ?? [];
+  return (
+    shape(r, [
+      "id",
+      "assignmentId",
+      "assessmentHandoffId",
+      "baseReportRevision",
+      "request",
+      "requestedPaths",
+      "paths",
+      "requestedResults",
+      "results",
+      "state",
+      "requestHandoffId",
+      "supplementHandoffId",
+      "supplementRevision",
+      "cancellationReason",
+    ]) &&
+    identity(r.id) &&
+    identity(r.assignmentId) &&
+    identity(r.assessmentHandoffId) &&
+    Number.isSafeInteger(r.baseReportRevision) &&
+    r.baseReportRevision > 0 &&
+    text(r.request) &&
+    Array.isArray(r.requestedPaths) &&
+    r.requestedPaths.length <= 32 &&
+    r.requestedPaths.every((path) => text(path, 4096)) &&
+    new Set(r.requestedPaths).size === r.requestedPaths.length &&
+    Array.isArray(r.paths) &&
+    r.paths.length <= 32 &&
+    r.paths.every((path) => text(path, 4096)) &&
+    new Set(r.paths).size === r.paths.length &&
+    Array.isArray(requestedResults) &&
+    requestedResults.length <= 32 &&
+    requestedResults.every((id) => text(id, 256)) &&
+    new Set(requestedResults).size === requestedResults.length &&
+    Array.isArray(results) &&
+    results.length <= 32 &&
+    results.every(
+      (grant) =>
+        shape(grant, ["id", "sha256"]) &&
+        text(grant.id, 256) &&
+        typeof grant.sha256 === "string" &&
+        /^[a-f0-9]{64}$/.test(grant.sha256),
+    ) &&
+    new Set(results.map((grant) => grant.id)).size === results.length &&
+    ["requested", "reading", "returning", "completed", "cancelled"].includes(r.state) &&
+    identity(r.requestHandoffId) &&
+    (r.supplementHandoffId === undefined || identity(r.supplementHandoffId)) &&
+    Number.isSafeInteger(r.supplementRevision) &&
+    r.supplementRevision >= 0 &&
+    (r.cancellationReason === undefined || text(r.cancellationReason, 2048))
+  );
+};
+const handoff = (h) =>
+  shape(h, [
+    "id",
+    "kind",
+    "assignmentId",
+    "text",
+    "from",
+    "to",
+    "executionId",
+    "toolCallId",
+    "basisUserEntryId",
+    "state",
+    "reportRevision",
+    "outcome",
+    "limitations",
+    "reason",
+  ]) &&
+  identity(h.id) &&
+  identity(h.assignmentId) &&
+  text(h.text) &&
+  identity(h.executionId) &&
+  identity(h.toolCallId) &&
+  nullableId(h.basisUserEntryId) &&
+  ["delegate", "return", "recovery-request", "recovery-return"].includes(h.kind) &&
+  PROFILES.includes(h.from) &&
+  PROFILES.includes(h.to) &&
+  h.from !== h.to &&
+  ["pending", "configured", "blocked", "superseded"].includes(h.state) &&
+  Number.isSafeInteger(h.reportRevision) &&
+  h.reportRevision >= 0 &&
+  Array.isArray(h.limitations) &&
+  h.limitations.length <= 32 &&
+  h.limitations.every((x) => text(x, 2048)) &&
+  (h.reason === undefined || text(h.reason, 4096)) &&
+  (h.outcome === undefined || ["completed", "partial", "blocked"].includes(h.outcome)) &&
+  (["delegate", "recovery-request"].includes(h.kind)
+    ? h.outcome === undefined && h.reportRevision === 0
+    : ["completed", "partial", "blocked"].includes(h.outcome) && h.reportRevision > 0);
+const fields = {
+  "execution-interrupted": ["executionId", "reason"],
+  "assignment-resumed": ["assignmentId", "basisUserEntryId"],
+  control: ["control", "profile", "reason"],
+  "profile-overrides": ["overrides", "reason"],
+  "delegation-override": ["delegation", "reason"],
+  "execution-opened": ["execution"],
+  "execution-bound": ["executionId", "assistantEntryId", "resultEntryIds", "outcome"],
+  "delegate-accepted": ["unit", "assignment", "handoff", "replacement"],
+  "return-accepted": ["handoff"],
+  "recovery-request-accepted": ["recovery", "handoff"],
+  "recovery-supplement-accepted": ["recoveryId", "handoff"],
+  "recovery-cancelled": ["recoveryId", "reason"],
+  "handoff-retry-requested": [
+    "handoffId",
+    "attemptId",
+    "executionId",
+    "toolCallId",
+    "reportRevision",
+    "selectionRevision",
+  ],
+  "handoff-prepared": ["handoffId", "reservation"],
+  "handoff-state": ["handoffId", "state", "reason", "observedPair"],
+  "selection-changed": ["assignmentId", "selection"],
+  "assessment-suspended": ["handoffId", "reason", "basisUserEntryId", "problems"],
+  "assessment-resumed": ["handoffId", "basisUserEntryId", "reservation"],
+  "unit-closed": ["unitId", "outcome", "assessment", "supersededAssignmentId"],
+  "sources-exposed": ["executionId", "view", "sources"],
+  "request-observed": ["executionId", "profile", "boundary", "manifestHash", "handoffId", "selectionRevision"],
+};
+export function parseRoutingEvent(raw) {
+  check(
+    isObject(raw) &&
+      raw.version === 2 &&
+      identity(raw.eventId) &&
+      identity(raw.operationId) &&
+      identity(raw.stepId) &&
+      identity(raw.recordedSessionId) &&
+      isObject(raw.data),
+    "invalid_event",
+  );
+  check(
+    Object.keys(raw).every((k) =>
+      ["version", "eventId", "operationId", "stepId", "recordedSessionId", "data"].includes(k),
+    ),
+    "invalid_event_field",
+  );
+  const d = raw.data;
+  check(
+    typeof d.type === "string" &&
+      Object.hasOwn(fields, d.type) &&
+      Object.keys(d).every((k) => k === "type" || fields[d.type].includes(k)),
+    "invalid_event_type",
+  );
+  check(Buffer.byteLength(JSON.stringify(raw)) <= 512 * 1024, "event_too_large");
+  if (d.handoffId !== undefined) check(identity(d.handoffId), "invalid_handoff_id");
+  switch (d.type) {
+    case "execution-interrupted":
+      check(identity(d.executionId) && text(d.reason, 4096), "invalid_interruption");
+      break;
+    case "assignment-resumed":
+      check(identity(d.assignmentId) && nullableId(d.basisUserEntryId), "invalid_resume");
+      break;
+    case "control":
+      check(
+        ["automatic", "manual", "inactive"].includes(d.control) &&
+          text(d.reason, 4096) &&
+          (d.profile === undefined || PROFILES.includes(d.profile)) &&
+          (d.control === "inactive" || PROFILES.includes(d.profile)),
+        "invalid_control",
+      );
+      break;
+    case "profile-overrides":
+      check(profileOverrides(d.overrides) && text(d.reason, 4096), "invalid_profile_overrides");
+      break;
+    case "delegation-override":
+      check(
+        (d.delegation === null || DELEGATION_MODES.includes(d.delegation)) && text(d.reason, 4096),
+        "invalid_delegation_override",
+      );
+      break;
+    case "execution-opened": {
+      const e = d.execution;
+      check(
+        shape(e, ["id", "profile", "assignmentId", "recoveryId", "basisUserEntryId", "pair", "resultEntryIds"]) &&
+          identity(e.id) &&
+          ["solo", ...PROFILES].includes(e.profile) &&
+          nullableId(e.basisUserEntryId) &&
+          pair(e.pair) &&
+          uniqueStrings(e.resultEntryIds) &&
+          e.resultEntryIds.length === 0 &&
+          (e.assignmentId === undefined || identity(e.assignmentId)) &&
+          (e.recoveryId === undefined || identity(e.recoveryId)),
+        "invalid_execution",
+      );
+      break;
+    }
+    case "execution-bound":
+      check(
+        identity(d.executionId) &&
+          identity(d.assistantEntryId) &&
+          uniqueStrings(d.resultEntryIds) &&
+          ["completed", "failed", "aborted"].includes(d.outcome),
+        "invalid_binding",
+      );
+      break;
+    case "delegate-accepted": {
+      const u = d.unit,
+        a = d.assignment,
+        r = d.replacement;
+      check(
+        shape(u, ["id", "objective", "state", "assignmentIds"]) &&
+          identity(u.id) &&
+          text(u.objective) &&
+          u.state === "open" &&
+          uniqueStrings(u.assignmentIds),
+        "invalid_unit",
+      );
+      check(
+        shape(a, ["id", "unitId", "contract", "basisUserEntryId", "state", "delegateHandoffId"]) &&
+          identity(a.id) &&
+          identity(a.unitId) &&
+          text(a.contract) &&
+          nullableId(a.basisUserEntryId) &&
+          a.state === "outstanding" &&
+          identity(a.delegateHandoffId) &&
+          handoff(d.handoff),
+        "invalid_assignment",
+      );
+      check(
+        r === undefined ||
+          (shape(r, ["assignmentId", "supersededHandoffId", "reason"]) &&
+            identity(r.assignmentId) &&
+            text(r.reason, 2048) &&
+            (r.supersededHandoffId === undefined || identity(r.supersededHandoffId))),
+        "invalid_replacement",
+      );
+      break;
+    }
+    case "return-accepted":
+      check(handoff(d.handoff) && d.handoff.kind === "return", "invalid_return");
+      break;
+    case "recovery-request-accepted":
+      check(
+        recovery(d.recovery) &&
+          handoff(d.handoff) &&
+          d.handoff.kind === "recovery-request" &&
+          d.recovery.requestHandoffId === d.handoff.id &&
+          d.recovery.assignmentId === d.handoff.assignmentId &&
+          d.recovery.request === d.handoff.text,
+        "invalid_recovery_request",
+      );
+      break;
+    case "recovery-supplement-accepted":
+      check(
+        identity(d.recoveryId) && handoff(d.handoff) && d.handoff.kind === "recovery-return",
+        "invalid_recovery_supplement",
+      );
+      break;
+    case "recovery-cancelled":
+      check(identity(d.recoveryId) && text(d.reason, 2048), "invalid_recovery_cancellation");
+      break;
+    case "handoff-retry-requested":
+      check(
+        identity(d.attemptId) &&
+          identity(d.executionId) &&
+          identity(d.toolCallId) &&
+          Number.isSafeInteger(d.reportRevision) &&
+          d.reportRevision > 0 &&
+          Number.isSafeInteger(d.selectionRevision) &&
+          d.selectionRevision >= 0,
+        "invalid_retry",
+      );
+      break;
+    case "handoff-prepared":
+      check(reservation(d.reservation), "invalid_reservation");
+      break;
+    case "handoff-state":
+      check(
+        ["configured", "blocked", "superseded"].includes(d.state) &&
+          (d.reason === undefined || text(d.reason, 4096)) &&
+          (d.observedPair === undefined || pair(d.observedPair)),
+        "invalid_delivery",
+      );
+      break;
+    case "selection-changed": {
+      const s = d.selection;
+      check(
+        identity(d.assignmentId) &&
+          shape(s, ["revision", "selected", "unresolved", "withdrawals"]) &&
+          Number.isSafeInteger(s.revision) &&
+          s.revision > 0 &&
+          uniqueStrings(s.selected) &&
+          problemList(s.unresolved) &&
+          Array.isArray(s.withdrawals) &&
+          s.withdrawals.every((w) => shape(w, ["ref", "reason"]) && identity(w.ref) && text(w.reason, 2048)),
+        "invalid_selection",
+      );
+      break;
+    }
+    case "assessment-suspended":
+      check(
+        ["user-attention", "delivery-gap"].includes(d.reason) &&
+          nullableId(d.basisUserEntryId) &&
+          problemList(d.problems),
+        "invalid_suspension",
+      );
+      break;
+    case "assessment-resumed":
+      check(
+        nullableId(d.basisUserEntryId) && (d.reservation === undefined || reservation(d.reservation)),
+        "invalid_resumption",
+      );
+      break;
+    case "unit-closed":
+      check(
+        identity(d.unitId) &&
+          ["accepted", "cancelled", "deferred"].includes(d.outcome) &&
+          text(d.assessment) &&
+          (d.supersededAssignmentId === undefined || identity(d.supersededAssignmentId)),
+        "invalid_closure",
+      );
+      break;
+    case "request-observed":
+      check(
+        identity(d.executionId) &&
+          ["solo", ...PROFILES].includes(d.profile) &&
+          ["assembled", "payload-hook"].includes(d.boundary) &&
+          /^[a-f0-9]{64}$/.test(d.manifestHash) &&
+          (d.selectionRevision === undefined ||
+            (Number.isSafeInteger(d.selectionRevision) && d.selectionRevision >= 0)),
+        "invalid_request_observation",
+      );
+      break;
+    case "sources-exposed":
+      check(
+        identity(d.executionId) &&
+          ["solo", ...PROFILES].includes(d.view) &&
+          Array.isArray(d.sources) &&
+          d.sources.every((s) => shape(s, ["ref", "bodyHash"]) && identity(s.ref) && /^[a-f0-9]{64}$/.test(s.bodyHash)),
+        "invalid_exposure",
+      );
+      break;
+  }
+  return structuredClone(raw);
+}

package/pi-extension/dist/{session-sources/events.js → cognitive-routing/event-store.js}

@@ -1,14 +1,15 @@
 import { randomUUID } from "node:crypto";
-import { activeReadOnlySessionBranch, readOnlySessionSnapshot, ReadOnlySessionError } from "./read-only-session.js";
 import {
-  ROUTING_ENTRY,
-  RoutingError,
-  canonical,
-  eventKey,
-  eventValue,
-  requireCondition as check,
-} from "../cognitive-routing-v2/types.js";
-import { parseRoutingEvent, reduce, replay } from "../cognitive-routing-v2/state.js";
+  activeReadOnlySessionBranch,
+  persistedTailMatches,
+  readOnlySessionSnapshot,
+  ReadOnlySessionError,
+} from "../host/read-only-session.js";
+import { ROUTING_ENTRY, RoutingError, canonical, eventKey, eventValue, requireCondition as check } from "./types.js";
+import { parseRoutingEvent } from "./event-schema.js";
+import { reduce, replay } from "./state.js";
+import { sessionStage, stagedFor } from "../host/staging.js";
+import { cachedBranch } from "../host/branch.js";
 export class EventStore {
   pi;
   reader;
@@ -25,11 +26,19 @@ export class EventStore {
   get blocked() {
     return this.fault ?? (!this.ready ? "acknowledgment_unclassified" : undefined);
   }
+  /** Routing state from the branch plus control changes staged since the last prompt. */
   state() {
-    const entries = this.reader.getBranch();
+    const staged = stagedFor(this.reader)?.events ?? [];
+    return staged.reduce((state, event) => reduce(state, event), this.branchState());
+  }
+  branchState() {
+    const entries = cachedBranch(this.reader);
+    // The cached branch is the same frozen array until the leaf moves, so an unchanged branch needs no comparison.
+    if (entries === this.cachedEntries) return this.cachedState;
     const prefix =
       this.cachedEntries.length <= entries.length && this.cachedEntries.every((entry, i) => entry === entries[i]);
-    if (!prefix) {
+    // An empty cache is a trivial prefix; the copying reducer would make that first fold quadratic.
+    if (!prefix || this.cachedEntries.length === 0) {
       this.cachedState = replay(entries);
     } else {
       let state = this.cachedState;
@@ -38,7 +47,7 @@ export class EventStore {
           state = reduce(state, parseRoutingEvent(entry.data));
       this.cachedState = state;
     }
-    this.cachedEntries = [...entries];
+    this.cachedEntries = entries;
     return this.cachedState;
   }
   block(reason) {
@@ -58,7 +67,7 @@ export class EventStore {
   async reconcile() {
     const acknowledged = this.ready && !this.fault;
     this.ready = false;
-    const branch = this.reader.getBranch();
+    const branch = cachedBranch(this.reader);
     const live = branch.filter((e) => e.type === "custom" && e.customType === ROUTING_ENTRY);
     try {
       const ids = new Set();
@@ -115,6 +124,20 @@ export class EventStore {
       }
       const path = this.reader.getSessionFile();
       check(path, "persisted_snapshot_unavailable", "Existing or uncertain routing state needs a persisted snapshot.");
+      // Pi parsed this file when it opened the session and appends each entry synchronously after that. When the
+      // bytes appended since that trust point are exactly Pi's in-memory entries, the live branch is persisted and
+      // no snapshot is needed. Otherwise the full comparison below decides.
+      if (await persistedTailMatches(this.reader)) {
+        const known = new Map();
+        for (const entry of live) {
+          const event = parseRoutingEvent(entry.data);
+          known.set(eventKey(event), { value: eventValue(event), eventId: event.eventId, entryId: entry.id });
+        }
+        this.acknowledge(known);
+        this.cachedEntries = branch;
+        this.cachedState = replay(branch);
+        return;
+      }
       const snapshot = await readOnlySessionSnapshot(path);
       check(snapshot.sessionId === this.reader.getSessionId(), "session_identity_changed");
       const persistedBranch = activeReadOnlySessionBranch(snapshot, this.reader.getLeafId());
@@ -123,14 +146,16 @@ export class EventStore {
         "snapshot_ancestry_mismatch",
         "Live branch is not the complete persisted ancestry for its claimed leaf.",
       );
-      replay(persistedBranch);
+      const replayed = replay(persistedBranch);
       const disk = new Map(persistedBranch.map((e) => [e.id, e]));
       const known = new Map();
       // Compare the live selected ancestry, not whichever branch was last written on disk.
       for (const entry of branch) {
         const captured = disk.get(entry.id);
+        // Both copies come from the same JSON, so their keys share an order and the native serializer decides
+        // equality at a fraction of canonical()'s cost; canonical() settles only a mismatch in key order.
         check(
-          captured && canonical(captured) === canonical(entry),
+          captured && (JSON.stringify(captured) === JSON.stringify(entry) || canonical(captured) === canonical(entry)),
           "snapshot_divergence",
           "Live ancestry is not fully established by the fresh persisted snapshot; reconcile the session before retrying.",
         );
@@ -139,18 +164,10 @@ export class EventStore {
           known.set(eventKey(event), { value: eventValue(event), eventId: event.eventId, entryId: entry.id });
         }
       }
-      for (const [key, event] of this.attempted)
-        check(
-          known.get(key)?.value === eventValue(event) && known.get(key)?.eventId === event.eventId,
-          "append_acknowledgment_uncertain",
-          "The attempted routing event is not established by persisted readback.",
-        );
-      this.observed = known;
-      this.attempted.clear();
-      this.fault = undefined;
-      this.ready = true;
-      this.cachedEntries = [];
-      this.cachedState = replay([]);
+      this.acknowledge(known);
+      // The live branch matches the persisted one entry for entry, so its replay is the state the next read needs.
+      this.cachedEntries = branch;
+      this.cachedState = replayed;
     } catch (error) {
       this.block(
         error instanceof RoutingError
@@ -162,20 +179,44 @@ export class EventStore {
       throw error;
     }
   }
-  append(event) {
+  /** Accept persisted events as observed; every attempted (uncertain) append must be among them. */
+  acknowledge(known) {
+    for (const [key, event] of this.attempted)
+      check(
+        known.get(key)?.value === eventValue(event) && known.get(key)?.eventId === event.eventId,
+        "append_acknowledgment_uncertain",
+        "The attempted routing event is not established by persisted readback.",
+      );
+    this.observed = known;
+    this.attempted.clear();
+    this.fault = undefined;
+    this.ready = true;
+  }
+  /** Append an event; with `stage`, hold it for the next prompt instead of writing it to the session now. */
+  append(event, stage = false) {
     const value = parseRoutingEvent(event),
       key = eventKey(value),
       body = eventValue(value);
     if (this.blocked) throw new RoutingError("acknowledgment_uncertain", this.blocked);
+    const staged = stage ? sessionStage(this.reader) : stagedFor(this.reader);
     const prior = this.state().events.get(key);
+    if (prior && staged?.events.some((event) => eventKey(event) === key)) {
+      check(eventValue(prior) === body, "operation_conflict");
+      return prior;
+    }
+    if (stage && staged) {
+      reduce(this.state(), value);
+      staged.events.push(value);
+      return value;
+    }
+    // Replay applies events in session order, so anything staged is written before a later event.
+    if (staged?.events.length) this.writeStaged(staged.events.splice(0));
     if (prior) {
       check(eventValue(prior) === body, "operation_conflict");
-      const occurrence = this.reader
-        .getBranch()
-        .find(
-          (entry) =>
-            entry.type === "custom" && entry.customType === ROUTING_ENTRY && entry.data?.eventId === prior.eventId,
-        );
+      const occurrence = cachedBranch(this.reader).find(
+        (entry) =>
+          entry.type === "custom" && entry.customType === ROUTING_ENTRY && entry.data?.eventId === prior.eventId,
+      );
       if (
         this.observed.get(key)?.value !== body ||
         this.observed.get(key)?.eventId !== prior.eventId ||
@@ -187,23 +228,23 @@ export class EventStore {
       return prior;
     }
     // Validate the whole state transition before mutating Pi's session memory.
-    const candidate = reduce(this.state(), value);
+    const candidate = reduce(this.branchState(), value);
     const baseline = this.cachedEntries;
     this.attempted.set(key, value);
     try {
       this.pi.appendEntry(ROUTING_ENTRY, value);
-      const found = this.reader
-        .getBranch()
-        .find((e) => e.type === "custom" && e.customType === ROUTING_ENTRY && e.data?.eventId === value.eventId);
+      const found = cachedBranch(this.reader).find(
+        (e) => e.type === "custom" && e.customType === ROUTING_ENTRY && e.data?.eventId === value.eventId,
+      );
       check(found && eventValue(parseRoutingEvent(found.data)) === body, "append_not_observed");
       this.observed.set(key, { value: body, eventId: value.eventId, entryId: found.id });
       this.attempted.delete(key);
-      const branch = this.reader.getBranch();
+      const branch = cachedBranch(this.reader);
       const delta = branch.slice(baseline.length).filter((e) => e.type === "custom" && e.customType === ROUTING_ENTRY);
       if (baseline.every((e, i) => branch[i] === e) && delta.length === 1 && delta[0].id === found.id) {
         // Publish the already validated candidate only after acknowledged native append.
         this.cachedState = candidate;
-        this.cachedEntries = [...branch];
+        this.cachedEntries = branch;
       }
       return value;
     } catch (error) {
@@ -214,4 +255,26 @@ export class EventStore {
       );
     }
   }
+  /**
+   * Write the net effect of control changes staged since the last prompt: at most one delegation, one profile
+   * override, and one control event, each only when it changes the recorded state.
+   */
+  writeStaged(events) {
+    if (!events.length) return;
+    const staged = events;
+    const before = this.branchState();
+    const after = staged.reduce((state, event) => reduce(state, event), before);
+    const last = (type) => staged.filter((event) => event.data.type === type).at(-1);
+    if (before.delegationOverride !== after.delegationOverride) this.append(last("delegation-override"));
+    const changed = Object.fromEntries(
+      [...new Set([...before.profileOverrides.keys(), ...after.profileOverrides.keys()])]
+        .filter(
+          (profile) =>
+            canonical(before.profileOverrides.get(profile)) !== canonical(after.profileOverrides.get(profile)),
+        )
+        .map((profile) => [profile, after.profileOverrides.get(profile) ?? null]),
+    );
+    if (Object.keys(changed).length) this.append(this.make({ ...last("profile-overrides").data, overrides: changed }));
+    if (before.control !== after.control || before.profile !== after.profile) this.append(last("control"));
+  }
 }