@abelxiaoxing/cadence 1.1.0 → 1.2.0

package/src/index.ts

@@ -31,6 +31,7 @@ import { loadAgentDefinitions } from "./agent-registry.ts";
 import { runChildSession } from "./child-session.ts";
 import {
   type AtomicVerificationContract,
+  type DesignEvidenceResult,
   isValidRelativePath,
   LIMITS,
   type StructuredVerificationContract,
@@ -41,11 +42,16 @@ import {
   assessDeliveryTraceability,
   compileImplementPlan,
   DeliveryValidationError,
+  type GateApprovalProof,
   IMPLEMENT_PLAN_SCHEMA_VERSION,
   parseGateAReceipt,
   parseImplementPlan,
   parseReadyReceipt,
 } from "./delivery-compiler.ts";
+import {
+  DesignController,
+  validateDesignControlRequest,
+} from "./design-control.ts";
 import { canonicalJson, hashCanonicalValue } from "./implement-graph.ts";
 import { BubblewrapIsolationBackend } from "./isolation-backend.ts";
 import { PACKET_ACTIONS, PacketRuntime } from "./packet-runtime.ts";
@@ -75,10 +81,12 @@ import {
 } from "./verification-capability.ts";
 import {
   openDurableWorkflowEngine,
+  type WorkflowAvailableDelivery,
   type WorkflowDeliverySource,
 } from "./workflow-engine.ts";
 
 export const DISPATCH_TOOL = "abel_dispatch";
+const DESIGN_PARENT_READ_TOOLS = new Set(["read", "grep", "find", "ls"]);
 
 const PACKAGE_ROOT = dirname(dirname(fileURLToPath(import.meta.url)));
 const ELIGIBLE_PROMPTS = [
@@ -109,6 +117,10 @@ const DESIGN_REQUEST_SCHEMA = {
       enum: ["design-explorer"],
       description: "Exact package-owned read-only Agent role.",
     },
+    runId: {
+      type: "string",
+      description: "Durable Design run identity returned by Design start.",
+    },
     id: {
       type: "string",
       description:
@@ -182,6 +194,7 @@ const DESIGN_REQUEST_SCHEMA = {
   required: [
     "stage",
     "role",
+    "runId",
     "id",
     "phase",
     "objective",
@@ -193,6 +206,145 @@ const DESIGN_REQUEST_SCHEMA = {
   additionalProperties: false,
 } as const;
 
+const DESIGN_CONTROL_REQUEST_SCHEMA = {
+  description:
+    "One closed durable Design control operation. The operation body is strict and idempotent by operationId.",
+  oneOf: [
+    ...(["requirement", "change"] as const).map((field) => ({
+      type: "object" as const,
+      properties: {
+        operation: { type: "string" as const, enum: ["start"] },
+        operationId: { type: "string" as const },
+        [field]: {
+          type: "string" as const,
+          description:
+            field === "requirement"
+              ? "Raw Design requirement; code derives and stores only its stable hash."
+              : "Existing canonical change name to resume or revise.",
+        },
+      },
+      required: ["operation", "operationId", field],
+      additionalProperties: false,
+    })),
+    {
+      type: "object",
+      properties: {
+        operation: { type: "string", enum: ["status"] },
+        runId: { type: "string" },
+      },
+      required: ["operation", "runId"],
+      additionalProperties: false,
+    },
+    {
+      type: "object",
+      properties: {
+        operation: { type: "string", enum: ["bind-change"] },
+        runId: { type: "string" },
+        operationId: { type: "string" },
+        change: { type: "string" },
+      },
+      required: ["operation", "runId", "operationId", "change"],
+      additionalProperties: false,
+    },
+    {
+      type: "object",
+      properties: {
+        operation: { type: "string", enum: ["record-decision"] },
+        runId: { type: "string" },
+        operationId: { type: "string" },
+        decisionId: { type: "string" },
+        category: { type: "string", enum: ["behavior", "technical"] },
+        contract: {
+          type: "string",
+          description:
+            "Normalized decision contract; code hashes it and never persists the text.",
+        },
+        refs: { type: "array", items: { type: "string" } },
+      },
+      required: [
+        "operation",
+        "runId",
+        "operationId",
+        "decisionId",
+        "category",
+        "contract",
+        "refs",
+      ],
+      additionalProperties: false,
+    },
+    {
+      type: "object",
+      properties: {
+        operation: { type: "string", enum: ["approve-gate"] },
+        runId: { type: "string" },
+        operationId: { type: "string" },
+        gate: { type: "string", enum: ["gate-a"] },
+        contract: {
+          type: "string",
+          description:
+            "Complete approved WHAT contract; code hashes it and never persists the text.",
+        },
+      },
+      required: ["operation", "runId", "operationId", "gate", "contract"],
+      additionalProperties: false,
+    },
+    {
+      type: "object",
+      properties: {
+        operation: { type: "string", enum: ["approve-gate"] },
+        runId: { type: "string" },
+        operationId: { type: "string" },
+        gate: { type: "string", enum: ["gate-b"] },
+      },
+      required: ["operation", "runId", "operationId", "gate"],
+      additionalProperties: false,
+    },
+    {
+      type: "object",
+      properties: {
+        operation: { type: "string", enum: ["write-artifact"] },
+        runId: { type: "string" },
+        operationId: { type: "string" },
+        path: {
+          type: "string",
+          description: "Allowed path relative to this Design change root.",
+        },
+        content: {
+          type: "string",
+          maxLength: 16 * 1024 * 1024,
+          description: "Exact bounded UTF-8 artifact content.",
+        },
+      },
+      required: ["operation", "runId", "operationId", "path", "content"],
+      additionalProperties: false,
+    },
+    {
+      type: "object",
+      properties: {
+        operation: { type: "string", enum: ["delete-artifact"] },
+        runId: { type: "string" },
+        operationId: { type: "string" },
+        path: {
+          type: "string",
+          description: "Allowed path relative to this Design change root.",
+        },
+      },
+      required: ["operation", "runId", "operationId", "path"],
+      additionalProperties: false,
+    },
+    ...(["compile-plan", "finalize-delivery"] as const).map((operation) => ({
+      type: "object" as const,
+      properties: {
+        operation: { type: "string" as const, enum: [operation] },
+        runId: { type: "string" as const },
+        operationId: { type: "string" as const },
+      },
+      required: ["operation", "runId", "operationId"],
+      additionalProperties: false,
+    })),
+  ],
+} as const;
+
 function invokedPrompt(text: string): EligiblePrompt | undefined {
   const name = text.match(/^\/([^\s]+)(?:\s|$)/)?.[1];
   return ELIGIBLE_PROMPTS.find((candidate) => candidate === name);
@@ -273,6 +425,12 @@ export interface WorkflowControlEngine {
     signal?: AbortSignal,
     onActivity?: (event: WorkflowActivityUpdate) => void,
   ): Promise<Record<string, unknown>>;
+  executeDesign?(request: unknown): Promise<Record<string, unknown>>;
+  assertDesignRun?(runId: string): void;
+  recordDesignEvidence?(input: {
+    runId: string;
+    evidence: DesignEvidenceResult;
+  }): unknown;
   close(): void | Promise<void>;
 }
 
@@ -297,12 +455,40 @@ export interface PackageDeliverySourceOptions {
     consumerRoot: string,
     change: string,
   ) => Promise<OpenSpecDeliveryInspection>;
+  verifyGateProof?: (input: {
+    change: string;
+    gate: "gate-a" | "gate-b";
+    proof: GateApprovalProof;
+  }) => boolean;
+  verifyFinalizedDelivery?: (input: {
+    change: string;
+    deliveryRevision: number;
+    receiptHash: string;
+    gateA: GateApprovalProof;
+    gateB: GateApprovalProof;
+    planCanonicalHash: string;
+  }) => boolean;
+}
+
+export interface PackageWorkflowDeliverySource extends WorkflowDeliverySource {
+  discoverLatest(input: {
+    stage: "abel-implement";
+    change: string;
+  }): Promise<WorkflowAvailableDelivery | undefined>;
 }
 
 function sha256(value: Uint8Array | string): string {
   return createHash("sha256").update(value).digest("hex");
 }
 
+function proofVerified(verify: () => boolean): boolean {
+  try {
+    return verify() === true;
+  } catch {
+    return false;
+  }
+}
+
 function normalizedTrackingArtifact(
   bytes: Uint8Array,
   taskIds: readonly string[],
@@ -449,9 +635,80 @@ export async function inspectOpenSpecDelivery(
 export function packageDeliverySource(
   consumerRoot: string,
   options: PackageDeliverySourceOptions = {},
-): WorkflowDeliverySource {
+): PackageWorkflowDeliverySource {
   const inspect = options.inspectOpenSpec ?? inspectOpenSpecDelivery;
+  const verifyGateProof = options.verifyGateProof;
+  const verifyFinalizedDelivery = options.verifyFinalizedDelivery;
   return {
+    async discoverLatest(input) {
+      if (
+        input.stage !== "abel-implement" ||
+        !verifyGateProof ||
+        !verifyFinalizedDelivery
+      ) {
+        return undefined;
+      }
+      try {
+        const changeRoot = `openspec/changes/${input.change}`;
+        const receiptBytes = readPackageDeliveryFile(
+          consumerRoot,
+          `${changeRoot}/ready.yaml`,
+          4 * 1024 * 1024,
+        );
+        const receipt = parseReadyReceipt(receiptBytes);
+        if (receipt.change !== input.change) return undefined;
+        const gateABytes = readPackageDeliveryFile(
+          consumerRoot,
+          `${changeRoot}/${receipt.approvals.gateA.path}`,
+          4 * 1024 * 1024,
+        );
+        if (sha256(gateABytes) !== receipt.approvals.gateA.rawSha256) {
+          return undefined;
+        }
+        const gateA = parseGateAReceipt(gateABytes);
+        if (
+          gateA.change !== receipt.change ||
+          gateA.schema !== receipt.schema ||
+          !proofVerified(() =>
+            verifyGateProof({
+              change: input.change,
+              gate: "gate-a",
+              proof: gateA.approval,
+            }),
+          ) ||
+          !proofVerified(() =>
+            verifyGateProof({
+              change: input.change,
+              gate: "gate-b",
+              proof: receipt.approvals.gateB,
+            }),
+          )
+        ) {
+          return undefined;
+        }
+        const receiptHash = sha256(receiptBytes);
+        if (
+          !proofVerified(() =>
+            verifyFinalizedDelivery({
+              change: input.change,
+              deliveryRevision: receipt.deliveryRevision,
+              receiptHash,
+              gateA: gateA.approval,
+              gateB: receipt.approvals.gateB,
+              planCanonicalHash: receipt.plan.canonicalHash,
+            }),
+          )
+        ) {
+          return undefined;
+        }
+        return {
+          deliveryRevision: receipt.deliveryRevision,
+          receiptHash,
+        };
+      } catch {
+        return undefined;
+      }
+    },
     async load(input) {
       if (input.stage !== "abel-implement") {
         throw new DeliveryValidationError(["delivery-stage-invalid"]);
@@ -567,6 +824,19 @@ export function packageDeliverySource(
         }
       }
       if (gateA) {
+        if (!verifyGateProof) {
+          diagnostics.add("delivery-gate-proof-verifier-unavailable");
+        } else if (
+          !proofVerified(() =>
+            verifyGateProof({
+              change: input.change,
+              gate: "gate-a",
+              proof: gateA.approval,
+            }),
+          )
+        ) {
+          diagnostics.add("delivery-gate-a-proof-invalid");
+        }
         for (const artifact of gateA.artifacts) {
           if (artifactHashes.get(artifact.path) !== artifact.rawSha256) {
             diagnostics.add(
@@ -575,6 +845,36 @@ export function packageDeliverySource(
           }
         }
       }
+      if (
+        verifyGateProof &&
+        !proofVerified(() =>
+          verifyGateProof({
+            change: input.change,
+            gate: "gate-b",
+            proof: receipt.approvals.gateB,
+          }),
+        )
+      ) {
+        diagnostics.add("delivery-gate-b-proof-invalid");
+      }
+      if (gateA) {
+        if (!verifyFinalizedDelivery) {
+          diagnostics.add("delivery-finalization-verifier-unavailable");
+        } else if (
+          !proofVerified(() =>
+            verifyFinalizedDelivery({
+              change: input.change,
+              deliveryRevision: receipt.deliveryRevision,
+              receiptHash,
+              gateA: gateA.approval,
+              gateB: receipt.approvals.gateB,
+              planCanonicalHash: receipt.plan.canonicalHash,
+            }),
+          )
+        ) {
+          diagnostics.add("delivery-finalization-proof-invalid");
+        }
+      }
 
       let plan: ReturnType<typeof parseImplementPlan> | undefined;
       let planBytes: Buffer | undefined;
@@ -675,7 +975,7 @@ export function packageDeliverySource(
           }
         }
       }
-      if (!plan || diagnostics.size > 0) {
+      if (!plan || !gateA || diagnostics.size > 0) {
         throw new DeliveryValidationError([...diagnostics]);
       }
       return {
@@ -684,6 +984,10 @@ export function packageDeliverySource(
         revision: receipt.deliveryRevision,
         receiptHash,
         plan,
+        approvalProofs: {
+          gateA: structuredClone(gateA.approval),
+          gateB: structuredClone(receipt.approvals.gateB),
+        },
       };
     },
   };
@@ -1169,12 +1473,18 @@ export function openPackageWorkflowControlEngine(
   const routeResolution = loadRoutePolicy({
     cwd: consumerRoot,
     home: homedir(),
+    ...(initialContext.model ? { parentModel: initialContext.model } : {}),
   });
   const stateRoot = resolveStateRoot({
     consumerRoot,
     xdgStateHome: process.env.XDG_STATE_HOME,
   });
   const contexts = new AsyncLocalStorage<ExtensionContext>();
+  const design = DesignController.open({
+    consumerRoot,
+    stateRoot,
+    inspectOpenSpec: inspectOpenSpecDelivery,
+  });
   const implementationAgent = loadAgentDefinitions().find(
     (agent) => agent.role === "implementation-worker",
   );
@@ -1184,7 +1494,10 @@ export function openPackageWorkflowControlEngine(
   const engine = openDurableWorkflowEngine({
     consumerRoot,
     stateRoot,
-    deliverySource: packageDeliverySource(consumerRoot),
+    deliverySource: packageDeliverySource(consumerRoot, {
+      verifyGateProof: (input) => design.verifyGateProof(input),
+      verifyFinalizedDelivery: (input) => design.verifyFinalizedDelivery(input),
+    }),
     routePolicy: routeResolution.ok
       ? routeResolution.policy
       : unavailableRoutePolicy(),
@@ -1357,6 +1670,7 @@ export function openPackageWorkflowControlEngine(
       const operationRouteResolution = loadRoutePolicy({
         cwd: consumerRoot,
         home: homedir(),
+        ...(context.model ? { parentModel: context.model } : {}),
       });
       engine.updateRoutePolicy(
         operationRouteResolution.ok
@@ -1382,9 +1696,19 @@ export function openPackageWorkflowControlEngine(
         return { ...outcome, routePolicy };
       });
     },
+    executeDesign(request: unknown) {
+      return design.execute(request);
+    },
+    assertDesignRun(runId: string) {
+      design.assertDesignRun(runId);
+    },
+    recordDesignEvidence(input) {
+      return design.recordEvidence(input);
+    },
     async close() {
       contexts.disable();
       await engine.close();
+      design.close();
     },
   };
 }
@@ -1410,6 +1734,7 @@ export function registerWorkflowControl(
   const engines = new Map<string, Promise<WorkflowControlEngine>>();
   let pendingPrompt: EligiblePrompt | undefined;
   let activePrompt: EligiblePrompt | undefined;
+  let designToolSnapshot: string[] | undefined;
   const activation = new (class implements Activation {
     state: Activation["state"] = "inactive";
 
@@ -1460,222 +1785,322 @@ export function registerWorkflowControl(
       ),
     );
   };
+  const restoreDesignTools = () => {
+    if (!designToolSnapshot) return;
+    const snapshot = designToolSnapshot;
+    designToolSnapshot = undefined;
+    pi.setActiveTools([...snapshot]);
+  };
+  const enforceDesignTools = () => {
+    if (!designToolSnapshot) {
+      designToolSnapshot = pi
+        .getActiveTools()
+        .filter((name) => name !== DISPATCH_TOOL);
+    }
+    pi.setActiveTools(
+      activateTool(
+        designToolSnapshot.filter((name) => DESIGN_PARENT_READ_TOOLS.has(name)),
+        DISPATCH_TOOL,
+      ),
+    );
+  };
   const deactivate = () => {
     activation.drain();
+    if (designToolSnapshot) {
+      restoreDesignTools();
+      return;
+    }
     const active = pi.getActiveTools();
     if (active.includes(DISPATCH_TOOL)) {
       pi.setActiveTools(deactivateTool(active, DISPATCH_TOOL));
     }
   };
+  const deactivateStage = async () => {
+    activePrompt = undefined;
+    await packetRuntime.drain();
+    deactivate();
+    parentPayloadBridge.clear();
+  };
 
-  pi.registerTool({
-    name: DISPATCH_TOOL,
-    label: "Abel Control",
-    description:
-      "Private stage-bound Abel workflow control. Implement accepts durable change commands; Design and Diagnose accept bounded packet commands.",
-    executionMode: "parallel",
-    parameters: {
-      type: "object",
-      properties: {
-        version: { type: "integer", enum: [2] },
-        command: {
-          type: "string",
-          enum: ["start", "status", "resume", "rebind", "cancel", "discard"],
-        },
-        stage: {
-          type: "string",
-          enum: ["abel-design", "abel-implement"],
-        },
-        change: { type: "string" },
-        provisionalKey: { type: "string" },
-        operationId: { type: "string" },
-        deliveryRevision: { type: "integer", minimum: 1 },
-        receiptHash: { type: "string" },
-        routeId: { type: "string" },
-        action: { type: "string", enum: [...PACKET_ACTIONS] },
-        request: {
-          anyOf: [DESIGN_REQUEST_SCHEMA, GENERIC_REQUEST_SCHEMA],
-        },
+  // Pi validates tool arguments before execute(), so expose only the schema
+  // legal for the active stage instead of one ambiguous command/packet union.
+  const CONTROL_COMMAND_PARAMETERS = {
+    type: "object",
+    properties: {
+      command: {
+        type: "string",
+        enum: ["start", "status", "resume", "rebind", "cancel", "discard"],
       },
-      additionalProperties: false,
-      oneOf: [
-        {
-          required: ["version", "command", "stage"],
-          not: {
-            anyOf: [{ required: ["action"] }, { required: ["request"] }],
-          },
-        },
-        {
-          required: ["action"],
-          not: {
-            anyOf: [
-              { required: ["version"] },
-              { required: ["command"] },
-              { required: ["stage"] },
-              { required: ["change"] },
-              { required: ["provisionalKey"] },
-              { required: ["operationId"] },
-              { required: ["deliveryRevision"] },
-              { required: ["receiptHash"] },
-              { required: ["routeId"] },
-            ],
-          },
-        },
-      ],
+      stage: {
+        type: "string",
+        enum: ["abel-implement"],
+      },
+      change: { type: "string" },
+      operationId: { type: "string" },
+      deliveryRevision: { type: "integer", minimum: 1 },
+      receiptHash: { type: "string" },
+      routeId: { type: "string" },
     },
-    async execute(
-      toolCallId: string,
-      params: unknown,
-      signal: AbortSignal | undefined,
-      onUpdate: AgentToolUpdateCallback<unknown> | undefined,
-      ctx: ExtensionContext,
-    ) {
-      const record =
-        params && typeof params === "object" && !Array.isArray(params)
-          ? (params as Record<string, unknown>)
-          : undefined;
-      if (record && typeof record.action === "string") {
-        if (
-          Object.keys(record).some(
-            (key) => key !== "action" && key !== "request",
-          )
-        ) {
-          throw new Error("control-envelope-ambiguous");
+    required: ["command", "stage"],
+    additionalProperties: false,
+  } as const;
+  const PACKET_PARAMETERS = {
+    type: "object",
+    properties: {
+      action: { type: "string", enum: [...PACKET_ACTIONS, "design"] },
+      request: {
+        anyOf: [
+          DESIGN_REQUEST_SCHEMA,
+          DESIGN_CONTROL_REQUEST_SCHEMA,
+          GENERIC_REQUEST_SCHEMA,
+        ],
+      },
+    },
+    required: ["action"],
+    additionalProperties: false,
+  } as const;
+  let registeredParameterKind: "command" | "packet" | undefined;
+
+  const registerDispatchTool = (kind: "command" | "packet") => {
+    if (registeredParameterKind === kind) return;
+    registeredParameterKind = kind;
+    pi.registerTool({
+      name: DISPATCH_TOOL,
+      label: "Abel Control",
+      description:
+        kind === "command"
+          ? "Private stage-bound Abel workflow control. Accepts durable Implement change commands."
+          : "Private stage-bound Abel packet control. Accepts bounded Design and Diagnose packet operations.",
+      executionMode: "parallel",
+      parameters:
+        kind === "command" ? CONTROL_COMMAND_PARAMETERS : PACKET_PARAMETERS,
+      async execute(
+        toolCallId: string,
+        params: unknown,
+        signal: AbortSignal | undefined,
+        onUpdate: AgentToolUpdateCallback<unknown> | undefined,
+        ctx: ExtensionContext,
+      ) {
+        const record =
+          params && typeof params === "object" && !Array.isArray(params)
+            ? (params as Record<string, unknown>)
+            : undefined;
+        if (record && typeof record.action === "string") {
+          if (
+            Object.keys(record).some(
+              (key) => key !== "action" && key !== "request",
+            )
+          ) {
+            throw new Error("control-envelope-ambiguous");
+          }
+          if (
+            activePrompt !== "abel-design" &&
+            activePrompt !== "abel-diagnose"
+          ) {
+            throw new Error("stage-control-mismatch");
+          }
+          if (record.action === "design") {
+            if (activePrompt !== "abel-design") {
+              throw new Error("stage-control-mismatch");
+            }
+            const designRequest = validateDesignControlRequest(record.request);
+            if (!designRequest.ok) throw new Error(designRequest.code);
+            const engine = await engineFor(ctx);
+            if (!engine.executeDesign)
+              throw new Error("design-control-unavailable");
+            const payload = await engine.executeDesign(designRequest.value);
+            if (
+              designRequest.value.operation === "finalize-delivery" &&
+              payload.state === "completed"
+            ) {
+              await deactivateStage();
+            }
+            return {
+              content: [
+                { type: "text" as const, text: JSON.stringify(payload) },
+              ],
+              details: payload,
+            };
+          }
+          let designEngine: WorkflowControlEngine | undefined;
+          if (record.action === "run") {
+            const packet = packetRuntime.validateRequest(record.request);
+            if (!packet.ok || packet.value.stage !== activePrompt) {
+              throw new Error(
+                packet.ok ? "stage-control-mismatch" : packet.reason,
+              );
+            }
+            if (packet.value.stage === "abel-design") {
+              designEngine = await engineFor(ctx);
+              if (!designEngine.assertDesignRun || !packet.value.runId) {
+                throw new Error("design-control-unavailable");
+              }
+              designEngine.assertDesignRun(packet.value.runId);
+            }
+          }
+          const operation = {
+            request: record.request,
+          };
+          const tuiRun = ctx.mode === "tui" && record.action === "run";
+          const packetPayload = tuiRun
+            ? await packetRuntime.execute(
+                record.action,
+                operation,
+                ctx,
+                signal,
+                activity.observe(
+                  toolCallId,
+                  onUpdate as ((result: unknown) => void) | undefined,
+                ),
+              )
+            : await packetRuntime.execute(
+                record.action,
+                operation,
+                ctx,
+                signal,
+              );
+          let payload:
+            | typeof packetPayload
+            | (typeof packetPayload & {
+                recordedEvidence: unknown;
+              }) = packetPayload;
+          if (
+            packetPayload.ok &&
+            record.action === "run" &&
+            activePrompt === "abel-design"
+          ) {
+            const packet = packetRuntime.validateRequest(record.request);
+            if (
+              !packet.ok ||
+              !packet.value.runId ||
+              !designEngine?.recordDesignEvidence
+            ) {
+              throw new Error("design-control-unavailable");
+            }
+            const recordedEvidence = designEngine.recordDesignEvidence({
+              runId: packet.value.runId,
+              evidence: packetPayload.result as DesignEvidenceResult,
+            });
+            payload = { ...packetPayload, recordedEvidence };
+          }
+          if (record.action === "finish" && packetPayload.ok) {
+            await deactivateStage();
+          }
+          const display = tuiRun
+            ? activity.finalize(toolCallId, payload)
+            : undefined;
+          const { payload: publicPayload, usage } = splitUsage(payload);
+          const details = display
+            ? {
+                ...(publicPayload as Record<string, unknown>),
+                [ACTIVITY_DETAILS_KEY]: display,
+              }
+            : publicPayload;
+          return {
+            content: [
+              { type: "text" as const, text: JSON.stringify(publicPayload) },
+            ],
+            details,
+            ...(usage === undefined ? {} : { usage }),
+          };
+        }
+        const validation = validateControlCommand(params);
+        if (!validation.ok) {
+          const error = new Error(validation.code);
+          error.name = "ControlCommandError";
+          throw error;
         }
         if (
-          activePrompt !== "abel-design" &&
-          activePrompt !== "abel-diagnose"
+          activePrompt !== undefined &&
+          (activePrompt === "abel-diagnose" ||
+            validation.value.stage !== activePrompt)
         ) {
           throw new Error("stage-control-mismatch");
         }
-        if (record.action === "run") {
-          const packet = packetRuntime.validateRequest(record.request);
-          if (!packet.ok || packet.value.stage !== activePrompt) {
-            throw new Error(
-              packet.ok ? "stage-control-mismatch" : packet.reason,
+        const engine = await engineFor(ctx);
+        const tuiControl = ctx.mode === "tui";
+        if (tuiControl) {
+          try {
+            activity.beginWorkflow(
+              toolCallId,
+              validation.value,
+              onUpdate as ((result: unknown) => void) | undefined,
             );
+          } catch {
+            // Presentation state must never alter workflow execution.
           }
         }
-        const operation = {
-          request: record.request,
-        };
-        const tuiRun = ctx.mode === "tui" && record.action === "run";
-        const payload = tuiRun
-          ? await packetRuntime.execute(
-              record.action,
-              operation,
-              ctx,
-              signal,
-              activity.observe(
-                toolCallId,
-                onUpdate as ((result: unknown) => void) | undefined,
-              ),
-            )
-          : await packetRuntime.execute(record.action, operation, ctx, signal);
-        const display = tuiRun
-          ? activity.finalize(toolCallId, payload)
-          : undefined;
-        const { payload: publicPayload, usage } = splitUsage(payload);
-        const details = display
-          ? {
-              ...(publicPayload as Record<string, unknown>),
-              [ACTIVITY_DETAILS_KEY]: display,
-            }
-          : publicPayload;
-        return {
-          content: [
-            { type: "text" as const, text: JSON.stringify(publicPayload) },
-          ],
-          details,
-          ...(usage === undefined ? {} : { usage }),
-        };
-      }
-      const validation = validateControlCommand(params);
-      if (!validation.ok) {
-        const error = new Error(validation.code);
-        error.name = "ControlCommandError";
-        throw error;
-      }
-      if (
-        activePrompt !== undefined &&
-        (activePrompt === "abel-diagnose" ||
-          validation.value.stage !== activePrompt)
-      ) {
-        throw new Error("stage-control-mismatch");
-      }
-      const engine = await engineFor(ctx);
-      const tuiControl = ctx.mode === "tui";
-      if (tuiControl) {
+        let payload: Record<string, unknown>;
         try {
-          activity.beginWorkflow(
-            toolCallId,
+          payload = await engine.execute(
             validation.value,
-            onUpdate as ((result: unknown) => void) | undefined,
-          );
-        } catch {
-          // Presentation state must never alter workflow execution.
-        }
-      }
-      let payload: Record<string, unknown>;
-      try {
-        payload = await engine.execute(
-          validation.value,
-          ctx,
-          signal,
-          tuiControl
-            ? (event) => {
-                try {
-                  activity.updateWorkflow(toolCallId, validation.value, event);
-                } catch {
-                  // Presentation state must never alter workflow execution.
+            ctx,
+            signal,
+            tuiControl
+              ? (event) => {
+                  try {
+                    activity.updateWorkflow(
+                      toolCallId,
+                      validation.value,
+                      event,
+                    );
+                  } catch {
+                    // Presentation state must never alter workflow execution.
+                  }
                 }
-              }
-            : undefined,
-        );
-      } catch (error) {
-        if (tuiControl) activity.failWorkflow(toolCallId);
-        throw error;
-      }
-      let display: ReturnType<typeof activity.finalizeWorkflow>;
-      if (tuiControl) {
-        try {
-          display = activity.finalizeWorkflow(
-            toolCallId,
-            validation.value,
-            payload,
+              : undefined,
           );
-        } catch {
-          activity.failWorkflow(toolCallId);
+        } catch (error) {
+          if (tuiControl) activity.failWorkflow(toolCallId);
+          throw error;
         }
-      }
-      return {
-        content: [{ type: "text" as const, text: JSON.stringify(payload) }],
-        details: display
-          ? { ...payload, [ACTIVITY_DETAILS_KEY]: display }
-          : payload,
-      };
-    },
-    renderCall(args: unknown, theme: Theme, context: unknown) {
-      return renderActivityCall(
-        args,
-        theme,
-        context as Parameters<typeof renderActivityCall>[2],
-      );
-    },
-    renderResult(
-      result: AgentToolResult<unknown>,
-      options: ToolRenderResultOptions,
-      theme: Theme,
-      context: unknown,
-    ) {
-      return renderActivityResult(
-        result,
-        options,
-        theme,
-        context as Parameters<typeof renderActivityResult>[3],
-      );
-    },
-  } as never);
+        if (
+          ["completed", "discarded", "rejected"].includes(String(payload.state))
+        ) {
+          await deactivateStage();
+        }
+        let display: ReturnType<typeof activity.finalizeWorkflow>;
+        if (tuiControl) {
+          try {
+            display = activity.finalizeWorkflow(
+              toolCallId,
+              validation.value,
+              payload,
+            );
+          } catch {
+            activity.failWorkflow(toolCallId);
+          }
+        }
+        return {
+          content: [{ type: "text" as const, text: JSON.stringify(payload) }],
+          details: display
+            ? { ...payload, [ACTIVITY_DETAILS_KEY]: display }
+            : payload,
+        };
+      },
+      renderCall(args: unknown, theme: Theme, context: unknown) {
+        return renderActivityCall(
+          args,
+          theme,
+          context as Parameters<typeof renderActivityCall>[2],
+        );
+      },
+      renderResult(
+        result: AgentToolResult<unknown>,
+        options: ToolRenderResultOptions,
+        theme: Theme,
+        context: unknown,
+      ) {
+        return renderActivityResult(
+          result,
+          options,
+          theme,
+          context as Parameters<typeof renderActivityResult>[3],
+        );
+      },
+    } as never);
+  };
+  registerDispatchTool("packet");
 
   pi.on("input", (event) => {
     pendingPrompt = invokedPrompt(event.text);
@@ -1687,6 +2112,10 @@ export function registerWorkflowControl(
     const verified =
       prompt && isVerifiedStageInvocation(pi, activation, prompt, event.prompt);
     if (verified) {
+      if (activePrompt === "abel-design" && prompt !== "abel-design") {
+        restoreDesignTools();
+      }
+      registerDispatchTool(prompt === "abel-implement" ? "command" : "packet");
       activePrompt = prompt;
       const sessionId = ctx.sessionManager?.getSessionId?.();
       if (typeof sessionId === "string") {
@@ -1699,6 +2128,7 @@ export function registerWorkflowControl(
       parentPayloadBridge.install(ctx.model, ctx.modelRegistry);
     }
     if (prompt) activateDispatcher(pi, activation, prompt, event.prompt);
+    if (verified && prompt === "abel-design") enforceDesignTools();
   });
   pi.on("model_select", (event, ctx) => {
     const sessionId = ctx.sessionManager?.getSessionId?.();