@trigger.dev/core 3.0.0-beta.3 → 3.0.0-beta.31

package/dist/v3/dev/index.d.mts

@@ -0,0 +1,28 @@
+import { R as RuntimeManager, T as TaskRunExecutionResult, B as BatchTaskRunExecutionResult, b as TaskRunContext, p as TaskRunExecution } from '../../manager-WNMVbgHf.mjs';
+import 'zod';
+
+declare class DevRuntimeManager implements RuntimeManager {
+    _taskWaits: Map<string, {
+        resolve: (value: TaskRunExecutionResult) => void;
+    }>;
+    _batchWaits: Map<string, {
+        resolve: (value: BatchTaskRunExecutionResult) => void;
+        reject: (err?: any) => void;
+    }>;
+    _pendingCompletionNotifications: Map<string, TaskRunExecutionResult>;
+    disable(): void;
+    waitForDuration(ms: number): Promise<void>;
+    waitUntil(date: Date): Promise<void>;
+    waitForTask(params: {
+        id: string;
+        ctx: TaskRunContext;
+    }): Promise<TaskRunExecutionResult>;
+    waitForBatch(params: {
+        id: string;
+        runs: string[];
+        ctx: TaskRunContext;
+    }): Promise<BatchTaskRunExecutionResult>;
+    resumeTask(completion: TaskRunExecutionResult, execution: TaskRunExecution): void;
+}
+
+export { DevRuntimeManager };

package/dist/v3/dev/index.d.ts

@@ -0,0 +1,28 @@
+import { R as RuntimeManager, T as TaskRunExecutionResult, B as BatchTaskRunExecutionResult, b as TaskRunContext, p as TaskRunExecution } from '../../manager-WNMVbgHf.js';
+import 'zod';
+
+declare class DevRuntimeManager implements RuntimeManager {
+    _taskWaits: Map<string, {
+        resolve: (value: TaskRunExecutionResult) => void;
+    }>;
+    _batchWaits: Map<string, {
+        resolve: (value: BatchTaskRunExecutionResult) => void;
+        reject: (err?: any) => void;
+    }>;
+    _pendingCompletionNotifications: Map<string, TaskRunExecutionResult>;
+    disable(): void;
+    waitForDuration(ms: number): Promise<void>;
+    waitUntil(date: Date): Promise<void>;
+    waitForTask(params: {
+        id: string;
+        ctx: TaskRunContext;
+    }): Promise<TaskRunExecutionResult>;
+    waitForBatch(params: {
+        id: string;
+        runs: string[];
+        ctx: TaskRunContext;
+    }): Promise<BatchTaskRunExecutionResult>;
+    resumeTask(completion: TaskRunExecutionResult, execution: TaskRunExecution): void;
+}
+
+export { DevRuntimeManager };

package/dist/v3/dev/index.js

@@ -0,0 +1,93 @@
+'use strict';
+
+var promises = require('timers/promises');
+
+var __defProp = Object.defineProperty;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __publicField = (obj, key, value) => {
+  __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
+  return value;
+};
+async function unboundedTimeout(delay = 0, value, options) {
+  const maxDelay = 2147483647;
+  const fullTimeouts = Math.floor(delay / maxDelay);
+  const remainingDelay = delay % maxDelay;
+  let lastTimeoutResult = await promises.setTimeout(remainingDelay, value, options);
+  for (let i = 0; i < fullTimeouts; i++) {
+    lastTimeoutResult = await promises.setTimeout(maxDelay, value, options);
+  }
+  return lastTimeoutResult;
+}
+__name(unboundedTimeout, "unboundedTimeout");
+
+// src/v3/runtime/devRuntimeManager.ts
+var _DevRuntimeManager = class _DevRuntimeManager {
+  constructor() {
+    __publicField(this, "_taskWaits", /* @__PURE__ */ new Map());
+    __publicField(this, "_batchWaits", /* @__PURE__ */ new Map());
+    __publicField(this, "_pendingCompletionNotifications", /* @__PURE__ */ new Map());
+  }
+  disable() {
+  }
+  async waitForDuration(ms) {
+    await unboundedTimeout(ms);
+  }
+  async waitUntil(date) {
+    return this.waitForDuration(date.getTime() - Date.now());
+  }
+  async waitForTask(params) {
+    const pendingCompletion = this._pendingCompletionNotifications.get(params.id);
+    if (pendingCompletion) {
+      this._pendingCompletionNotifications.delete(params.id);
+      return pendingCompletion;
+    }
+    const promise = new Promise((resolve) => {
+      this._taskWaits.set(params.id, {
+        resolve
+      });
+    });
+    return await promise;
+  }
+  async waitForBatch(params) {
+    if (!params.runs.length) {
+      return Promise.resolve({
+        id: params.id,
+        items: []
+      });
+    }
+    const promise = Promise.all(params.runs.map((runId) => {
+      return new Promise((resolve, reject) => {
+        const pendingCompletion = this._pendingCompletionNotifications.get(runId);
+        if (pendingCompletion) {
+          this._pendingCompletionNotifications.delete(runId);
+          resolve(pendingCompletion);
+          return;
+        }
+        this._taskWaits.set(runId, {
+          resolve
+        });
+      });
+    }));
+    const results = await promise;
+    return {
+      id: params.id,
+      items: results
+    };
+  }
+  resumeTask(completion, execution) {
+    const wait = this._taskWaits.get(execution.run.id);
+    if (!wait) {
+      this._pendingCompletionNotifications.set(execution.run.id, completion);
+      return;
+    }
+    wait.resolve(completion);
+    this._taskWaits.delete(execution.run.id);
+  }
+};
+__name(_DevRuntimeManager, "DevRuntimeManager");
+var DevRuntimeManager = _DevRuntimeManager;
+
+exports.DevRuntimeManager = DevRuntimeManager;
+//# sourceMappingURL=out.js.map
+//# sourceMappingURL=index.js.map

