pi-subagents 0.73.1 → 0.74.0

package/src/workflows/workflow-revival.js

@@ -0,0 +1,111 @@
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { writePrivateAtomicJson } from "../shared/atomic-json.js";
+import { readStatus } from "../shared/utils.js";
+/**
+ * A top-level resume of a failed async workflow child starts a detached run. These records
+ * link that run back to its workflow key without writing the workflow's own status or receipt.
+ * The source run is terminal, so the live workflow host never writes its directory.
+ */
+const REVIVAL_LINK_FILE = "workflow-revival.json";
+/** The same link, kept in the revived run's directory so reviving it again keeps the key. */
+export const REVIVAL_ORIGIN_FILE = "workflow-revival-origin.json";
+/** Recording refuses a deeper link, so readers bounded by this always reach the chain's end. */
+const MAX_REVIVAL_DEPTH = 16;
+const KEY_PATTERN = /^[A-Za-z0-9][A-Za-z0-9._-]{0,127}$/;
+function isRunId(value) {
+    return typeof value === "string" && value.length > 0 && value.length <= 255 && path.basename(value) === value && value !== "." && value !== "..";
+}
+function readLink(file) {
+    let value;
+    try {
+        value = JSON.parse(fs.readFileSync(file, "utf-8"));
+    }
+    catch {
+        return undefined;
+    }
+    if (!value || typeof value !== "object" || Array.isArray(value))
+        return undefined;
+    const link = value;
+    if (link.version !== 1 || !isRunId(link.workflowRunId) || typeof link.workflowKey !== "string" || !KEY_PATTERN.test(link.workflowKey) || !isRunId(link.sourceRunId) || !isRunId(link.revivedRunId))
+        return undefined;
+    if (!Number.isInteger(link.depth) || link.depth < 1 || link.depth > MAX_REVIVAL_DEPTH)
+        return undefined;
+    return { version: 1, workflowRunId: link.workflowRunId, workflowKey: link.workflowKey, sourceRunId: link.sourceRunId, revivedRunId: link.revivedRunId, depth: link.depth };
+}
+/**
+ * Links a detached revival to the workflow key of its source run when the source failed and is
+ * an async workflow child or an earlier revival of one. Throws when the source status cannot be
+ * read or the key's revival chain is full; the caller reports that on the revive receipt.
+ */
+export function recordWorkflowRevival(asyncDirRoot, sourceRunId, revivedRunId) {
+    const sourceDir = path.join(asyncDirRoot, sourceRunId);
+    const status = readStatus(sourceDir);
+    if (status?.runId !== sourceRunId || status.state !== "failed")
+        return;
+    const origin = isRunId(status.parentWorkflowRunId) && typeof status.workflowKey === "string" && KEY_PATTERN.test(status.workflowKey)
+        ? { workflowRunId: status.parentWorkflowRunId, workflowKey: status.workflowKey, depth: 0 }
+        : readLink(path.join(sourceDir, REVIVAL_ORIGIN_FILE));
+    if (!origin || ("revivedRunId" in origin && origin.revivedRunId !== sourceRunId))
+        return;
+    if (origin.depth >= MAX_REVIVAL_DEPTH)
+        throw new Error(`workflow key '${origin.workflowKey}' of ${origin.workflowRunId} already has ${MAX_REVIVAL_DEPTH} chained revivals.`);
+    const link = { version: 1, workflowRunId: origin.workflowRunId, workflowKey: origin.workflowKey, sourceRunId, revivedRunId, depth: origin.depth + 1 };
+    const originFile = path.join(asyncDirRoot, revivedRunId, REVIVAL_ORIGIN_FILE);
+    writePrivateAtomicJson(originFile, link);
+    try {
+        writePrivateAtomicJson(path.join(sourceDir, REVIVAL_LINK_FILE), link);
+    }
+    catch (error) {
+        // Keep the origin exactly when the published link names this revival: an unlinked origin
+        // would only pin the run against retention, and a linked one protects the key's chain.
+        if (readLink(path.join(sourceDir, REVIVAL_LINK_FILE))?.revivedRunId !== revivedRunId)
+            fs.rmSync(originFile, { force: true });
+        throw error;
+    }
+}
+/**
+ * Follows revival links from one run of a workflow key by exact reads. Returns undefined when
+ * the run was never revived; malformed or mismatched links end the chain.
+ */
+export function projectWorkflowKeyRevival(asyncDirRoot, workflowRunId, workflowKey, runId) {
+    if (!isRunId(runId))
+        return undefined;
+    const revivedRunIds = [];
+    let current = runId;
+    while (revivedRunIds.length < MAX_REVIVAL_DEPTH) {
+        const link = readLink(path.join(asyncDirRoot, current, REVIVAL_LINK_FILE));
+        if (!link || link.sourceRunId !== current || link.workflowRunId !== workflowRunId || link.workflowKey !== workflowKey)
+            break;
+        revivedRunIds.push(link.revivedRunId);
+        current = link.revivedRunId;
+    }
+    if (revivedRunIds.length === 0)
+        return undefined;
+    let status = null;
+    try {
+        status = readStatus(path.join(asyncDirRoot, current));
+    }
+    catch {
+        // Readers show "status unavailable" instead of failing the workflow's status or notice.
+    }
+    return { revivedRunIds, latestRunId: current, ...(status?.runId === current ? { state: status.state } : {}) };
+}
+/** `Revived → <run> [→ <run>]: <state>`, shown next to the key's original failure. */
+export function formatWorkflowKeyRevival(revival) {
+    return `Revived → ${revival.revivedRunIds.join(" → ")}: ${revival.state === "complete" ? "completed" : revival.state ?? "status unavailable"}`;
+}
+/**
+ * Adds each failed child's revivals to its lineage, so the receipt's latest run for that key is
+ * the newest revival. The child's own result and evidence stay as they were.
+ */
+export function withWorkflowRevivals(asyncDirRoot, workflowRunId, children) {
+    return children.map((child) => {
+        if (child.ok || child.state === "running" || !child.runId)
+            return child;
+        const lineage = child.continuation?.runIds ?? [child.runId];
+        const revival = projectWorkflowKeyRevival(asyncDirRoot, workflowRunId, child.key, lineage.at(-1) ?? child.runId);
+        return revival ? { ...child, continuation: { runIds: [...lineage, ...revival.revivedRunIds] }, revival } : child;
+    });
+}
+//# sourceMappingURL=workflow-revival.js.map

