@trigger.dev/core 3.0.0-beta.37 → 3.0.0-beta.39

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

@@ -1,5 +1,5 @@
-import { R as RuntimeManager, T as TaskRunExecutionResult, B as BatchTaskRunExecutionResult, a as TaskRunContext } from '../../manager-JkbddlcO.mjs';
-import { P as ProdWorkerToChildMessages, a as ProdChildToWorkerMessages } from '../../messages-9lty-Du5.mjs';
+import { R as RuntimeManager, T as TaskRunExecutionResult, B as BatchTaskRunExecutionResult, a as TaskRunContext } from '../../manager-S98VaLUy.mjs';
+import { P as ProdWorkerToChildMessages, a as ProdChildToWorkerMessages } from '../../messages--WkQvA2l.mjs';
 import { ZodIpcConnection } from '../zodIpc.mjs';
 import 'zod';
 import '../zodSocket.mjs';

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

@@ -1,5 +1,5 @@
-import { R as RuntimeManager, T as TaskRunExecutionResult, B as BatchTaskRunExecutionResult, a as TaskRunContext } from '../../manager-JkbddlcO.js';
-import { P as ProdWorkerToChildMessages, a as ProdChildToWorkerMessages } from '../../messages-9lty-Du5.js';
+import { R as RuntimeManager, T as TaskRunExecutionResult, B as BatchTaskRunExecutionResult, a as TaskRunContext } from '../../manager-S98VaLUy.js';
+import { P as ProdWorkerToChildMessages, a as ProdChildToWorkerMessages } from '../../messages--WkQvA2l.js';
 import { ZodIpcConnection } from '../zodIpc.js';
 import 'zod';
 import '../zodSocket.js';

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/v3/utils/platform.ts","../../../src/v3/utils/globals.ts","../../../src/v3/clock/simpleClock.ts","../../../src/v3/clock/index.ts","../../../src/v3/clock-api.ts","../../../src/v3/logger/taskLogger.ts","../../../src/v3/logger/index.ts","../../../src/v3/logger-api.ts","../../../src/v3/utils/timers.ts","../../../src/v3/runtime/prodRuntimeManager.ts"],"names":["_globalThis","globalThis","global","GLOBAL_TRIGGER_DOT_DEV_KEY","Symbol","for","_global","registerGlobal","type","instance","allowOverride","api","err","Error","getGlobal","unregisterGlobal","PreciseDate","SimpleClock","preciseNow","now","nowStruct","toStruct","seconds","nanos","reset","API_NAME","SIMPLE_CLOCK","ClockAPI","getInstance","_instance","setGlobalClock","clock","SeverityNumber","NoopTaskLogger","debug","log","info","warn","error","trace","name","fn","NOOP_TASK_LOGGER","LoggerAPI","disable","setGlobalTaskLogger","taskLogger","message","metadata","logger","setInterval","setTimeout","unboundedTimeout","delay","value","options","maxDelay","fullTimeouts","Math","floor","remainingDelay","lastTimeoutResult","i","checkpointSafeTimeout","scanIntervalMs","start","Date","ProdRuntimeManager","constructor","ipc","_taskWaits","Map","_batchWaits","waitForDuration","ms","internalTimeout","checkpointSafeInternalTimeout","waitThresholdInMs","externalResume","Promise","resolve","reject","_waitForDuration","willCheckpointAndRestore","sendWithAck","send","race","checkpointCanceled","version","reason","resumeAfterDuration","undefined","waitUntil","date","getTime","waitForTask","params","promise","set","id","friendlyId","result","waitForBatch","runs","length","items","all","map","runId","batchFriendlyId","runFriendlyIds","results","resumeTask","completion","wait","get","delete"],"mappings":";;;;;;;;;;;;;;;;;AAAO,IAAMA,cAAc,OAAOC,eAAe,WAAWA,aAAaC;;;ACOzE,IAAMC,6BAA6BC,OAAOC,IAAI,oBAAoB;AAElE,IAAMC,UAAUN;AAET,SAASO,eACdC,MACAC,UACAC,gBAAgB,OACP;AACT,QAAMC,MAAOL,QAAQH,0BAAAA,IAA8BG,QAAQH,0BAAAA,KAA+B,CAAC;AAE3F,MAAI,CAACO,iBAAiBC,IAAIH,IAAAA,GAAO;AAE/B,UAAMI,MAAM,IAAIC,MAAM,yDAAyDL,IAAAA,EAAM;AACrF,WAAO;EACT;AAEAG,MAAIH,IAAAA,IAAQC;AAEZ,SAAO;AACT;AAhBgBF;AAkBT,SAASO,UACdN,MAC0C;AAC1C,SAAOF,QAAQH,0BAAAA,IAA8BK,IAAAA;AAC/C;AAJgBM;AAMT,SAASC,iBAAiBP,MAAoC;AACnE,QAAMG,MAAML,QAAQH,0BAAAA;AAEpB,MAAIQ,KAAK;AACP,WAAOA,IAAIH,IAAAA;EACb;AACF;AANgBO;;;ACnChB,SAASC,mBAAmB;AAGrB,IAAMC,eAAN,MAAMA,aAAAA;EACXC,aAA+B;AAC7B,UAAMC,MAAM,IAAIH,YAAAA;AAChB,UAAMI,YAAYD,IAAIE,SAAQ;AAE9B,WAAO;MAACD,UAAUE;MAASF,UAAUG;;EACvC;EAEAC,QAAQ;EAER;AACF;AAXaP;AAAN,IAAMA,cAAN;;;ACHP,IAAMQ,WAAW;AAMjB,IAAMC,eAAe,IAAIT,YAAAA;AANzB;AAQO,IAAMU,YAAN,MAAMA,UAAAA;EAGX,cAAsB;AAsBtB;EAtBuB;EAEvB,OAAcC,cAAwB;AACpC,QAAI,CAAC,KAAKC,WAAW;AACnB,WAAKA,YAAY,IAAIF,UAAAA;IACvB;AAEA,WAAO,KAAKE;EACd;EAEOC,eAAeC,QAAuB;AAC3C,WAAOxB,eAAekB,UAAUM,MAAAA;EAClC;EAEOb,aAAwB;AAC7B,WAAO,sBAAK,wBAAL,WAAiBA,WAAU;EACpC;EAEOM,QAAc;AACnB,0BAAK,wBAAL,WAAiBA,MAAK;EACxB;AAKF;AAHE;cAAS,kCAAU;AACjB,SAAOV,UAAUW,QAAAA,KAAaC;AAChC,GAFS;AAzBEC;AAAN,IAAMA,WAAN;;;ACJA,IAAMI,QAAQJ,SAASC,YAAW;;;ACHzC,SAAiBI,sBAAsB;AAgGhC,IAAMC,kBAAN,MAAMA,gBAAAA;EACXC,QAAQ;EAAC;EACTC,MAAM;EAAC;EACPC,OAAO;EAAC;EACRC,OAAO;EAAC;EACRC,QAAQ;EAAC;EACTC,MAASC,MAAcC,IAA4C;AACjE,WAAOA,GAAG,CAAC,CAAA;EACb;AACF;AATaR;AAAN,IAAMA,iBAAN;;;AC7FP,IAAMR,YAAW;AAEjB,IAAMiB,mBAAmB,IAAIT,eAAAA;AAN7B;AAQO,IAAMU,aAAN,MAAMA,WAAAA;EAGX,cAAsB;AA0CtB;EA1CuB;EAEvB,OAAcf,cAAyB;AACrC,QAAI,CAAC,KAAKC,WAAW;AACnB,WAAKA,YAAY,IAAIc,WAAAA;IACvB;AAEA,WAAO,KAAKd;EACd;EAEOe,UAAU;AACf7B,qBAAiBU,SAAAA;EACnB;EAEOoB,oBAAoBC,YAAiC;AAC1D,WAAOvC,eAAekB,WAAUqB,UAAAA;EAClC;EAEOZ,MAAMa,SAAiBC,UAAoC;AAChE,0BAAK,kCAAL,WAAsBd,MAAMa,SAASC,QAAAA;EACvC;EAEOb,IAAIY,SAAiBC,UAAoC;AAC9D,0BAAK,kCAAL,WAAsBb,IAAIY,SAASC,QAAAA;EACrC;EAEOZ,KAAKW,SAAiBC,UAAoC;AAC/D,0BAAK,kCAAL,WAAsBZ,KAAKW,SAASC,QAAAA;EACtC;EAEOX,KAAKU,SAAiBC,UAAoC;AAC/D,0BAAK,kCAAL,WAAsBX,KAAKU,SAASC,QAAAA;EACtC;EAEOV,MAAMS,SAAiBC,UAAoC;AAChE,0BAAK,kCAAL,WAAsBV,MAAMS,SAASC,QAAAA;EACvC;EAEOT,MAASC,MAAcC,IAA4C;AACxE,WAAO,sBAAK,kCAAL,WAAsBF,MAAMC,MAAMC,EAAAA;EAC3C;AAKF;AAHE;mBAAc,kCAAe;AAC3B,SAAO3B,UAAUW,SAAAA,KAAaiB;AAChC,GAFc;AA7CHC;AAAN,IAAMA,YAAN;;;ACJA,IAAMM,SAASN,UAAUf,YAAW;;;ACH3C,SAASsB,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;AAmBtB,eAAsBW,sBAAsBV,QAAgB,GAAkB;AAC5E,QAAMW,iBAAiB;AAGvB,mBAAiBC,SAASf,YAAYc,gBAAgBE,KAAK/C,IAAG,CAAA,GAAK;AACjE,QAAI+C,KAAK/C,IAAG,IAAK8C,QAAQZ,OAAO;AAC9B;IACF;EACF;AACF;AATsBU;;;ACJf,IAAMI,sBAAN,MAAMA,oBAAAA;EAYXC,YACUC,KAIAd,UAAqC,CAAC,GAC9C;eALQc;mBAIAd;SAhBVe,aAAgF,oBAAIC,IAAAA;SAEpFC,cAGI,oBAAID,IAAAA;EAYL;EAEH3B,UAAgB;EAEhB;EAEA,MAAM6B,gBAAgBC,IAA2B;AAC/C,UAAMvD,MAAM+C,KAAK/C,IAAG;AAEpB,UAAMwD,kBAAkBvB,iBAAiBsB,IAAI,UAAA;AAC7C,UAAME,gCAAgCb,sBAAsBW,EAAAA;AAE5D,QAAIA,MAAM,KAAKG,mBAAmB;AAChC,YAAMF;AACN;IACF;AAEA,UAAMG,iBAAiB,IAAIC,QAAoB,CAACC,SAASC,WAAW;AAClE,WAAKC,mBAAmB;QAAEF;QAASC;MAAO;IAC5C,CAAA;AAEA,UAAM,EAAEE,yBAAwB,IAAK,MAAM,KAAKd,IAAIe,YAClD,qBACA;MACEV;MACAvD;IACF,GACA,IAAA;AAGF,QAAI,CAACgE,0BAA0B;AAC7B,YAAMR;AACN;IACF;AAEA,SAAKN,IAAIgB,KAAK,wBAAwB,CAAC,CAAA;AAIvC,UAAMN,QAAQO,KAAK;MAACX;MAAiBC;KAA8B;AAGnE7C,UAAMP,MAAK;AAEX,QAAI;AAEF,YAAM,EAAE+D,oBAAoBC,QAAO,IAAK,MAAM,KAAKnB,IAAIe,YACrD,qBACA;QACEI,SAAS;QACTC,QAAQ;MACV,GACA,IAAA;AAGF,UAAIF,oBAAoB;AAEtB;MACF;IACF,SAASjD,OAAO;AAEdW,aAAOf,MAAM,qCAAqC;QAAEI;MAAM,CAAA;AAC1D;IACF;AAGA,UAAMwC;EACR;EAEAY,sBAA4B;AAC1B,QAAI,CAAC,KAAKR,kBAAkB;AAC1B;IACF;AAGAnD,UAAMP,MAAK;AAEX,SAAK0D,iBAAiBF,QAAQ,UAAA;AAC9B,SAAKE,mBAAmBS;EAC1B;EAEA,MAAMC,UAAUC,MAA2B;AACzC,WAAO,KAAKpB,gBAAgBoB,KAAKC,QAAO,IAAK5B,KAAK/C,IAAG,CAAA;EACvD;EAEA,MAAM4E,YAAYC,QAA8E;AAC9F,UAAMC,UAAU,IAAIlB,QAAgC,CAACC,YAAY;AAC/D,WAAKV,WAAW4B,IAAIF,OAAOG,IAAI;QAAEnB;MAAQ,CAAA;IAC3C,CAAA;AAEA,UAAM,KAAKX,IAAIgB,KAAK,iBAAiB;MACnCe,YAAYJ,OAAOG;IACrB,CAAA;AAEA,UAAME,SAAS,MAAMJ;AAErBlE,UAAMP,MAAK;AAEX,WAAO6E;EACT;EAEA,MAAMC,aAAaN,QAIsB;AACvC,QAAI,CAACA,OAAOO,KAAKC,QAAQ;AACvB,aAAOzB,QAAQC,QAAQ;QAAEmB,IAAIH,OAAOG;QAAIM,OAAO,CAAA;MAAG,CAAA;IACpD;AAEA,UAAMR,UAAUlB,QAAQ2B,IACtBV,OAAOO,KAAKI,IAAI,CAACC,UAAU;AACzB,aAAO,IAAI7B,QAAgC,CAACC,SAASC,WAAW;AAC9D,aAAKX,WAAW4B,IAAIU,OAAO;UAAE5B;QAAQ,CAAA;MACvC,CAAA;IACF,CAAA,CAAA;AAGF,UAAM,KAAKX,IAAIgB,KAAK,kBAAkB;MACpCwB,iBAAiBb,OAAOG;MACxBW,gBAAgBd,OAAOO;IACzB,CAAA;AAEA,UAAMQ,UAAU,MAAMd;AAEtBlE,UAAMP,MAAK;AAEX,WAAO;MACL2E,IAAIH,OAAOG;MACXM,OAAOM;IACT;EACF;EAEAC,WAAWC,YAA0C;AACnD,UAAMC,OAAO,KAAK5C,WAAW6C,IAAIF,WAAWd,EAAE;AAE9C,QAAI,CAACe,MAAM;AACT;IACF;AAEAA,SAAKlC,QAAQiC,UAAAA;AAEb,SAAK3C,WAAW8C,OAAOH,WAAWd,EAAE;EACtC;EAEA,IAAYtB,oBAA4B;AACtC,WAAO,KAAKtB,QAAQsB,qBAAqB;EAC3C;AACF;AAtKaV;AAAN,IAAMA,qBAAN","sourcesContent":["export const _globalThis = typeof globalThis === \"object\" ? globalThis : global;\n","import { ApiClientConfiguration } from \"../apiClientManager/types\";\nimport { Clock } from \"../clock/clock\";\nimport type { RuntimeManager } from \"../runtime/manager\";\nimport { TaskCatalog } from \"../task-catalog/catalog\";\nimport { TaskContext } from \"../taskContext/types\";\nimport { _globalThis } from \"./platform\";\n\nconst GLOBAL_TRIGGER_DOT_DEV_KEY = Symbol.for(`dev.trigger.ts.api`);\n\nconst _global = _globalThis as TriggerDotDevGlobal;\n\nexport function registerGlobal<Type extends keyof TriggerDotDevGlobalAPI>(\n  type: Type,\n  instance: TriggerDotDevGlobalAPI[Type],\n  allowOverride = false\n): boolean {\n  const api = (_global[GLOBAL_TRIGGER_DOT_DEV_KEY] = _global[GLOBAL_TRIGGER_DOT_DEV_KEY] ?? {});\n\n  if (!allowOverride && api[type]) {\n    // already registered an API of this type\n    const err = new Error(`trigger.dev: Attempted duplicate registration of API: ${type}`);\n    return false;\n  }\n\n  api[type] = instance;\n\n  return true;\n}\n\nexport function getGlobal<Type extends keyof TriggerDotDevGlobalAPI>(\n  type: Type\n): TriggerDotDevGlobalAPI[Type] | undefined {\n  return _global[GLOBAL_TRIGGER_DOT_DEV_KEY]?.[type];\n}\n\nexport function unregisterGlobal(type: keyof TriggerDotDevGlobalAPI) {\n  const api = _global[GLOBAL_TRIGGER_DOT_DEV_KEY];\n\n  if (api) {\n    delete api[type];\n  }\n}\n\ntype TriggerDotDevGlobal = {\n  [GLOBAL_TRIGGER_DOT_DEV_KEY]?: TriggerDotDevGlobalAPI;\n};\n\ntype TriggerDotDevGlobalAPI = {\n  runtime?: RuntimeManager;\n  logger?: any;\n  clock?: Clock;\n  [\"task-catalog\"]?: TaskCatalog;\n  [\"task-context\"]?: TaskContext;\n  [\"api-client\"]?: ApiClientConfiguration;\n};\n","import { PreciseDate } from \"@google-cloud/precise-date\";\nimport { Clock } from \"./clock\";\n\nexport class SimpleClock implements Clock {\n  preciseNow(): [number, number] {\n    const now = new PreciseDate();\n    const nowStruct = now.toStruct();\n\n    return [nowStruct.seconds, nowStruct.nanos];\n  }\n\n  reset() {\n    // do nothing\n  }\n}\n","const API_NAME = \"clock\";\n\nimport { getGlobal, registerGlobal } from \"../utils/globals\";\nimport type { Clock, ClockTime } from \"./clock\";\nimport { SimpleClock } from \"./simpleClock\";\n\nconst SIMPLE_CLOCK = new SimpleClock();\n\nexport class ClockAPI {\n  private static _instance?: ClockAPI;\n\n  private constructor() {}\n\n  public static getInstance(): ClockAPI {\n    if (!this._instance) {\n      this._instance = new ClockAPI();\n    }\n\n    return this._instance;\n  }\n\n  public setGlobalClock(clock: Clock): boolean {\n    return registerGlobal(API_NAME, clock);\n  }\n\n  public preciseNow(): ClockTime {\n    return this.#getClock().preciseNow();\n  }\n\n  public reset(): void {\n    this.#getClock().reset();\n  }\n\n  #getClock(): Clock {\n    return getGlobal(API_NAME) ?? SIMPLE_CLOCK;\n  }\n}\n","// Split module-level variable definition into separate files to allow\n// tree-shaking on each api instance.\nimport { ClockAPI } from \"./clock\";\n/** Entrypoint for clock API */\nexport const clock = ClockAPI.getInstance();\n","import { Attributes, Span, SpanOptions } from \"@opentelemetry/api\";\nimport { Logger, SeverityNumber } from \"@opentelemetry/api-logs\";\nimport { iconStringForSeverity } from \"../icons\";\nimport { SemanticInternalAttributes } from \"../semanticInternalAttributes\";\nimport { TriggerTracer } from \"../tracer\";\nimport { flattenAttributes } from \"../utils/flattenAttributes\";\nimport { ClockTime } from \"../clock/clock\";\nimport { clock } from \"../clock-api\";\n\nexport type LogLevel = \"none\" | \"error\" | \"warn\" | \"info\" | \"debug\" | \"log\";\n\nexport const logLevels: Array<LogLevel> = [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\nexport type TaskLoggerConfig = {\n  logger: Logger;\n  tracer: TriggerTracer;\n  level: LogLevel;\n};\n\nexport interface TaskLogger {\n  debug(message: string, properties?: Record<string, unknown>): void;\n  log(message: string, properties?: Record<string, unknown>): void;\n  info(message: string, properties?: Record<string, unknown>): void;\n  warn(message: string, properties?: Record<string, unknown>): void;\n  error(message: string, properties?: Record<string, unknown>): void;\n  trace<T>(name: string, fn: (span: Span) => Promise<T>, options?: SpanOptions): Promise<T>;\n}\n\nexport class OtelTaskLogger implements TaskLogger {\n  private readonly _level: number;\n\n  constructor(private readonly _config: TaskLoggerConfig) {\n    this._level = logLevels.indexOf(_config.level);\n  }\n\n  debug(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 4) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"debug\", SeverityNumber.DEBUG, properties);\n  }\n\n  log(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 3) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"log\", SeverityNumber.INFO, properties);\n  }\n\n  info(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 3) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"info\", SeverityNumber.INFO, properties);\n  }\n\n  warn(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 2) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"warn\", SeverityNumber.WARN, properties);\n  }\n\n  error(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 1) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"error\", SeverityNumber.ERROR, properties);\n  }\n\n  #emitLog(\n    message: string,\n    timestamp: ClockTime,\n    severityText: string,\n    severityNumber: SeverityNumber,\n    properties?: Record<string, unknown>\n  ) {\n    let attributes: Attributes = { ...flattenAttributes(safeJsonProcess(properties)) };\n\n    const icon = iconStringForSeverity(severityNumber);\n    if (icon !== undefined) {\n      attributes[SemanticInternalAttributes.STYLE_ICON] = icon;\n    }\n\n    this._config.logger.emit({\n      severityNumber,\n      severityText,\n      body: message,\n      attributes,\n      timestamp,\n    });\n  }\n\n  trace<T>(name: string, fn: (span: Span) => Promise<T>, options?: SpanOptions): Promise<T> {\n    return this._config.tracer.startActiveSpan(name, fn, options);\n  }\n\n  #getTimestampInHrTime(): ClockTime {\n    return clock.preciseNow();\n  }\n}\n\nexport class NoopTaskLogger implements TaskLogger {\n  debug() {}\n  log() {}\n  info() {}\n  warn() {}\n  error() {}\n  trace<T>(name: string, fn: (span: Span) => Promise<T>): Promise<T> {\n    return fn({} as Span);\n  }\n}\n\nfunction safeJsonProcess(value?: Record<string, unknown>): Record<string, unknown> | undefined {\n  try {\n    return JSON.parse(JSON.stringify(value, jsonErrorReplacer));\n  } catch {\n    return value;\n  }\n}\n\nfunction jsonErrorReplacer(key: string, value: unknown) {\n  if (value instanceof Error) {\n    return {\n      name: value.name,\n      message: value.message,\n      stack: value.stack,\n    };\n  }\n\n  return value;\n}\n","import { NoopTaskLogger, TaskLogger } from \"./taskLogger\";\nimport { getGlobal, registerGlobal, unregisterGlobal } from \"../utils/globals\";\nimport { Span } from \"@opentelemetry/api\";\n\nconst API_NAME = \"logger\";\n\nconst NOOP_TASK_LOGGER = new NoopTaskLogger();\n\nexport class LoggerAPI implements TaskLogger {\n  private static _instance?: LoggerAPI;\n\n  private constructor() {}\n\n  public static getInstance(): LoggerAPI {\n    if (!this._instance) {\n      this._instance = new LoggerAPI();\n    }\n\n    return this._instance;\n  }\n\n  public disable() {\n    unregisterGlobal(API_NAME);\n  }\n\n  public setGlobalTaskLogger(taskLogger: TaskLogger): boolean {\n    return registerGlobal(API_NAME, taskLogger);\n  }\n\n  public debug(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().debug(message, metadata);\n  }\n\n  public log(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().log(message, metadata);\n  }\n\n  public info(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().info(message, metadata);\n  }\n\n  public warn(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().warn(message, metadata);\n  }\n\n  public error(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().error(message, metadata);\n  }\n\n  public trace<T>(name: string, fn: (span: Span) => Promise<T>): Promise<T> {\n    return this.#getTaskLogger().trace(name, fn);\n  }\n\n  #getTaskLogger(): TaskLogger {\n    return getGlobal(API_NAME) ?? NOOP_TASK_LOGGER;\n  }\n}\n","// Split module-level variable definition into separate files to allow\n// tree-shaking on each api instance.\nimport { LoggerAPI } from \"./logger\";\n/** Entrypoint for logger API */\nexport const logger = LoggerAPI.getInstance();\n","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 { clock } from \"../clock-api\";\nimport { logger } from \"../logger-api\";\nimport {\n  BatchTaskRunExecutionResult,\n  ProdChildToWorkerMessages,\n  ProdWorkerToChildMessages,\n  TaskRunContext,\n  TaskRunExecution,\n  TaskRunExecutionResult,\n} from \"../schemas\";\nimport { checkpointSafeTimeout, unboundedTimeout } from \"../utils/timers\";\nimport { ZodIpcConnection } from \"../zodIpc\";\nimport { RuntimeManager } from \"./manager\";\n\nexport type ProdRuntimeManagerOptions = {\n  waitThresholdInMs?: number;\n};\n\nexport class ProdRuntimeManager 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  _waitForDuration:\n    | { resolve: (value: \"external\") => void; reject: (err?: any) => void }\n    | undefined;\n\n  constructor(\n    private ipc: ZodIpcConnection<\n      typeof ProdWorkerToChildMessages,\n      typeof ProdChildToWorkerMessages\n    >,\n    private options: ProdRuntimeManagerOptions = {}\n  ) {}\n\n  disable(): void {\n    // do nothing\n  }\n\n  async waitForDuration(ms: number): Promise<void> {\n    const now = Date.now();\n\n    const internalTimeout = unboundedTimeout(ms, \"internal\" as const);\n    const checkpointSafeInternalTimeout = checkpointSafeTimeout(ms);\n\n    if (ms <= this.waitThresholdInMs) {\n      await internalTimeout;\n      return;\n    }\n\n    const externalResume = new Promise<\"external\">((resolve, reject) => {\n      this._waitForDuration = { resolve, reject };\n    });\n\n    const { willCheckpointAndRestore } = await this.ipc.sendWithAck(\n      \"WAIT_FOR_DURATION\",\n      {\n        ms,\n        now,\n      },\n      31_000\n    );\n\n    if (!willCheckpointAndRestore) {\n      await internalTimeout;\n      return;\n    }\n\n    this.ipc.send(\"READY_FOR_CHECKPOINT\", {});\n\n    // internalTimeout acts as a backup and will be accurate if the checkpoint never happens\n    // checkpointSafeInternalTimeout is accurate even after non-simulated restores\n    await Promise.race([internalTimeout, checkpointSafeInternalTimeout]);\n\n    // Resets the clock to the current time\n    clock.reset();\n\n    try {\n      // The coordinator should cancel any in-progress checkpoints\n      const { checkpointCanceled, version } = await this.ipc.sendWithAck(\n        \"CANCEL_CHECKPOINT\",\n        {\n          version: \"v2\",\n          reason: \"WAIT_FOR_DURATION\",\n        },\n        31_000\n      );\n\n      if (checkpointCanceled) {\n        // There won't be a checkpoint or external resume and we've already completed our internal timeout\n        return;\n      }\n    } catch (error) {\n      // If the cancellation times out, we will proceed as if the checkpoint was canceled\n      logger.debug(\"Checkpoint cancellation timed out\", { error });\n      return;\n    }\n\n    // No checkpoint was canceled, so we were checkpointed. We need to wait for the external resume message.\n    await externalResume;\n  }\n\n  resumeAfterDuration(): void {\n    if (!this._waitForDuration) {\n      return;\n    }\n\n    // Resets the clock to the current time\n    clock.reset();\n\n    this._waitForDuration.resolve(\"external\");\n    this._waitForDuration = undefined;\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 promise = new Promise<TaskRunExecutionResult>((resolve) => {\n      this._taskWaits.set(params.id, { resolve });\n    });\n\n    await this.ipc.send(\"WAIT_FOR_TASK\", {\n      friendlyId: params.id,\n    });\n\n    const result = await promise;\n\n    clock.reset();\n\n    return result;\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          this._taskWaits.set(runId, { resolve });\n        });\n      })\n    );\n\n    await this.ipc.send(\"WAIT_FOR_BATCH\", {\n      batchFriendlyId: params.id,\n      runFriendlyIds: params.runs,\n    });\n\n    const results = await promise;\n\n    clock.reset();\n\n    return {\n      id: params.id,\n      items: results,\n    };\n  }\n\n  resumeTask(completion: TaskRunExecutionResult): void {\n    const wait = this._taskWaits.get(completion.id);\n\n    if (!wait) {\n      return;\n    }\n\n    wait.resolve(completion);\n\n    this._taskWaits.delete(completion.id);\n  }\n\n  private get waitThresholdInMs(): number {\n    return this.options.waitThresholdInMs ?? 30_000;\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/v3/utils/platform.ts","../../../src/v3/utils/globals.ts","../../../src/v3/clock/simpleClock.ts","../../../src/v3/clock/index.ts","../../../src/v3/clock-api.ts","../../../src/v3/logger/taskLogger.ts","../../../src/v3/logger/index.ts","../../../src/v3/logger-api.ts","../../../src/v3/utils/timers.ts","../../../src/v3/runtime/prodRuntimeManager.ts"],"names":["_globalThis","globalThis","global","GLOBAL_TRIGGER_DOT_DEV_KEY","Symbol","for","_global","registerGlobal","type","instance","allowOverride","api","err","Error","getGlobal","unregisterGlobal","PreciseDate","SimpleClock","preciseNow","now","nowStruct","toStruct","seconds","nanos","reset","API_NAME","SIMPLE_CLOCK","ClockAPI","getInstance","_instance","setGlobalClock","clock","SeverityNumber","NoopTaskLogger","debug","log","info","warn","error","trace","name","fn","NOOP_TASK_LOGGER","LoggerAPI","disable","setGlobalTaskLogger","taskLogger","message","metadata","logger","setInterval","setTimeout","unboundedTimeout","delay","value","options","maxDelay","fullTimeouts","Math","floor","remainingDelay","lastTimeoutResult","i","checkpointSafeTimeout","scanIntervalMs","start","Date","ProdRuntimeManager","constructor","ipc","_taskWaits","Map","_batchWaits","waitForDuration","ms","internalTimeout","checkpointSafeInternalTimeout","waitThresholdInMs","externalResume","Promise","resolve","reject","_waitForDuration","willCheckpointAndRestore","sendWithAck","send","race","checkpointCanceled","version","reason","resumeAfterDuration","undefined","waitUntil","date","getTime","waitForTask","params","promise","set","id","friendlyId","result","waitForBatch","runs","length","items","all","map","runId","batchFriendlyId","runFriendlyIds","results","resumeTask","completion","wait","get","delete"],"mappings":";;;;;;;;;;;;;;;;;AAAO,IAAMA,cAAc,OAAOC,eAAe,WAAWA,aAAaC;;;ACQzE,IAAMC,6BAA6BC,OAAOC,IAAI,oBAAoB;AAElE,IAAMC,UAAUN;AAET,SAASO,eACdC,MACAC,UACAC,gBAAgB,OACP;AACT,QAAMC,MAAOL,QAAQH,0BAAAA,IAA8BG,QAAQH,0BAAAA,KAA+B,CAAC;AAE3F,MAAI,CAACO,iBAAiBC,IAAIH,IAAAA,GAAO;AAE/B,UAAMI,MAAM,IAAIC,MAAM,yDAAyDL,IAAAA,EAAM;AACrF,WAAO;EACT;AAEAG,MAAIH,IAAAA,IAAQC;AAEZ,SAAO;AACT;AAhBgBF;AAkBT,SAASO,UACdN,MAC0C;AAC1C,SAAOF,QAAQH,0BAAAA,IAA8BK,IAAAA;AAC/C;AAJgBM;AAMT,SAASC,iBAAiBP,MAAoC;AACnE,QAAMG,MAAML,QAAQH,0BAAAA;AAEpB,MAAIQ,KAAK;AACP,WAAOA,IAAIH,IAAAA;EACb;AACF;AANgBO;;;ACpChB,SAASC,mBAAmB;AAGrB,IAAMC,eAAN,MAAMA,aAAAA;EACXC,aAA+B;AAC7B,UAAMC,MAAM,IAAIH,YAAAA;AAChB,UAAMI,YAAYD,IAAIE,SAAQ;AAE9B,WAAO;MAACD,UAAUE;MAASF,UAAUG;;EACvC;EAEAC,QAAQ;EAER;AACF;AAXaP;AAAN,IAAMA,cAAN;;;ACHP,IAAMQ,WAAW;AAMjB,IAAMC,eAAe,IAAIT,YAAAA;AANzB;AAQO,IAAMU,YAAN,MAAMA,UAAAA;EAGX,cAAsB;AAsBtB;EAtBuB;EAEvB,OAAcC,cAAwB;AACpC,QAAI,CAAC,KAAKC,WAAW;AACnB,WAAKA,YAAY,IAAIF,UAAAA;IACvB;AAEA,WAAO,KAAKE;EACd;EAEOC,eAAeC,QAAuB;AAC3C,WAAOxB,eAAekB,UAAUM,MAAAA;EAClC;EAEOb,aAAwB;AAC7B,WAAO,sBAAK,wBAAL,WAAiBA,WAAU;EACpC;EAEOM,QAAc;AACnB,0BAAK,wBAAL,WAAiBA,MAAK;EACxB;AAKF;AAHE;cAAS,kCAAU;AACjB,SAAOV,UAAUW,QAAAA,KAAaC;AAChC,GAFS;AAzBEC;AAAN,IAAMA,WAAN;;;ACJA,IAAMI,QAAQJ,SAASC,YAAW;;;ACHzC,SAAiBI,sBAAsB;AAgGhC,IAAMC,kBAAN,MAAMA,gBAAAA;EACXC,QAAQ;EAAC;EACTC,MAAM;EAAC;EACPC,OAAO;EAAC;EACRC,OAAO;EAAC;EACRC,QAAQ;EAAC;EACTC,MAASC,MAAcC,IAA4C;AACjE,WAAOA,GAAG,CAAC,CAAA;EACb;AACF;AATaR;AAAN,IAAMA,iBAAN;;;AC7FP,IAAMR,YAAW;AAEjB,IAAMiB,mBAAmB,IAAIT,eAAAA;AAN7B;AAQO,IAAMU,aAAN,MAAMA,WAAAA;EAGX,cAAsB;AA0CtB;EA1CuB;EAEvB,OAAcf,cAAyB;AACrC,QAAI,CAAC,KAAKC,WAAW;AACnB,WAAKA,YAAY,IAAIc,WAAAA;IACvB;AAEA,WAAO,KAAKd;EACd;EAEOe,UAAU;AACf7B,qBAAiBU,SAAAA;EACnB;EAEOoB,oBAAoBC,YAAiC;AAC1D,WAAOvC,eAAekB,WAAUqB,UAAAA;EAClC;EAEOZ,MAAMa,SAAiBC,UAAoC;AAChE,0BAAK,kCAAL,WAAsBd,MAAMa,SAASC,QAAAA;EACvC;EAEOb,IAAIY,SAAiBC,UAAoC;AAC9D,0BAAK,kCAAL,WAAsBb,IAAIY,SAASC,QAAAA;EACrC;EAEOZ,KAAKW,SAAiBC,UAAoC;AAC/D,0BAAK,kCAAL,WAAsBZ,KAAKW,SAASC,QAAAA;EACtC;EAEOX,KAAKU,SAAiBC,UAAoC;AAC/D,0BAAK,kCAAL,WAAsBX,KAAKU,SAASC,QAAAA;EACtC;EAEOV,MAAMS,SAAiBC,UAAoC;AAChE,0BAAK,kCAAL,WAAsBV,MAAMS,SAASC,QAAAA;EACvC;EAEOT,MAASC,MAAcC,IAA4C;AACxE,WAAO,sBAAK,kCAAL,WAAsBF,MAAMC,MAAMC,EAAAA;EAC3C;AAKF;AAHE;mBAAc,kCAAe;AAC3B,SAAO3B,UAAUW,SAAAA,KAAaiB;AAChC,GAFc;AA7CHC;AAAN,IAAMA,YAAN;;;ACJA,IAAMM,SAASN,UAAUf,YAAW;;;ACH3C,SAASsB,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;AAmBtB,eAAsBW,sBAAsBV,QAAgB,GAAkB;AAC5E,QAAMW,iBAAiB;AAGvB,mBAAiBC,SAASf,YAAYc,gBAAgBE,KAAK/C,IAAG,CAAA,GAAK;AACjE,QAAI+C,KAAK/C,IAAG,IAAK8C,QAAQZ,OAAO;AAC9B;IACF;EACF;AACF;AATsBU;;;ACJf,IAAMI,sBAAN,MAAMA,oBAAAA;EAYXC,YACUC,KAIAd,UAAqC,CAAC,GAC9C;eALQc;mBAIAd;SAhBVe,aAAgF,oBAAIC,IAAAA;SAEpFC,cAGI,oBAAID,IAAAA;EAYL;EAEH3B,UAAgB;EAEhB;EAEA,MAAM6B,gBAAgBC,IAA2B;AAC/C,UAAMvD,MAAM+C,KAAK/C,IAAG;AAEpB,UAAMwD,kBAAkBvB,iBAAiBsB,IAAI,UAAA;AAC7C,UAAME,gCAAgCb,sBAAsBW,EAAAA;AAE5D,QAAIA,MAAM,KAAKG,mBAAmB;AAChC,YAAMF;AACN;IACF;AAEA,UAAMG,iBAAiB,IAAIC,QAAoB,CAACC,SAASC,WAAW;AAClE,WAAKC,mBAAmB;QAAEF;QAASC;MAAO;IAC5C,CAAA;AAEA,UAAM,EAAEE,yBAAwB,IAAK,MAAM,KAAKd,IAAIe,YAClD,qBACA;MACEV;MACAvD;IACF,GACA,IAAA;AAGF,QAAI,CAACgE,0BAA0B;AAC7B,YAAMR;AACN;IACF;AAEA,SAAKN,IAAIgB,KAAK,wBAAwB,CAAC,CAAA;AAIvC,UAAMN,QAAQO,KAAK;MAACX;MAAiBC;KAA8B;AAGnE7C,UAAMP,MAAK;AAEX,QAAI;AAEF,YAAM,EAAE+D,oBAAoBC,QAAO,IAAK,MAAM,KAAKnB,IAAIe,YACrD,qBACA;QACEI,SAAS;QACTC,QAAQ;MACV,GACA,IAAA;AAGF,UAAIF,oBAAoB;AAEtB;MACF;IACF,SAASjD,OAAO;AAEdW,aAAOf,MAAM,qCAAqC;QAAEI;MAAM,CAAA;AAC1D;IACF;AAGA,UAAMwC;EACR;EAEAY,sBAA4B;AAC1B,QAAI,CAAC,KAAKR,kBAAkB;AAC1B;IACF;AAGAnD,UAAMP,MAAK;AAEX,SAAK0D,iBAAiBF,QAAQ,UAAA;AAC9B,SAAKE,mBAAmBS;EAC1B;EAEA,MAAMC,UAAUC,MAA2B;AACzC,WAAO,KAAKpB,gBAAgBoB,KAAKC,QAAO,IAAK5B,KAAK/C,IAAG,CAAA;EACvD;EAEA,MAAM4E,YAAYC,QAA8E;AAC9F,UAAMC,UAAU,IAAIlB,QAAgC,CAACC,YAAY;AAC/D,WAAKV,WAAW4B,IAAIF,OAAOG,IAAI;QAAEnB;MAAQ,CAAA;IAC3C,CAAA;AAEA,UAAM,KAAKX,IAAIgB,KAAK,iBAAiB;MACnCe,YAAYJ,OAAOG;IACrB,CAAA;AAEA,UAAME,SAAS,MAAMJ;AAErBlE,UAAMP,MAAK;AAEX,WAAO6E;EACT;EAEA,MAAMC,aAAaN,QAIsB;AACvC,QAAI,CAACA,OAAOO,KAAKC,QAAQ;AACvB,aAAOzB,QAAQC,QAAQ;QAAEmB,IAAIH,OAAOG;QAAIM,OAAO,CAAA;MAAG,CAAA;IACpD;AAEA,UAAMR,UAAUlB,QAAQ2B,IACtBV,OAAOO,KAAKI,IAAI,CAACC,UAAU;AACzB,aAAO,IAAI7B,QAAgC,CAACC,SAASC,WAAW;AAC9D,aAAKX,WAAW4B,IAAIU,OAAO;UAAE5B;QAAQ,CAAA;MACvC,CAAA;IACF,CAAA,CAAA;AAGF,UAAM,KAAKX,IAAIgB,KAAK,kBAAkB;MACpCwB,iBAAiBb,OAAOG;MACxBW,gBAAgBd,OAAOO;IACzB,CAAA;AAEA,UAAMQ,UAAU,MAAMd;AAEtBlE,UAAMP,MAAK;AAEX,WAAO;MACL2E,IAAIH,OAAOG;MACXM,OAAOM;IACT;EACF;EAEAC,WAAWC,YAA0C;AACnD,UAAMC,OAAO,KAAK5C,WAAW6C,IAAIF,WAAWd,EAAE;AAE9C,QAAI,CAACe,MAAM;AACT;IACF;AAEAA,SAAKlC,QAAQiC,UAAAA;AAEb,SAAK3C,WAAW8C,OAAOH,WAAWd,EAAE;EACtC;EAEA,IAAYtB,oBAA4B;AACtC,WAAO,KAAKtB,QAAQsB,qBAAqB;EAC3C;AACF;AAtKaV;AAAN,IAAMA,qBAAN","sourcesContent":["export const _globalThis = typeof globalThis === \"object\" ? globalThis : global;\n","import { ApiClientConfiguration } from \"../apiClientManager/types\";\nimport { Clock } from \"../clock/clock\";\nimport type { RuntimeManager } from \"../runtime/manager\";\nimport { TaskCatalog } from \"../task-catalog/catalog\";\nimport { TaskContext } from \"../taskContext/types\";\nimport { UsageManager } from \"../usage/types\";\nimport { _globalThis } from \"./platform\";\n\nconst GLOBAL_TRIGGER_DOT_DEV_KEY = Symbol.for(`dev.trigger.ts.api`);\n\nconst _global = _globalThis as TriggerDotDevGlobal;\n\nexport function registerGlobal<Type extends keyof TriggerDotDevGlobalAPI>(\n  type: Type,\n  instance: TriggerDotDevGlobalAPI[Type],\n  allowOverride = false\n): boolean {\n  const api = (_global[GLOBAL_TRIGGER_DOT_DEV_KEY] = _global[GLOBAL_TRIGGER_DOT_DEV_KEY] ?? {});\n\n  if (!allowOverride && api[type]) {\n    // already registered an API of this type\n    const err = new Error(`trigger.dev: Attempted duplicate registration of API: ${type}`);\n    return false;\n  }\n\n  api[type] = instance;\n\n  return true;\n}\n\nexport function getGlobal<Type extends keyof TriggerDotDevGlobalAPI>(\n  type: Type\n): TriggerDotDevGlobalAPI[Type] | undefined {\n  return _global[GLOBAL_TRIGGER_DOT_DEV_KEY]?.[type];\n}\n\nexport function unregisterGlobal(type: keyof TriggerDotDevGlobalAPI) {\n  const api = _global[GLOBAL_TRIGGER_DOT_DEV_KEY];\n\n  if (api) {\n    delete api[type];\n  }\n}\n\ntype TriggerDotDevGlobal = {\n  [GLOBAL_TRIGGER_DOT_DEV_KEY]?: TriggerDotDevGlobalAPI;\n};\n\ntype TriggerDotDevGlobalAPI = {\n  runtime?: RuntimeManager;\n  logger?: any;\n  clock?: Clock;\n  usage?: UsageManager;\n  [\"task-catalog\"]?: TaskCatalog;\n  [\"task-context\"]?: TaskContext;\n  [\"api-client\"]?: ApiClientConfiguration;\n};\n","import { PreciseDate } from \"@google-cloud/precise-date\";\nimport { Clock } from \"./clock\";\n\nexport class SimpleClock implements Clock {\n  preciseNow(): [number, number] {\n    const now = new PreciseDate();\n    const nowStruct = now.toStruct();\n\n    return [nowStruct.seconds, nowStruct.nanos];\n  }\n\n  reset() {\n    // do nothing\n  }\n}\n","const API_NAME = \"clock\";\n\nimport { getGlobal, registerGlobal } from \"../utils/globals\";\nimport type { Clock, ClockTime } from \"./clock\";\nimport { SimpleClock } from \"./simpleClock\";\n\nconst SIMPLE_CLOCK = new SimpleClock();\n\nexport class ClockAPI {\n  private static _instance?: ClockAPI;\n\n  private constructor() {}\n\n  public static getInstance(): ClockAPI {\n    if (!this._instance) {\n      this._instance = new ClockAPI();\n    }\n\n    return this._instance;\n  }\n\n  public setGlobalClock(clock: Clock): boolean {\n    return registerGlobal(API_NAME, clock);\n  }\n\n  public preciseNow(): ClockTime {\n    return this.#getClock().preciseNow();\n  }\n\n  public reset(): void {\n    this.#getClock().reset();\n  }\n\n  #getClock(): Clock {\n    return getGlobal(API_NAME) ?? SIMPLE_CLOCK;\n  }\n}\n","// Split module-level variable definition into separate files to allow\n// tree-shaking on each api instance.\nimport { ClockAPI } from \"./clock\";\n/** Entrypoint for clock API */\nexport const clock = ClockAPI.getInstance();\n","import { Attributes, Span, SpanOptions } from \"@opentelemetry/api\";\nimport { Logger, SeverityNumber } from \"@opentelemetry/api-logs\";\nimport { iconStringForSeverity } from \"../icons\";\nimport { SemanticInternalAttributes } from \"../semanticInternalAttributes\";\nimport { TriggerTracer } from \"../tracer\";\nimport { flattenAttributes } from \"../utils/flattenAttributes\";\nimport { ClockTime } from \"../clock/clock\";\nimport { clock } from \"../clock-api\";\n\nexport type LogLevel = \"none\" | \"error\" | \"warn\" | \"info\" | \"debug\" | \"log\";\n\nexport const logLevels: Array<LogLevel> = [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\nexport type TaskLoggerConfig = {\n  logger: Logger;\n  tracer: TriggerTracer;\n  level: LogLevel;\n};\n\nexport interface TaskLogger {\n  debug(message: string, properties?: Record<string, unknown>): void;\n  log(message: string, properties?: Record<string, unknown>): void;\n  info(message: string, properties?: Record<string, unknown>): void;\n  warn(message: string, properties?: Record<string, unknown>): void;\n  error(message: string, properties?: Record<string, unknown>): void;\n  trace<T>(name: string, fn: (span: Span) => Promise<T>, options?: SpanOptions): Promise<T>;\n}\n\nexport class OtelTaskLogger implements TaskLogger {\n  private readonly _level: number;\n\n  constructor(private readonly _config: TaskLoggerConfig) {\n    this._level = logLevels.indexOf(_config.level);\n  }\n\n  debug(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 4) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"debug\", SeverityNumber.DEBUG, properties);\n  }\n\n  log(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 3) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"log\", SeverityNumber.INFO, properties);\n  }\n\n  info(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 3) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"info\", SeverityNumber.INFO, properties);\n  }\n\n  warn(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 2) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"warn\", SeverityNumber.WARN, properties);\n  }\n\n  error(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 1) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"error\", SeverityNumber.ERROR, properties);\n  }\n\n  #emitLog(\n    message: string,\n    timestamp: ClockTime,\n    severityText: string,\n    severityNumber: SeverityNumber,\n    properties?: Record<string, unknown>\n  ) {\n    let attributes: Attributes = { ...flattenAttributes(safeJsonProcess(properties)) };\n\n    const icon = iconStringForSeverity(severityNumber);\n    if (icon !== undefined) {\n      attributes[SemanticInternalAttributes.STYLE_ICON] = icon;\n    }\n\n    this._config.logger.emit({\n      severityNumber,\n      severityText,\n      body: message,\n      attributes,\n      timestamp,\n    });\n  }\n\n  trace<T>(name: string, fn: (span: Span) => Promise<T>, options?: SpanOptions): Promise<T> {\n    return this._config.tracer.startActiveSpan(name, fn, options);\n  }\n\n  #getTimestampInHrTime(): ClockTime {\n    return clock.preciseNow();\n  }\n}\n\nexport class NoopTaskLogger implements TaskLogger {\n  debug() {}\n  log() {}\n  info() {}\n  warn() {}\n  error() {}\n  trace<T>(name: string, fn: (span: Span) => Promise<T>): Promise<T> {\n    return fn({} as Span);\n  }\n}\n\nfunction safeJsonProcess(value?: Record<string, unknown>): Record<string, unknown> | undefined {\n  try {\n    return JSON.parse(JSON.stringify(value, jsonErrorReplacer));\n  } catch {\n    return value;\n  }\n}\n\nfunction jsonErrorReplacer(key: string, value: unknown) {\n  if (value instanceof Error) {\n    return {\n      name: value.name,\n      message: value.message,\n      stack: value.stack,\n    };\n  }\n\n  return value;\n}\n","import { NoopTaskLogger, TaskLogger } from \"./taskLogger\";\nimport { getGlobal, registerGlobal, unregisterGlobal } from \"../utils/globals\";\nimport { Span } from \"@opentelemetry/api\";\n\nconst API_NAME = \"logger\";\n\nconst NOOP_TASK_LOGGER = new NoopTaskLogger();\n\nexport class LoggerAPI implements TaskLogger {\n  private static _instance?: LoggerAPI;\n\n  private constructor() {}\n\n  public static getInstance(): LoggerAPI {\n    if (!this._instance) {\n      this._instance = new LoggerAPI();\n    }\n\n    return this._instance;\n  }\n\n  public disable() {\n    unregisterGlobal(API_NAME);\n  }\n\n  public setGlobalTaskLogger(taskLogger: TaskLogger): boolean {\n    return registerGlobal(API_NAME, taskLogger);\n  }\n\n  public debug(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().debug(message, metadata);\n  }\n\n  public log(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().log(message, metadata);\n  }\n\n  public info(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().info(message, metadata);\n  }\n\n  public warn(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().warn(message, metadata);\n  }\n\n  public error(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().error(message, metadata);\n  }\n\n  public trace<T>(name: string, fn: (span: Span) => Promise<T>): Promise<T> {\n    return this.#getTaskLogger().trace(name, fn);\n  }\n\n  #getTaskLogger(): TaskLogger {\n    return getGlobal(API_NAME) ?? NOOP_TASK_LOGGER;\n  }\n}\n","// Split module-level variable definition into separate files to allow\n// tree-shaking on each api instance.\nimport { LoggerAPI } from \"./logger\";\n/** Entrypoint for logger API */\nexport const logger = LoggerAPI.getInstance();\n","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 { clock } from \"../clock-api\";\nimport { logger } from \"../logger-api\";\nimport {\n  BatchTaskRunExecutionResult,\n  ProdChildToWorkerMessages,\n  ProdWorkerToChildMessages,\n  TaskRunContext,\n  TaskRunExecution,\n  TaskRunExecutionResult,\n} from \"../schemas\";\nimport { checkpointSafeTimeout, unboundedTimeout } from \"../utils/timers\";\nimport { ZodIpcConnection } from \"../zodIpc\";\nimport { RuntimeManager } from \"./manager\";\n\nexport type ProdRuntimeManagerOptions = {\n  waitThresholdInMs?: number;\n};\n\nexport class ProdRuntimeManager 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  _waitForDuration:\n    | { resolve: (value: \"external\") => void; reject: (err?: any) => void }\n    | undefined;\n\n  constructor(\n    private ipc: ZodIpcConnection<\n      typeof ProdWorkerToChildMessages,\n      typeof ProdChildToWorkerMessages\n    >,\n    private options: ProdRuntimeManagerOptions = {}\n  ) {}\n\n  disable(): void {\n    // do nothing\n  }\n\n  async waitForDuration(ms: number): Promise<void> {\n    const now = Date.now();\n\n    const internalTimeout = unboundedTimeout(ms, \"internal\" as const);\n    const checkpointSafeInternalTimeout = checkpointSafeTimeout(ms);\n\n    if (ms <= this.waitThresholdInMs) {\n      await internalTimeout;\n      return;\n    }\n\n    const externalResume = new Promise<\"external\">((resolve, reject) => {\n      this._waitForDuration = { resolve, reject };\n    });\n\n    const { willCheckpointAndRestore } = await this.ipc.sendWithAck(\n      \"WAIT_FOR_DURATION\",\n      {\n        ms,\n        now,\n      },\n      31_000\n    );\n\n    if (!willCheckpointAndRestore) {\n      await internalTimeout;\n      return;\n    }\n\n    this.ipc.send(\"READY_FOR_CHECKPOINT\", {});\n\n    // internalTimeout acts as a backup and will be accurate if the checkpoint never happens\n    // checkpointSafeInternalTimeout is accurate even after non-simulated restores\n    await Promise.race([internalTimeout, checkpointSafeInternalTimeout]);\n\n    // Resets the clock to the current time\n    clock.reset();\n\n    try {\n      // The coordinator should cancel any in-progress checkpoints\n      const { checkpointCanceled, version } = await this.ipc.sendWithAck(\n        \"CANCEL_CHECKPOINT\",\n        {\n          version: \"v2\",\n          reason: \"WAIT_FOR_DURATION\",\n        },\n        31_000\n      );\n\n      if (checkpointCanceled) {\n        // There won't be a checkpoint or external resume and we've already completed our internal timeout\n        return;\n      }\n    } catch (error) {\n      // If the cancellation times out, we will proceed as if the checkpoint was canceled\n      logger.debug(\"Checkpoint cancellation timed out\", { error });\n      return;\n    }\n\n    // No checkpoint was canceled, so we were checkpointed. We need to wait for the external resume message.\n    await externalResume;\n  }\n\n  resumeAfterDuration(): void {\n    if (!this._waitForDuration) {\n      return;\n    }\n\n    // Resets the clock to the current time\n    clock.reset();\n\n    this._waitForDuration.resolve(\"external\");\n    this._waitForDuration = undefined;\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 promise = new Promise<TaskRunExecutionResult>((resolve) => {\n      this._taskWaits.set(params.id, { resolve });\n    });\n\n    await this.ipc.send(\"WAIT_FOR_TASK\", {\n      friendlyId: params.id,\n    });\n\n    const result = await promise;\n\n    clock.reset();\n\n    return result;\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          this._taskWaits.set(runId, { resolve });\n        });\n      })\n    );\n\n    await this.ipc.send(\"WAIT_FOR_BATCH\", {\n      batchFriendlyId: params.id,\n      runFriendlyIds: params.runs,\n    });\n\n    const results = await promise;\n\n    clock.reset();\n\n    return {\n      id: params.id,\n      items: results,\n    };\n  }\n\n  resumeTask(completion: TaskRunExecutionResult): void {\n    const wait = this._taskWaits.get(completion.id);\n\n    if (!wait) {\n      return;\n    }\n\n    wait.resolve(completion);\n\n    this._taskWaits.delete(completion.id);\n  }\n\n  private get waitThresholdInMs(): number {\n    return this.options.waitThresholdInMs ?? 30_000;\n  }\n}\n"]}

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/v3/utils/platform.ts","../../../src/v3/utils/globals.ts","../../../src/v3/clock/simpleClock.ts","../../../src/v3/clock/index.ts","../../../src/v3/clock-api.ts","../../../src/v3/logger/taskLogger.ts","../../../src/v3/logger/index.ts","../../../src/v3/logger-api.ts","../../../src/v3/utils/timers.ts","../../../src/v3/runtime/prodRuntimeManager.ts"],"names":["_globalThis","globalThis","global","GLOBAL_TRIGGER_DOT_DEV_KEY","Symbol","for","_global","registerGlobal","type","instance","allowOverride","api","err","Error","getGlobal","unregisterGlobal","PreciseDate","SimpleClock","preciseNow","now","nowStruct","toStruct","seconds","nanos","reset","API_NAME","SIMPLE_CLOCK","ClockAPI","getInstance","_instance","setGlobalClock","clock","SeverityNumber","NoopTaskLogger","debug","log","info","warn","error","trace","name","fn","NOOP_TASK_LOGGER","LoggerAPI","disable","setGlobalTaskLogger","taskLogger","message","metadata","logger","setInterval","setTimeout","unboundedTimeout","delay","value","options","maxDelay","fullTimeouts","Math","floor","remainingDelay","lastTimeoutResult","i","checkpointSafeTimeout","scanIntervalMs","start","Date","ProdRuntimeManager","constructor","ipc","_taskWaits","Map","_batchWaits","waitForDuration","ms","internalTimeout","checkpointSafeInternalTimeout","waitThresholdInMs","externalResume","Promise","resolve","reject","_waitForDuration","willCheckpointAndRestore","sendWithAck","send","race","checkpointCanceled","version","reason","resumeAfterDuration","undefined","waitUntil","date","getTime","waitForTask","params","promise","set","id","friendlyId","result","waitForBatch","runs","length","items","all","map","runId","batchFriendlyId","runFriendlyIds","results","resumeTask","completion","wait","get","delete"],"mappings":";;;;;;;;;;;;;;;;;AAAO,IAAMA,cAAc,OAAOC,eAAe,WAAWA,aAAaC;;;ACOzE,IAAMC,6BAA6BC,OAAOC,IAAI,oBAAoB;AAElE,IAAMC,UAAUN;AAET,SAASO,eACdC,MACAC,UACAC,gBAAgB,OACP;AACT,QAAMC,MAAOL,QAAQH,0BAAAA,IAA8BG,QAAQH,0BAAAA,KAA+B,CAAC;AAE3F,MAAI,CAACO,iBAAiBC,IAAIH,IAAAA,GAAO;AAE/B,UAAMI,MAAM,IAAIC,MAAM,yDAAyDL,IAAAA,EAAM;AACrF,WAAO;EACT;AAEAG,MAAIH,IAAAA,IAAQC;AAEZ,SAAO;AACT;AAhBgBF;AAkBT,SAASO,UACdN,MAC0C;AAC1C,SAAOF,QAAQH,0BAAAA,IAA8BK,IAAAA;AAC/C;AAJgBM;AAMT,SAASC,iBAAiBP,MAAoC;AACnE,QAAMG,MAAML,QAAQH,0BAAAA;AAEpB,MAAIQ,KAAK;AACP,WAAOA,IAAIH,IAAAA;EACb;AACF;AANgBO;;;ACnChB,SAASC,mBAAmB;AAGrB,IAAMC,eAAN,MAAMA,aAAAA;EACXC,aAA+B;AAC7B,UAAMC,MAAM,IAAIH,YAAAA;AAChB,UAAMI,YAAYD,IAAIE,SAAQ;AAE9B,WAAO;MAACD,UAAUE;MAASF,UAAUG;;EACvC;EAEAC,QAAQ;EAER;AACF;AAXaP;AAAN,IAAMA,cAAN;;;ACHP,IAAMQ,WAAW;AAMjB,IAAMC,eAAe,IAAIT,YAAAA;AANzB;AAQO,IAAMU,YAAN,MAAMA,UAAAA;EAGX,cAAsB;AAsBtB;EAtBuB;EAEvB,OAAcC,cAAwB;AACpC,QAAI,CAAC,KAAKC,WAAW;AACnB,WAAKA,YAAY,IAAIF,UAAAA;IACvB;AAEA,WAAO,KAAKE;EACd;EAEOC,eAAeC,QAAuB;AAC3C,WAAOxB,eAAekB,UAAUM,MAAAA;EAClC;EAEOb,aAAwB;AAC7B,WAAO,sBAAK,wBAAL,WAAiBA,WAAU;EACpC;EAEOM,QAAc;AACnB,0BAAK,wBAAL,WAAiBA,MAAK;EACxB;AAKF;AAHE;cAAS,kCAAU;AACjB,SAAOV,UAAUW,QAAAA,KAAaC;AAChC,GAFS;AAzBEC;AAAN,IAAMA,WAAN;;;ACJA,IAAMI,QAAQJ,SAASC,YAAW;;;ACHzC,SAAiBI,sBAAsB;AAgGhC,IAAMC,kBAAN,MAAMA,gBAAAA;EACXC,QAAQ;EAAC;EACTC,MAAM;EAAC;EACPC,OAAO;EAAC;EACRC,OAAO;EAAC;EACRC,QAAQ;EAAC;EACTC,MAASC,MAAcC,IAA4C;AACjE,WAAOA,GAAG,CAAC,CAAA;EACb;AACF;AATaR;AAAN,IAAMA,iBAAN;;;AC7FP,IAAMR,YAAW;AAEjB,IAAMiB,mBAAmB,IAAIT,eAAAA;AAN7B;AAQO,IAAMU,aAAN,MAAMA,WAAAA;EAGX,cAAsB;AA0CtB;EA1CuB;EAEvB,OAAcf,cAAyB;AACrC,QAAI,CAAC,KAAKC,WAAW;AACnB,WAAKA,YAAY,IAAIc,WAAAA;IACvB;AAEA,WAAO,KAAKd;EACd;EAEOe,UAAU;AACf7B,qBAAiBU,SAAAA;EACnB;EAEOoB,oBAAoBC,YAAiC;AAC1D,WAAOvC,eAAekB,WAAUqB,UAAAA;EAClC;EAEOZ,MAAMa,SAAiBC,UAAoC;AAChE,0BAAK,kCAAL,WAAsBd,MAAMa,SAASC,QAAAA;EACvC;EAEOb,IAAIY,SAAiBC,UAAoC;AAC9D,0BAAK,kCAAL,WAAsBb,IAAIY,SAASC,QAAAA;EACrC;EAEOZ,KAAKW,SAAiBC,UAAoC;AAC/D,0BAAK,kCAAL,WAAsBZ,KAAKW,SAASC,QAAAA;EACtC;EAEOX,KAAKU,SAAiBC,UAAoC;AAC/D,0BAAK,kCAAL,WAAsBX,KAAKU,SAASC,QAAAA;EACtC;EAEOV,MAAMS,SAAiBC,UAAoC;AAChE,0BAAK,kCAAL,WAAsBV,MAAMS,SAASC,QAAAA;EACvC;EAEOT,MAASC,MAAcC,IAA4C;AACxE,WAAO,sBAAK,kCAAL,WAAsBF,MAAMC,MAAMC,EAAAA;EAC3C;AAKF;AAHE;mBAAc,kCAAe;AAC3B,SAAO3B,UAAUW,SAAAA,KAAaiB;AAChC,GAFc;AA7CHC;AAAN,IAAMA,YAAN;;;ACJA,IAAMM,SAASN,UAAUf,YAAW;;;ACH3C,SAASsB,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;AAmBtB,eAAsBW,sBAAsBV,QAAgB,GAAkB;AAC5E,QAAMW,iBAAiB;AAGvB,mBAAiBC,SAASf,YAAYc,gBAAgBE,KAAK/C,IAAG,CAAA,GAAK;AACjE,QAAI+C,KAAK/C,IAAG,IAAK8C,QAAQZ,OAAO;AAC9B;IACF;EACF;AACF;AATsBU;;;ACJf,IAAMI,sBAAN,MAAMA,oBAAAA;EAYXC,YACUC,KAIAd,UAAqC,CAAC,GAC9C;eALQc;mBAIAd;SAhBVe,aAAgF,oBAAIC,IAAAA;SAEpFC,cAGI,oBAAID,IAAAA;EAYL;EAEH3B,UAAgB;EAEhB;EAEA,MAAM6B,gBAAgBC,IAA2B;AAC/C,UAAMvD,MAAM+C,KAAK/C,IAAG;AAEpB,UAAMwD,kBAAkBvB,iBAAiBsB,IAAI,UAAA;AAC7C,UAAME,gCAAgCb,sBAAsBW,EAAAA;AAE5D,QAAIA,MAAM,KAAKG,mBAAmB;AAChC,YAAMF;AACN;IACF;AAEA,UAAMG,iBAAiB,IAAIC,QAAoB,CAACC,SAASC,WAAW;AAClE,WAAKC,mBAAmB;QAAEF;QAASC;MAAO;IAC5C,CAAA;AAEA,UAAM,EAAEE,yBAAwB,IAAK,MAAM,KAAKd,IAAIe,YAClD,qBACA;MACEV;MACAvD;IACF,GACA,IAAA;AAGF,QAAI,CAACgE,0BAA0B;AAC7B,YAAMR;AACN;IACF;AAEA,SAAKN,IAAIgB,KAAK,wBAAwB,CAAC,CAAA;AAIvC,UAAMN,QAAQO,KAAK;MAACX;MAAiBC;KAA8B;AAGnE7C,UAAMP,MAAK;AAEX,QAAI;AAEF,YAAM,EAAE+D,oBAAoBC,QAAO,IAAK,MAAM,KAAKnB,IAAIe,YACrD,qBACA;QACEI,SAAS;QACTC,QAAQ;MACV,GACA,IAAA;AAGF,UAAIF,oBAAoB;AAEtB;MACF;IACF,SAASjD,OAAO;AAEdW,aAAOf,MAAM,qCAAqC;QAAEI;MAAM,CAAA;AAC1D;IACF;AAGA,UAAMwC;EACR;EAEAY,sBAA4B;AAC1B,QAAI,CAAC,KAAKR,kBAAkB;AAC1B;IACF;AAGAnD,UAAMP,MAAK;AAEX,SAAK0D,iBAAiBF,QAAQ,UAAA;AAC9B,SAAKE,mBAAmBS;EAC1B;EAEA,MAAMC,UAAUC,MAA2B;AACzC,WAAO,KAAKpB,gBAAgBoB,KAAKC,QAAO,IAAK5B,KAAK/C,IAAG,CAAA;EACvD;EAEA,MAAM4E,YAAYC,QAA8E;AAC9F,UAAMC,UAAU,IAAIlB,QAAgC,CAACC,YAAY;AAC/D,WAAKV,WAAW4B,IAAIF,OAAOG,IAAI;QAAEnB;MAAQ,CAAA;IAC3C,CAAA;AAEA,UAAM,KAAKX,IAAIgB,KAAK,iBAAiB;MACnCe,YAAYJ,OAAOG;IACrB,CAAA;AAEA,UAAME,SAAS,MAAMJ;AAErBlE,UAAMP,MAAK;AAEX,WAAO6E;EACT;EAEA,MAAMC,aAAaN,QAIsB;AACvC,QAAI,CAACA,OAAOO,KAAKC,QAAQ;AACvB,aAAOzB,QAAQC,QAAQ;QAAEmB,IAAIH,OAAOG;QAAIM,OAAO,CAAA;MAAG,CAAA;IACpD;AAEA,UAAMR,UAAUlB,QAAQ2B,IACtBV,OAAOO,KAAKI,IAAI,CAACC,UAAU;AACzB,aAAO,IAAI7B,QAAgC,CAACC,SAASC,WAAW;AAC9D,aAAKX,WAAW4B,IAAIU,OAAO;UAAE5B;QAAQ,CAAA;MACvC,CAAA;IACF,CAAA,CAAA;AAGF,UAAM,KAAKX,IAAIgB,KAAK,kBAAkB;MACpCwB,iBAAiBb,OAAOG;MACxBW,gBAAgBd,OAAOO;IACzB,CAAA;AAEA,UAAMQ,UAAU,MAAMd;AAEtBlE,UAAMP,MAAK;AAEX,WAAO;MACL2E,IAAIH,OAAOG;MACXM,OAAOM;IACT;EACF;EAEAC,WAAWC,YAA0C;AACnD,UAAMC,OAAO,KAAK5C,WAAW6C,IAAIF,WAAWd,EAAE;AAE9C,QAAI,CAACe,MAAM;AACT;IACF;AAEAA,SAAKlC,QAAQiC,UAAAA;AAEb,SAAK3C,WAAW8C,OAAOH,WAAWd,EAAE;EACtC;EAEA,IAAYtB,oBAA4B;AACtC,WAAO,KAAKtB,QAAQsB,qBAAqB;EAC3C;AACF;AAtKaV;AAAN,IAAMA,qBAAN","sourcesContent":["export const _globalThis = typeof globalThis === \"object\" ? globalThis : global;\n","import { ApiClientConfiguration } from \"../apiClientManager/types\";\nimport { Clock } from \"../clock/clock\";\nimport type { RuntimeManager } from \"../runtime/manager\";\nimport { TaskCatalog } from \"../task-catalog/catalog\";\nimport { TaskContext } from \"../taskContext/types\";\nimport { _globalThis } from \"./platform\";\n\nconst GLOBAL_TRIGGER_DOT_DEV_KEY = Symbol.for(`dev.trigger.ts.api`);\n\nconst _global = _globalThis as TriggerDotDevGlobal;\n\nexport function registerGlobal<Type extends keyof TriggerDotDevGlobalAPI>(\n  type: Type,\n  instance: TriggerDotDevGlobalAPI[Type],\n  allowOverride = false\n): boolean {\n  const api = (_global[GLOBAL_TRIGGER_DOT_DEV_KEY] = _global[GLOBAL_TRIGGER_DOT_DEV_KEY] ?? {});\n\n  if (!allowOverride && api[type]) {\n    // already registered an API of this type\n    const err = new Error(`trigger.dev: Attempted duplicate registration of API: ${type}`);\n    return false;\n  }\n\n  api[type] = instance;\n\n  return true;\n}\n\nexport function getGlobal<Type extends keyof TriggerDotDevGlobalAPI>(\n  type: Type\n): TriggerDotDevGlobalAPI[Type] | undefined {\n  return _global[GLOBAL_TRIGGER_DOT_DEV_KEY]?.[type];\n}\n\nexport function unregisterGlobal(type: keyof TriggerDotDevGlobalAPI) {\n  const api = _global[GLOBAL_TRIGGER_DOT_DEV_KEY];\n\n  if (api) {\n    delete api[type];\n  }\n}\n\ntype TriggerDotDevGlobal = {\n  [GLOBAL_TRIGGER_DOT_DEV_KEY]?: TriggerDotDevGlobalAPI;\n};\n\ntype TriggerDotDevGlobalAPI = {\n  runtime?: RuntimeManager;\n  logger?: any;\n  clock?: Clock;\n  [\"task-catalog\"]?: TaskCatalog;\n  [\"task-context\"]?: TaskContext;\n  [\"api-client\"]?: ApiClientConfiguration;\n};\n","import { PreciseDate } from \"@google-cloud/precise-date\";\nimport { Clock } from \"./clock\";\n\nexport class SimpleClock implements Clock {\n  preciseNow(): [number, number] {\n    const now = new PreciseDate();\n    const nowStruct = now.toStruct();\n\n    return [nowStruct.seconds, nowStruct.nanos];\n  }\n\n  reset() {\n    // do nothing\n  }\n}\n","const API_NAME = \"clock\";\n\nimport { getGlobal, registerGlobal } from \"../utils/globals\";\nimport type { Clock, ClockTime } from \"./clock\";\nimport { SimpleClock } from \"./simpleClock\";\n\nconst SIMPLE_CLOCK = new SimpleClock();\n\nexport class ClockAPI {\n  private static _instance?: ClockAPI;\n\n  private constructor() {}\n\n  public static getInstance(): ClockAPI {\n    if (!this._instance) {\n      this._instance = new ClockAPI();\n    }\n\n    return this._instance;\n  }\n\n  public setGlobalClock(clock: Clock): boolean {\n    return registerGlobal(API_NAME, clock);\n  }\n\n  public preciseNow(): ClockTime {\n    return this.#getClock().preciseNow();\n  }\n\n  public reset(): void {\n    this.#getClock().reset();\n  }\n\n  #getClock(): Clock {\n    return getGlobal(API_NAME) ?? SIMPLE_CLOCK;\n  }\n}\n","// Split module-level variable definition into separate files to allow\n// tree-shaking on each api instance.\nimport { ClockAPI } from \"./clock\";\n/** Entrypoint for clock API */\nexport const clock = ClockAPI.getInstance();\n","import { Attributes, Span, SpanOptions } from \"@opentelemetry/api\";\nimport { Logger, SeverityNumber } from \"@opentelemetry/api-logs\";\nimport { iconStringForSeverity } from \"../icons\";\nimport { SemanticInternalAttributes } from \"../semanticInternalAttributes\";\nimport { TriggerTracer } from \"../tracer\";\nimport { flattenAttributes } from \"../utils/flattenAttributes\";\nimport { ClockTime } from \"../clock/clock\";\nimport { clock } from \"../clock-api\";\n\nexport type LogLevel = \"none\" | \"error\" | \"warn\" | \"info\" | \"debug\" | \"log\";\n\nexport const logLevels: Array<LogLevel> = [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\nexport type TaskLoggerConfig = {\n  logger: Logger;\n  tracer: TriggerTracer;\n  level: LogLevel;\n};\n\nexport interface TaskLogger {\n  debug(message: string, properties?: Record<string, unknown>): void;\n  log(message: string, properties?: Record<string, unknown>): void;\n  info(message: string, properties?: Record<string, unknown>): void;\n  warn(message: string, properties?: Record<string, unknown>): void;\n  error(message: string, properties?: Record<string, unknown>): void;\n  trace<T>(name: string, fn: (span: Span) => Promise<T>, options?: SpanOptions): Promise<T>;\n}\n\nexport class OtelTaskLogger implements TaskLogger {\n  private readonly _level: number;\n\n  constructor(private readonly _config: TaskLoggerConfig) {\n    this._level = logLevels.indexOf(_config.level);\n  }\n\n  debug(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 4) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"debug\", SeverityNumber.DEBUG, properties);\n  }\n\n  log(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 3) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"log\", SeverityNumber.INFO, properties);\n  }\n\n  info(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 3) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"info\", SeverityNumber.INFO, properties);\n  }\n\n  warn(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 2) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"warn\", SeverityNumber.WARN, properties);\n  }\n\n  error(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 1) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"error\", SeverityNumber.ERROR, properties);\n  }\n\n  #emitLog(\n    message: string,\n    timestamp: ClockTime,\n    severityText: string,\n    severityNumber: SeverityNumber,\n    properties?: Record<string, unknown>\n  ) {\n    let attributes: Attributes = { ...flattenAttributes(safeJsonProcess(properties)) };\n\n    const icon = iconStringForSeverity(severityNumber);\n    if (icon !== undefined) {\n      attributes[SemanticInternalAttributes.STYLE_ICON] = icon;\n    }\n\n    this._config.logger.emit({\n      severityNumber,\n      severityText,\n      body: message,\n      attributes,\n      timestamp,\n    });\n  }\n\n  trace<T>(name: string, fn: (span: Span) => Promise<T>, options?: SpanOptions): Promise<T> {\n    return this._config.tracer.startActiveSpan(name, fn, options);\n  }\n\n  #getTimestampInHrTime(): ClockTime {\n    return clock.preciseNow();\n  }\n}\n\nexport class NoopTaskLogger implements TaskLogger {\n  debug() {}\n  log() {}\n  info() {}\n  warn() {}\n  error() {}\n  trace<T>(name: string, fn: (span: Span) => Promise<T>): Promise<T> {\n    return fn({} as Span);\n  }\n}\n\nfunction safeJsonProcess(value?: Record<string, unknown>): Record<string, unknown> | undefined {\n  try {\n    return JSON.parse(JSON.stringify(value, jsonErrorReplacer));\n  } catch {\n    return value;\n  }\n}\n\nfunction jsonErrorReplacer(key: string, value: unknown) {\n  if (value instanceof Error) {\n    return {\n      name: value.name,\n      message: value.message,\n      stack: value.stack,\n    };\n  }\n\n  return value;\n}\n","import { NoopTaskLogger, TaskLogger } from \"./taskLogger\";\nimport { getGlobal, registerGlobal, unregisterGlobal } from \"../utils/globals\";\nimport { Span } from \"@opentelemetry/api\";\n\nconst API_NAME = \"logger\";\n\nconst NOOP_TASK_LOGGER = new NoopTaskLogger();\n\nexport class LoggerAPI implements TaskLogger {\n  private static _instance?: LoggerAPI;\n\n  private constructor() {}\n\n  public static getInstance(): LoggerAPI {\n    if (!this._instance) {\n      this._instance = new LoggerAPI();\n    }\n\n    return this._instance;\n  }\n\n  public disable() {\n    unregisterGlobal(API_NAME);\n  }\n\n  public setGlobalTaskLogger(taskLogger: TaskLogger): boolean {\n    return registerGlobal(API_NAME, taskLogger);\n  }\n\n  public debug(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().debug(message, metadata);\n  }\n\n  public log(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().log(message, metadata);\n  }\n\n  public info(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().info(message, metadata);\n  }\n\n  public warn(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().warn(message, metadata);\n  }\n\n  public error(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().error(message, metadata);\n  }\n\n  public trace<T>(name: string, fn: (span: Span) => Promise<T>): Promise<T> {\n    return this.#getTaskLogger().trace(name, fn);\n  }\n\n  #getTaskLogger(): TaskLogger {\n    return getGlobal(API_NAME) ?? NOOP_TASK_LOGGER;\n  }\n}\n","// Split module-level variable definition into separate files to allow\n// tree-shaking on each api instance.\nimport { LoggerAPI } from \"./logger\";\n/** Entrypoint for logger API */\nexport const logger = LoggerAPI.getInstance();\n","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 { clock } from \"../clock-api\";\nimport { logger } from \"../logger-api\";\nimport {\n  BatchTaskRunExecutionResult,\n  ProdChildToWorkerMessages,\n  ProdWorkerToChildMessages,\n  TaskRunContext,\n  TaskRunExecution,\n  TaskRunExecutionResult,\n} from \"../schemas\";\nimport { checkpointSafeTimeout, unboundedTimeout } from \"../utils/timers\";\nimport { ZodIpcConnection } from \"../zodIpc\";\nimport { RuntimeManager } from \"./manager\";\n\nexport type ProdRuntimeManagerOptions = {\n  waitThresholdInMs?: number;\n};\n\nexport class ProdRuntimeManager 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  _waitForDuration:\n    | { resolve: (value: \"external\") => void; reject: (err?: any) => void }\n    | undefined;\n\n  constructor(\n    private ipc: ZodIpcConnection<\n      typeof ProdWorkerToChildMessages,\n      typeof ProdChildToWorkerMessages\n    >,\n    private options: ProdRuntimeManagerOptions = {}\n  ) {}\n\n  disable(): void {\n    // do nothing\n  }\n\n  async waitForDuration(ms: number): Promise<void> {\n    const now = Date.now();\n\n    const internalTimeout = unboundedTimeout(ms, \"internal\" as const);\n    const checkpointSafeInternalTimeout = checkpointSafeTimeout(ms);\n\n    if (ms <= this.waitThresholdInMs) {\n      await internalTimeout;\n      return;\n    }\n\n    const externalResume = new Promise<\"external\">((resolve, reject) => {\n      this._waitForDuration = { resolve, reject };\n    });\n\n    const { willCheckpointAndRestore } = await this.ipc.sendWithAck(\n      \"WAIT_FOR_DURATION\",\n      {\n        ms,\n        now,\n      },\n      31_000\n    );\n\n    if (!willCheckpointAndRestore) {\n      await internalTimeout;\n      return;\n    }\n\n    this.ipc.send(\"READY_FOR_CHECKPOINT\", {});\n\n    // internalTimeout acts as a backup and will be accurate if the checkpoint never happens\n    // checkpointSafeInternalTimeout is accurate even after non-simulated restores\n    await Promise.race([internalTimeout, checkpointSafeInternalTimeout]);\n\n    // Resets the clock to the current time\n    clock.reset();\n\n    try {\n      // The coordinator should cancel any in-progress checkpoints\n      const { checkpointCanceled, version } = await this.ipc.sendWithAck(\n        \"CANCEL_CHECKPOINT\",\n        {\n          version: \"v2\",\n          reason: \"WAIT_FOR_DURATION\",\n        },\n        31_000\n      );\n\n      if (checkpointCanceled) {\n        // There won't be a checkpoint or external resume and we've already completed our internal timeout\n        return;\n      }\n    } catch (error) {\n      // If the cancellation times out, we will proceed as if the checkpoint was canceled\n      logger.debug(\"Checkpoint cancellation timed out\", { error });\n      return;\n    }\n\n    // No checkpoint was canceled, so we were checkpointed. We need to wait for the external resume message.\n    await externalResume;\n  }\n\n  resumeAfterDuration(): void {\n    if (!this._waitForDuration) {\n      return;\n    }\n\n    // Resets the clock to the current time\n    clock.reset();\n\n    this._waitForDuration.resolve(\"external\");\n    this._waitForDuration = undefined;\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 promise = new Promise<TaskRunExecutionResult>((resolve) => {\n      this._taskWaits.set(params.id, { resolve });\n    });\n\n    await this.ipc.send(\"WAIT_FOR_TASK\", {\n      friendlyId: params.id,\n    });\n\n    const result = await promise;\n\n    clock.reset();\n\n    return result;\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          this._taskWaits.set(runId, { resolve });\n        });\n      })\n    );\n\n    await this.ipc.send(\"WAIT_FOR_BATCH\", {\n      batchFriendlyId: params.id,\n      runFriendlyIds: params.runs,\n    });\n\n    const results = await promise;\n\n    clock.reset();\n\n    return {\n      id: params.id,\n      items: results,\n    };\n  }\n\n  resumeTask(completion: TaskRunExecutionResult): void {\n    const wait = this._taskWaits.get(completion.id);\n\n    if (!wait) {\n      return;\n    }\n\n    wait.resolve(completion);\n\n    this._taskWaits.delete(completion.id);\n  }\n\n  private get waitThresholdInMs(): number {\n    return this.options.waitThresholdInMs ?? 30_000;\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/v3/utils/platform.ts","../../../src/v3/utils/globals.ts","../../../src/v3/clock/simpleClock.ts","../../../src/v3/clock/index.ts","../../../src/v3/clock-api.ts","../../../src/v3/logger/taskLogger.ts","../../../src/v3/logger/index.ts","../../../src/v3/logger-api.ts","../../../src/v3/utils/timers.ts","../../../src/v3/runtime/prodRuntimeManager.ts"],"names":["_globalThis","globalThis","global","GLOBAL_TRIGGER_DOT_DEV_KEY","Symbol","for","_global","registerGlobal","type","instance","allowOverride","api","err","Error","getGlobal","unregisterGlobal","PreciseDate","SimpleClock","preciseNow","now","nowStruct","toStruct","seconds","nanos","reset","API_NAME","SIMPLE_CLOCK","ClockAPI","getInstance","_instance","setGlobalClock","clock","SeverityNumber","NoopTaskLogger","debug","log","info","warn","error","trace","name","fn","NOOP_TASK_LOGGER","LoggerAPI","disable","setGlobalTaskLogger","taskLogger","message","metadata","logger","setInterval","setTimeout","unboundedTimeout","delay","value","options","maxDelay","fullTimeouts","Math","floor","remainingDelay","lastTimeoutResult","i","checkpointSafeTimeout","scanIntervalMs","start","Date","ProdRuntimeManager","constructor","ipc","_taskWaits","Map","_batchWaits","waitForDuration","ms","internalTimeout","checkpointSafeInternalTimeout","waitThresholdInMs","externalResume","Promise","resolve","reject","_waitForDuration","willCheckpointAndRestore","sendWithAck","send","race","checkpointCanceled","version","reason","resumeAfterDuration","undefined","waitUntil","date","getTime","waitForTask","params","promise","set","id","friendlyId","result","waitForBatch","runs","length","items","all","map","runId","batchFriendlyId","runFriendlyIds","results","resumeTask","completion","wait","get","delete"],"mappings":";;;;;;;;;;;;;;;;;AAAO,IAAMA,cAAc,OAAOC,eAAe,WAAWA,aAAaC;;;ACQzE,IAAMC,6BAA6BC,OAAOC,IAAI,oBAAoB;AAElE,IAAMC,UAAUN;AAET,SAASO,eACdC,MACAC,UACAC,gBAAgB,OACP;AACT,QAAMC,MAAOL,QAAQH,0BAAAA,IAA8BG,QAAQH,0BAAAA,KAA+B,CAAC;AAE3F,MAAI,CAACO,iBAAiBC,IAAIH,IAAAA,GAAO;AAE/B,UAAMI,MAAM,IAAIC,MAAM,yDAAyDL,IAAAA,EAAM;AACrF,WAAO;EACT;AAEAG,MAAIH,IAAAA,IAAQC;AAEZ,SAAO;AACT;AAhBgBF;AAkBT,SAASO,UACdN,MAC0C;AAC1C,SAAOF,QAAQH,0BAAAA,IAA8BK,IAAAA;AAC/C;AAJgBM;AAMT,SAASC,iBAAiBP,MAAoC;AACnE,QAAMG,MAAML,QAAQH,0BAAAA;AAEpB,MAAIQ,KAAK;AACP,WAAOA,IAAIH,IAAAA;EACb;AACF;AANgBO;;;ACpChB,SAASC,mBAAmB;AAGrB,IAAMC,eAAN,MAAMA,aAAAA;EACXC,aAA+B;AAC7B,UAAMC,MAAM,IAAIH,YAAAA;AAChB,UAAMI,YAAYD,IAAIE,SAAQ;AAE9B,WAAO;MAACD,UAAUE;MAASF,UAAUG;;EACvC;EAEAC,QAAQ;EAER;AACF;AAXaP;AAAN,IAAMA,cAAN;;;ACHP,IAAMQ,WAAW;AAMjB,IAAMC,eAAe,IAAIT,YAAAA;AANzB;AAQO,IAAMU,YAAN,MAAMA,UAAAA;EAGX,cAAsB;AAsBtB;EAtBuB;EAEvB,OAAcC,cAAwB;AACpC,QAAI,CAAC,KAAKC,WAAW;AACnB,WAAKA,YAAY,IAAIF,UAAAA;IACvB;AAEA,WAAO,KAAKE;EACd;EAEOC,eAAeC,QAAuB;AAC3C,WAAOxB,eAAekB,UAAUM,MAAAA;EAClC;EAEOb,aAAwB;AAC7B,WAAO,sBAAK,wBAAL,WAAiBA,WAAU;EACpC;EAEOM,QAAc;AACnB,0BAAK,wBAAL,WAAiBA,MAAK;EACxB;AAKF;AAHE;cAAS,kCAAU;AACjB,SAAOV,UAAUW,QAAAA,KAAaC;AAChC,GAFS;AAzBEC;AAAN,IAAMA,WAAN;;;ACJA,IAAMI,QAAQJ,SAASC,YAAW;;;ACHzC,SAAiBI,sBAAsB;AAgGhC,IAAMC,kBAAN,MAAMA,gBAAAA;EACXC,QAAQ;EAAC;EACTC,MAAM;EAAC;EACPC,OAAO;EAAC;EACRC,OAAO;EAAC;EACRC,QAAQ;EAAC;EACTC,MAASC,MAAcC,IAA4C;AACjE,WAAOA,GAAG,CAAC,CAAA;EACb;AACF;AATaR;AAAN,IAAMA,iBAAN;;;AC7FP,IAAMR,YAAW;AAEjB,IAAMiB,mBAAmB,IAAIT,eAAAA;AAN7B;AAQO,IAAMU,aAAN,MAAMA,WAAAA;EAGX,cAAsB;AA0CtB;EA1CuB;EAEvB,OAAcf,cAAyB;AACrC,QAAI,CAAC,KAAKC,WAAW;AACnB,WAAKA,YAAY,IAAIc,WAAAA;IACvB;AAEA,WAAO,KAAKd;EACd;EAEOe,UAAU;AACf7B,qBAAiBU,SAAAA;EACnB;EAEOoB,oBAAoBC,YAAiC;AAC1D,WAAOvC,eAAekB,WAAUqB,UAAAA;EAClC;EAEOZ,MAAMa,SAAiBC,UAAoC;AAChE,0BAAK,kCAAL,WAAsBd,MAAMa,SAASC,QAAAA;EACvC;EAEOb,IAAIY,SAAiBC,UAAoC;AAC9D,0BAAK,kCAAL,WAAsBb,IAAIY,SAASC,QAAAA;EACrC;EAEOZ,KAAKW,SAAiBC,UAAoC;AAC/D,0BAAK,kCAAL,WAAsBZ,KAAKW,SAASC,QAAAA;EACtC;EAEOX,KAAKU,SAAiBC,UAAoC;AAC/D,0BAAK,kCAAL,WAAsBX,KAAKU,SAASC,QAAAA;EACtC;EAEOV,MAAMS,SAAiBC,UAAoC;AAChE,0BAAK,kCAAL,WAAsBV,MAAMS,SAASC,QAAAA;EACvC;EAEOT,MAASC,MAAcC,IAA4C;AACxE,WAAO,sBAAK,kCAAL,WAAsBF,MAAMC,MAAMC,EAAAA;EAC3C;AAKF;AAHE;mBAAc,kCAAe;AAC3B,SAAO3B,UAAUW,SAAAA,KAAaiB;AAChC,GAFc;AA7CHC;AAAN,IAAMA,YAAN;;;ACJA,IAAMM,SAASN,UAAUf,YAAW;;;ACH3C,SAASsB,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;AAmBtB,eAAsBW,sBAAsBV,QAAgB,GAAkB;AAC5E,QAAMW,iBAAiB;AAGvB,mBAAiBC,SAASf,YAAYc,gBAAgBE,KAAK/C,IAAG,CAAA,GAAK;AACjE,QAAI+C,KAAK/C,IAAG,IAAK8C,QAAQZ,OAAO;AAC9B;IACF;EACF;AACF;AATsBU;;;ACJf,IAAMI,sBAAN,MAAMA,oBAAAA;EAYXC,YACUC,KAIAd,UAAqC,CAAC,GAC9C;eALQc;mBAIAd;SAhBVe,aAAgF,oBAAIC,IAAAA;SAEpFC,cAGI,oBAAID,IAAAA;EAYL;EAEH3B,UAAgB;EAEhB;EAEA,MAAM6B,gBAAgBC,IAA2B;AAC/C,UAAMvD,MAAM+C,KAAK/C,IAAG;AAEpB,UAAMwD,kBAAkBvB,iBAAiBsB,IAAI,UAAA;AAC7C,UAAME,gCAAgCb,sBAAsBW,EAAAA;AAE5D,QAAIA,MAAM,KAAKG,mBAAmB;AAChC,YAAMF;AACN;IACF;AAEA,UAAMG,iBAAiB,IAAIC,QAAoB,CAACC,SAASC,WAAW;AAClE,WAAKC,mBAAmB;QAAEF;QAASC;MAAO;IAC5C,CAAA;AAEA,UAAM,EAAEE,yBAAwB,IAAK,MAAM,KAAKd,IAAIe,YAClD,qBACA;MACEV;MACAvD;IACF,GACA,IAAA;AAGF,QAAI,CAACgE,0BAA0B;AAC7B,YAAMR;AACN;IACF;AAEA,SAAKN,IAAIgB,KAAK,wBAAwB,CAAC,CAAA;AAIvC,UAAMN,QAAQO,KAAK;MAACX;MAAiBC;KAA8B;AAGnE7C,UAAMP,MAAK;AAEX,QAAI;AAEF,YAAM,EAAE+D,oBAAoBC,QAAO,IAAK,MAAM,KAAKnB,IAAIe,YACrD,qBACA;QACEI,SAAS;QACTC,QAAQ;MACV,GACA,IAAA;AAGF,UAAIF,oBAAoB;AAEtB;MACF;IACF,SAASjD,OAAO;AAEdW,aAAOf,MAAM,qCAAqC;QAAEI;MAAM,CAAA;AAC1D;IACF;AAGA,UAAMwC;EACR;EAEAY,sBAA4B;AAC1B,QAAI,CAAC,KAAKR,kBAAkB;AAC1B;IACF;AAGAnD,UAAMP,MAAK;AAEX,SAAK0D,iBAAiBF,QAAQ,UAAA;AAC9B,SAAKE,mBAAmBS;EAC1B;EAEA,MAAMC,UAAUC,MAA2B;AACzC,WAAO,KAAKpB,gBAAgBoB,KAAKC,QAAO,IAAK5B,KAAK/C,IAAG,CAAA;EACvD;EAEA,MAAM4E,YAAYC,QAA8E;AAC9F,UAAMC,UAAU,IAAIlB,QAAgC,CAACC,YAAY;AAC/D,WAAKV,WAAW4B,IAAIF,OAAOG,IAAI;QAAEnB;MAAQ,CAAA;IAC3C,CAAA;AAEA,UAAM,KAAKX,IAAIgB,KAAK,iBAAiB;MACnCe,YAAYJ,OAAOG;IACrB,CAAA;AAEA,UAAME,SAAS,MAAMJ;AAErBlE,UAAMP,MAAK;AAEX,WAAO6E;EACT;EAEA,MAAMC,aAAaN,QAIsB;AACvC,QAAI,CAACA,OAAOO,KAAKC,QAAQ;AACvB,aAAOzB,QAAQC,QAAQ;QAAEmB,IAAIH,OAAOG;QAAIM,OAAO,CAAA;MAAG,CAAA;IACpD;AAEA,UAAMR,UAAUlB,QAAQ2B,IACtBV,OAAOO,KAAKI,IAAI,CAACC,UAAU;AACzB,aAAO,IAAI7B,QAAgC,CAACC,SAASC,WAAW;AAC9D,aAAKX,WAAW4B,IAAIU,OAAO;UAAE5B;QAAQ,CAAA;MACvC,CAAA;IACF,CAAA,CAAA;AAGF,UAAM,KAAKX,IAAIgB,KAAK,kBAAkB;MACpCwB,iBAAiBb,OAAOG;MACxBW,gBAAgBd,OAAOO;IACzB,CAAA;AAEA,UAAMQ,UAAU,MAAMd;AAEtBlE,UAAMP,MAAK;AAEX,WAAO;MACL2E,IAAIH,OAAOG;MACXM,OAAOM;IACT;EACF;EAEAC,WAAWC,YAA0C;AACnD,UAAMC,OAAO,KAAK5C,WAAW6C,IAAIF,WAAWd,EAAE;AAE9C,QAAI,CAACe,MAAM;AACT;IACF;AAEAA,SAAKlC,QAAQiC,UAAAA;AAEb,SAAK3C,WAAW8C,OAAOH,WAAWd,EAAE;EACtC;EAEA,IAAYtB,oBAA4B;AACtC,WAAO,KAAKtB,QAAQsB,qBAAqB;EAC3C;AACF;AAtKaV;AAAN,IAAMA,qBAAN","sourcesContent":["export const _globalThis = typeof globalThis === \"object\" ? globalThis : global;\n","import { ApiClientConfiguration } from \"../apiClientManager/types\";\nimport { Clock } from \"../clock/clock\";\nimport type { RuntimeManager } from \"../runtime/manager\";\nimport { TaskCatalog } from \"../task-catalog/catalog\";\nimport { TaskContext } from \"../taskContext/types\";\nimport { UsageManager } from \"../usage/types\";\nimport { _globalThis } from \"./platform\";\n\nconst GLOBAL_TRIGGER_DOT_DEV_KEY = Symbol.for(`dev.trigger.ts.api`);\n\nconst _global = _globalThis as TriggerDotDevGlobal;\n\nexport function registerGlobal<Type extends keyof TriggerDotDevGlobalAPI>(\n  type: Type,\n  instance: TriggerDotDevGlobalAPI[Type],\n  allowOverride = false\n): boolean {\n  const api = (_global[GLOBAL_TRIGGER_DOT_DEV_KEY] = _global[GLOBAL_TRIGGER_DOT_DEV_KEY] ?? {});\n\n  if (!allowOverride && api[type]) {\n    // already registered an API of this type\n    const err = new Error(`trigger.dev: Attempted duplicate registration of API: ${type}`);\n    return false;\n  }\n\n  api[type] = instance;\n\n  return true;\n}\n\nexport function getGlobal<Type extends keyof TriggerDotDevGlobalAPI>(\n  type: Type\n): TriggerDotDevGlobalAPI[Type] | undefined {\n  return _global[GLOBAL_TRIGGER_DOT_DEV_KEY]?.[type];\n}\n\nexport function unregisterGlobal(type: keyof TriggerDotDevGlobalAPI) {\n  const api = _global[GLOBAL_TRIGGER_DOT_DEV_KEY];\n\n  if (api) {\n    delete api[type];\n  }\n}\n\ntype TriggerDotDevGlobal = {\n  [GLOBAL_TRIGGER_DOT_DEV_KEY]?: TriggerDotDevGlobalAPI;\n};\n\ntype TriggerDotDevGlobalAPI = {\n  runtime?: RuntimeManager;\n  logger?: any;\n  clock?: Clock;\n  usage?: UsageManager;\n  [\"task-catalog\"]?: TaskCatalog;\n  [\"task-context\"]?: TaskContext;\n  [\"api-client\"]?: ApiClientConfiguration;\n};\n","import { PreciseDate } from \"@google-cloud/precise-date\";\nimport { Clock } from \"./clock\";\n\nexport class SimpleClock implements Clock {\n  preciseNow(): [number, number] {\n    const now = new PreciseDate();\n    const nowStruct = now.toStruct();\n\n    return [nowStruct.seconds, nowStruct.nanos];\n  }\n\n  reset() {\n    // do nothing\n  }\n}\n","const API_NAME = \"clock\";\n\nimport { getGlobal, registerGlobal } from \"../utils/globals\";\nimport type { Clock, ClockTime } from \"./clock\";\nimport { SimpleClock } from \"./simpleClock\";\n\nconst SIMPLE_CLOCK = new SimpleClock();\n\nexport class ClockAPI {\n  private static _instance?: ClockAPI;\n\n  private constructor() {}\n\n  public static getInstance(): ClockAPI {\n    if (!this._instance) {\n      this._instance = new ClockAPI();\n    }\n\n    return this._instance;\n  }\n\n  public setGlobalClock(clock: Clock): boolean {\n    return registerGlobal(API_NAME, clock);\n  }\n\n  public preciseNow(): ClockTime {\n    return this.#getClock().preciseNow();\n  }\n\n  public reset(): void {\n    this.#getClock().reset();\n  }\n\n  #getClock(): Clock {\n    return getGlobal(API_NAME) ?? SIMPLE_CLOCK;\n  }\n}\n","// Split module-level variable definition into separate files to allow\n// tree-shaking on each api instance.\nimport { ClockAPI } from \"./clock\";\n/** Entrypoint for clock API */\nexport const clock = ClockAPI.getInstance();\n","import { Attributes, Span, SpanOptions } from \"@opentelemetry/api\";\nimport { Logger, SeverityNumber } from \"@opentelemetry/api-logs\";\nimport { iconStringForSeverity } from \"../icons\";\nimport { SemanticInternalAttributes } from \"../semanticInternalAttributes\";\nimport { TriggerTracer } from \"../tracer\";\nimport { flattenAttributes } from \"../utils/flattenAttributes\";\nimport { ClockTime } from \"../clock/clock\";\nimport { clock } from \"../clock-api\";\n\nexport type LogLevel = \"none\" | \"error\" | \"warn\" | \"info\" | \"debug\" | \"log\";\n\nexport const logLevels: Array<LogLevel> = [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\nexport type TaskLoggerConfig = {\n  logger: Logger;\n  tracer: TriggerTracer;\n  level: LogLevel;\n};\n\nexport interface TaskLogger {\n  debug(message: string, properties?: Record<string, unknown>): void;\n  log(message: string, properties?: Record<string, unknown>): void;\n  info(message: string, properties?: Record<string, unknown>): void;\n  warn(message: string, properties?: Record<string, unknown>): void;\n  error(message: string, properties?: Record<string, unknown>): void;\n  trace<T>(name: string, fn: (span: Span) => Promise<T>, options?: SpanOptions): Promise<T>;\n}\n\nexport class OtelTaskLogger implements TaskLogger {\n  private readonly _level: number;\n\n  constructor(private readonly _config: TaskLoggerConfig) {\n    this._level = logLevels.indexOf(_config.level);\n  }\n\n  debug(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 4) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"debug\", SeverityNumber.DEBUG, properties);\n  }\n\n  log(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 3) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"log\", SeverityNumber.INFO, properties);\n  }\n\n  info(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 3) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"info\", SeverityNumber.INFO, properties);\n  }\n\n  warn(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 2) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"warn\", SeverityNumber.WARN, properties);\n  }\n\n  error(message: string, properties?: Record<string, unknown>) {\n    if (this._level < 1) return; // [\"none\", \"error\", \"warn\", \"info\", \"debug\"];\n\n    this.#emitLog(message, this.#getTimestampInHrTime(), \"error\", SeverityNumber.ERROR, properties);\n  }\n\n  #emitLog(\n    message: string,\n    timestamp: ClockTime,\n    severityText: string,\n    severityNumber: SeverityNumber,\n    properties?: Record<string, unknown>\n  ) {\n    let attributes: Attributes = { ...flattenAttributes(safeJsonProcess(properties)) };\n\n    const icon = iconStringForSeverity(severityNumber);\n    if (icon !== undefined) {\n      attributes[SemanticInternalAttributes.STYLE_ICON] = icon;\n    }\n\n    this._config.logger.emit({\n      severityNumber,\n      severityText,\n      body: message,\n      attributes,\n      timestamp,\n    });\n  }\n\n  trace<T>(name: string, fn: (span: Span) => Promise<T>, options?: SpanOptions): Promise<T> {\n    return this._config.tracer.startActiveSpan(name, fn, options);\n  }\n\n  #getTimestampInHrTime(): ClockTime {\n    return clock.preciseNow();\n  }\n}\n\nexport class NoopTaskLogger implements TaskLogger {\n  debug() {}\n  log() {}\n  info() {}\n  warn() {}\n  error() {}\n  trace<T>(name: string, fn: (span: Span) => Promise<T>): Promise<T> {\n    return fn({} as Span);\n  }\n}\n\nfunction safeJsonProcess(value?: Record<string, unknown>): Record<string, unknown> | undefined {\n  try {\n    return JSON.parse(JSON.stringify(value, jsonErrorReplacer));\n  } catch {\n    return value;\n  }\n}\n\nfunction jsonErrorReplacer(key: string, value: unknown) {\n  if (value instanceof Error) {\n    return {\n      name: value.name,\n      message: value.message,\n      stack: value.stack,\n    };\n  }\n\n  return value;\n}\n","import { NoopTaskLogger, TaskLogger } from \"./taskLogger\";\nimport { getGlobal, registerGlobal, unregisterGlobal } from \"../utils/globals\";\nimport { Span } from \"@opentelemetry/api\";\n\nconst API_NAME = \"logger\";\n\nconst NOOP_TASK_LOGGER = new NoopTaskLogger();\n\nexport class LoggerAPI implements TaskLogger {\n  private static _instance?: LoggerAPI;\n\n  private constructor() {}\n\n  public static getInstance(): LoggerAPI {\n    if (!this._instance) {\n      this._instance = new LoggerAPI();\n    }\n\n    return this._instance;\n  }\n\n  public disable() {\n    unregisterGlobal(API_NAME);\n  }\n\n  public setGlobalTaskLogger(taskLogger: TaskLogger): boolean {\n    return registerGlobal(API_NAME, taskLogger);\n  }\n\n  public debug(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().debug(message, metadata);\n  }\n\n  public log(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().log(message, metadata);\n  }\n\n  public info(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().info(message, metadata);\n  }\n\n  public warn(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().warn(message, metadata);\n  }\n\n  public error(message: string, metadata?: Record<string, unknown>) {\n    this.#getTaskLogger().error(message, metadata);\n  }\n\n  public trace<T>(name: string, fn: (span: Span) => Promise<T>): Promise<T> {\n    return this.#getTaskLogger().trace(name, fn);\n  }\n\n  #getTaskLogger(): TaskLogger {\n    return getGlobal(API_NAME) ?? NOOP_TASK_LOGGER;\n  }\n}\n","// Split module-level variable definition into separate files to allow\n// tree-shaking on each api instance.\nimport { LoggerAPI } from \"./logger\";\n/** Entrypoint for logger API */\nexport const logger = LoggerAPI.getInstance();\n","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 { clock } from \"../clock-api\";\nimport { logger } from \"../logger-api\";\nimport {\n  BatchTaskRunExecutionResult,\n  ProdChildToWorkerMessages,\n  ProdWorkerToChildMessages,\n  TaskRunContext,\n  TaskRunExecution,\n  TaskRunExecutionResult,\n} from \"../schemas\";\nimport { checkpointSafeTimeout, unboundedTimeout } from \"../utils/timers\";\nimport { ZodIpcConnection } from \"../zodIpc\";\nimport { RuntimeManager } from \"./manager\";\n\nexport type ProdRuntimeManagerOptions = {\n  waitThresholdInMs?: number;\n};\n\nexport class ProdRuntimeManager 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  _waitForDuration:\n    | { resolve: (value: \"external\") => void; reject: (err?: any) => void }\n    | undefined;\n\n  constructor(\n    private ipc: ZodIpcConnection<\n      typeof ProdWorkerToChildMessages,\n      typeof ProdChildToWorkerMessages\n    >,\n    private options: ProdRuntimeManagerOptions = {}\n  ) {}\n\n  disable(): void {\n    // do nothing\n  }\n\n  async waitForDuration(ms: number): Promise<void> {\n    const now = Date.now();\n\n    const internalTimeout = unboundedTimeout(ms, \"internal\" as const);\n    const checkpointSafeInternalTimeout = checkpointSafeTimeout(ms);\n\n    if (ms <= this.waitThresholdInMs) {\n      await internalTimeout;\n      return;\n    }\n\n    const externalResume = new Promise<\"external\">((resolve, reject) => {\n      this._waitForDuration = { resolve, reject };\n    });\n\n    const { willCheckpointAndRestore } = await this.ipc.sendWithAck(\n      \"WAIT_FOR_DURATION\",\n      {\n        ms,\n        now,\n      },\n      31_000\n    );\n\n    if (!willCheckpointAndRestore) {\n      await internalTimeout;\n      return;\n    }\n\n    this.ipc.send(\"READY_FOR_CHECKPOINT\", {});\n\n    // internalTimeout acts as a backup and will be accurate if the checkpoint never happens\n    // checkpointSafeInternalTimeout is accurate even after non-simulated restores\n    await Promise.race([internalTimeout, checkpointSafeInternalTimeout]);\n\n    // Resets the clock to the current time\n    clock.reset();\n\n    try {\n      // The coordinator should cancel any in-progress checkpoints\n      const { checkpointCanceled, version } = await this.ipc.sendWithAck(\n        \"CANCEL_CHECKPOINT\",\n        {\n          version: \"v2\",\n          reason: \"WAIT_FOR_DURATION\",\n        },\n        31_000\n      );\n\n      if (checkpointCanceled) {\n        // There won't be a checkpoint or external resume and we've already completed our internal timeout\n        return;\n      }\n    } catch (error) {\n      // If the cancellation times out, we will proceed as if the checkpoint was canceled\n      logger.debug(\"Checkpoint cancellation timed out\", { error });\n      return;\n    }\n\n    // No checkpoint was canceled, so we were checkpointed. We need to wait for the external resume message.\n    await externalResume;\n  }\n\n  resumeAfterDuration(): void {\n    if (!this._waitForDuration) {\n      return;\n    }\n\n    // Resets the clock to the current time\n    clock.reset();\n\n    this._waitForDuration.resolve(\"external\");\n    this._waitForDuration = undefined;\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 promise = new Promise<TaskRunExecutionResult>((resolve) => {\n      this._taskWaits.set(params.id, { resolve });\n    });\n\n    await this.ipc.send(\"WAIT_FOR_TASK\", {\n      friendlyId: params.id,\n    });\n\n    const result = await promise;\n\n    clock.reset();\n\n    return result;\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          this._taskWaits.set(runId, { resolve });\n        });\n      })\n    );\n\n    await this.ipc.send(\"WAIT_FOR_BATCH\", {\n      batchFriendlyId: params.id,\n      runFriendlyIds: params.runs,\n    });\n\n    const results = await promise;\n\n    clock.reset();\n\n    return {\n      id: params.id,\n      items: results,\n    };\n  }\n\n  resumeTask(completion: TaskRunExecutionResult): void {\n    const wait = this._taskWaits.get(completion.id);\n\n    if (!wait) {\n      return;\n    }\n\n    wait.resolve(completion);\n\n    this._taskWaits.delete(completion.id);\n  }\n\n  private get waitThresholdInMs(): number {\n    return this.options.waitThresholdInMs ?? 30_000;\n  }\n}\n"]}

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

