@abelxiaoxing/cadence 1.0.2 → 1.0.4

package/src/worker.ts

@@ -1,163 +1,94 @@
-// Process-local logical Worker registry for private orchestration. A Worker
-// is keyed by stable task identity. Its immutable task contract is separate
-// from the current phase request, snapshot, verification, and launch budget.
-// No credential, transcript, provider diagnostic, or unbounded retry ledger
-// is retained.
-
-import type { RequestEnvelope, VerificationContract } from "./contracts.ts";
-
-export interface WorkerContract {
-  stage: string;
-  role: string;
-  taskId: string;
-  objective: string;
-  roots: string[];
-  agents: string;
-  contractText: string;
-  read: string[];
-  conflicts: string[];
-  resources: string[];
-  output: string;
-  currentPhase: WorkerPhaseContract;
-}
-
-export interface WorkerTaskContract {
-  stage: string;
-  role: string;
-  taskId: string;
-  objective: string;
-  roots: string[];
-  agents: string;
-  contractText: string;
-  read: string[];
-  conflicts: string[];
-  resources: string[];
-  output: string;
-}
-
-export interface WorkerPhaseContract {
-  requestId: string;
-  phase: string;
-  readSet: string[];
-  writeSet: string[];
-  conflicts: string[];
-  resources: string[];
-  verificationLock?: string;
-  snapshot?: unknown;
-  verification?: VerificationContract;
-  correctionIndex: 0 | 1;
-}
-
-export interface LogicalWorker {
-  identity: string;
-  taskContract: WorkerTaskContract;
-  currentPhase: WorkerPhaseContract;
-  state: WorkerState;
+// Process-local task registry for the private Implement runtime.
+
+import type {
+  CandidateFailure,
+  ImplementationPhase,
+  PhaseAttempt,
+  TaskBoundary,
+  TaskFailure,
+} from "./contracts.ts";
+import type { ConflictDeclaration } from "./scheduler.ts";
+import type { SubagentEndpoint } from "./subagent-endpoint.ts";
+
+export type TaskState =
+  | {
+      kind: "ready";
+      phase: ImplementationPhase;
+      launchIndex: 0 | 1;
+      correction?: CandidateFailure;
+    }
+  | {
+      kind: "candidate-pending";
+      phase: ImplementationPhase;
+      launchIndex: 0 | 1;
+      originRequestId: string;
+      resultId: string;
+    }
+  | {
+      kind: "agents-checkpoint-pending";
+      finalPhase: "green" | "refactor";
+      attemptIndex: 0 | 1;
+    }
+  | { kind: "blocked"; phase: ImplementationPhase; failure: TaskFailure }
+  | { kind: "completed"; finalPhase: "green" | "refactor" };
+
+export interface TaskRecord {
+  key: string;
+  workspaceRoot: string;
+  workerIdentity: string;
+  subagentEndpoint: Readonly<SubagentEndpoint> | null;
+  boundary: TaskBoundary;
+  conflict: ConflictDeclaration;
+  state: TaskState;
 }
 
-export type WorkerState =
-  | { kind: "ready" }
-  | { kind: "candidate-pending" }
-  | { kind: "phase-applied" }
-  | { kind: "artifact-correction-pending"; rejection: string }
-  | { kind: "stale-redispatch-pending" }
-  | { kind: "blocked"; reason: "artifact" | "mechanical" };
-
-export function contractOf(envelope: RequestEnvelope): WorkerContract {
-  return {
-    stage: envelope.stage,
-    role: envelope.role,
-    taskId: envelope.taskId ?? envelope.id,
-    objective: envelope.objective,
-    roots: [...envelope.roots],
-    agents: envelope.context.agents,
-    contractText: envelope.context.contract,
-    read: [...envelope.declared.read],
-    conflicts: [...envelope.declared.conflicts],
-    resources: [...envelope.declared.resources],
-    output: envelope.output,
-    currentPhase: phaseOf(envelope, 0),
-  };
+export function taskRecordKey(
+  workspaceRoot: string,
+  changeId: string,
+  taskId: string,
+): string {
+  return `${workspaceRoot}\0${changeId}\0${taskId}`;
 }
 
-export function sameContract(
-  a: WorkerTaskContract,
-  b: WorkerTaskContract,
-): boolean {
-  if (
-    a.stage !== b.stage ||
-    a.role !== b.role ||
-    a.taskId !== b.taskId ||
-    a.objective !== b.objective ||
-    a.agents !== b.agents ||
-    a.contractText !== b.contractText ||
-    a.output !== b.output
-  )
-    return false;
-  return (
-    arrayEqual(a.roots, b.roots) &&
-    arrayEqual(a.read, b.read) &&
-    arrayEqual(a.conflicts, b.conflicts) &&
-    arrayEqual(a.resources, b.resources)
-  );
-}
-
-export function samePhaseContract(
-  a: WorkerPhaseContract,
-  b: WorkerPhaseContract,
-): boolean {
-  return (
-    a.phase === b.phase &&
-    arrayEqual(a.readSet, b.readSet) &&
-    arrayEqual(a.writeSet, b.writeSet) &&
-    arrayEqual(a.conflicts, b.conflicts) &&
-    arrayEqual(a.resources, b.resources) &&
-    a.verificationLock === b.verificationLock &&
-    JSON.stringify(a.verification) === JSON.stringify(b.verification)
-  );
-}
-
-function phaseOf(
-  envelope: RequestEnvelope,
-  correctionIndex: 0 | 1,
-): WorkerPhaseContract {
+export function taskConflictOf(boundary: TaskBoundary): ConflictDeclaration {
+  const phases = [
+    boundary.phases.red,
+    boundary.phases.green,
+    ...(boundary.phases.refactor ? [boundary.phases.refactor] : []),
+  ];
+  const write = new Set(phases.flatMap((phase) => phase.write));
+  if (boundary.agents.impact !== "none" && boundary.agents.target) {
+    write.add(boundary.agents.target);
+  }
   return {
-    requestId: envelope.id,
-    phase: envelope.phase,
-    readSet: [...envelope.declared.read],
-    writeSet: [...envelope.declared.write],
-    conflicts: [...envelope.declared.conflicts],
-    resources: [...envelope.declared.resources],
-    verificationLock: envelope.declared.verificationLock,
-    snapshot:
-      envelope.snapshot === undefined
-        ? undefined
-        : structuredClone(envelope.snapshot),
-    verification:
-      envelope.verification === undefined
-        ? undefined
-        : {
-            ...envelope.verification,
-            argv: [...envelope.verification.argv],
-          },
-    correctionIndex,
+    taskId: boundary.taskId,
+    read: [...new Set(phases.flatMap((phase) => phase.read))],
+    write: [...write],
+    conflicts: [...boundary.scheduling.conflicts],
+    resources: [...boundary.scheduling.resources],
+    verificationLocks: [
+      ...new Set(
+        phases.flatMap((phase) =>
+          phase.verificationLock ? [phase.verificationLock] : [],
+        ),
+      ),
+    ],
   };
 }
 
-function arrayEqual(a: string[], b: string[]): boolean {
-  return a.length === b.length && a.every((value, index) => value === b[index]);
-}
-
-// Stable provider/model/API/thinking identity fingerprint for a phase. Each
-// child attempt re-derives it while the phase-local auth boundary is called
-// anew, so the identity stays pinned across fresh phases.
+// Stable provider/model/API identity fingerprint. Fresh auth may be acquired
+// for each phase, while the selected execution identity remains pinned.
 export function workerIdentity(model: {
   provider?: unknown;
+  api?: unknown;
   baseUrl?: unknown;
   id?: unknown;
   name?: unknown;
 }): string {
-  const parts: string[] = [`p:${String(model.provider ?? "")}`];
+  const parts: string[] = [
+    `p:${String(model.provider ?? "")}`,
+    `a:${String(model.api ?? "")}`,
+  ];
   if (model.baseUrl !== undefined) parts.push(`u:${String(model.baseUrl)}`);
   if (typeof model.id === "string") parts.push(`id:${model.id}`);
   if (typeof model.name === "string") parts.push(`n:${model.name}`);
@@ -165,65 +96,60 @@ export function workerIdentity(model: {
 }
 
 export class WorkerRegistry {
-  private readonly workers = new Map<string, LogicalWorker>();
+  private readonly tasks = new Map<string, TaskRecord>();
 
-  has(id: string): boolean {
-    return this.workers.has(id);
+  has(key: string): boolean {
+    return this.tasks.has(key);
   }
 
-  get(id: string): LogicalWorker | undefined {
-    return this.workers.get(id);
+  get(key: string): TaskRecord | undefined {
+    return this.tasks.get(key);
   }
 
-  pin(contract: WorkerContract, identity: string): LogicalWorker {
-    const currentPhase = clonePhase(contract.currentPhase);
-    const worker: LogicalWorker = {
-      identity,
-      taskContract: cloneTask(contract),
-      currentPhase,
-      state: { kind: "ready" },
-    };
-    this.workers.set(contract.taskId, worker);
-    return worker;
+  find(workspaceRoot: string, taskId: string): TaskRecord | undefined {
+    const matches = [...this.tasks.values()].filter(
+      (record) =>
+        record.workspaceRoot === workspaceRoot &&
+        record.boundary.taskId === taskId,
+    );
+    return matches.length === 1 ? matches[0] : undefined;
   }
 
-  setCurrentPhase(
-    taskId: string,
-    contract: WorkerContract,
-    correctionIndex: 0 | 1,
-  ): LogicalWorker | undefined {
-    const worker = this.workers.get(taskId);
-    if (!worker) return undefined;
-    const currentPhase = {
-      ...clonePhase(contract.currentPhase),
-      correctionIndex,
+  open(
+    boundary: TaskBoundary,
+    identity: string,
+    workspaceRoot: string,
+    attempt: PhaseAttempt,
+    subagentEndpoint: Readonly<SubagentEndpoint> | null = null,
+  ): TaskRecord {
+    const key = taskRecordKey(
+      workspaceRoot,
+      boundary.changeId,
+      boundary.taskId,
+    );
+    if (this.tasks.has(key)) throw new Error("duplicate task open");
+    const storedBoundary = structuredClone(boundary);
+    const record: TaskRecord = {
+      key,
+      workspaceRoot,
+      workerIdentity: identity,
+      subagentEndpoint:
+        subagentEndpoint === null
+          ? null
+          : Object.freeze({ ...subagentEndpoint }),
+      boundary: storedBoundary,
+      conflict: taskConflictOf(storedBoundary),
+      state: { kind: "ready", phase: attempt.phase, launchIndex: 0 },
     };
-    worker.currentPhase = currentPhase;
-    worker.state = { kind: "ready" };
-    return worker;
+    this.tasks.set(key, record);
+    return record;
   }
 
-  clear(): void {
-    this.workers.clear();
+  values(): TaskRecord[] {
+    return [...this.tasks.values()];
   }
-}
 
-function cloneTask(contract: WorkerContract): WorkerTaskContract {
-  return {
-    stage: contract.stage,
-    role: contract.role,
-    taskId: contract.taskId,
-    objective: contract.objective,
-    roots: [...contract.roots],
-    agents: contract.agents,
-    contractText: contract.contractText,
-    read: [...contract.read],
-    conflicts: [...contract.conflicts],
-    resources: [...contract.resources],
-    output: contract.output,
-  };
-}
-
-function clonePhase(phase: WorkerPhaseContract): WorkerPhaseContract {
-  return structuredClone(phase);
+  clear(): void {
+    this.tasks.clear();
+  }
 }