@agimon-ai/po-workflow 0.0.0-alpha.16 → 0.0.0-alpha.17

package/dist/extensions/server.cjs

@@ -3508,9 +3508,15 @@ function createWorkflowServerRuntime(host, serverHost) {
 	let control;
 	let controlDisposers = [];
 	const disposeControl = () => {
-		for (const dispose of controlDisposers.splice(0)) dispose();
-		control?.dispose();
+		const owned = control;
 		control = void 0;
+		const failures = [];
+		for (const dispose of [...controlDisposers.splice(0), () => owned?.dispose()]) try {
+			dispose();
+		} catch (error) {
+			failures.push(error);
+		}
+		if (failures.length) throw new AggregateError(failures, "Workflow control cleanup failed.");
 	};
 	let modeOwner;
 	const modeSelected = () => (host.context.selection.state?.["minor-mode"] ?? []).includes(WORKFLOW_MODE_ID);
@@ -4027,18 +4033,25 @@ function createWorkflowServerRuntime(host, serverHost) {
 			pendingCompletions.clear();
 			narrationReady = false;
 			narration = void 0;
-			await ownedReadiness?.dispose();
-			disposeControl();
-			lifecycle?.dispose();
-			stopReleaseRequests?.();
-			stopPeerReady?.();
-			if (runsTimer !== void 0) clearTimeout(runsTimer);
-			runsTimer = void 0;
-			runsAfter = void 0;
-			await launcher.dispose();
-			runProvider?.dispose();
-			runProvider = void 0;
-			await telemetry.shutdown();
+			await (0, _agimon_ai_po_core_extensionFile.runCleanups)([
+				() => ownedReadiness?.dispose(),
+				disposeControl,
+				() => lifecycle?.dispose(),
+				() => stopReleaseRequests?.(),
+				() => stopPeerReady?.(),
+				() => {
+					if (runsTimer !== void 0) clearTimeout(runsTimer);
+					runsTimer = void 0;
+					runsAfter = void 0;
+				},
+				() => launcher.dispose(),
+				() => {
+					const provider = runProvider;
+					runProvider = void 0;
+					provider?.dispose();
+				},
+				() => telemetry.shutdown()
+			], "Workflow cleanup failed.");
 		}
 	};
 }