@workglow/task-graph 0.2.30 → 0.2.32

package/dist/node.js

@@ -153,7 +153,10 @@ var TransformRegistry = {
   }
 };
 var TRANSFORM_DEFS = createServiceToken("transform.defs");
-globalServiceRegistry.registerIfAbsent(TRANSFORM_DEFS, () => TransformRegistry.all, true);
+function registerTransformDefaults(registry = globalServiceRegistry) {
+  registry.registerIfAbsent(TRANSFORM_DEFS, () => TransformRegistry.all, true);
+}
+registerTransformDefaults();
 
 // src/task-graph/Dataflow.ts
 var DATAFLOW_ALL_PORTS = "*";
@@ -1281,7 +1284,7 @@ import {
   getTelemetryProvider as getTelemetryProvider2,
   globalServiceRegistry as globalServiceRegistry3,
   ResourceScope as ResourceScope2,
-  ServiceRegistry as ServiceRegistry2,
+  ServiceRegistry as ServiceRegistry3,
   SpanStatusCode as SpanStatusCode2,
   uuid4 as uuid44
 } from "@workglow/util";
@@ -3936,7 +3939,7 @@ class TaskGraphRunner {
     if (config?.registry !== undefined) {
       this.registry = config.registry;
     } else if (this.registry === undefined) {
-      this.registry = new ServiceRegistry2(globalServiceRegistry3.container.createChildContainer());
+      this.registry = new ServiceRegistry3(globalServiceRegistry3.container.createChildContainer());
     }
     if (config?.resourceScope !== undefined) {
       this.resourceScope = config.resourceScope;
@@ -6436,6 +6439,111 @@ function registerBuiltInTransforms() {
   for (const t of all)
     TransformRegistry.registerTransform(t);
 }
+// src/task/EntitlementProfile.ts
+import { createServiceToken as createServiceToken5 } from "@workglow/util";
+var STATIC_SIGNAL_SOURCE = Object.freeze({
+  subscribe(_listener) {
+    return () => {};
+  }
+});
+function createPolicyProfile(name, policy, options = {}) {
+  const enforcer = createPolicyEnforcer(policy, options.resolver);
+  const signalSource = options.signalSource ?? STATIC_SIGNAL_SOURCE;
+  const listeners = new Set;
+  const lastOutcome = new Map;
+  const lastEntitlement = new Map;
+  let disposed = false;
+  function key(e) {
+    const resources = e.resources === undefined ? null : [...e.resources].sort();
+    return `${e.id}|${JSON.stringify(resources)}`;
+  }
+  async function evaluate(e) {
+    if (e.optional)
+      return "granted";
+    const denials = await enforcer.checkAll({ entitlements: [e] });
+    return denials.length === 0 ? "granted" : "denied";
+  }
+  async function emitFlipFor(e) {
+    const k = key(e);
+    const previous = lastOutcome.get(k);
+    if (previous === undefined)
+      return;
+    const current = await evaluate(e);
+    if (disposed)
+      return;
+    if (current === previous)
+      return;
+    lastOutcome.set(k, current);
+    const event = {
+      kind: current === "granted" ? "granted" : "revoked",
+      entitlement: e
+    };
+    for (const l of listeners) {
+      try {
+        l(event);
+      } catch (err) {
+        console.error(`[EntitlementProfile:${name}] listener threw:`, err);
+      }
+    }
+  }
+  function safeEmitFlipFor(e) {
+    emitFlipFor(e).catch((err) => {
+      console.error(`[EntitlementProfile:${name}] error evaluating signal:`, err);
+    });
+  }
+  const sourceUnsub = signalSource.subscribe((signal) => {
+    if (disposed)
+      return;
+    if (signal.kind === "reload") {
+      for (const [, e] of lastEntitlement) {
+        safeEmitFlipFor(e);
+      }
+    } else {
+      safeEmitFlipFor(signal.entitlement);
+    }
+  });
+  const profile = {
+    name,
+    checkAll: enforcer.checkAll.bind(enforcer),
+    checkTask: enforcer.checkTask.bind(enforcer),
+    surface: () => policy.grant,
+    async requestEntitlement(required) {
+      if (required.optional) {
+        return { outcome: "granted", entitlement: required };
+      }
+      const denials = await enforcer.checkAll({ entitlements: [required] });
+      const k = key(required);
+      lastEntitlement.set(k, required);
+      if (denials.length === 0) {
+        lastOutcome.set(k, "granted");
+        return { outcome: "granted", entitlement: required };
+      }
+      lastOutcome.set(k, "denied");
+      return { outcome: "denied", denial: denials[0] };
+    },
+    subscribe(listener) {
+      listeners.add(listener);
+      let unsubbed = false;
+      return () => {
+        if (unsubbed)
+          return;
+        unsubbed = true;
+        listeners.delete(listener);
+      };
+    },
+    async dispose() {
+      if (disposed)
+        return;
+      disposed = true;
+      sourceUnsub();
+      listeners.clear();
+      lastOutcome.clear();
+      lastEntitlement.clear();
+    }
+  };
+  return profile;
+}
+var ENTITLEMENT_PROFILE = createServiceToken5("workglow.entitlementProfile");
 // src/task/EntitlementProfiles.ts
 var BROWSER_GRANTS = [
   { id: Entitlements.NETWORK_HTTP },
@@ -6470,8 +6578,8 @@ var PROFILE_GRANTS = {
 function createProfilePolicy(profile) {
   return { deny: [], grant: PROFILE_GRANTS[profile], ask: [] };
 }
-function createProfileEnforcer(profile, resolver) {
-  return createPolicyEnforcer(createProfilePolicy(profile), resolver);
+function createProfileEnforcer(profile, options) {
+  return createPolicyProfile(profile, createProfilePolicy(profile), options);
 }
 function getProfileGrants(profile) {
   return PROFILE_GRANTS[profile];
@@ -8057,15 +8165,15 @@ import {
   JobQueueServer
 } from "@workglow/job-queue";
 import { InMemoryQueueStorage } from "@workglow/job-queue";
-import { createServiceToken as createServiceToken5, globalServiceRegistry as globalServiceRegistry4 } from "@workglow/util";
-var JOB_QUEUE_FACTORY = createServiceToken5("taskgraph.jobQueueFactory");
+import { createServiceToken as createServiceToken6, globalServiceRegistry as globalServiceRegistry4 } from "@workglow/util";
+var JOB_QUEUE_FACTORY = createServiceToken6("taskgraph.jobQueueFactory");
 var defaultJobQueueFactory = async ({
   queueName,
   jobClass,
   options
 }) => {
   const storage = options?.storage ?? new InMemoryQueueStorage(queueName);
-  await storage.setupDatabase();
+  await storage.migrate();
   const server = new JobQueueServer(jobClass, {
     storage,
     queueName,
@@ -8099,7 +8207,7 @@ function createJobQueueFactoryWithOptions(defaultOptions = {}) {
       ...options ?? {}
     };
     const storage = mergedOptions.storage ?? new InMemoryQueueStorage(queueName);
-    await storage.setupDatabase();
+    await storage.migrate();
     const server = new JobQueueServer(jobClass, {
       storage,
       queueName,
@@ -8323,7 +8431,7 @@ queueMicrotask(() => {
 });
 // src/task/TaskRegistry.ts
 import {
-  createServiceToken as createServiceToken6,
+  createServiceToken as createServiceToken7,
   getLogger as getLogger9,
   globalServiceRegistry as globalServiceRegistry5,
   registerInputCompactor,
@@ -8363,8 +8471,7 @@ var TaskRegistry = {
   registerTask,
   unregisterTask
 };
-var TASK_CONSTRUCTORS = createServiceToken6("task.constructors");
-globalServiceRegistry5.registerIfAbsent(TASK_CONSTRUCTORS, () => TaskRegistry.all, true);
+var TASK_CONSTRUCTORS = createServiceToken7("task.constructors");
 function getGlobalTaskConstructors() {
   return globalServiceRegistry5.get(TASK_CONSTRUCTORS);
 }
@@ -8391,8 +8498,7 @@ function resolveTaskFromRegistry(id, _format, registry) {
     ...configSchema ? { configSchema } : {}
   };
 }
-registerInputResolver("tasks", resolveTaskFromRegistry);
-registerInputCompactor("tasks", (value, _format, registry) => {
+function compactTask(value, _format, registry) {
   if (typeof value === "object" && value !== null && "name" in value) {
     const name = value.name;
     if (typeof name !== "string")
@@ -8402,7 +8508,13 @@ registerInputCompactor("tasks", (value, _format, registry) => {
     return ctor ? name : undefined;
   }
   return;
-});
+}
+function registerTaskDefaults(registry = globalServiceRegistry5) {
+  registry.registerIfAbsent(TASK_CONSTRUCTORS, () => TaskRegistry.all, true);
+  registerInputResolver("tasks", resolveTaskFromRegistry, registry);
+  registerInputCompactor("tasks", compactTask, registry);
+}
+registerTaskDefaults();
 
 // src/task/TaskJSON.ts
 var createSingleTaskFromJSON = (item, registry, options) => {
@@ -8528,8 +8640,8 @@ var registerBaseTasks = () => {
   return tasks;
 };
 // src/storage/TaskGraphRepository.ts
-import { createServiceToken as createServiceToken7, EventEmitter as EventEmitter7 } from "@workglow/util";
-var TASK_GRAPH_REPOSITORY = createServiceToken7("taskgraph.taskGraphRepository");
+import { createServiceToken as createServiceToken8, EventEmitter as EventEmitter7 } from "@workglow/util";
+var TASK_GRAPH_REPOSITORY = createServiceToken8("taskgraph.taskGraphRepository");
 
 class TaskGraphRepository {
   type = "TaskGraphRepository";
@@ -8712,6 +8824,8 @@ export {
   resolveIterationBound,
   resetMethodNameCache,
   removeIterationProperties,
+  registerTransformDefaults,
+  registerTaskDefaults,
   registerPortCodec,
   registerJobQueueFactory,
   registerBuiltInTransforms,
@@ -8777,6 +8891,7 @@ export {
   createScopedEnforcer,
   createProfilePolicy,
   createProfileEnforcer,
+  createPolicyProfile,
   createPolicyEnforcer,
   createJobQueueFactoryWithOptions,
   createGraphFromGraphJSON,
@@ -8841,6 +8956,7 @@ export {
   TASK_CONSTRUCTORS,
   StreamPump,
   StreamProcessor,
+  STATIC_SIGNAL_SOURCE,
   SERVER_GRANTS,
   RunScheduler,
   RunContext,
@@ -8866,6 +8982,7 @@ export {
   Entitlements,
   EdgeMaterializer,
   ENTITLEMENT_RESOLVER,
+  ENTITLEMENT_PROFILE,
   ENTITLEMENT_ENFORCER,
   EMPTY_POLICY,
   EMPTY_ENTITLEMENTS,
@@ -8884,4 +9001,4 @@ export {
   BROWSER_GRANTS
 };
 
-//# debugId=DAC45C811156BEA364756E2164756E21
+//# debugId=BF168934B50892FF64756E2164756E21