package/dist/v3/dev/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/v3/utils/timers.ts","../../../src/v3/runtime/devRuntimeManager.ts"],"names":["setInterval","setTimeout","unboundedTimeout","delay","value","options","maxDelay","fullTimeouts","Math","floor","remainingDelay","lastTimeoutResult","i","DevRuntimeManager","_taskWaits","Map","_batchWaits","_pendingCompletionNotifications","disable","waitForDuration","ms","waitUntil","date","getTime","Date","now","waitForTask","params","pendingCompletion","get","id","delete","promise","Promise","resolve","set","waitForBatch","runs","length","items","all","map","runId","reject","results","resumeTask","completion","execution","wait","run"],"mappings":";;;;;;;;;AACA,SAASA,aAAaC,kBAAkB;AAExC,eAAsBC,iBACpBC,QAAgB,GAChBC,OACAC,SACY;AACZ,QAAMC,WAAW;AAEjB,QAAMC,eAAeC,KAAKC,MAAMN,QAAQG,QAAAA;AACxC,QAAMI,iBAAiBP,QAAQG;AAE/B,MAAIK,oBAAoB,MAAMV,WAAWS,gBAAgBN,OAAOC,OAAAA;AAEhE,WAASO,IAAI,GAAGA,IAAIL,cAAcK,KAAK;AACrCD,wBAAoB,MAAMV,WAAWK,UAAUF,OAAOC,OAAAA;EACxD;AAEA,SAAOM;AACT;AAjBsBT;;;ACMf,IAAMW,qBAAN,MAAMA,mBAAAA;EAAN;AACLC,sCAAgF,oBAAIC,IAAAA;AAEpFC,uCAGI,oBAAID,IAAAA;AAERE,2DAAuE,oBAAIF,IAAAA;;EAE3EG,UAAgB;EAEhB;EAEA,MAAMC,gBAAgBC,IAA2B;AAC/C,UAAMlB,iBAAiBkB,EAAAA;EACzB;EAEA,MAAMC,UAAUC,MAA2B;AACzC,WAAO,KAAKH,gBAAgBG,KAAKC,QAAO,IAAKC,KAAKC,IAAG,CAAA;EACvD;EAEA,MAAMC,YAAYC,QAA8E;AAC9F,UAAMC,oBAAoB,KAAKX,gCAAgCY,IAAIF,OAAOG,EAAE;AAE5E,QAAIF,mBAAmB;AACrB,WAAKX,gCAAgCc,OAAOJ,OAAOG,EAAE;AAErD,aAAOF;IACT;AAEA,UAAMI,UAAU,IAAIC,QAAgC,CAACC,YAAY;AAC/D,WAAKpB,WAAWqB,IAAIR,OAAOG,IAAI;QAAEI;MAAQ,CAAA;IAC3C,CAAA;AAEA,WAAO,MAAMF;EACf;EAEA,MAAMI,aAAaT,QAIsB;AACvC,QAAI,CAACA,OAAOU,KAAKC,QAAQ;AACvB,aAAOL,QAAQC,QAAQ;QAAEJ,IAAIH,OAAOG;QAAIS,OAAO,CAAA;MAAG,CAAA;IACpD;AAEA,UAAMP,UAAUC,QAAQO,IACtBb,OAAOU,KAAKI,IAAI,CAACC,UAAU;AACzB,aAAO,IAAIT,QAAgC,CAACC,SAASS,WAAW;AAC9D,cAAMf,oBAAoB,KAAKX,gCAAgCY,IAAIa,KAAAA;AAEnE,YAAId,mBAAmB;AACrB,eAAKX,gCAAgCc,OAAOW,KAAAA;AAE5CR,kBAAQN,iBAAAA;AAER;QACF;AAEA,aAAKd,WAAWqB,IAAIO,OAAO;UAAER;QAAQ,CAAA;MACvC,CAAA;IACF,CAAA,CAAA;AAGF,UAAMU,UAAU,MAAMZ;AAEtB,WAAO;MACLF,IAAIH,OAAOG;MACXS,OAAOK;IACT;EACF;EAEAC,WAAWC,YAAoCC,WAAmC;AAChF,UAAMC,OAAO,KAAKlC,WAAWe,IAAIkB,UAAUE,IAAInB,EAAE;AAEjD,QAAI,CAACkB,MAAM;AAET,WAAK/B,gCAAgCkB,IAAIY,UAAUE,IAAInB,IAAIgB,UAAAA;AAE3D;IACF;AAEAE,SAAKd,QAAQY,UAAAA;AAEb,SAAKhC,WAAWiB,OAAOgB,UAAUE,IAAInB,EAAE;EACzC;AACF;AAvFajB;AAAN,IAAMA,oBAAN","sourcesContent":["import { TimerOptions } from \"node:timers\";\nimport { setInterval, setTimeout } from \"node:timers/promises\";\n\nexport async function unboundedTimeout<T = void>(\n  delay: number = 0,\n  value?: T,\n  options?: TimerOptions\n): Promise<T> {\n  const maxDelay = 2147483647; // Highest value that will fit in a 32-bit signed integer\n\n  const fullTimeouts = Math.floor(delay / maxDelay);\n  const remainingDelay = delay % maxDelay;\n\n  let lastTimeoutResult = await setTimeout(remainingDelay, value, options);\n\n  for (let i = 0; i < fullTimeouts; i++) {\n    lastTimeoutResult = await setTimeout(maxDelay, value, options);\n  }\n\n  return lastTimeoutResult;\n}\n\nexport async function checkpointSafeTimeout(delay: number = 0): Promise<void> {\n  const scanIntervalMs = 1000;\n\n  // Every scanIntervalMs, check if delay has elapsed\n  for await (const start of setInterval(scanIntervalMs, Date.now())) {\n    if (Date.now() - start > delay) {\n      break;\n    }\n  }\n}\n","import {\n  BatchTaskRunExecutionResult,\n  TaskRunContext,\n  TaskRunExecution,\n  TaskRunExecutionResult,\n} from \"../schemas\";\nimport { RuntimeManager } from \"./manager\";\nimport { unboundedTimeout } from \"../utils/timers\";\n\nexport class DevRuntimeManager implements RuntimeManager {\n  _taskWaits: Map<string, { resolve: (value: TaskRunExecutionResult) => void }> = new Map();\n\n  _batchWaits: Map<\n    string,\n    { resolve: (value: BatchTaskRunExecutionResult) => void; reject: (err?: any) => void }\n  > = new Map();\n\n  _pendingCompletionNotifications: Map<string, TaskRunExecutionResult> = new Map();\n\n  disable(): void {\n    // do nothing\n  }\n\n  async waitForDuration(ms: number): Promise<void> {\n    await unboundedTimeout(ms);\n  }\n\n  async waitUntil(date: Date): Promise<void> {\n    return this.waitForDuration(date.getTime() - Date.now());\n  }\n\n  async waitForTask(params: { id: string; ctx: TaskRunContext }): Promise<TaskRunExecutionResult> {\n    const pendingCompletion = this._pendingCompletionNotifications.get(params.id);\n\n    if (pendingCompletion) {\n      this._pendingCompletionNotifications.delete(params.id);\n\n      return pendingCompletion;\n    }\n\n    const promise = new Promise<TaskRunExecutionResult>((resolve) => {\n      this._taskWaits.set(params.id, { resolve });\n    });\n\n    return await promise;\n  }\n\n  async waitForBatch(params: {\n    id: string;\n    runs: string[];\n    ctx: TaskRunContext;\n  }): Promise<BatchTaskRunExecutionResult> {\n    if (!params.runs.length) {\n      return Promise.resolve({ id: params.id, items: [] });\n    }\n\n    const promise = Promise.all(\n      params.runs.map((runId) => {\n        return new Promise<TaskRunExecutionResult>((resolve, reject) => {\n          const pendingCompletion = this._pendingCompletionNotifications.get(runId);\n\n          if (pendingCompletion) {\n            this._pendingCompletionNotifications.delete(runId);\n\n            resolve(pendingCompletion);\n\n            return;\n          }\n\n          this._taskWaits.set(runId, { resolve });\n        });\n      })\n    );\n\n    const results = await promise;\n\n    return {\n      id: params.id,\n      items: results,\n    };\n  }\n\n  resumeTask(completion: TaskRunExecutionResult, execution: TaskRunExecution): void {\n    const wait = this._taskWaits.get(execution.run.id);\n\n    if (!wait) {\n      // We need to store the completion in case the task is awaited later\n      this._pendingCompletionNotifications.set(execution.run.id, completion);\n\n      return;\n    }\n\n    wait.resolve(completion);\n\n    this._taskWaits.delete(execution.run.id);\n  }\n}\n"]}

