@agimon-ai/po-workflow 0.0.0-alpha.83 → 0.0.0-alpha.85

package/dist/extensions/server.cjs

@@ -7186,6 +7186,7 @@ function createWorkflowServerRuntime(host, serverHost, runtime, resolveSecrets,
 	*/
 	const launchIn = async (owner, parameters) => {
 		if (serverHost.context.requestApi === void 0) return textResult(`Error: this host cannot reach session ${owner} to launch the workflow there.`, true);
+		watchOwner(owner);
 		const response = await serverHost.context.requestApi({
 			scope: "session",
 			sessionId: owner
@@ -7593,7 +7594,7 @@ function createWorkflowServerRuntime(host, serverHost, runtime, resolveSecrets,
 			deferred: report.deferred ?? []
 		};
 	};
-	const policyHandles = /* @__PURE__ */ new Set();
+	const policies = /* @__PURE__ */ new Map();
 	const admissions = createWorkflowAdmissions({
 		sessionId: host.context.sessionId,
 		journal: featureFor(legacyHome).registry,
@@ -7602,12 +7603,24 @@ function createWorkflowServerRuntime(host, serverHost, runtime, resolveSecrets,
 		registerPolicy(coordinator, callback, options) {
 			assertAvailable();
 			if (!publicServices) throw new Error("Workflow policy registration is unavailable.");
-			const handle = publicServices.registerPolicy(host.context.sessionId, coordinator, callback, options);
-			policyHandles.add(handle);
+			if (policies.has(coordinator)) throw new Error("Workflow policy coordinator is already registered for this session.");
+			const registration = {
+				callback,
+				options: {
+					...options,
+					environmentKeys: [...options.environmentKeys]
+				}
+			};
+			registration.binding = {
+				service: publicServices,
+				handle: publicServices.registerPolicy(host.context.sessionId, coordinator, callback, registration.options)
+			};
+			policies.set(coordinator, registration);
 			queueMicrotask(deliverSettlements);
 			return { dispose() {
-				handle.dispose();
-				policyHandles.delete(handle);
+				registration.binding?.handle.dispose();
+				delete registration.binding;
+				if (policies.get(coordinator) === registration) policies.delete(coordinator);
 			} };
 		},
 		async execute(request, admission) {
@@ -7617,6 +7630,7 @@ function createWorkflowServerRuntime(host, serverHost, runtime, resolveSecrets,
 				const ownerId = record.admission?.ownerSessionId ?? request.previousRun.ownerSessionId;
 				const owner = publicServices?.owner(ownerId);
 				if (!owner) throw new Error("Workflow recovery owner is unavailable; admission remains pending.");
+				watchOwner(ownerId);
 				await owner.recover(request.previousRun, request.options?.runner, admission);
 			} else {
 				const input = await admissionInput(request);
@@ -7624,6 +7638,7 @@ function createWorkflowServerRuntime(host, serverHost, runtime, resolveSecrets,
 				if (ownerId !== void 0) {
 					const owner = publicServices?.owner(ownerId);
 					if (!owner) throw new Error("Workflow launch owner is unavailable; admission remains pending.");
+					watchOwner(ownerId);
 					await owner.launch(input, {
 						...admission,
 						ownerSessionId: ownerId
@@ -7809,11 +7824,27 @@ function createWorkflowServerRuntime(host, serverHost, runtime, resolveSecrets,
 								})).isError) throw new Error("Workflow recovery did not acknowledge admission.");
 							}
 						});
-						return () => {
+						const detach = () => {
+							for (const registration of policies.values()) {
+								if (registration.binding?.service !== service) continue;
+								registration.binding.handle.dispose();
+								delete registration.binding;
+							}
 							owner.dispose();
 							service.withdraw(host.context.sessionId);
 							if (publicServices === service) publicServices = void 0;
 						};
+						try {
+							for (const [coordinator, registration] of policies) registration.binding = {
+								service,
+								handle: service.registerPolicy(host.context.sessionId, coordinator, registration.callback, registration.options)
+							};
+							queueMicrotask(deliverSettlements);
+						} catch (error) {
+							detach();
+							throw error;
+						}
+						return detach;
 					});
 				});
 			}
@@ -8030,8 +8061,8 @@ function createWorkflowServerRuntime(host, serverHost, runtime, resolveSecrets,
 			disposed = true;
 			publicServices?.withdraw(host.context.sessionId);
 			startupAbort.abort();
-			for (const handle of policyHandles) handle.dispose();
-			policyHandles.clear();
+			for (const registration of policies.values()) registration.binding?.handle.dispose();
+			policies.clear();
 			if (deliveryRetry !== void 0) clearTimeout(deliveryRetry);
 			deliveryRetry = void 0;
 			pendingCompletions.clear();