@trigger.dev/sdk 3.0.0-beta.2 → 3.0.0-beta.21

package/dist/v3/index.mjs

@@ -1,57 +1,65 @@
 import { SpanKind, trace, context, SpanStatusCode } from '@opentelemetry/api';
-import { SEMATTRS_HTTP_STATUS_CODE, SEMATTRS_HTTP_METHOD, SEMATTRS_HTTP_URL, SEMATTRS_HTTP_HOST, SEMATTRS_HTTP_SCHEME, SEMATTRS_HTTP_RESPONSE_CONTENT_LENGTH, SEMATTRS_MESSAGING_OPERATION, SEMATTRS_MESSAGING_DESTINATION, SEMATTRS_MESSAGING_SYSTEM } from '@opentelemetry/semantic-conventions';
-import { TriggerTracer, runtime, SemanticInternalAttributes, accessoryAttributes, defaultRetryOptions, conditionallyImportPacket, parsePacket, createErrorTaskError, apiClientManager, calculateNextRetryDelay, defaultFetchRetryOptions, calculateResetAt, eventFilterMatches, flattenAttributes, taskContextManager } from '@trigger.dev/core/v3';
-export { logger } from '@trigger.dev/core/v3';
+import { SEMATTRS_MESSAGING_OPERATION, SEMATTRS_MESSAGING_DESTINATION, SEMATTRS_MESSAGING_SYSTEM, SEMATTRS_HTTP_STATUS_CODE, SEMATTRS_HTTP_METHOD, SEMATTRS_HTTP_URL, SEMATTRS_HTTP_HOST, SEMATTRS_HTTP_SCHEME, SEMATTRS_HTTP_RESPONSE_CONTENT_LENGTH } from '@opentelemetry/semantic-conventions';
+import { TriggerTracer, runtime, SemanticInternalAttributes, accessoryAttributes, apiClientManager, taskCatalog, stringifyIO, taskContext, logger, defaultRetryOptions, conditionallyImportPacket, parsePacket, createErrorTaskError, calculateNextRetryDelay, defaultFetchRetryOptions, calculateResetAt, eventFilterMatches, flattenAttributes } from '@trigger.dev/core/v3';
+export { APIError, AuthenticationError, BadRequestError, ConflictError, InternalServerError, NotFoundError, PermissionDeniedError, RateLimitError, UnprocessableEntityError, logger } from '@trigger.dev/core/v3';
 import { AsyncLocalStorage } from 'node:async_hooks';
 
 var __defProp = Object.defineProperty;
 var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
 var __publicField = (obj, key, value) => {
   __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
   return value;
 };
 
 // package.json
-var version = "3.0.0-beta.2";
+var version = "3.0.0-beta.21";
 var tracer = new TriggerTracer({
   name: "@trigger.dev/sdk",
   version
 });
 
 // src/v3/shared.ts