package/dist/v3/dev/index.mjs

@@ -0,0 +1,91 @@
+import { setTimeout } from 'node:timers/promises';
+
+var __defProp = Object.defineProperty;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __publicField = (obj, key, value) => {
+  __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
+  return value;
+};
+async function unboundedTimeout(delay = 0, value, options) {
+  const maxDelay = 2147483647;
+  const fullTimeouts = Math.floor(delay / maxDelay);
+  const remainingDelay = delay % maxDelay;
+  let lastTimeoutResult = await setTimeout(remainingDelay, value, options);
+  for (let i = 0; i < fullTimeouts; i++) {
+    lastTimeoutResult = await setTimeout(maxDelay, value, options);
+  }
+  return lastTimeoutResult;
+}
+__name(unboundedTimeout, "unboundedTimeout");
+
+// src/v3/runtime/devRuntimeManager.ts
+var _DevRuntimeManager = class _DevRuntimeManager {
+  constructor() {
+    __publicField(this, "_taskWaits", /* @__PURE__ */ new Map());
+    __publicField(this, "_batchWaits", /* @__PURE__ */ new Map());
+    __publicField(this, "_pendingCompletionNotifications", /* @__PURE__ */ new Map());
+  }
+  disable() {
+  }
+  async waitForDuration(ms) {
+    await unboundedTimeout(ms);
+  }
+  async waitUntil(date) {
+    return this.waitForDuration(date.getTime() - Date.now());
+  }
+  async waitForTask(params) {
+    const pendingCompletion = this._pendingCompletionNotifications.get(params.id);
+    if (pendingCompletion) {
+      this._pendingCompletionNotifications.delete(params.id);
+      return pendingCompletion;
+    }
+    const promise = new Promise((resolve) => {
+      this._taskWaits.set(params.id, {
+        resolve
+      });
+    });
+    return await promise;
+  }
+  async waitForBatch(params) {
+    if (!params.runs.length) {
+      return Promise.resolve({
+        id: params.id,
+        items: []
+      });
+    }
+    const promise = Promise.all(params.runs.map((runId) => {
+      return new Promise((resolve, reject) => {
+        const pendingCompletion = this._pendingCompletionNotifications.get(runId);
+        if (pendingCompletion) {
+          this._pendingCompletionNotifications.delete(runId);
+          resolve(pendingCompletion);
+          return;
+        }
+        this._taskWaits.set(runId, {
+          resolve
+        });
+      });
+    }));
+    const results = await promise;
+    return {
+      id: params.id,
+      items: results
+    };
+  }
+  resumeTask(completion, execution) {
+    const wait = this._taskWaits.get(execution.run.id);
+    if (!wait) {
+      this._pendingCompletionNotifications.set(execution.run.id, completion);
+      return;
+    }
+    wait.resolve(completion);
+    this._taskWaits.delete(execution.run.id);
+  }
+};
+__name(_DevRuntimeManager, "DevRuntimeManager");
+var DevRuntimeManager = _DevRuntimeManager;
+
+export { DevRuntimeManager };
+//# sourceMappingURL=out.js.map
+//# sourceMappingURL=index.mjs.map