package/src/workflows/workflow-revival.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workflow-revival.js","sourceRoot":"","sources":["../../../src/workflows/workflow-revival.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAElE,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAGhD;;;;GAIG;AACH,MAAM,iBAAiB,GAAG,uBAAuB,CAAC;AAClD,6FAA6F;AAC7F,MAAM,CAAC,MAAM,mBAAmB,GAAG,8BAA8B,CAAC;AAClE,gGAAgG;AAChG,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAC7B,MAAM,WAAW,GAAG,oCAAoC,CAAC;AAoBzD,SAAS,OAAO,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,KAAK,IAAI,KAAK,KAAK,GAAG,IAAI,KAAK,KAAK,IAAI,CAAC;AAClJ,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC7B,IAAI,KAAc,CAAC;IACnB,IAAI,CAAC;QACJ,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC;IACpD,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,SAAS,CAAC;IAClB,CAAC;IACD,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAClF,MAAM,IAAI,GAAG,KAAgC,CAAC;IAC9C,IAAI,IAAI,CAAC,OAAO,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,OAAO,IAAI,CAAC,WAAW,KAAK,QAAQ,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC;QAAE,OAAO,SAAS,CAAC;IACrN,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAK,IAAI,CAAC,KAAgB,GAAG,CAAC,IAAK,IAAI,CAAC,KAAgB,GAAG,iBAAiB;QAAE,OAAO,SAAS,CAAC;IAChI,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,KAAe,EAAE,CAAC;AACtL,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,qBAAqB,CAAC,YAAoB,EAAE,WAAmB,EAAE,YAAoB;IACpG,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;IACvD,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;IACrC,IAAI,MAAM,EAAE,KAAK,KAAK,WAAW,IAAI,MAAM,CAAC,KAAK,KAAK,QAAQ;QAAE,OAAO;IACvE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,mBAAmB,CAAC,IAAI,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ,IAAI,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;QACnI,CAAC,CAAC,EAAE,aAAa,EAAE,MAAM,CAAC,mBAAmB,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC,EAAE;QAC1F,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC,CAAC;IACvD,IAAI,CAAC,MAAM,IAAI,CAAC,cAAc,IAAI,MAAM,IAAI,MAAM,CAAC,YAAY,KAAK,WAAW,CAAC;QAAE,OAAO;IACzF,IAAI,MAAM,CAAC,KAAK,IAAI,iBAAiB;QAAE,MAAM,IAAI,KAAK,CAAC,iBAAiB,MAAM,CAAC,WAAW,QAAQ,MAAM,CAAC,aAAa,gBAAgB,iBAAiB,oBAAoB,CAAC,CAAC;IAC7K,MAAM,IAAI,GAAwB,EAAE,OAAO,EAAE,CAAC,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,WAAW,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC;IAC3K,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,YAAY,EAAE,mBAAmB,CAAC,CAAC;IAC9E,sBAAsB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;IACzC,IAAI,CAAC;QACJ,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,iBAAiB,CAAC,EAAE,IAAI,CAAC,CAAC;IACvE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,yFAAyF;QACzF,uFAAuF;QACvF,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC,EAAE,YAAY,KAAK,YAAY;YAAE,EAAE,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7H,MAAM,KAAK,CAAC;IACb,CAAC;AACF,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,yBAAyB,CAAC,YAAoB,EAAE,aAAqB,EAAE,WAAmB,EAAE,KAAa;IACxH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACtC,MAAM,aAAa,GAAa,EAAE,CAAC;IACnC,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,OAAO,aAAa,CAAC,MAAM,GAAG,iBAAiB,EAAE,CAAC;QACjD,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,iBAAiB,CAAC,CAAC,CAAC;QAC3E,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,WAAW,KAAK,OAAO,IAAI,IAAI,CAAC,aAAa,KAAK,aAAa,IAAI,IAAI,CAAC,WAAW,KAAK,WAAW;YAAE,MAAM;QAC7H,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACtC,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC;IAC7B,CAAC;IACD,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACjD,IAAI,MAAM,GAAuB,IAAI,CAAC;IACtC,IAAI,CAAC;QACJ,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC;IACvD,CAAC;IAAC,MAAM,CAAC;QACR,wFAAwF;IACzF,CAAC;IACD,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,KAAK,KAAK,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;AAC/G,CAAC;AAED,sFAAsF;AACtF,MAAM,UAAU,wBAAwB,CAAC,OAA2B;IACnE,OAAO,aAAa,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,OAAO,CAAC,KAAK,KAAK,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,IAAI,oBAAoB,EAAE,CAAC;AAChJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAAsC,YAAoB,EAAE,aAAqB,EAAE,QAAa;IACnI,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QAC7B,IAAI,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,KAAK;YAAE,OAAO,KAAK,CAAC;QACxE,MAAM,OAAO,GAAG,KAAK,CAAC,YAAY,EAAE,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAG,yBAAyB,CAAC,YAAY,EAAE,aAAa,EAAE,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC;QACjH,OAAO,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,KAAK,EAAE,YAAY,EAAE,EAAE,MAAM,EAAE,CAAC,GAAG,OAAO,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;IAClH,CAAC,CAAC,CAAC;AACJ,CAAC"}