chore: import upstream snapshot with attribution

2026-07-13 13:32:57 +08:00
commit cd420f9332
4811 changed files with 884702 additions and 0 deletions
@@ -0,0 +1,96 @@
+import { packetRequiresOffloading, stringifyIO } from "@trigger.dev/core/v3";
+import { type AuthenticatedEnvironment } from "~/services/apiAuth.server";
+import { BaseService } from "./baseService.server";
+import { env } from "~/env.server";
+import { handleMetadataPacket } from "~/utils/packets";
+import { type RunTemplateData } from "../taskRunTemplate";
+
+export class TaskRunTemplateService extends BaseService {
+  public async call(environment: AuthenticatedEnvironment, data: RunTemplateData) {
+    const { triggerSource } = data;
+
+    switch (triggerSource) {
+      case "STANDARD": {
+        const packet = { data: JSON.stringify(data.payload), dataType: "application/json" };
+
+        const { needsOffloading } = packetRequiresOffloading(
+          packet,
+          env.TASK_PAYLOAD_OFFLOAD_THRESHOLD
+        );
+
+        if (needsOffloading) {
+          // we currently disallow large payloads in task run templates
+          throw new Error("Payload too large");
+        }
+
+        const metadataPacket = data.metadata
+          ? handleMetadataPacket(
+              data.metadata,
+              "application/json",
+              env.TASK_RUN_METADATA_MAXIMUM_SIZE
+            )
+          : undefined;
+
+        const taskRunTemplate = await this._prisma.taskRunTemplate.create({
+          data: {
+            taskSlug: data.taskIdentifier,
+            triggerSource: "STANDARD",
+            label: data.label,
+            payload: packet.data,
+            payloadType: packet.dataType,
+            metadata: metadataPacket?.data,
+            metadataType: metadataPacket?.dataType,
+            queue: data.queue,
+            ttlSeconds: data.ttlSeconds,
+            concurrencyKey: data.concurrencyKey,
+            maxAttempts: data.maxAttempts,
+            maxDurationSeconds: data.maxDurationSeconds,
+            tags: data.tags ?? [],
+            machinePreset: data.machine,
+            projectId: environment.projectId,
+            organizationId: environment.organizationId,
+          },
+        });
+
+        return taskRunTemplate;
+      }
+      case "SCHEDULED": {
+        const payload = {
+          scheduleId: "sched_1234",
+          type: "IMPERATIVE",
+          timestamp: data.timestamp,
+          lastTimestamp: data.lastTimestamp,
+          timezone: data.timezone,
+          externalId: data.externalId,
+          upcoming: [],
+        };
+        const payloadPacket = await stringifyIO(payload);
+
+        const taskRunTemplate = await this._prisma.taskRunTemplate.create({
+          data: {
+            taskSlug: data.taskIdentifier,
+            triggerSource: "SCHEDULED",
+            label: data.label,
+            payload: payloadPacket.data,
+            payloadType: payloadPacket.dataType,
+            queue: data.queue,
+            ttlSeconds: data.ttlSeconds,
+            concurrencyKey: data.concurrencyKey,
+            maxAttempts: data.maxAttempts,
+            maxDurationSeconds: data.maxDurationSeconds,
+            tags: data.tags ?? [],
+            machinePreset: data.machine,
+            projectId: environment.projectId,
+            organizationId: environment.organizationId,
+          },
+        });
+
+        return taskRunTemplate;
+      }
+      default: {
+        triggerSource satisfies never;
+        throw new Error("Invalid trigger source");
+      }
+    }
+  }
+}