@fitlab-ai/agent-infra 0.12.0 → 0.12.2

package/lib/agent-clients/adapters/codex-lifecycle/recovery.ts

@@ -1,418 +0,0 @@
-import fs from 'node:fs';
-
-import { createCodexLifecycleStore, readCodexLifecycleActivationEvidence } from './store.ts';
-import type { CodexLifecycleActivationEvidence, StoredCodexLifecycle } from './store.ts';
-import { managedDelegationRole } from '../../../task/delegation-receipts.ts';
-import type { DelegationReceipt } from '../../../task/delegation-receipts.ts';
-import { normalizeAgentToken } from '../../tokens.ts';
-import { locateActivityLog, pairEntries, appendActivityEntry } from '../../../task/activity-log.ts';
-import type { ActivityLogSection, StepRow } from '../../../task/activity-log.ts';
-import { captureTaskWriteMetadata, writeTask } from '../../../task/write.ts';
-import { parseTypedTaskFrontmatter } from '../../../task/frontmatter.ts';
-import { resolveTaskRef } from '../../../task/resolve-ref.ts';
-import {
-  recoverActivatedOrchestrationDelegation,
-  readRun
-} from '../../../task/orchestration.ts';
-import type { OrchestrationOptions, OrchestrationRun } from '../../../task/orchestration.ts';
-import type {
-  AgentClientLifecycleRecoveryRequest,
-  AgentClientLifecycleRecoveryResult
-} from '../../adapter.ts';
-
-const ACTIONS = {
-  analysis: 'Analyze Task',
-  'review-analysis': 'Review Analysis',
-  plan: 'Plan Task',
-  'review-plan': 'Review Plan',
-  code: 'Code Task',
-  'review-code': 'Review Code'
-} as const;
-
-type RecoveryStage = keyof typeof ACTIONS;
-type LifecycleRecoveryRequest = AgentClientLifecycleRecoveryRequest;
-type LifecycleRecoveryResult = AgentClientLifecycleRecoveryResult;
-type LifecycleRecoveryOptions = Readonly<{
-  repoRoot?: string;
-  now?: () => string;
-  orchestration?: Pick<OrchestrationOptions, 'diagnosticLog' | 'now'>;
-  lifecycleStore?: ReturnType<typeof createCodexLifecycleStore>;
-  writeTask?: typeof writeTask;
-}>;
-
-type RecoveryCandidate = Readonly<{
-  receipt: DelegationReceipt;
-  row: StepRow;
-}>;
-
-function result(
-  request: LifecycleRecoveryRequest,
-  status: LifecycleRecoveryResult['status'],
-  extra: Partial<LifecycleRecoveryResult> = {}
-): LifecycleRecoveryResult {
-  return {
-    status,
-    changed: status === 'applied',
-    targetState: 'active',
-    requestRef: request.taskRef,
-    intent: 'recover-started',
-    taskId: null,
-    receiptId: null,
-    childId: null,
-    error: null,
-    ...extra
-  };
-}
-
-function failure(
-  request: LifecycleRecoveryRequest,
-  status: 'owner-unknown' | 'conflict',
-  code: string,
-  message: string,
-  extra: Partial<LifecycleRecoveryResult> = {}
-): LifecycleRecoveryResult {
-  return result(request, status, { error: { code, message }, ...extra });
-}
-
-function actionPattern(receipt: Pick<DelegationReceipt, 'stage' | 'round'>): RegExp | null {
-  const action = ACTIONS[receipt.stage as RecoveryStage];
-  if (!action) return null;
-  const escaped = action.replace(/[.*+?^${}()|[\]\\]/gu, '\\$&');
-  return new RegExp(`^${escaped} \\(Round ${receipt.round}(?:, (?:fix for review-code(?:-r[2-9]|-r[1-9]\\d+)?.md|decision II-[1-9]\\d*))?\\)$`, 'u');
-}
-
-function openRows(section: ActivityLogSection): StepRow[] {
-  return pairEntries([...section.entries]).filter((row) => row.started !== '' && row.done === '');
-}
-
-function rowForReceipt(rows: readonly StepRow[], receipt: DelegationReceipt): StepRow[] {
-  const pattern = actionPattern(receipt);
-  return pattern ? rows.filter((row) => pattern.test(row.step)) : [];
-}
-
-function readLifecycleStore(
-  options: LifecycleRecoveryOptions,
-  repoRoot: string,
-  taskId: string
-): ReturnType<typeof createCodexLifecycleStore> {
-  return options.lifecycleStore ?? createCodexLifecycleStore({
-    repoRoot,
-    taskId,
-    cliVersion: 'recovery',
-    now: options.now
-  });
-}
-
-function readStoredEvidence(
-  store: ReturnType<typeof createCodexLifecycleStore>,
-  childId: string
-): StoredCodexLifecycle | { missing: true } | { error: Error } {
-  try {
-    return store.read(childId);
-  } catch (error) {
-    const message = error instanceof Error ? error.message : String(error);
-    if (message.includes('was not found uniquely')) return { missing: true };
-    return { error: error instanceof Error ? error : new Error(message) };
-  }
-}
-
-function validateStopRecord(
-  record: StoredCodexLifecycle,
-  receipt: DelegationReceipt
-): { ok: true } | { ok: false; code: string; message: string } {
-  if (!stopRecordMatchesReceipt(record, receipt)) {
-    return { ok: false, code: 'RECOVERY_TASK_BINDING_MISMATCH', message: 'lifecycle evidence belongs to a different task receipt' };
-  }
-  if (!stopEvidenceMatchesReceipt(record, receipt)) {
-    return { ok: false, code: 'RECOVERY_STOP_EVIDENCE_INVALID', message: 'Codex stop evidence does not match the activated receipt' };
-  }
-  if (record.consumer !== null && record.consumer !== receipt.id) {
-    return { ok: false, code: 'RECOVERY_CONSUMER_CONFLICT', message: `Codex lifecycle evidence was consumed by '${record.consumer}'` };
-  }
-  return { ok: true };
-}
-
-function stopRecordMatchesReceipt(record: StoredCodexLifecycle, receipt: DelegationReceipt): boolean {
-  return record.taskBinding?.runId === receipt.runId
-    && record.taskBinding.receiptId === receipt.id;
-}
-
-function stopEvidenceMatchesReceipt(record: StoredCodexLifecycle, receipt: DelegationReceipt): boolean {
-  const start = record.state.startEvidence;
-  const stop = record.state.stopEvidence;
-  const host = readCodexLifecycleActivationEvidence(receipt);
-  if (!stopRecordHasRequiredEvidence(record, receipt, start, stop, host)) return false;
-  return spawnMatchesReceipt(record, receipt, host!, start!)
-    && childMatchesReceipt(record, receipt, start!, stop!)
-    && hostEvidenceMatches(record, receipt, host!, start!);
-}
-
-function stopRecordHasRequiredEvidence(
-  record: StoredCodexLifecycle,
-  receipt: DelegationReceipt,
-  start: StoredCodexLifecycle['state']['startEvidence'],
-  stop: StoredCodexLifecycle['state']['stopEvidence'],
-  host: CodexLifecycleActivationEvidence | null
-): boolean {
-  return receipt.client === 'codex'
-    && host?.kind === 'codex-lifecycle-v2'
-    && Boolean(start && stop && record.state.spawn && record.state.child && record.state.stop)
-    && record.state.status === 'stop-ready'
-    && stop?.terminalStatus === 'completed'
-    && stop.hookStopObserved === true;
-}
-
-function spawnMatchesReceipt(
-  record: StoredCodexLifecycle,
-  receipt: DelegationReceipt,
-  host: CodexLifecycleActivationEvidence,
-  start: NonNullable<StoredCodexLifecycle['state']['startEvidence']>
-): boolean {
-  const spawn = record.state.spawn;
-  return Boolean(spawn)
-    && spawn!.sessionId === start.parentThreadId
-    && spawn!.turnId === host.capabilityTurnId
-    && spawn!.toolUseId === host.spawnToolUseId
-    && spawn!.nativeAgent === start.nativeAgent
-    && spawn!.hookDefinitionHash === host.hookDefinitionHash
-    && spawn!.requestedModel === receipt.requestedModel
-    && spawn!.requestedReasoningEffort === receipt.requestedReasoningEffort
-    && spawn!.taskBinding?.runId === receipt.runId
-    && spawn!.taskBinding.receiptId === receipt.id;
-}
-
-function childMatchesReceipt(
-  record: StoredCodexLifecycle,
-  receipt: DelegationReceipt,
-  start: NonNullable<StoredCodexLifecycle['state']['startEvidence']>,
-  stop: NonNullable<StoredCodexLifecycle['state']['stopEvidence']>
-): boolean {
-  const child = record.state.child;
-  const stopState = record.state.stop;
-  return Boolean(child && stopState)
-    && child!.sessionId === start.parentThreadId
-    && stopState!.turnId === child!.turnId
-    && (stopState!.source === 'parent-rollout'
-      ? stop.turnId === record.state.terminal?.turnId
-      : child!.turnId === stop.turnId)
-    && child!.childThreadId === receipt.childId
-    && child!.nativeAgent === start.nativeAgent
-    && stopState!.childThreadId === receipt.childId
-    && stopState!.nativeAgent === start.nativeAgent
-    && start.childThreadId === receipt.childId
-    && start.parentThreadId === receipt.parentId
-    && managedDelegationRole(start.nativeAgent) === receipt.role;
-}
-
-function hostEvidenceMatches(
-  record: StoredCodexLifecycle,
-  receipt: DelegationReceipt,
-  host: CodexLifecycleActivationEvidence,
-  start: NonNullable<StoredCodexLifecycle['state']['startEvidence']>
-): boolean {
-  return start.hookDefinitionHash === host.hookDefinitionHash
-    && host.capabilitySessionId === start.parentThreadId
-    && host.capabilityTurnId === record.state.spawn?.turnId
-    && host.spawnToolUseId === start.spawnToolUseId
-    && Boolean(host.spawnObservedAt)
-    && host.spawnObservedAt === record.spawnObservedAt
-    && hostRevisionMatches(record, receipt, host);
-}
-
-function hostRevisionMatches(record: StoredCodexLifecycle, receipt: DelegationReceipt, host: CodexLifecycleActivationEvidence): boolean {
-  return host.startRevision >= 1
-    && record.revision > host.startRevision
-    && record.state.child?.childThreadId === receipt.childId
-    && record.state.stop?.childThreadId === receipt.childId;
-}
-
-function abortedCandidates(run: OrchestrationRun, rows: readonly StepRow[]): RecoveryCandidate[] {
-  const candidates = run.receipts.flatMap((receipt) => {
-    if (receipt.status !== 'aborted' || receipt.activatedAt === null || receipt.client !== 'codex') return [];
-    const matches = rowForReceipt(rows, receipt);
-    return matches.length === 1 ? [{ receipt, row: matches[0]! }] : [];
-  });
-  return candidates;
-}
-
-function appendRecoveryLog(
-  request: LifecycleRecoveryRequest,
-  taskId: string,
-  section: ActivityLogSection,
-  candidate: RecoveryCandidate,
-  repoRoot: string,
-  writer: typeof writeTask
-): LifecycleRecoveryResult {
-  const metadata = captureTaskWriteMetadata();
-  const body = appendActivityEntry(section, {
-    time: metadata.timestamp,
-    step: `${candidate.row.step} [aborted]`,
-    agent: request.agent,
-    note: `terminated Codex delegation; receipt=${candidate.receipt.id}; child=${candidate.receipt.childId}`
-  });
-  const written = writer({
-    taskRef: taskId,
-    expectedState: 'active',
-    mutations: [{ kind: 'section', aliases: ['活动日志', 'Activity Log'], heading: section.heading, body }]
-  }, { repoRoot, metadataProvider: () => metadata });
-  if (written.status === 'failed') {
-    return failure(request, 'owner-unknown', 'RECOVERY_LOG_WRITE_FAILED', written.error.message, {
-      taskId, receiptId: candidate.receipt.id, childId: candidate.receipt.childId
-    });
-  }
-  return result(request, 'applied', {
-    taskId, receiptId: candidate.receipt.id, childId: candidate.receipt.childId
-  });
-}
-
-function recoverStartedLifecycle(
-  request: LifecycleRecoveryRequest,
-  options: LifecycleRecoveryOptions = {}
-): LifecycleRecoveryResult {
-  if (!request || request.intent !== 'recover-started' || request.auto !== true || normalizeAgentToken(request.agent) !== 'codex') {
-    return failure(request, 'conflict', 'RECOVERY_PAYLOAD_INVALID', 'automatic recovery requires a task and the Codex agent');
-  }
-  const resolved = resolveTaskRef(request.taskRef, { repoRoot: options.repoRoot });
-  if (!resolved.ok) return failure(request, 'conflict', resolved.code, resolved.message);
-  const taskId = resolved.taskId;
-  let content: string;
-  try {
-    content = fs.readFileSync(resolved.taskMdPath, 'utf8');
-    parseTypedTaskFrontmatter(content);
-  } catch (error) {
-    return failure(request, 'conflict', 'RECOVERY_TASK_INVALID', error instanceof Error ? error.message : String(error), { taskId });
-  }
-  const section = locateActivityLog(content);
-  if (!section) return failure(request, 'conflict', 'RECOVERY_LOG_INVALID', 'task has no unique Activity Log section', { taskId });
-  let run: OrchestrationRun | null;
-  try {
-    run = readRun(resolved.taskDir, options.orchestration);
-  } catch (error) {
-    return failure(request, 'owner-unknown', 'RECOVERY_ORCHESTRATION_UNKNOWN', error instanceof Error ? error.message : String(error), { taskId });
-  }
-  if (!run) return result(request, 'no-op', { taskId });
-
-  const rows = openRows(section);
-  const pending = run.pendingDelegation;
-  if (!pending) return recoverAbortedLifecycle(request, taskId, section, resolved.repoRoot, run, rows, options);
-  return recoverActiveLifecycle(request, taskId, section, resolved.repoRoot, run, rows, pending, options);
-}
-
-function recoverAbortedLifecycle(
-  request: LifecycleRecoveryRequest,
-  taskId: string,
-  section: ActivityLogSection,
-  repoRoot: string,
-  run: OrchestrationRun,
-  rows: readonly StepRow[],
-  options: LifecycleRecoveryOptions
-): LifecycleRecoveryResult {
-  const candidates = abortedCandidates(run, rows);
-  if (candidates.length === 0) return result(request, 'no-op', { taskId });
-  if (candidates.length > 1) {
-    return failure(request, 'conflict', 'RECOVERY_CANDIDATE_AMBIGUOUS', 'more than one aborted delegation has an open lifecycle row', { taskId });
-  }
-  return appendRecoveryLog(request, taskId, section, candidates[0]!, repoRoot, options.writeTask ?? writeTask);
-}
-
-function recoverActiveLifecycle(
-  request: LifecycleRecoveryRequest,
-  taskId: string,
-  section: ActivityLogSection,
-  repoRoot: string,
-  run: OrchestrationRun,
-  rows: readonly StepRow[],
-  pending: DelegationReceipt,
-  options: LifecycleRecoveryOptions
-): LifecycleRecoveryResult {
-  if (pending.status !== 'activated') return result(request, 'no-op', { taskId });
-  if (pending.client !== 'codex' || run.status !== 'running' || run.pause !== null) {
-    return failure(request, 'conflict', 'RECOVERY_ORCHESTRATION_INVALID', 'activated recovery requires one running Codex delegation', {
-      taskId, receiptId: pending.id, childId: pending.childId
-    });
-  }
-  const rowResult = matchingActiveRecoveryRow(request, taskId, rows, pending);
-  if ('result' in rowResult) return rowResult.result;
-  const evidenceResult = consumeRecoveryEvidence(request, taskId, repoRoot, pending, options);
-  if ('result' in evidenceResult) return evidenceResult.result;
-  return finishActiveRecovery(request, taskId, section, repoRoot, pending, rowResult.row, options);
-}
-
-function matchingActiveRecoveryRow(
-  request: LifecycleRecoveryRequest,
-  taskId: string,
-  rows: readonly StepRow[],
-  pending: DelegationReceipt
-): { row: StepRow } | { result: LifecycleRecoveryResult } {
-  const matchingRows = rowForReceipt(rows, pending);
-  if (matchingRows.length !== 1 || normalizeAgentToken(matchingRows[0]!.agent) !== 'codex' || matchingRows[0]!.note !== 'started') {
-    return { result: failure(request, 'conflict', 'RECOVERY_LOG_CONFLICT', 'activated delegation does not have one matching open lifecycle row', {
-      taskId, receiptId: pending.id, childId: pending.childId
-    }) };
-  }
-  if (!pending.childId) {
-    return { result: failure(request, 'conflict', 'RECOVERY_DELEGATION_INVALID', 'activated delegation has no child identity', { taskId, receiptId: pending.id }) };
-  }
-  return { row: matchingRows[0]! };
-}
-
-function consumeRecoveryEvidence(
-  request: LifecycleRecoveryRequest,
-  taskId: string,
-  repoRoot: string,
-  pending: DelegationReceipt,
-  options: LifecycleRecoveryOptions
-): { consumed: StoredCodexLifecycle } | { result: LifecycleRecoveryResult } {
-  const store = readLifecycleStore(options, repoRoot, taskId);
-  const stored = readStoredEvidence(store, pending.childId!);
-  if ('error' in stored) return { result: failure(request, 'owner-unknown', 'RECOVERY_STORE_UNKNOWN', stored.error.message, { taskId, receiptId: pending.id, childId: pending.childId }) };
-  if ('missing' in stored) return { result: failure(request, 'owner-unknown', 'RECOVERY_STOP_EVIDENCE_MISSING', 'matching Codex lifecycle stop evidence is missing', { taskId, receiptId: pending.id, childId: pending.childId }) };
-  const evidence = validateStopRecord(stored, pending);
-  if (!evidence.ok) return { result: failure(request, evidence.code === 'RECOVERY_CONSUMER_CONFLICT' ? 'conflict' : 'owner-unknown', evidence.code, evidence.message, { taskId, receiptId: pending.id, childId: pending.childId }) };
-  let consumed = stored;
-  if (stored.consumer === null) {
-    try {
-      consumed = store.consume(pending.childId!, pending.id, stored.state.startEvidence?.hookDefinitionHash, {
-        taskId: pending.taskId, runId: pending.runId, receiptId: pending.id
-      });
-    } catch (error) {
-      return { result: failure(request, 'owner-unknown', 'RECOVERY_EVIDENCE_CONSUME_FAILED', error instanceof Error ? error.message : String(error), { taskId, receiptId: pending.id, childId: pending.childId }) };
-    }
-  }
-  return consumed.consumedAt
-    ? { consumed }
-    : { result: failure(request, 'owner-unknown', 'RECOVERY_EVIDENCE_INVALID', 'consumed stop evidence has no timestamp', { taskId, receiptId: pending.id, childId: pending.childId }) };
-}
-
-function finishActiveRecovery(
-  request: LifecycleRecoveryRequest,
-  taskId: string,
-  section: ActivityLogSection,
-  repoRoot: string,
-  pending: DelegationReceipt,
-  row: StepRow,
-  options: LifecycleRecoveryOptions
-): LifecycleRecoveryResult {
-  const recovered = recoverActivatedOrchestrationDelegation(taskId, {
-    receiptId: pending.id,
-    stage: pending.stage,
-    round: pending.round,
-    artifact: pending.artifact,
-    startedAgent: request.agent,
-    childId: pending.childId!
-  }, { ...options.orchestration, repoRoot, now: options.now });
-  if (recovered.status === 'failed' || !recovered.run) {
-    return failure(request, 'owner-unknown', recovered.error?.code ?? 'RECOVERY_ORCHESTRATION_FAILED', recovered.error?.message ?? 'orchestration recovery failed', { taskId, receiptId: pending.id, childId: pending.childId });
-  }
-  return appendRecoveryLog(request, taskId, section, { receipt: recovered.run.receipts.at(-1)!, row }, repoRoot, options.writeTask ?? writeTask);
-}
-
-function recoverStartedLifecycleFromAdapter(
-  request: LifecycleRecoveryRequest,
-  options: LifecycleRecoveryOptions = {}
-): LifecycleRecoveryResult {
-  return recoverStartedLifecycle(request, options);
-}
-
-export { recoverStartedLifecycle, recoverStartedLifecycleFromAdapter };
-export type { LifecycleRecoveryOptions, LifecycleRecoveryRequest, LifecycleRecoveryResult };