agentweaver 0.1.14 → 0.1.16

package/dist/flow-state.js

@@ -1,16 +1,27 @@
+import { randomUUID } from "node:crypto";
 import { existsSync, readFileSync, rmSync, writeFileSync } from "node:fs";
 import { ensureScopeWorkspaceDir, flowStateFile } from "./artifacts.js";
 import { TaskRunnerError } from "./errors.js";
-const FLOW_STATE_SCHEMA_VERSION = 1;
+import { isFlowRunResumeEnvelope } from "./pipeline/flow-run-resume.js";
+import { resolveStoredExecutionRoutingSnapshot, singleLaunchProfileExecutionRouting } from "./runtime/execution-routing.js";
+const FLOW_STATE_SCHEMA_VERSION = 2;
 function nowIso8601() {
     return new Date().toISOString();
 }
+function ensurePublicationRunId(executionState) {
+    executionState.publicationRunId ??= randomUUID();
+    return executionState.publicationRunId;
+}
 export function stripExecutionStatePayload(executionState) {
+    ensurePublicationRunId(executionState);
     return {
+        ...(executionState.runId ? { runId: executionState.runId } : {}),
+        ...(executionState.publicationRunId ? { publicationRunId: executionState.publicationRunId } : {}),
         flowKind: executionState.flowKind,
         flowVersion: executionState.flowVersion,
         terminated: executionState.terminated,
         ...(executionState.terminationReason ? { terminationReason: executionState.terminationReason } : {}),
+        ...(executionState.terminationOutcome ? { terminationOutcome: executionState.terminationOutcome } : {}),
         phases: executionState.phases.map((phase) => ({
             id: phase.id,
             status: phase.status,
@@ -22,6 +33,7 @@ export function stripExecutionStatePayload(executionState) {
                 status: step.status,
                 ...(step.outputs ? { outputs: step.outputs } : {}),
                 ...(step.value !== undefined ? { value: step.value } : {}),
+                ...(step.publishedArtifacts ? { publishedArtifacts: step.publishedArtifacts } : {}),
                 ...(step.startedAt ? { startedAt: step.startedAt } : {}),
                 ...(step.finishedAt ? { finishedAt: step.finishedAt } : {}),
                 ...(step.stopFlow !== undefined ? { stopFlow: step.stopFlow } : {}),
@@ -29,7 +41,10 @@ export function stripExecutionStatePayload(executionState) {
         })),
     };
 }
-export function createFlowRunState(scopeKey, flowId, executionState, jiraRef, launchProfile) {
+export function createFlowRunState(scopeKey, flowId, executionState, jiraRef, launchProfile, executionRouting, selectedRoutingPreset) {
+    ensurePublicationRunId(executionState);
+    const effectiveExecutionRouting = executionRouting ?? (launchProfile ? singleLaunchProfileExecutionRouting(launchProfile) : undefined);
+    const effectiveLaunchProfile = launchProfile ?? effectiveExecutionRouting?.defaultRoute;
     return {
         schemaVersion: FLOW_STATE_SCHEMA_VERSION,
         flowId,
@@ -38,10 +53,52 @@ export function createFlowRunState(scopeKey, flowId, executionState, jiraRef, la
         status: "pending",
         currentStep: null,
         updatedAt: nowIso8601(),
-        ...(launchProfile ? { launchProfile } : {}),
+        ...(effectiveLaunchProfile ? { launchProfile: effectiveLaunchProfile } : {}),
+        ...(effectiveExecutionRouting ? { executionRouting: effectiveExecutionRouting, routingFingerprint: effectiveExecutionRouting.fingerprint } : {}),
+        ...(selectedRoutingPreset ? { selectedRoutingPreset } : {}),
         executionState: stripExecutionStatePayload(executionState),
     };
 }
+function upgradeFlowRunStateV1(state) {
+    const executionRouting = state.launchProfile ? singleLaunchProfileExecutionRouting(state.launchProfile) : undefined;
+    return {
+        ...state,
+        schemaVersion: FLOW_STATE_SCHEMA_VERSION,
+        ...(executionRouting ? { executionRouting, routingFingerprint: executionRouting.fingerprint } : {}),
+        ...(executionRouting ? { selectedRoutingPreset: { kind: "custom", label: "Legacy launch profile" } } : {}),
+    };
+}
+function normalizeFlowRunState(raw, flowId, filePath) {
+    if (!raw || typeof raw !== "object") {
+        throw new TaskRunnerError(`Invalid flow state file format: ${filePath}`);
+    }
+    const schemaVersion = raw.schemaVersion;
+    let state;
+    if (schemaVersion === 1) {
+        state = upgradeFlowRunStateV1(raw);
+    }
+    else if (schemaVersion === FLOW_STATE_SCHEMA_VERSION) {
+        state = raw;
+    }
+    else {
+        throw new TaskRunnerError(`Unsupported flow state schema in ${filePath}: ${String(schemaVersion ?? "unknown")}`);
+    }
+    if (state.flowId !== flowId) {
+        throw new TaskRunnerError(`Flow state ${filePath} belongs to flow '${state.flowId}', expected '${flowId}'`);
+    }
+    if (state.executionRouting) {
+        const executionRouting = resolveStoredExecutionRoutingSnapshot(state.executionRouting);
+        state.executionRouting = executionRouting;
+        state.routingFingerprint = executionRouting.fingerprint;
+        state.launchProfile = executionRouting.defaultRoute;
+    }
+    else if (state.launchProfile) {
+        const executionRouting = singleLaunchProfileExecutionRouting(state.launchProfile);
+        state.executionRouting = executionRouting;
+        state.routingFingerprint = executionRouting.fingerprint;
+    }
+    return state;
+}
 export function loadFlowRunState(scopeKey, flowId) {
     const filePath = flowStateFile(scopeKey, flowId);
     if (!existsSync(filePath)) {
@@ -54,20 +111,20 @@ export function loadFlowRunState(scopeKey, flowId) {
     catch (error) {
         throw new TaskRunnerError(`Failed to parse flow state file ${filePath}: ${error.message}`);
     }
-    if (!raw || typeof raw !== "object") {
-        throw new TaskRunnerError(`Invalid flow state file format: ${filePath}`);
-    }
-    const state = raw;
-    if (state.schemaVersion !== FLOW_STATE_SCHEMA_VERSION) {
-        throw new TaskRunnerError(`Unsupported flow state schema in ${filePath}: ${state.schemaVersion}`);
-    }
-    if (state.flowId !== flowId) {
-        throw new TaskRunnerError(`Flow state ${filePath} belongs to flow '${state.flowId}', expected '${flowId}'`);
-    }
-    return state;
+    return normalizeFlowRunState(raw, flowId, filePath);
 }
 export function saveFlowRunState(state) {
     state.updatedAt = nowIso8601();
+    state.schemaVersion = FLOW_STATE_SCHEMA_VERSION;
+    if (state.executionRouting) {
+        state.executionRouting = resolveStoredExecutionRoutingSnapshot(state.executionRouting);
+        state.routingFingerprint = state.executionRouting.fingerprint;
+        state.launchProfile = state.executionRouting.defaultRoute;
+    }
+    else if (state.launchProfile) {
+        state.executionRouting = singleLaunchProfileExecutionRouting(state.launchProfile);
+        state.routingFingerprint = state.executionRouting.fingerprint;
+    }
     ensureScopeWorkspaceDir(state.scopeKey);
     writeFileSync(flowStateFile(state.scopeKey, state.flowId), `${JSON.stringify({
         ...state,
@@ -101,10 +158,12 @@ function normalizeStepState(step) {
     if (step.status !== "running") {
         return step;
     }
-    const { finishedAt: _finishedAt, outputs: _outputs, value: _value, stopFlow: _stopFlow, ...rest } = step;
+    const resumeValue = isFlowRunResumeEnvelope(step.value) ? step.value : undefined;
+    const { finishedAt: _finishedAt, outputs: _outputs, value: _value, publishedArtifacts: _publishedArtifacts, stopFlow: _stopFlow, ...rest } = step;
     return {
         ...rest,
         status: "pending",
+        ...(resumeValue ? { value: resumeValue } : {}),
     };
 }
 function normalizePhaseState(phase) {
@@ -171,6 +230,7 @@ export function prepareFlowStateForResume(state) {
     state.currentStep = null;
     state.executionState = {
         ...state.executionState,
+        publicationRunId: randomUUID(),
         terminated: false,
         phases: state.executionState.phases.map(normalizePhaseState),
     };