@workglow/task-graph 0.0.64 → 0.0.66

package/dist/bun.js

@@ -2243,6 +2243,30 @@ class TaskGraph {
       unsubscribes.forEach((unsub) => unsub());
     };
   }
+  subscribeToTaskProgress(callback) {
+    const unsubscribes = [];
+    const tasks = this.getTasks();
+    tasks.forEach((task) => {
+      const unsub = task.subscribe("progress", (progress, message, ...args) => {
+        callback(task.config.id, progress, message, ...args);
+      });
+      unsubscribes.push(unsub);
+    });
+    const handleTaskAdded = (taskId) => {
+      const task = this.getTask(taskId);
+      if (!task || typeof task.subscribe !== "function")
+        return;
+      const unsub = task.subscribe("progress", (progress, message, ...args) => {
+        callback(task.config.id, progress, message, ...args);
+      });
+      unsubscribes.push(unsub);
+    };
+    const graphUnsub = this.subscribe("task_added", handleTaskAdded);
+    unsubscribes.push(graphUnsub);
+    return () => {
+      unsubscribes.forEach((unsub) => unsub());
+    };
+  }
   subscribeToDataflowStatus(callback) {
     const unsubscribes = [];
     const dataflows = this.getDataflows();
@@ -2778,6 +2802,9 @@ class TaskGraphTabularRepository extends TaskGraphRepository {
     super();
     this.tabularRepository = tabularRepository;
   }
+  async setupDatabase() {
+    await this.tabularRepository.setupDatabase?.();
+  }
   async saveTaskGraph(key, output) {
     const value = JSON.stringify(output.toJSON());
     await this.tabularRepository.put({ key, value });
@@ -2823,6 +2850,9 @@ class TaskOutputTabularRepository extends TaskOutputRepository {
     this.tabularRepository = tabularRepository;
     this.outputCompression = outputCompression;
   }
+  async setupDatabase() {
+    await this.tabularRepository.setupDatabase?.();
+  }
   async keyFromInputs(inputs) {
     return await makeFingerprint(inputs);
   }
@@ -2939,4 +2969,4 @@ export {
   ArrayTask
 };
 
-//# debugId=54877F3DCD961A9A64756E2164756E21
+//# debugId=27CCA088711684C364756E2164756E21