Create run templates in a new service

Author: myftija

apps/webapp/app/v3/services/taskRunTemplate.server.ts

@@ -0,0 +1,88 @@
+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")
+          : undefined;
+
+        const taskRunTemplate = await this._prisma.taskRunTemplate.create({
+          data: {
+            taskSlug: data.taskIdentifier,
+            label: data.label,
+            payload: packet.data,
+            payloadType: packet.dataType,
+            metadata: metadataPacket?.data,
+            metadataType: metadataPacket?.dataType,
+            queue: data.queue ?? "default",
+            delaySeconds: data.delaySeconds,
+            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,
+            label: data.label,
+            payload: payloadPacket.data,
+            payloadType: payloadPacket.dataType,
+            queue: data.queue ?? "default",
+            delaySeconds: data.delaySeconds,
+            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");
+      }
+    }
+  }
+}

apps/webapp/app/v3/services/testTask.server.ts

@@ -7,8 +7,9 @@ import { TriggerTaskService } from "./triggerTask.server";
 export class TestTaskService extends BaseService {
   public async call(environment: AuthenticatedEnvironment, data: TestTaskData) {
     const triggerTaskService = new TriggerTaskService();
+    const { triggerSource } = data;
 
-    switch (data.triggerSource) {
+    switch (triggerSource) {
       case "STANDARD": {
         const result = await triggerTaskService.call(data.taskIdentifier, environment, {
           payload: data.payload,
@@ -72,8 +73,10 @@ export class TestTaskService extends BaseService {
 
         return result?.run;
       }
-      default:
+      default: {
+        triggerSource satisfies never;
         throw new Error("Invalid trigger source");
+      }
     }
   }
 }

apps/webapp/app/v3/taskRunTemplate.ts

@@ -0,0 +1,10 @@
+import { z } from "zod";
+import { TestTaskData } from "./testTask";
+
+export const RunTemplateData = TestTaskData.and(
+  z.object({
+    label: z.string(),
+  })
+);
+
+export type RunTemplateData = z.infer<typeof RunTemplateData>;