@@ -1,12 +1,12 @@
 import * as logsAPI from '@opentelemetry/api-logs';
 import { TracingSDK } from '../otel/index.mjs';
 export { TracingDiagnosticLogLevel, recordSpanException } from '../otel/index.mjs';
-import { b as TaskRunExecution, T as TaskRunExecutionResult } from '../../manager-JkbddlcO.mjs';
-export { R as RuntimeManager } from '../../manager-JkbddlcO.mjs';
-import { B as BackgroundWorkerProperties } from '../../messages-9lty-Du5.mjs';
-import { C as Config, T as TaskFileMetadata, a as TaskMetadataWithFilePath } from '../../schemas-r4ZP9S-F.mjs';
-import { T as TriggerTracer, P as ProjectConfig, H as HandleErrorFunction, a as TaskMetadataWithFunctions, C as Clock, b as ClockTime, c as TaskCatalog } from '../../catalog-XTlJQaMn.mjs';
-export { O as OtelTaskLogger, l as logLevels } from '../../catalog-XTlJQaMn.mjs';
+import { b as TaskRunExecution, T as TaskRunExecutionResult } from '../../manager-S98VaLUy.mjs';
+export { R as RuntimeManager } from '../../manager-S98VaLUy.mjs';
+import { B as BackgroundWorkerProperties } from '../../messages--WkQvA2l.mjs';
+import { C as Config, T as TaskFileMetadata, a as TaskMetadataWithFilePath } from '../../schemas-Sb0sJcEt.mjs';
+import { T as TriggerTracer, P as ProjectConfig, H as HandleErrorFunction, a as TaskMetadataWithFunctions, U as UsageMeasurement, C as Clock, b as ClockTime, c as TaskCatalog, d as UsageManager, e as UsageSample } from '../../catalog-Gjy5NtAB.mjs';
+export { O as OtelTaskLogger, l as logLevels, u as usage } from '../../catalog-Gjy5NtAB.mjs';
 import { PreciseDate } from '@google-cloud/precise-date';
 import { LogRecordProcessor, LogRecord } from '@opentelemetry/sdk-logs';
 import { SpanProcessor, Span } from '@opentelemetry/sdk-trace-base';
