@wichayutdew/pi-workflows 0.2.2 → 0.2.3

package/src/engine/transitions.ts

@@ -51,11 +51,25 @@ export function pauseRun(
       status: 'paused',
       pausedFrom: run.status,
       pauseReason: reason || `Paused during step "${run.currentStepId}"`,
+      failedStepId: undefined,
     },
     now,
   );
 }
 
+export function failRun(
+  run: WorkflowRun,
+  reason: string,
+  now: number,
+): WorkflowRun {
+  const paused = pauseRun(run, reason, now);
+  if (paused.status !== 'paused') return paused;
+  return {
+    ...paused,
+    failedStepId: paused.currentStepId,
+  };
+}
+
 export function resumeRun(run: WorkflowRun, now: number): WorkflowRun {
   if (run.status !== 'paused') return run;
   return withUpdate(
@@ -64,6 +78,7 @@ export function resumeRun(run: WorkflowRun, now: number): WorkflowRun {
       status: run.pausedFrom ?? (run.pendingGate ? 'awaiting-gate' : 'running'),
       pauseReason: undefined,
       pausedFrom: undefined,
+      failedStepId: undefined,
     },
     now,
   );
@@ -80,6 +95,7 @@ export function abortRun(
       status: 'aborted',
       pauseReason: reason || 'Aborted by user',
       pausedFrom: undefined,
+      failedStepId: undefined,
       pendingGate: undefined,
     },
     now,
@@ -166,6 +182,7 @@ export function advanceRun(
       stepHandoff: summary,
       lastSummary: summary,
       gateFeedback: '',
+      ...(nextStep.gate ? { reviewedArtifact: '' } : {}),
       ...(overVisitLimit
         ? {
             pausedFrom: 'running' as const,
@@ -334,6 +351,31 @@ function retainedReviewedArtifact(
   return sourceRetained ? reviewedArtifact : '';
 }
 
+function refreshApprovedGateHistory(
+  run: WorkflowRun,
+  workflow: LoadedWorkflow,
+): WorkflowRun {
+  const reviewedArtifact = run.reviewedArtifact ?? '';
+  if (!reviewedArtifact) return run;
+  let changed = false;
+  const history = run.history.map((entry) => {
+    const gate = workflow.definition.steps[entry.stepId]?.gate;
+    const currentDigest = workflow.stepDigests[entry.stepId];
+    if (
+      gate &&
+      currentDigest &&
+      entry.outcome === gate.approvedOutcome &&
+      entry.summary === reviewedArtifact &&
+      entry.stepDigest !== currentDigest
+    ) {
+      changed = true;
+      return { ...entry, stepDigest: currentDigest };
+    }
+    return entry;
+  });
+  return changed ? { ...run, history } : run;
+}
+
 export function reconcileRun(
   run: WorkflowRun,
   workflow: LoadedWorkflow,
@@ -354,12 +396,13 @@ export function reconcileRun(
       error: `current step "${run.currentStepId}" was removed; abort or restore the configuration`,
     };
   }
+  const reconciledRun = refreshApprovedGateHistory(run, workflow);
 
-  const changedHistoryIndex = run.history.findIndex(
+  const changedHistoryIndex = reconciledRun.history.findIndex(
     (entry) => workflow.stepDigests[entry.stepId] !== entry.stepDigest,
   );
   if (changedHistoryIndex >= 0) {
-    const changedEntry = run.history[changedHistoryIndex];
+    const changedEntry = reconciledRun.history[changedHistoryIndex];
     if (!changedEntry || !workflow.definition.steps[changedEntry.stepId]) {
       return {
         changed: true,
@@ -367,19 +410,19 @@ export function reconcileRun(
           'a completed step was removed; abort or restore the configuration',
       };
     }
-    const retainedHistory = run.history.slice(0, changedHistoryIndex);
+    const retainedHistory = reconciledRun.history.slice(0, changedHistoryIndex);
     const restartedStep = changedEntry.stepId;
     const stepHandoff = retainedHistory.at(-1)?.summary ?? '';
     const reviewedArtifact = retainedReviewedArtifact(
       workflow,
-      run,
+      reconciledRun,
       retainedHistory,
     );
     return {
       changed: true,
       restartedStep,
       run: withUpdate(
-        run,
+        reconciledRun,
         {
           workflowDigest: workflow.digest,
           status: 'paused',
@@ -392,6 +435,7 @@ export function reconcileRun(
           lastSummary: stepHandoff,
           pendingGate: undefined,
           pausedFrom: 'running',
+          failedStepId: undefined,
           pauseReason: `Configuration changed; restarted step "${restartedStep}"`,
           gateFeedback: '',
         },
@@ -401,12 +445,12 @@ export function reconcileRun(
   }
 
   const currentDigest = workflow.stepDigests[run.currentStepId] ?? '';
-  const currentChanged = currentDigest !== run.currentStepDigest;
+  const currentChanged = currentDigest !== reconciledRun.currentStepDigest;
   return {
     changed: true,
     ...(currentChanged ? { restartedStep: run.currentStepId } : {}),
     run: withUpdate(
-      run,
+      reconciledRun,
       {
         workflowDigest: workflow.digest,
         currentStepDigest: currentDigest,
@@ -415,6 +459,7 @@ export function reconcileRun(
               status: 'paused' as const,
               pendingGate: undefined,
               pausedFrom: 'running' as const,
+              failedStepId: undefined,
               pauseReason: `Configuration changed; restarted step "${run.currentStepId}"`,
               gateFeedback: '',
             }