@agimon-ai/doompi-workflow 0.0.1-alpha.95 → 0.0.1-alpha.97

package/dist/extensions/pi.cjs

@@ -3321,6 +3321,27 @@ const ERROR_COLOUR = "error";
 * user in a session that will not close.
 */
 const SHUTDOWN_FINALIZE_TIMEOUT_MS = 1e4;
+/**
+* Pi awaits extension disposal before it exits the process, so any unbounded
+* wait in dispose keeps an `--auto-stop` session alive after it asked to stop.
+*/
+const SHUTDOWN_STEP_TIMEOUT_MS = 5e3;
+/**
+* Rejects as soon as the signal aborts, so a registry read that never answers
+* cannot hold the readiness task, and every disposal waiting on it, open.
+*/
+function untilAborted(work, signal) {
+	if (signal.aborted) return Promise.reject(signal.reason);
+	return new Promise((resolveWork, rejectWork) => {
+		const onAbort = () => rejectWork(signal.reason);
+		signal.addEventListener("abort", onAbort, { once: true });
+		work.then(resolveWork, rejectWork).finally(() => signal.removeEventListener("abort", onAbort));
+	});
+}
+/** Resolves when the work does or the timeout passes; a rejection still propagates. */
+async function withinTimeout(work, timeoutMs) {
+	await Promise.race([work, new Promise((settle) => setTimeout(settle, timeoutMs).unref?.())]);
+}
 const FOLLOW_INTERVAL_MS = 2e3;
 const TAIL_LINES = 24;
 /** Below these a resized run has no room left to draw, so it keeps its own size and the panel clips. */
@@ -4939,10 +4960,10 @@ function createWorkflowPiRuntime(pi, options) {
 				const retired = () => ownReadinessAbort.signal.aborted || !ownsSession();
 				signal.throwIfAborted();
 				if (retired()) return { value: void 0 };
-				await startRunControl(ctx);
+				await untilAborted(startRunControl(ctx), signal);
 				signal.throwIfAborted();
 				if (retired()) return { value: void 0 };
-				await refreshStatus(ctx, false);
+				await untilAborted(refreshStatus(ctx, false), signal);
 				signal.throwIfAborted();
 				if (retired()) return { value: void 0 };
 				lifecycleNarrationReady = true;
@@ -4976,13 +4997,13 @@ function createWorkflowPiRuntime(pi, options) {
 			const ownedReadiness = standaloneReadiness;
 			standaloneReadiness = void 0;
 			if (ownedReadiness) await runCleanup("dispose standalone workflow readiness", () => ownedReadiness.dispose());
-			else if (pendingReadiness) await Promise.allSettled([pendingReadiness.then((handle) => handle.wait())]);
+			else if (pendingReadiness) await withinTimeout(Promise.allSettled([pendingReadiness.then((handle) => handle.wait())]), SHUTDOWN_STEP_TIMEOUT_MS);
 			if (pendingInlineRuns.size > 0) {
 				for (const service of [runService, recoverService]) await runCleanup("interrupt an inline workflow", () => service.interrupt("SIGTERM", {
 					phase: "workflow",
 					source: "pi-session-shutdown"
 				}));
-				await Promise.race([Promise.allSettled(pendingInlineRuns), new Promise((settle) => setTimeout(settle, SHUTDOWN_FINALIZE_TIMEOUT_MS).unref?.())]);
+				await withinTimeout(Promise.allSettled(pendingInlineRuns), SHUTDOWN_FINALIZE_TIMEOUT_MS);
 			}
 			if (monitorTimer) clearInterval(monitorTimer);
 			monitorTimer = void 0;
@@ -5017,13 +5038,13 @@ function createWorkflowPiRuntime(pi, options) {
 			const pendingCount = pendingInlineRuns.size;
 			const activeTelemetry = telemetry;
 			telemetry = void 0;
-			if (activeTelemetry) {
+			if (activeTelemetry) await runCleanup("stop workflow telemetry", () => withinTimeout((async () => {
 				await runCleanup("record the workflow session finish", () => activeTelemetry.recordEvent("doom_workflow.session_finished", {
 					"workflow.pending_count": pendingCount,
 					outcome: pendingCount === 0 ? "completed" : "interrupted"
 				}));
-				await runCleanup("stop workflow telemetry", () => activeTelemetry.shutdown());
-			}
+				await activeTelemetry.shutdown();
+			})(), SHUTDOWN_STEP_TIMEOUT_MS));
 			pendingInlineRuns.clear();
 			pendingTerminalRuns.clear();
 			deliveredTerminalRuns.clear();