@workglow/task-graph 0.2.31 → 0.2.32

package/dist/bun.js

@@ -6440,6 +6440,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 },
@@ -6474,8 +6579,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];
@@ -8061,15 +8166,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,
@@ -8103,7 +8208,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,
@@ -8327,7 +8432,7 @@ queueMicrotask(() => {
 });
 // src/task/TaskRegistry.ts
 import {
-  createServiceToken as createServiceToken6,
+  createServiceToken as createServiceToken7,
   getLogger as getLogger9,
   globalServiceRegistry as globalServiceRegistry5,
   registerInputCompactor,
@@ -8367,7 +8472,7 @@ var TaskRegistry = {
   registerTask,
   unregisterTask
 };
-var TASK_CONSTRUCTORS = createServiceToken6("task.constructors");
+var TASK_CONSTRUCTORS = createServiceToken7("task.constructors");
 function getGlobalTaskConstructors() {
   return globalServiceRegistry5.get(TASK_CONSTRUCTORS);
 }
@@ -8536,8 +8641,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";
@@ -8787,6 +8892,7 @@ export {
   createScopedEnforcer,
   createProfilePolicy,
   createProfileEnforcer,
+  createPolicyProfile,
   createPolicyEnforcer,
   createJobQueueFactoryWithOptions,
   createGraphFromGraphJSON,
@@ -8851,6 +8957,7 @@ export {
   TASK_CONSTRUCTORS,
   StreamPump,
   StreamProcessor,
+  STATIC_SIGNAL_SOURCE,
   SERVER_GRANTS,
   RunScheduler,
   RunContext,
@@ -8876,6 +8983,7 @@ export {
   Entitlements,
   EdgeMaterializer,
   ENTITLEMENT_RESOLVER,
+  ENTITLEMENT_PROFILE,
   ENTITLEMENT_ENFORCER,
   EMPTY_POLICY,
   EMPTY_ENTITLEMENTS,
@@ -8894,4 +9002,4 @@ export {
   BROWSER_GRANTS
 };
 
-//# debugId=70AF17E20B1A815F64756E2164756E21
+//# debugId=896F2287B3DD5F3464756E2164756E21