@trigger.dev/core 0.0.0-v3-canary-20240322220011 → 0.0.0-v3-canary-20240325170325

package/dist/v3/index.mjs

@@ -248,7 +248,8 @@ var BackgroundWorkerServerMessages = z.discriminatedUnion("type", [
     id: z.string(),
     image: z.string(),
     envId: z.string(),
-    runId: z.string()
+    runId: z.string(),
+    version: z.string()
   })
 ]);
 var serverWebsocketMessages = {
@@ -348,17 +349,62 @@ var RateLimitOptions = z.discriminatedUnion("type", [
   SlidingWindowRateLimit
 ]);
 var RetryOptions = z.object({
+  /** The number of attempts before giving up */
   maxAttempts: z.number().int().optional(),
+  /** The exponential factor to use when calculating the next retry time.
+  *
+  * Each subsequent retry will be calculated as `previousTimeout * factor`
+  */
   factor: z.number().optional(),
+  /** The minimum time to wait before retrying */
   minTimeoutInMs: z.number().int().optional(),
+  /** The maximum time to wait before retrying */
   maxTimeoutInMs: z.number().int().optional(),
+  /** Randomize the timeout between retries.
+  *
+  * This can be useful to prevent the thundering herd problem where all retries happen at the same time.
+  */
   randomize: z.boolean().optional()
 });
 var QueueOptions = z.object({
-  /** @deprecated This feature is coming soon */
-  rateLimit: RateLimitOptions.optional(),
+  /** You can define a shared queue and then pass the name in to your task.
+     * 
+     * @example
+     * 
+     * ```ts
+     * const myQueue = queue({
+        name: "my-queue",
+        concurrencyLimit: 1,
+      });
+  
+      export const task1 = task({
+        id: "task-1",
+        queue: {
+          name: "my-queue",
+        },
+        run: async (payload: { message: string }) => {
+          // ...
+        },
+      });
+  
+      export const task2 = task({
+        id: "task-2",
+        queue: {
+          name: "my-queue",
+        },
+        run: async (payload: { message: string }) => {
+          // ...
+        },
+      });
+     * ```
+     */
+  name: z.string().optional(),
+  /** An optional property that specifies the maximum number of concurrent run executions.
+  *
+  * If this property is omitted, the task can potentially use up the full concurrency of an environment. */
   concurrencyLimit: z.number().int().min(1).max(1e3).optional(),
-  name: z.string().optional()
+  /** @deprecated This feature is coming soon */
+  rateLimit: RateLimitOptions.optional()
 });
 var TaskMetadata = z.object({
   id: z.string(),
@@ -435,10 +481,21 @@ var ProdChildToWorkerMessages = {
   READY_TO_DISPOSE: {
     message: z.undefined()
   },
+  READY_FOR_CHECKPOINT: {
+    message: z.object({
+      version: z.literal("v1").default("v1")
+    })
+  },
+  CANCEL_CHECKPOINT: {
+    message: z.object({
+      version: z.literal("v1").default("v1")
+    })
+  },
   WAIT_FOR_DURATION: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      ms: z.number()
+      ms: z.number(),
+      now: z.number()
     }),
     callback: z.object({
       willCheckpointAndRestore: z.boolean()
@@ -447,14 +504,14 @@ var ProdChildToWorkerMessages = {
   WAIT_FOR_TASK: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      id: z.string()
+      friendlyId: z.string()
     })
   },
   WAIT_FOR_BATCH: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      id: z.string(),
-      runs: z.string().array()
+      batchFriendlyId: z.string(),
+      runFriendlyIds: z.string().array()
     })
   },
   UNCAUGHT_EXCEPTION: {
@@ -668,6 +725,11 @@ var Machine = z.object({
   cpu: z.string().default("1").optional(),
   memory: z.string().default("500Mi").optional()
 });
+var WaitReason = z.enum([
+  "WAIT_FOR_DURATION",
+  "WAIT_FOR_TASK",
+  "WAIT_FOR_BATCH"
+]);
 var ProviderToPlatformMessages = {
   LOG: {
     message: z.object({
@@ -698,7 +760,7 @@ var PlatformToProviderMessages = {
     message: z.object({
       version: z.literal("v1").default("v1"),
       imageTag: z.string(),
-      contentHash: z.string(),
+      shortCode: z.string(),
       envId: z.string(),
       apiKey: z.string(),
       apiUrl: z.string()
@@ -729,13 +791,13 @@ var PlatformToProviderMessages = {
       version: z.literal("v1").default("v1"),
       checkpointId: z.string(),
       runId: z.string(),
-      attemptId: z.string(),
       type: z.enum([
         "DOCKER",
         "KUBERNETES"
       ]),
       location: z.string(),
-      reason: z.string().optional()
+      reason: z.string().optional(),
+      imageRef: z.string()
     })
   },
   DELETE: {
@@ -787,8 +849,8 @@ var CoordinatorToPlatformMessages = {
   READY_FOR_EXECUTION: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      attemptId: z.string(),
-      runId: z.string()
+      runId: z.string(),
+      totalCompletions: z.number()
     }),
     callback: z.discriminatedUnion("success", [
       z.object({
@@ -803,19 +865,19 @@ var CoordinatorToPlatformMessages = {
   READY_FOR_RESUME: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      attemptId: z.string(),
-      type: z.enum([
-        "WAIT_FOR_DURATION",
-        "WAIT_FOR_TASK",
-        "WAIT_FOR_BATCH"
-      ])
+      attemptFriendlyId: z.string(),
+      type: WaitReason
     })
   },
   TASK_RUN_COMPLETED: {
     message: z.object({
       version: z.literal("v1").default("v1"),
       execution: ProdTaskRunExecution,
-      completion: TaskRunExecutionResult
+      completion: TaskRunExecutionResult,
+      checkpoint: z.object({
+        docker: z.boolean(),
+        location: z.string()
+      }).optional()
     })
   },
   TASK_HEARTBEAT: {
@@ -827,21 +889,23 @@ var CoordinatorToPlatformMessages = {
   CHECKPOINT_CREATED: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      attemptId: z.string(),
+      attemptFriendlyId: z.string(),
       docker: z.boolean(),
       location: z.string(),
       reason: z.discriminatedUnion("type", [
         z.object({
           type: z.literal("WAIT_FOR_DURATION"),
-          ms: z.number()
+          ms: z.number(),
+          now: z.number()
         }),
         z.object({
           type: z.literal("WAIT_FOR_BATCH"),
-          id: z.string()
+          batchFriendlyId: z.string(),
+          runFriendlyIds: z.string().array()
         }),
         z.object({
           type: z.literal("WAIT_FOR_TASK"),
-          id: z.string()
+          friendlyId: z.string()
         }),
         z.object({
           type: z.literal("RETRYING_AFTER_FAILURE"),
@@ -863,10 +927,12 @@ var CoordinatorToPlatformMessages = {
   }
 };
 var PlatformToCoordinatorMessages = {
-  RESUME: {
+  RESUME_AFTER_DEPENDENCY: {
     message: z.object({
       version: z.literal("v1").default("v1"),
+      runId: z.string(),
       attemptId: z.string(),
+      attemptFriendlyId: z.string(),
       completions: TaskRunExecutionResult.array(),
       executions: TaskRunExecution.array()
     })
@@ -874,13 +940,21 @@ var PlatformToCoordinatorMessages = {
   RESUME_AFTER_DURATION: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      attemptId: z.string()
+      attemptId: z.string(),
+      attemptFriendlyId: z.string()
     })
   },
   REQUEST_ATTEMPT_CANCELLATION: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      attemptId: z.string()
+      attemptId: z.string(),
+      attemptFriendlyId: z.string()
+    })
+  },
+  READY_FOR_RETRY: {
+    message: z.object({
+      version: z.literal("v1").default("v1"),
+      runId: z.string()
     })
   }
 };
@@ -947,19 +1021,25 @@ var ProdWorkerToCoordinatorMessages = {
   READY_FOR_EXECUTION: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      attemptId: z.string(),
-      runId: z.string()
+      runId: z.string(),
+      totalCompletions: z.number()
     })
   },
   READY_FOR_RESUME: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      attemptId: z.string(),
-      type: z.enum([
-        "WAIT_FOR_DURATION",
-        "WAIT_FOR_TASK",
-        "WAIT_FOR_BATCH"
-      ])
+      attemptFriendlyId: z.string(),
+      type: WaitReason
+    })
+  },
+  READY_FOR_CHECKPOINT: {
+    message: z.object({
+      version: z.literal("v1").default("v1")
+    })
+  },
+  CANCEL_CHECKPOINT: {
+    message: z.object({
+      version: z.literal("v1").default("v1")
     })
   },
   TASK_HEARTBEAT: {
@@ -975,14 +1055,16 @@ var ProdWorkerToCoordinatorMessages = {
       completion: TaskRunExecutionResult
     }),
     callback: z.object({
-      didCheckpoint: z.boolean(),
+      willCheckpointAndRestore: z.boolean(),
       shouldExit: z.boolean()
     })
   },
   WAIT_FOR_DURATION: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      ms: z.number()
+      ms: z.number(),
+      now: z.number(),
+      attemptFriendlyId: z.string()
     }),
     callback: z.object({
       willCheckpointAndRestore: z.boolean()
@@ -991,7 +1073,9 @@ var ProdWorkerToCoordinatorMessages = {
   WAIT_FOR_TASK: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      id: z.string()
+      friendlyId: z.string(),
+      // This is the attempt that is waiting
+      attemptFriendlyId: z.string()
     }),
     callback: z.object({
       willCheckpointAndRestore: z.boolean()
@@ -1000,8 +1084,10 @@ var ProdWorkerToCoordinatorMessages = {
   WAIT_FOR_BATCH: {
     message: z.object({
       version: z.literal("v1").default("v1"),
-      id: z.string(),
-      runs: z.string().array()
+      batchFriendlyId: z.string(),
+      runFriendlyIds: z.string().array(),
+      // This is the attempt that is waiting
+      attemptFriendlyId: z.string()
     }),
     callback: z.object({
       willCheckpointAndRestore: z.boolean()
@@ -1020,7 +1106,7 @@ var ProdWorkerToCoordinatorMessages = {
   }
 };
 var CoordinatorToProdWorkerMessages = {
-  RESUME: {
+  RESUME_AFTER_DEPENDENCY: {
     message: z.object({
       version: z.literal("v1").default("v1"),
       attemptId: z.string(),
@@ -1050,6 +1136,12 @@ var CoordinatorToProdWorkerMessages = {
     message: z.object({
       version: z.literal("v1").default("v1")
     })
+  },
+  READY_FOR_RETRY: {
+    message: z.object({
+      version: z.literal("v1").default("v1"),
+      runId: z.string()
+    })
   }
 };
 var ProdWorkerSocketData = z.object({
@@ -1057,9 +1149,10 @@ var ProdWorkerSocketData = z.object({
   projectRef: z.string(),
   envId: z.string(),
   runId: z.string(),
-  attemptId: z.string(),
+  attemptFriendlyId: z.string().optional(),
   podName: z.string(),
-  deploymentId: z.string()
+  deploymentId: z.string(),
+  deploymentVersion: z.string()
 });
 var PRIMARY_VARIANT = "primary";
 var Variant = z.enum([
@@ -1193,12 +1286,22 @@ var FetchRetryStrategy = z.discriminatedUnion("strategy", [
   FetchRetryHeadersStrategy,
   FetchRetryBackoffStrategy
 ]);
-var FetchRetryOptions = z.record(FetchRetryStrategy);
+var FetchRetryByStatusOptions = z.record(z.string(), FetchRetryStrategy);
 var FetchTimeoutOptions = z.object({
   /** The maximum time to wait for the request to complete. */
   durationInMs: z.number().optional(),
   retry: RetryOptions.optional()
 });
+var FetchRetryOptions = z.object({
+  /** The retrying strategy for specific status codes. */
+  byStatus: FetchRetryByStatusOptions.optional(),
+  /** The timeout options for the request. */
+  timeout: RetryOptions.optional(),
+  /**
+  * The retrying strategy for connection errors.
+  */
+  connectionError: RetryOptions.optional()
+});
 var ExceptionEventProperties = z.object({
   type: z.string().optional(),
   message: z.string().optional(),
@@ -1237,6 +1340,18 @@ function isCancellationSpanEvent(event) {
   return event.name === "cancellation";
 }
 __name(isCancellationSpanEvent, "isCancellationSpanEvent");
+var SpanMessagingEvent = z.object({
+  system: z.string().optional(),
+  client_id: z.string().optional(),
+  operation: z.enum([
+    "publish",
+    "create",
+    "receive",
+    "deliver"
+  ]),
+  message: z.any(),
+  destination: z.string().optional()
+});
 
 // src/zodfetch.ts
 async function zodfetch(schema, url, requestInit) {
@@ -1317,9 +1432,9 @@ function flattenAttributes(obj, prefix) {
     if (Array.isArray(value)) {
       for (let i = 0; i < value.length; i++) {
         if (typeof value[i] === "object" && value[i] !== null) {
-          Object.assign(result, flattenAttributes(value[i], `${newPrefix}.${i}`));
+          Object.assign(result, flattenAttributes(value[i], `${newPrefix}.[${i}]`));
         } else {
-          result[`${newPrefix}.${i}`] = value[i];
+          result[`${newPrefix}.[${i}]`] = value[i];
         }
       }
     } else if (isRecord(value)) {
@@ -1338,32 +1453,45 @@ function isRecord(value) {
 }
 __name(isRecord, "isRecord");
 function unflattenAttributes(obj) {
-  if (obj === null || obj === void 0 || typeof obj === "string" || typeof obj === "number" || typeof obj === "boolean" || Array.isArray(obj)) {
+  if (typeof obj !== "object" || obj === null || Array.isArray(obj)) {
     return obj;
   }
   const result = {};
   for (const [key, value] of Object.entries(obj)) {
-    const parts = key.split(".");
+    const parts = key.split(".").reduce((acc, part) => {
+      if (detectIsArrayIndex(part)) {
+        acc.push(part);
+      } else {
+        acc.push(...part.split(/\.\[(.*?)\]/).filter(Boolean));
+      }
+      return acc;
+    }, []);
     let current = result;
     for (let i = 0; i < parts.length - 1; i++) {
       const part = parts[i];
-      if (typeof part === "string") {
-        const nextPart = parts[i + 1];
-        const isArray = nextPart ? parseInt(nextPart, 10).toString() === nextPart : false;
-        if (current[part] == null) {
-          current[part] = isArray ? [] : {};
-        }
-        current = current[part];
+      const isArray = detectIsArrayIndex(part);
+      const cleanPart = isArray ? part.substring(1, part.length - 1) : part;
+      const nextIsArray = detectIsArrayIndex(parts[i + 1]);
+      if (!current[cleanPart]) {
+        current[cleanPart] = nextIsArray ? [] : {};
       }
+      current = current[cleanPart];
     }
     const lastPart = parts[parts.length - 1];
-    if (typeof lastPart === "string") {
-      current[lastPart] = value;
-    }
+    const cleanLastPart = detectIsArrayIndex(lastPart) ? parseInt(lastPart.substring(1, lastPart.length - 1), 10) : lastPart;
+    current[cleanLastPart] = value;
   }
   return result;
 }
 __name(unflattenAttributes, "unflattenAttributes");
+function detectIsArrayIndex(key) {
+  const match = key.match(/^\[(\d+)\]$/);
+  if (match) {
+    return true;
+  }
+  return false;
+}
+__name(detectIsArrayIndex, "detectIsArrayIndex");
 function primitiveValueOrflattenedAttributes(obj, prefix) {
   if (typeof obj === "string" || typeof obj === "number" || typeof obj === "boolean" || obj === null || obj === void 0) {
     return obj;
@@ -1420,7 +1548,8 @@ var SemanticInternalAttributes = {
   METADATA: "$metadata",
   TRIGGER: "$trigger",
   PAYLOAD: "$payload",
-  SHOW_ACTIONS: "show.actions",
+  SHOW: "$show",
+  SHOW_ACTIONS: "$show.actions",
   WORKER_ID: "worker.id",
   WORKER_VERSION: "worker.version",
   CLI_VERSION: "cli.version",
@@ -1688,14 +1817,17 @@ var _ZodMessageHandler = class _ZodMessageHandler {
     };
   }
   registerHandlers(emitter, logger2) {
-    const log = logger2 ?? console.log;
+    const log = logger2 ?? console;
     if (!__privateGet(this, _handlers)) {
-      log("No handlers provided");
+      log.info("No handlers provided");
       return;
     }
     for (const eventName of Object.keys(__privateGet(this, _schema))) {
       emitter.on(eventName, async (message, callback) => {
-        log(`handling ${eventName}`, message);
+        log.info(`handling ${eventName}`, {
+          payload: message,
+          hasCallback: !!callback
+        });
         let ack;
         if ("payload" in message) {
           ack = await this.handleMessage({
@@ -1813,27 +1945,37 @@ var _ZodSocketMessageHandler = class _ZodSocketMessageHandler {
     };
   }
   registerHandlers(emitter, logger2) {
-    const log = logger2 ?? console.log;
+    const log = logger2 ?? console;
     if (!__privateGet(this, _handlers2)) {
-      log("No handlers provided");
+      log.info("No handlers provided");
       return;
     }
     for (const eventName of Object.keys(__privateGet(this, _handlers2))) {
       emitter.on(eventName, async (message, callback) => {
-        log(`handling ${eventName}`, message);
+        log.info(`handling ${eventName}`, {
+          payload: message,
+          hasCallback: !!callback
+        });
         let ack;
-        if ("payload" in message) {
-          ack = await this.handleMessage({
-            type: eventName,
-            ...message
-          });
-        } else {
-          const { version, ...payload } = message;
-          ack = await this.handleMessage({
-            type: eventName,
-            version,
-            payload
+        try {
+          if ("payload" in message) {
+            ack = await this.handleMessage({
+              type: eventName,
+              ...message
+            });
+          } else {
+            const { version, ...payload } = message;
+            ack = await this.handleMessage({
+              type: eventName,
+              version,
+              payload
+            });
+          }
+        } catch (error) {
+          log.error("Error while handling message", {
+            error
           });
+          return;
         }
         if (callback && typeof callback === "function") {
           callback(ack);
@@ -1904,7 +2046,9 @@ var _ZodSocketConnection = class _ZodSocketConnection {
       },
       extraHeaders: opts.extraHeaders
     });
-    __privateSet(this, _logger, createLogger(`[${opts.namespace}][${this.socket.id}]`));
+    __privateSet(this, _logger, new SimpleStructuredLogger(opts.namespace, LogLevel.info, {
+      socketId: this.socket.id
+    }));
     __privateSet(this, _handler, new ZodSocketMessageHandler({
       schema: opts.serverMessages,
       handlers: opts.handlers
@@ -1915,19 +2059,22 @@ var _ZodSocketConnection = class _ZodSocketConnection {
       socket: this.socket
     }));
     this.socket.on("connect_error", async (error) => {
-      __privateGet(this, _logger).call(this, `connect_error: ${error}`);
+      __privateGet(this, _logger).error(`connect_error: ${error}`);
       if (opts.onError) {
         await opts.onError(this.socket, error, __privateGet(this, _logger));
       }
     });
     this.socket.on("connect", async () => {
-      __privateGet(this, _logger).call(this, "connect");
+      __privateGet(this, _logger).info("connect");
       if (opts.onConnection) {
         await opts.onConnection(this.socket, __privateGet(this, _handler), __privateGet(this, _sender2), __privateGet(this, _logger));
       }
     });
     this.socket.on("disconnect", async (reason, description) => {
-      __privateGet(this, _logger).call(this, "disconnect");
+      __privateGet(this, _logger).info("disconnect", {
+        reason,
+        description
+      });
       if (opts.onDisconnect) {
         await opts.onDisconnect(this.socket, reason, description, __privateGet(this, _logger));
       }
@@ -1951,16 +2098,79 @@ _handler = new WeakMap();
 _logger = new WeakMap();
 __name(_ZodSocketConnection, "ZodSocketConnection");
 var ZodSocketConnection = _ZodSocketConnection;
-function createLogger(prefix) {
-  return (...args) => console.log(prefix, ...args);
-}
-__name(createLogger, "createLogger");
 
 // src/v3/zodNamespace.ts
-var _handler2;
+var LogLevel;
+(function(LogLevel2) {
+  LogLevel2[LogLevel2["log"] = 0] = "log";
+  LogLevel2[LogLevel2["error"] = 1] = "error";
+  LogLevel2[LogLevel2["warn"] = 2] = "warn";
+  LogLevel2[LogLevel2["info"] = 3] = "info";
+  LogLevel2[LogLevel2["debug"] = 4] = "debug";
+})(LogLevel || (LogLevel = {}));
+var _structuredLog, structuredLog_fn;
+var _SimpleStructuredLogger = class _SimpleStructuredLogger {
+  constructor(name, level = [
+    "1",
+    "true"
+  ].includes(process.env.DEBUG ?? "") ? LogLevel.debug : LogLevel.info, fields) {
+    __privateAdd(this, _structuredLog);
+    this.name = name;
+    this.level = level;
+    this.fields = fields;
+  }
+  child(fields, level) {
+    return new _SimpleStructuredLogger(this.name, level, {
+      ...this.fields,
+      ...fields
+    });
+  }
+  log(message, ...args) {
+    if (this.level < LogLevel.log)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.log, message, "log", ...args);
+  }
+  error(message, ...args) {
+    if (this.level < LogLevel.error)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.error, message, "error", ...args);
+  }
+  warn(message, ...args) {
+    if (this.level < LogLevel.warn)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.warn, message, "warn", ...args);
+  }
+  info(message, ...args) {
+    if (this.level < LogLevel.info)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.info, message, "info", ...args);
+  }
+  debug(message, ...args) {
+    if (this.level < LogLevel.debug)
+      return;
+    __privateMethod(this, _structuredLog, structuredLog_fn).call(this, console.debug, message, "debug", ...args);
+  }
+};
+_structuredLog = new WeakSet();
+structuredLog_fn = /* @__PURE__ */ __name(function(loggerFunction, message, level, ...args) {
+  const structuredLog = {
+    ...args.length === 1 ? args[0] : args,
+    ...this.fields,
+    timestamp: /* @__PURE__ */ new Date(),
+    name: this.name,
+    message,
+    level
+  };
+  loggerFunction(JSON.stringify(structuredLog));
+}, "#structuredLog");
+__name(_SimpleStructuredLogger, "SimpleStructuredLogger");
+var SimpleStructuredLogger = _SimpleStructuredLogger;
+var _logger2, _handler2;
 var _ZodNamespace = class _ZodNamespace {
   constructor(opts) {
+    __privateAdd(this, _logger2, void 0);
     __privateAdd(this, _handler2, void 0);
+    __privateSet(this, _logger2, opts.logger ?? new SimpleStructuredLogger(opts.name));
     __privateSet(this, _handler2, new ZodSocketMessageHandler({
       schema: opts.clientMessages,
       handlers: opts.handlers
@@ -1982,7 +2192,10 @@ var _ZodNamespace = class _ZodNamespace {
     });
     if (opts.preAuth) {
       this.namespace.use(async (socket, next) => {
-        const logger2 = createLogger2(`[${opts.name}][${socket.id}][preAuth]`);
+        const logger2 = __privateGet(this, _logger2).child({
+          socketId: socket.id,
+          socketStage: "preAuth"
+        });
         if (typeof opts.preAuth === "function") {
           await opts.preAuth(socket, next, logger2);
         }
@@ -1990,34 +2203,43 @@ var _ZodNamespace = class _ZodNamespace {
     }
     if (opts.authToken) {
       this.namespace.use((socket, next) => {
-        const logger2 = createLogger2(`[${opts.name}][${socket.id}][auth]`);
+        const logger2 = __privateGet(this, _logger2).child({
+          socketId: socket.id,
+          socketStage: "auth"
+        });
         const { auth } = socket.handshake;
         if (!("token" in auth)) {
-          logger2("no token");
+          logger2.error("no token");
           return socket.disconnect(true);
         }
         if (auth.token !== opts.authToken) {
-          logger2("invalid token");
+          logger2.error("invalid token");
           return socket.disconnect(true);
         }
-        logger2("success");
+        logger2.info("success");
         next();
       });
     }
     if (opts.postAuth) {
       this.namespace.use(async (socket, next) => {
-        const logger2 = createLogger2(`[${opts.name}][${socket.id}][postAuth]`);
+        const logger2 = __privateGet(this, _logger2).child({
+          socketId: socket.id,
+          socketStage: "auth"
+        });
         if (typeof opts.postAuth === "function") {
           await opts.postAuth(socket, next, logger2);
         }
       });
     }
     this.namespace.on("connection", async (socket) => {
-      const logger2 = createLogger2(`[${opts.name}][${socket.id}]`);
-      logger2("connection");
+      const logger2 = __privateGet(this, _logger2).child({
+        socketId: socket.id,
+        socketStage: "connection"
+      });
+      logger2.info("connected");
       __privateGet(this, _handler2).registerHandlers(socket, logger2);
       socket.on("disconnect", async (reason, description) => {
-        logger2("disconnect", {
+        logger2.info("disconnect", {
           reason,
           description
         });
@@ -2026,7 +2248,9 @@ var _ZodNamespace = class _ZodNamespace {
         }
       });
       socket.on("error", async (error) => {
-        logger2("error", error);
+        logger2.error("error", {
+          error
+        });
         if (opts.onError) {
           await opts.onError(socket, error, logger2);
         }
@@ -2040,13 +2264,10 @@ var _ZodNamespace = class _ZodNamespace {
     return this.namespace.fetchSockets();
   }
 };
+_logger2 = new WeakMap();
 _handler2 = new WeakMap();
 __name(_ZodNamespace, "ZodNamespace");
 var ZodNamespace = _ZodNamespace;
-function createLogger2(prefix) {
-  return (...args) => console.log(prefix, ...args);
-}
-__name(createLogger2, "createLogger");
 var messageSchema2 = z.object({
   version: z.literal("v1").default("v1"),
   type: z.string(),
@@ -2821,6 +3042,7 @@ var _ProdRuntimeManager = class _ProdRuntimeManager {
   }
   async waitForDuration(ms) {
     let timeout;
+    const now = Date.now();
     const resolveAfterDuration = new Promise((resolve) => {
       timeout = setTimeout(resolve, ms);
     });
@@ -2835,13 +3057,19 @@ var _ProdRuntimeManager = class _ProdRuntimeManager {
       };
     });
     const { willCheckpointAndRestore } = await this.ipc.sendWithAck("WAIT_FOR_DURATION", {
-      ms
+      ms,
+      now
     });
     if (!willCheckpointAndRestore) {
       await resolveAfterDuration;
       return;
     }
-    await waitForRestore;
+    this.ipc.send("READY_FOR_CHECKPOINT", {});
+    await Promise.race([
+      waitForRestore,
+      resolveAfterDuration
+    ]);
+    this.ipc.send("CANCEL_CHECKPOINT", {});
     clearTimeout(timeout);
   }
   resumeAfterRestore() {
@@ -2862,7 +3090,7 @@ var _ProdRuntimeManager = class _ProdRuntimeManager {
       });
     });
     await this.ipc.send("WAIT_FOR_TASK", {
-      id: params.id
+      friendlyId: params.id
     });
     return await promise;
   }
@@ -2882,8 +3110,8 @@ var _ProdRuntimeManager = class _ProdRuntimeManager {
       });
     }));
     await this.ipc.send("WAIT_FOR_BATCH", {
-      id: params.id,
-      runs: params.runs
+      batchFriendlyId: params.id,
+      runFriendlyIds: params.runs
     });
     const results = await promise;
     return {
@@ -3176,6 +3404,16 @@ var defaultRetryOptions = {
   maxTimeoutInMs: 6e4,
   randomize: true
 };
+({
+  byStatus: {
+    "429,408,409,5xx": {
+      strategy: "backoff",
+      ...defaultRetryOptions
+    }
+  },
+  connectionError: defaultRetryOptions,
+  timeout: defaultRetryOptions
+});
 function calculateNextRetryDelay(options, attempt) {
   const opts = {
     ...defaultRetryOptions,
@@ -3747,21 +3985,17 @@ var TaskExecutor = _TaskExecutor;
 // package.json
 var dependencies = {
   "@google-cloud/precise-date": "^4.0.0",
-  "@opentelemetry/api": "^1.7.0",
+  "@opentelemetry/api": "^1.8.0",
   "@opentelemetry/api-logs": "^0.48.0",
-  "@opentelemetry/auto-instrumentations-node": "^0.40.3",
-  "@opentelemetry/exporter-collector": "^0.25.0",
-  "@opentelemetry/exporter-logs-otlp-http": "^0.48.0",
-  "@opentelemetry/exporter-trace-otlp-http": "^0.48.0",
-  "@opentelemetry/instrumentation": "^0.48.0",
-  "@opentelemetry/instrumentation-fetch": "^0.48.0",
-  "@opentelemetry/instrumentation-http": "^0.48.0",
-  "@opentelemetry/resources": "^1.21.0",
-  "@opentelemetry/sdk-logs": "^0.48.0",
-  "@opentelemetry/sdk-node": "^0.48.0",
-  "@opentelemetry/sdk-trace-base": "^1.21.0",
-  "@opentelemetry/sdk-trace-node": "^1.21.0",
-  "@opentelemetry/semantic-conventions": "^1.21.0",
+  "@opentelemetry/exporter-logs-otlp-http": "^0.49.1",
+  "@opentelemetry/exporter-trace-otlp-http": "^0.49.1",
+  "@opentelemetry/instrumentation": "^0.49.1",
+  "@opentelemetry/resources": "^1.22.0",
+  "@opentelemetry/sdk-logs": "^0.49.1",
+  "@opentelemetry/sdk-node": "^0.49.1",
+  "@opentelemetry/sdk-trace-base": "^1.22.0",
+  "@opentelemetry/sdk-trace-node": "^1.22.0",
+  "@opentelemetry/semantic-conventions": "^1.22.0",
   "humanize-duration": "^3.27.3",
   "socket.io": "^4.7.4",
   "socket.io-client": "^4.7.4",
@@ -3786,6 +4020,6 @@ function parseBatchTriggerTaskRequestBody(body) {
 }
 __name(parseBatchTriggerTaskRequestBody, "parseBatchTriggerTaskRequestBody");
 
-export { ApiClient, ApiClientManager, BackgroundWorkerClientMessages, BackgroundWorkerMetadata, BackgroundWorkerProperties, BackgroundWorkerServerMessages, BatchTaskRunExecutionResult, BatchTriggerTaskRequestBody, BatchTriggerTaskResponse, CancellationSpanEvent, ClientToSharedQueueMessages, Config, ConsoleInterceptor, CoordinatorToPlatformMessages, CoordinatorToProdWorkerMessages, CreateAuthorizationCodeResponseSchema, CreateBackgroundWorkerRequestBody, CreateBackgroundWorkerResponse, DevRuntimeManager, EventFilter, ExceptionEventProperties, ExceptionSpanEvent, ExternalBuildData, FetchRetryBackoffStrategy, FetchRetryHeadersStrategy, FetchRetryOptions, FetchRetryStrategy, FetchTimeoutOptions, FixedWindowRateLimit, GetBatchResponseBody, GetDeploymentResponseBody, GetEnvironmentVariablesResponseBody, GetPersonalAccessTokenRequestSchema, GetPersonalAccessTokenResponseSchema, GetProjectEnvResponse, GetProjectResponseBody, GetProjectsResponseBody, ImageDetailsMetadata, InitializeDeploymentRequestBody, InitializeDeploymentResponseBody, Machine, OtelTaskLogger, OtherSpanEvent, PRIMARY_VARIANT, PlatformToCoordinatorMessages, PlatformToProviderMessages, ProdChildToWorkerMessages, ProdRuntimeManager, ProdTaskRunExecution, ProdTaskRunExecutionPayload, ProdWorkerSocketData, ProdWorkerToChildMessages, ProdWorkerToCoordinatorMessages, ProviderToPlatformMessages, QueueOptions, RateLimitOptions, RetryOptions, SemanticInternalAttributes, SharedQueueToClientMessages, SlidingWindowRateLimit, SpanEvent, SpanEvents, StartDeploymentIndexingRequestBody, StartDeploymentIndexingResponseBody, TaskContextSpanProcessor, TaskEventStyle, TaskExecutor, TaskMetadata, TaskMetadataWithFilePath, TaskResource, TaskRun, TaskRunBuiltInError, TaskRunContext, TaskRunCustomErrorObject, TaskRunError, TaskRunErrorCodes, TaskRunExecution, TaskRunExecutionAttempt, TaskRunExecutionBatch, TaskRunExecutionEnvironment, TaskRunExecutionOrganization, TaskRunExecutionPayload, TaskRunExecutionProject, TaskRunExecutionQueue, TaskRunExecutionResult, TaskRunExecutionRetry, TaskRunExecutionTask, TaskRunFailedExecutionResult, TaskRunInternalError, TaskRunStringError, TaskRunSuccessfulExecutionResult, TracingSDK, TriggerTaskRequestBody, TriggerTaskResponse, TriggerTracer, UncaughtExceptionMessage, WhoAmIResponseSchema, ZodIpcConnection, ZodMessageHandler, ZodMessageSchema, ZodMessageSender, ZodNamespace, ZodSocketConnection, ZodSocketMessageHandler, ZodSocketMessageSender, accessoryAttributes, apiClientManager, calculateNextRetryDelay, calculatePreciseDateHrTime, calculateResetAt2 as calculateResetAt, childToWorkerMessages, clientWebsocketMessages, correctErrorStackTrace, createErrorTaskError, defaultRetryOptions, detectDependencyVersion, eventFilterMatches, flattenAttributes, formatDuration, formatDurationInDays, formatDurationMilliseconds, formatDurationNanoseconds, getEnvVar, iconStringForSeverity, isCancellationSpanEvent, isExceptionSpanEvent, logger, millisecondsToNanoseconds, nanosecondsToMilliseconds, omit, parseBatchTriggerTaskRequestBody, parseError, parseTriggerTaskRequestBody, preciseDateOriginNow, primitiveValueOrflattenedAttributes, recordSpanException, runtime, serverWebsocketMessages, stringPatternMatchers, taskContextManager, unflattenAttributes, workerToChildMessages };
+export { ApiClient, ApiClientManager, BackgroundWorkerClientMessages, BackgroundWorkerMetadata, BackgroundWorkerProperties, BackgroundWorkerServerMessages, BatchTaskRunExecutionResult, BatchTriggerTaskRequestBody, BatchTriggerTaskResponse, CancellationSpanEvent, ClientToSharedQueueMessages, Config, ConsoleInterceptor, CoordinatorToPlatformMessages, CoordinatorToProdWorkerMessages, CreateAuthorizationCodeResponseSchema, CreateBackgroundWorkerRequestBody, CreateBackgroundWorkerResponse, DevRuntimeManager, EventFilter, ExceptionEventProperties, ExceptionSpanEvent, ExternalBuildData, FetchRetryBackoffStrategy, FetchRetryByStatusOptions, FetchRetryHeadersStrategy, FetchRetryOptions, FetchRetryStrategy, FetchTimeoutOptions, FixedWindowRateLimit, GetBatchResponseBody, GetDeploymentResponseBody, GetEnvironmentVariablesResponseBody, GetPersonalAccessTokenRequestSchema, GetPersonalAccessTokenResponseSchema, GetProjectEnvResponse, GetProjectResponseBody, GetProjectsResponseBody, ImageDetailsMetadata, InitializeDeploymentRequestBody, InitializeDeploymentResponseBody, LogLevel, Machine, OtelTaskLogger, OtherSpanEvent, PRIMARY_VARIANT, PlatformToCoordinatorMessages, PlatformToProviderMessages, ProdChildToWorkerMessages, ProdRuntimeManager, ProdTaskRunExecution, ProdTaskRunExecutionPayload, ProdWorkerSocketData, ProdWorkerToChildMessages, ProdWorkerToCoordinatorMessages, ProviderToPlatformMessages, QueueOptions, RateLimitOptions, RetryOptions, SemanticInternalAttributes, SharedQueueToClientMessages, SimpleStructuredLogger, SlidingWindowRateLimit, SpanEvent, SpanEvents, SpanMessagingEvent, StartDeploymentIndexingRequestBody, StartDeploymentIndexingResponseBody, TaskContextSpanProcessor, TaskEventStyle, TaskExecutor, TaskMetadata, TaskMetadataWithFilePath, TaskResource, TaskRun, TaskRunBuiltInError, TaskRunContext, TaskRunCustomErrorObject, TaskRunError, TaskRunErrorCodes, TaskRunExecution, TaskRunExecutionAttempt, TaskRunExecutionBatch, TaskRunExecutionEnvironment, TaskRunExecutionOrganization, TaskRunExecutionPayload, TaskRunExecutionProject, TaskRunExecutionQueue, TaskRunExecutionResult, TaskRunExecutionRetry, TaskRunExecutionTask, TaskRunFailedExecutionResult, TaskRunInternalError, TaskRunStringError, TaskRunSuccessfulExecutionResult, TracingSDK, TriggerTaskRequestBody, TriggerTaskResponse, TriggerTracer, UncaughtExceptionMessage, WaitReason, WhoAmIResponseSchema, ZodIpcConnection, ZodMessageHandler, ZodMessageSchema, ZodMessageSender, ZodNamespace, ZodSocketConnection, ZodSocketMessageHandler, ZodSocketMessageSender, accessoryAttributes, apiClientManager, calculateNextRetryDelay, calculatePreciseDateHrTime, calculateResetAt2 as calculateResetAt, childToWorkerMessages, clientWebsocketMessages, correctErrorStackTrace, createErrorTaskError, defaultRetryOptions, detectDependencyVersion, eventFilterMatches, flattenAttributes, formatDuration, formatDurationInDays, formatDurationMilliseconds, formatDurationNanoseconds, getEnvVar, iconStringForSeverity, isCancellationSpanEvent, isExceptionSpanEvent, logger, millisecondsToNanoseconds, nanosecondsToMilliseconds, omit, parseBatchTriggerTaskRequestBody, parseError, parseTriggerTaskRequestBody, preciseDateOriginNow, primitiveValueOrflattenedAttributes, recordSpanException, runtime, serverWebsocketMessages, stringPatternMatchers, taskContextManager, unflattenAttributes, workerToChildMessages };
 //# sourceMappingURL=out.js.map
 //# sourceMappingURL=index.mjs.map