package/dist/v3/dev/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/v3/utils/timers.ts","../../../src/v3/runtime/devRuntimeManager.ts"],"names":["setInterval","setTimeout","unboundedTimeout","delay","value","options","maxDelay","fullTimeouts","Math","floor","remainingDelay","lastTimeoutResult","i","DevRuntimeManager","_taskWaits","Map","_batchWaits","_pendingCompletionNotifications","disable","waitForDuration","ms","waitUntil","date","getTime","Date","now","waitForTask","params","pendingCompletion","get","id","delete","promise","Promise","resolve","set","waitForBatch","runs","length","items","all","map","runId","reject","results","resumeTask","completion","execution","wait","run"],"mappings":";;;;;;;;;AACA,SAASA,aAAaC,kBAAkB;AAExC,eAAsBC,iBACpBC,QAAgB,GAChBC,OACAC,SACY;AACZ,QAAMC,WAAW;AAEjB,QAAMC,eAAeC,KAAKC,MAAMN,QAAQG,QAAAA;AACxC,QAAMI,iBAAiBP,QAAQG;AAE/B,MAAIK,oBAAoB,MAAMV,WAAWS,gBAAgBN,OAAOC,OAAAA;AAEhE,WAASO,IAAI,GAAGA,IAAIL,cAAcK,KAAK;AACrCD,wBAAoB,MAAMV,WAAWK,UAAUF,OAAOC,OAAAA;EACxD;AAEA,SAAOM;AACT;AAjBsBT;;;ACMf,IAAMW,qBAAN,MAAMA,mBAAAA;EAAN;AACLC,sCAAgF,oBAAIC,IAAAA;AAEpFC,uCAGI,oBAAID,IAAAA;AAERE,2DAAuE,oBAAIF,IAAAA;;EAE3EG,UAAgB;EAEhB;EAEA,MAAMC,gBAAgBC,IAA2B;AAC/C,UAAMlB,iBAAiBkB,EAAAA;EACzB;EAEA,MAAMC,UAAUC,MAA2B;AACzC,WAAO,KAAKH,gBAAgBG,KAAKC,QAAO,IAAKC,KAAKC,IAAG,CAAA;EACvD;EAEA,MAAMC,YAAYC,QAA8E;AAC9F,UAAMC,oBAAoB,KAAKX,gCAAgCY,IAAIF,OAAOG,EAAE;AAE5E,QAAIF,mBAAmB;AACrB,WAAKX,gCAAgCc,OAAOJ,OAAOG,EAAE;AAErD,aAAOF;IACT;AAEA,UAAMI,UAAU,IAAIC,QAAgC,CAACC,YAAY;AAC/D,WAAKpB,WAAWqB,IAAIR,OAAOG,IAAI;QAAEI;MAAQ,CAAA;IAC3C,CAAA;AAEA,WAAO,MAAMF;EACf;EAEA,MAAMI,aAAaT,QAIsB;AACvC,QAAI,CAACA,OAAOU,KAAKC,QAAQ;AACvB,aAAOL,QAAQC,QAAQ;QAAEJ,IAAIH,OAAOG;QAAIS,OAAO,CAAA;MAAG,CAAA;IACpD;AAEA,UAAMP,UAAUC,QAAQO,IACtBb,OAAOU,KAAKI,IAAI,CAACC,UAAU;AACzB,aAAO,IAAIT,QAAgC,CAACC,SAASS,WAAW;AAC9D,cAAMf,oBAAoB,KAAKX,gCAAgCY,IAAIa,KAAAA;AAEnE,YAAId,mBAAmB;AACrB,eAAKX,gCAAgCc,OAAOW,KAAAA;AAE5CR,kBAAQN,iBAAAA;AAER;QACF;AAEA,aAAKd,WAAWqB,IAAIO,OAAO;UAAER;QAAQ,CAAA;MACvC,CAAA;IACF,CAAA,CAAA;AAGF,UAAMU,UAAU,MAAMZ;AAEtB,WAAO;MACLF,IAAIH,OAAOG;MACXS,OAAOK;IACT;EACF;EAEAC,WAAWC,YAAoCC,WAAmC;AAChF,UAAMC,OAAO,KAAKlC,WAAWe,IAAIkB,UAAUE,IAAInB,EAAE;AAEjD,QAAI,CAACkB,MAAM;AAET,WAAK/B,gCAAgCkB,IAAIY,UAAUE,IAAInB,IAAIgB,UAAAA;AAE3D;IACF;AAEAE,SAAKd,QAAQY,UAAAA;AAEb,SAAKhC,WAAWiB,OAAOgB,UAAUE,IAAInB,EAAE;EACzC;AACF;AAvFajB;AAAN,IAAMA,oBAAN","sourcesContent":["import { TimerOptions } from \"node:timers\";\nimport { setInterval, setTimeout } from \"node:timers/promises\";\n\nexport async function unboundedTimeout<T = void>(\n  delay: number = 0,\n  value?: T,\n  options?: TimerOptions\n): Promise<T> {\n  const maxDelay = 2147483647; // Highest value that will fit in a 32-bit signed integer\n\n  const fullTimeouts = Math.floor(delay / maxDelay);\n  const remainingDelay = delay % maxDelay;\n\n  let lastTimeoutResult = await setTimeout(remainingDelay, value, options);\n\n  for (let i = 0; i < fullTimeouts; i++) {\n    lastTimeoutResult = await setTimeout(maxDelay, value, options);\n  }\n\n  return lastTimeoutResult;\n}\n\nexport async function checkpointSafeTimeout(delay: number = 0): Promise<void> {\n  const scanIntervalMs = 1000;\n\n  // Every scanIntervalMs, check if delay has elapsed\n  for await (const start of setInterval(scanIntervalMs, Date.now())) {\n    if (Date.now() - start > delay) {\n      break;\n    }\n  }\n}\n","import {\n  BatchTaskRunExecutionResult,\n  TaskRunContext,\n  TaskRunExecution,\n  TaskRunExecutionResult,\n} from \"../schemas\";\nimport { RuntimeManager } from \"./manager\";\nimport { unboundedTimeout } from \"../utils/timers\";\n\nexport class DevRuntimeManager implements RuntimeManager {\n  _taskWaits: Map<string, { resolve: (value: TaskRunExecutionResult) => void }> = new Map();\n\n  _batchWaits: Map<\n    string,\n    { resolve: (value: BatchTaskRunExecutionResult) => void; reject: (err?: any) => void }\n  > = new Map();\n\n  _pendingCompletionNotifications: Map<string, TaskRunExecutionResult> = new Map();\n\n  disable(): void {\n    // do nothing\n  }\n\n  async waitForDuration(ms: number): Promise<void> {\n    await unboundedTimeout(ms);\n  }\n\n  async waitUntil(date: Date): Promise<void> {\n    return this.waitForDuration(date.getTime() - Date.now());\n  }\n\n  async waitForTask(params: { id: string; ctx: TaskRunContext }): Promise<TaskRunExecutionResult> {\n    const pendingCompletion = this._pendingCompletionNotifications.get(params.id);\n\n    if (pendingCompletion) {\n      this._pendingCompletionNotifications.delete(params.id);\n\n      return pendingCompletion;\n    }\n\n    const promise = new Promise<TaskRunExecutionResult>((resolve) => {\n      this._taskWaits.set(params.id, { resolve });\n    });\n\n    return await promise;\n  }\n\n  async waitForBatch(params: {\n    id: string;\n    runs: string[];\n    ctx: TaskRunContext;\n  }): Promise<BatchTaskRunExecutionResult> {\n    if (!params.runs.length) {\n      return Promise.resolve({ id: params.id, items: [] });\n    }\n\n    const promise = Promise.all(\n      params.runs.map((runId) => {\n        return new Promise<TaskRunExecutionResult>((resolve, reject) => {\n          const pendingCompletion = this._pendingCompletionNotifications.get(runId);\n\n          if (pendingCompletion) {\n            this._pendingCompletionNotifications.delete(runId);\n\n            resolve(pendingCompletion);\n\n            return;\n          }\n\n          this._taskWaits.set(runId, { resolve });\n        });\n      })\n    );\n\n    const results = await promise;\n\n    return {\n      id: params.id,\n      items: results,\n    };\n  }\n\n  resumeTask(completion: TaskRunExecutionResult, execution: TaskRunExecution): void {\n    const wait = this._taskWaits.get(execution.run.id);\n\n    if (!wait) {\n      // We need to store the completion in case the task is awaited later\n      this._pendingCompletionNotifications.set(execution.run.id, completion);\n\n      return;\n    }\n\n    wait.resolve(completion);\n\n    this._taskWaits.delete(execution.run.id);\n  }\n}\n"]}