+function queue(options) {
+  return options;
+}
+__name(queue, "queue");
 function createTask(params) {
-  const task2 = {
-    trigger: async ({ payload, options }) => {
+  const task3 = {
+    id: params.id,
+    trigger: async (payload, options) => {
       const apiClient = apiClientManager.client;
       if (!apiClient) {
         throw apiClientMissingError();
       }
-      const taskMetadata = runtime.getTaskMetadata(params.id);
+      const taskMetadata = taskCatalog.getTaskMetadata(params.id);
+      const payloadPacket = await stringifyIO(payload);
       const handle = await tracer.startActiveSpan(taskMetadata ? "Trigger" : `${params.id} trigger()`, async (span) => {
         const response = await apiClient.triggerTask(params.id, {
-          payload,
+          payload: payloadPacket.data,
           options: {
             queue: params.queue,
             concurrencyKey: options?.concurrencyKey,
-            test: taskContextManager.ctx?.run.isTest
+            test: taskContext.ctx?.run.isTest,
+            payloadType: payloadPacket.dataType,
+            idempotencyKey: options?.idempotencyKey
           }
         }, {
           spanParentAsLink: true
         });
-        if (!response.ok) {
-          throw new Error(response.error);
-        }
-        span.setAttribute("messaging.message.id", response.data.id);
-        return response.data;
+        span.setAttribute("messaging.message.id", response.id);
+        return response;
       }, {
         kind: SpanKind.PRODUCER,
         attributes: {
           [SEMATTRS_MESSAGING_OPERATION]: "publish",
           [SemanticInternalAttributes.STYLE_ICON]: "trigger",
-          ["messaging.client_id"]: taskContextManager.worker?.id,
+          ["messaging.client_id"]: taskContext.worker?.id,
           [SEMATTRS_MESSAGING_DESTINATION]: params.queue?.name ?? params.id,
-          ["messaging.message.body.size"]: JSON.stringify(payload).length,
           [SEMATTRS_MESSAGING_SYSTEM]: "trigger.dev",
           ...taskMetadata ? accessoryAttributes({
             items: [
@@ -66,38 +74,39 @@ function createTask(params) {
       });
       return handle;
     },
-    batchTrigger: async ({ items }) => {
+    batchTrigger: async (items) => {
       const apiClient = apiClientManager.client;
       if (!apiClient) {
         throw apiClientMissingError();
       }
-      const taskMetadata = runtime.getTaskMetadata(params.id);
+      const taskMetadata = taskCatalog.getTaskMetadata(params.id);
       const response = await tracer.startActiveSpan(taskMetadata ? "Batch trigger" : `${params.id} batchTrigger()`, async (span) => {
         const response2 = await apiClient.batchTriggerTask(params.id, {
-          items: items.map((item) => ({
-            payload: item.payload,
-            options: {
-              queue: item.options?.queue ?? params.queue,
-              concurrencyKey: item.options?.concurrencyKey,
-              test: taskContextManager.ctx?.run.isTest
-            }
+          items: await Promise.all(items.map(async (item) => {
+            const payloadPacket = await stringifyIO(item.payload);
+            return {
+              payload: payloadPacket.data,
+              options: {
+                queue: item.options?.queue ?? params.queue,
+                concurrencyKey: item.options?.concurrencyKey,
+                test: taskContext.ctx?.run.isTest,
+                payloadType: payloadPacket.dataType,
+                idempotencyKey: item.options?.idempotencyKey
+              }
+            };
           }))
         }, {
           spanParentAsLink: true
         });
-        if (!response2.ok) {
-          throw new Error(response2.error);
-        }
-        span.setAttribute("messaging.message.id", response2.data.batchId);
-        return response2.data;
+        span.setAttribute("messaging.message.id", response2.batchId);
+        return response2;
       }, {
         kind: SpanKind.PRODUCER,
         attributes: {
           [SEMATTRS_MESSAGING_OPERATION]: "publish",
           ["messaging.batch.message_count"]: items.length,
-          ["messaging.client_id"]: taskContextManager.worker?.id,
+          ["messaging.client_id"]: taskContext.worker?.id,
           [SEMATTRS_MESSAGING_DESTINATION]: params.queue?.name ?? params.id,
-          ["messaging.message.body.size"]: items.map((item) => JSON.stringify(item.payload)).join("").length,
           [SEMATTRS_MESSAGING_SYSTEM]: "trigger.dev",
           [SemanticInternalAttributes.STYLE_ICON]: "trigger",
           ...taskMetadata ? accessoryAttributes({
@@ -113,8 +122,8 @@ function createTask(params) {
       });
       return response;
     },
-    triggerAndWait: async ({ payload, options }) => {
-      const ctx = taskContextManager.ctx;
+    triggerAndWait: async (payload, options) => {
+      const ctx = taskContext.ctx;
       if (!ctx) {
         throw new Error("triggerAndWait can only be used from inside a task.run()");
       }
@@ -122,37 +131,43 @@ function createTask(params) {
       if (!apiClient) {
         throw apiClientMissingError();
       }
-      const taskMetadata = runtime.getTaskMetadata(params.id);
+      const taskMetadata = taskCatalog.getTaskMetadata(params.id);
+      const payloadPacket = await stringifyIO(payload);
       return await tracer.startActiveSpan(taskMetadata ? "Trigger" : `${params.id} triggerAndWait()`, async (span) => {
         const response = await apiClient.triggerTask(params.id, {
-          payload,
+          payload: payloadPacket.data,
           options: {
             dependentAttempt: ctx.attempt.id,
-            lockToVersion: taskContextManager.worker?.version,
+            lockToVersion: taskContext.worker?.version,
             queue: params.queue,
             concurrencyKey: options?.concurrencyKey,
-            test: taskContextManager.ctx?.run.isTest
+            test: taskContext.ctx?.run.isTest,
+            payloadType: payloadPacket.dataType,
+            idempotencyKey: options?.idempotencyKey
           }
         });
-        if (!response.ok) {
-          throw new Error(response.error);
+        span.setAttribute("messaging.message.id", response.id);
+        if (options?.idempotencyKey) {
+          const result2 = await apiClient.getRunResult(response.id);
+          if (result2) {
+            logger.log(`Result reused from previous task run with idempotency key '${options.idempotencyKey}'.`, {
+              runId: response.id,
+              idempotencyKey: options.idempotencyKey
+            });
+            return await handleTaskRunExecutionResult(result2);
+          }
         }
-        span.setAttribute("messaging.message.id", response.data.id);
         const result = await runtime.waitForTask({
-          id: response.data.id,
+          id: response.id,
           ctx
         });
-        const runResult = await handleTaskRunExecutionResult(result);
-        if (!runResult.ok) {
-          throw runResult.error;
-        }
-        return runResult.output;
+        return await handleTaskRunExecutionResult(result);
       }, {
         kind: SpanKind.PRODUCER,
         attributes: {
           [SemanticInternalAttributes.STYLE_ICON]: "trigger",
           [SEMATTRS_MESSAGING_OPERATION]: "publish",
-          ["messaging.client_id"]: taskContextManager.worker?.id,
+          ["messaging.client_id"]: taskContext.worker?.id,
           [SEMATTRS_MESSAGING_DESTINATION]: params.queue?.name ?? params.id,
           [SEMATTRS_MESSAGING_SYSTEM]: "trigger.dev",
           ...taskMetadata ? accessoryAttributes({
@@ -167,8 +182,8 @@ function createTask(params) {
         }
       });
     },
-    batchTriggerAndWait: async ({ items }) => {
-      const ctx = taskContextManager.ctx;
+    batchTriggerAndWait: async (items) => {
+      const ctx = taskContext.ctx;
       if (!ctx) {
         throw new Error("batchTriggerAndWait can only be used from inside a task.run()");
       }
@@ -176,42 +191,78 @@ function createTask(params) {
       if (!apiClient) {
         throw apiClientMissingError();
       }
-      const taskMetadata = runtime.getTaskMetadata(params.id);
+      const taskMetadata = taskCatalog.getTaskMetadata(params.id);
       return await tracer.startActiveSpan(taskMetadata ? "Batch trigger" : `${params.id} batchTriggerAndWait()`, async (span) => {
         const response = await apiClient.batchTriggerTask(params.id, {
-          items: items.map((item) => ({
-            payload: item.payload,
-            options: {
-              lockToVersion: taskContextManager.worker?.version,
-              queue: item.options?.queue ?? params.queue,
-              concurrencyKey: item.options?.concurrencyKey,
-              test: taskContextManager.ctx?.run.isTest
-            }
+          items: await Promise.all(items.map(async (item) => {
+            const payloadPacket = await stringifyIO(item.payload);
+            return {
+              payload: payloadPacket.data,
+              options: {
+                lockToVersion: taskContext.worker?.version,
+                queue: item.options?.queue ?? params.queue,
+                concurrencyKey: item.options?.concurrencyKey,
+                test: taskContext.ctx?.run.isTest,
+                payloadType: payloadPacket.dataType,
+                idempotencyKey: item.options?.idempotencyKey
+              }
+            };
           })),
           dependentAttempt: ctx.attempt.id
         });
-        if (!response.ok) {
-          throw new Error(response.error);
+        span.setAttribute("messaging.message.id", response.batchId);
+        const getBatchResults = /* @__PURE__ */ __name(async () => {
+          const hasIdempotencyKey = items.some((item) => item.options?.idempotencyKey);
+          if (hasIdempotencyKey) {
+            const results = await apiClient.getBatchResults(response.batchId);
+            if (results) {
+              return results;
+            }
+          }
+          return {
+            id: response.batchId,
+            items: []
+          };
+        }, "getBatchResults");
+        const existingResults = await getBatchResults();
+        const incompleteRuns = response.runs.filter((runId) => !existingResults.items.some((item) => item.id === runId));
+        if (incompleteRuns.length === 0) {
+          logger.log(`Results reused from previous task runs because of the provided idempotency keys.`);
+          const runs3 = await handleBatchTaskRunExecutionResult(existingResults.items);
+          return {
+            id: existingResults.id,
+            runs: runs3
+          };
         }
-        span.setAttribute("messaging.message.id", response.data.batchId);
         const result = await runtime.waitForBatch({
-          id: response.data.batchId,
-          runs: response.data.runs,
+          id: response.batchId,
+          runs: incompleteRuns,
           ctx
         });
-        const runs = await handleBatchTaskRunExecutionResult(result.items);
+        const combinedItems = [];
+        for (const runId of response.runs) {
+          const existingItem = existingResults.items.find((item) => item.id === runId);
+          if (existingItem) {
+            combinedItems.push(existingItem);
+          } else {
+            const newItem = result.items.find((item) => item.id === runId);
+            if (newItem) {
+              combinedItems.push(newItem);
+            }
+          }
+        }
+        const runs2 = await handleBatchTaskRunExecutionResult(combinedItems);
         return {
           id: result.id,
-          runs
+          runs: runs2
         };
       }, {
         kind: SpanKind.PRODUCER,
         attributes: {
           [SEMATTRS_MESSAGING_OPERATION]: "publish",
           ["messaging.batch.message_count"]: items.length,
-          ["messaging.client_id"]: taskContextManager.worker?.id,
+          ["messaging.client_id"]: taskContext.worker?.id,
           [SEMATTRS_MESSAGING_DESTINATION]: params.queue?.name ?? params.id,
-          ["messaging.message.body.size"]: items.map((item) => JSON.stringify(item.payload)).join("").length,
           [SEMATTRS_MESSAGING_SYSTEM]: "trigger.dev",
           [SemanticInternalAttributes.STYLE_ICON]: "trigger",
           ...taskMetadata ? accessoryAttributes({
@@ -227,27 +278,27 @@ function createTask(params) {
       });
     }
   };
-  Object.defineProperty(task2, "__trigger", {
-    value: {
-      id: params.id,
-      packageVersion: version,
-      queue: params.queue,
-      retry: params.retry ? {
-        ...defaultRetryOptions,
-        ...params.retry
-      } : void 0,
-      machine: params.machine,
-      fns: {
-        run: params.run,
-        init: params.init,
-        cleanup: params.cleanup,
-        middleware: params.middleware,
-        handleError: params.handleError
-      }
-    },
-    enumerable: false
+  taskCatalog.registerTaskMetadata({
+    id: params.id,
+    packageVersion: version,
+    queue: params.queue,
+    retry: params.retry ? {
+      ...defaultRetryOptions,
+      ...params.retry
+    } : void 0,
+    machine: params.machine,
+    fns: {
+      run: params.run,
+      init: params.init,
+      cleanup: params.cleanup,
+      middleware: params.middleware,
+      handleError: params.handleError,
+      onSuccess: params.onSuccess,
+      onFailure: params.onFailure,
+      onStart: params.onStart
+    }
   });
-  return task2;
+  return task3;
 }
 __name(createTask, "createTask");
 async function handleBatchTaskRunExecutionResult(items) {
@@ -877,7 +928,104 @@ var retry = {
   fetch: retryFetch,
   interceptFetch
 };
+var runs = {
+  replay: replayRun,
+  cancel: cancelRun
+};
+async function replayRun(runId) {
+  const apiClient = apiClientManager.client;
+  if (!apiClient) {
+    throw apiClientMissingError();
+  }
+  return await apiClient.replayRun(runId);
+}
+__name(replayRun, "replayRun");
+async function cancelRun(runId) {
+  const apiClient = apiClientManager.client;
+  if (!apiClient) {
+    throw apiClientMissingError();
+  }
+  return await apiClient.cancelRun(runId);
+}
+__name(cancelRun, "cancelRun");
+
+// src/v3/schedules/index.ts
+var schedules_exports = {};
+__export(schedules_exports, {
+  activate: () => activate,
+  create: () => create,
+  deactivate: () => deactivate,
+  del: () => del,
+  list: () => list,
+  retrieve: () => retrieve,
+  task: () => task2,
+  update: () => update
+});
+function task2(params) {
+  const task3 = createTask(params);
+  taskCatalog.updateTaskMetadata(task3.id, {
+    triggerSource: "schedule"
+  });
+  return task3;
+}
+__name(task2, "task");
+async function create(options) {
+  const apiClient = apiClientManager.client;
+  if (!apiClient) {
+    throw apiClientMissingError();
+  }
+  return apiClient.createSchedule(options);
+}
+__name(create, "create");
+async function retrieve(scheduleId) {
+  const apiClient = apiClientManager.client;
+  if (!apiClient) {
+    throw apiClientMissingError();
+  }
+  return apiClient.retrieveSchedule(scheduleId);
+}
+__name(retrieve, "retrieve");
+async function update(scheduleId, options) {
+  const apiClient = apiClientManager.client;
+  if (!apiClient) {
+    throw apiClientMissingError();
+  }
+  return apiClient.updateSchedule(scheduleId, options);
+}
+__name(update, "update");
+async function del(scheduleId) {
+  const apiClient = apiClientManager.client;
+  if (!apiClient) {
+    throw apiClientMissingError();
+  }
+  return apiClient.deleteSchedule(scheduleId);
+}
+__name(del, "del");
+async function deactivate(scheduleId) {
+  const apiClient = apiClientManager.client;
+  if (!apiClient) {
+    throw apiClientMissingError();
+  }
+  return apiClient.deactivateSchedule(scheduleId);
+}
+__name(deactivate, "deactivate");
+async function activate(scheduleId) {
+  const apiClient = apiClientManager.client;
+  if (!apiClient) {
+    throw apiClientMissingError();
+  }
+  return apiClient.activateSchedule(scheduleId);
+}
+__name(activate, "activate");
+async function list(options) {
+  const apiClient = apiClientManager.client;
+  if (!apiClient) {
+    throw apiClientMissingError();
+  }
+  return apiClient.listSchedules(options);
+}
+__name(list, "list");
 
-export { InMemoryCache, createCache, retry, task, wait };
+export { InMemoryCache, createCache, queue, retry, runs, schedules_exports as schedules, task, wait };
 //# sourceMappingURL=out.js.map
 //# sourceMappingURL=index.mjs.map