@@ -46,7 +46,9 @@ declare class TaskExecutor {
     private _importedConfig;
     private _handleErrorFn;
     constructor(task: TaskMetadataWithFunctions, options: TaskExecutorOptions);
-    execute(execution: TaskRunExecution, worker: BackgroundWorkerProperties, traceContext: Record<string, unknown>): Promise<TaskRunExecutionResult>;
+    execute(execution: TaskRunExecution, worker: BackgroundWorkerProperties, traceContext: Record<string, unknown>, usage: UsageMeasurement): Promise<{
+        result: TaskRunExecutionResult;
+    }>;
 }
 
 type PreciseWallClockOptions = {
@@ -93,4 +95,48 @@ declare class TaskContextLogProcessor implements LogRecordProcessor {
     shutdown(): Promise<void>;
 }
 
-export { ConsoleInterceptor, PreciseWallClock as DurableClock, StandardTaskCatalog, TaskContextLogProcessor, TaskContextSpanProcessor, TaskExecutor, type TaskExecutorOptions, TracingSDK, getEnvVar };
+declare class DevUsageMeasurement implements UsageMeasurement {
+    readonly id: string;
+    private startedAt;
+    private _pauses;
+    private _endedAt;
+    constructor(id: string, startedAt?: ClockTime);
+    stop(): void;
+    sample(): UsageSample;
+    registerPause(pauseId: string, start: ClockTime, end?: ClockTime): void;
+}
+declare class DevUsageManager implements UsageManager {
+    private _firstMeasurement?;
+    private _currentMeasurements;
+    private _pauses;
+    disable(): void;
+    sample(): UsageSample | undefined;
+    start(): DevUsageMeasurement;
+    stop(measurement: DevUsageMeasurement): UsageSample;
+    pauseAsync<T>(cb: () => Promise<T>): Promise<T>;
+}
+
+type ProdUsageManagerOptions = {
+    heartbeatIntervalMs?: number;
+    url?: string;
+    jwt?: string;
+};
+declare class ProdUsageManager implements UsageManager {
+    #private;
+    private readonly delegageUsageManager;
+    private readonly options;
+    private _measurement;
+    private _abortController;
+    private _lastSample;
+    private _usageClient;
+    constructor(delegageUsageManager: UsageManager, options: ProdUsageManagerOptions);
+    get isReportingEnabled(): boolean;
+    disable(): void;
+    sample(): UsageSample | undefined;
+    start(): UsageMeasurement;
+    stop(measurement: UsageMeasurement): UsageSample;
+    pauseAsync<T>(cb: () => Promise<T>): Promise<T>;
+    flush(): Promise<void>;
+}
+
+export { ConsoleInterceptor, DevUsageManager, PreciseWallClock as DurableClock, ProdUsageManager, type ProdUsageManagerOptions, StandardTaskCatalog, TaskContextLogProcessor, TaskContextSpanProcessor, TaskExecutor, type TaskExecutorOptions, TracingSDK, getEnvVar };

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

@@ -1,12 +1,12 @@
 import * as logsAPI from '@opentelemetry/api-logs';
 import { TracingSDK } from '../otel/index.js';
 export { TracingDiagnosticLogLevel, recordSpanException } from '../otel/index.js';
-import { b as TaskRunExecution, T as TaskRunExecutionResult } from '../../manager-JkbddlcO.js';
-export { R as RuntimeManager } from '../../manager-JkbddlcO.js';
-import { B as BackgroundWorkerProperties } from '../../messages-9lty-Du5.js';
-import { C as Config, T as TaskFileMetadata, a as TaskMetadataWithFilePath } from '../../schemas-r4ZP9S-F.js';
-import { T as TriggerTracer, P as ProjectConfig, H as HandleErrorFunction, a as TaskMetadataWithFunctions, C as Clock, b as ClockTime, c as TaskCatalog } from '../../catalog-dRKTgwQ7.js';
-export { O as OtelTaskLogger, l as logLevels } from '../../catalog-dRKTgwQ7.js';
+import { b as TaskRunExecution, T as TaskRunExecutionResult } from '../../manager-S98VaLUy.js';
+export { R as RuntimeManager } from '../../manager-S98VaLUy.js';
+import { B as BackgroundWorkerProperties } from '../../messages--WkQvA2l.js';
+import { C as Config, T as TaskFileMetadata, a as TaskMetadataWithFilePath } from '../../schemas-Sb0sJcEt.js';
+import { T as TriggerTracer, P as ProjectConfig, H as HandleErrorFunction, a as TaskMetadataWithFunctions, U as UsageMeasurement, C as Clock, b as ClockTime, c as TaskCatalog, d as UsageManager, e as UsageSample } from '../../catalog-EP9DGAGm.js';
+export { O as OtelTaskLogger, l as logLevels, u as usage } from '../../catalog-EP9DGAGm.js';
 import { PreciseDate } from '@google-cloud/precise-date';
 import { LogRecordProcessor, LogRecord } from '@opentelemetry/sdk-logs';
 import { SpanProcessor, Span } from '@opentelemetry/sdk-trace-base';
@@ -46,7 +46,9 @@ declare class TaskExecutor {
     private _importedConfig;
     private _handleErrorFn;
     constructor(task: TaskMetadataWithFunctions, options: TaskExecutorOptions);
-    execute(execution: TaskRunExecution, worker: BackgroundWorkerProperties, traceContext: Record<string, unknown>): Promise<TaskRunExecutionResult>;
+    execute(execution: TaskRunExecution, worker: BackgroundWorkerProperties, traceContext: Record<string, unknown>, usage: UsageMeasurement): Promise<{
+        result: TaskRunExecutionResult;
+    }>;
 }
 
 type PreciseWallClockOptions = {
@@ -93,4 +95,48 @@ declare class TaskContextLogProcessor implements LogRecordProcessor {
     shutdown(): Promise<void>;
 }
 
-export { ConsoleInterceptor, PreciseWallClock as DurableClock, StandardTaskCatalog, TaskContextLogProcessor, TaskContextSpanProcessor, TaskExecutor, type TaskExecutorOptions, TracingSDK, getEnvVar };
+declare class DevUsageMeasurement implements UsageMeasurement {
+    readonly id: string;
+    private startedAt;
+    private _pauses;
+    private _endedAt;
+    constructor(id: string, startedAt?: ClockTime);
+    stop(): void;
+    sample(): UsageSample;
+    registerPause(pauseId: string, start: ClockTime, end?: ClockTime): void;
+}
+declare class DevUsageManager implements UsageManager {
+    private _firstMeasurement?;
+    private _currentMeasurements;
+    private _pauses;
+    disable(): void;
+    sample(): UsageSample | undefined;
+    start(): DevUsageMeasurement;
+    stop(measurement: DevUsageMeasurement): UsageSample;
+    pauseAsync<T>(cb: () => Promise<T>): Promise<T>;
+}
+
+type ProdUsageManagerOptions = {
+    heartbeatIntervalMs?: number;
+    url?: string;
+    jwt?: string;
+};
+declare class ProdUsageManager implements UsageManager {
+    #private;
+    private readonly delegageUsageManager;
+    private readonly options;
+    private _measurement;
+    private _abortController;
+    private _lastSample;
+    private _usageClient;
+    constructor(delegageUsageManager: UsageManager, options: ProdUsageManagerOptions);
+    get isReportingEnabled(): boolean;
+    disable(): void;
+    sample(): UsageSample | undefined;
+    start(): UsageMeasurement;
+    stop(measurement: UsageMeasurement): UsageSample;
+    pauseAsync<T>(cb: () => Promise<T>): Promise<T>;
+    flush(): Promise<void>;
+}
+
+export { ConsoleInterceptor, DevUsageManager, PreciseWallClock as DurableClock, ProdUsageManager, type ProdUsageManagerOptions, StandardTaskCatalog, TaskContextLogProcessor, TaskContextSpanProcessor, TaskExecutor, type TaskExecutorOptions, TracingSDK, getEnvVar };