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

package/dist/extensions/server.cjs

@@ -7593,7 +7593,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 +7602,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) {
@@ -7809,11 +7821,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 +8058,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();