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,129 @@
+import { ZodError } from "zod";
+import { CronPattern } from "../schedules";
+import { BaseService, ServiceValidationError } from "./baseService.server";
+import { getLimit } from "~/services/platform.v3.server";
+import { getTimezones } from "~/utils/timezones.server";
+import { env } from "~/env.server";
+import { type PrismaClientOrTransaction } from "@trigger.dev/database";
+
+type Schedule = {
+  cron: string;
+  timezone?: string;
+  taskIdentifier: string;
+  friendlyId?: string;
+};
+
+export class CheckScheduleService extends BaseService {
+  public async call(projectId: string, schedule: Schedule, environmentIds: string[]) {
+    //validate the cron expression
+    try {
+      CronPattern.parse(schedule.cron);
+    } catch (e) {
+      if (e instanceof ZodError) {
+        throw new ServiceValidationError(`Invalid cron expression: ${e.issues[0].message}`);
+      }
+
+      throw new ServiceValidationError(
+        `Invalid cron expression: ${e instanceof Error ? e.message : JSON.stringify(e)}`
+      );
+    }
+
+    //chek it's a valid timezone
+    if (schedule.timezone) {
+      const possibleTimezones = getTimezones();
+      if (!possibleTimezones.includes(schedule.timezone)) {
+        throw new ServiceValidationError(
+          `Invalid IANA timezone: '${schedule.timezone}'. View the list of valid timezones at ${env.APP_ORIGIN}/timezones`
+        );
+      }
+    }
+
+    //check the task exists
+    const task = await this._prisma.backgroundWorkerTask.findFirst({
+      where: {
+        slug: schedule.taskIdentifier,
+        projectId: projectId,
+      },
+      orderBy: {
+        createdAt: "desc",
+      },
+    });
+
+    if (!task) {
+      throw new ServiceValidationError(
+        `Task with identifier ${schedule.taskIdentifier} not found in project.`
+      );
+    }
+
+    if (task.triggerSource !== "SCHEDULED") {
+      throw new ServiceValidationError(
+        `Task with identifier ${schedule.taskIdentifier} is not a scheduled task.`
+      );
+    }
+
+    //check they're within their limit
+    const project = await this._prisma.project.findFirst({
+      where: {
+        id: projectId,
+      },
+      select: {
+        organizationId: true,
+        environments: {
+          select: {
+            id: true,
+            type: true,
+            archivedAt: true,
+          },
+        },
+      },
+    });
+
+    if (!project) {
+      throw new ServiceValidationError("Project not found");
+    }
+
+    const environments = project.environments.filter((env) => environmentIds.includes(env.id));
+    if (environments.some((env) => env.archivedAt)) {
+      throw new ServiceValidationError("Can't add or edit a schedule for an archived branch");
+    }
+
+    //if creating a schedule, check they're under the limits
+    if (!schedule.friendlyId) {
+      const limit = await getLimit(project.organizationId, "schedules", 100_000_000);
+      const schedulesCount = await CheckScheduleService.getUsedSchedulesCount({
+        prisma: this._prisma,
+        projectId,
+      });
+
+      if (schedulesCount >= limit) {
+        throw new ServiceValidationError(
+          `You have created ${schedulesCount}/${limit} schedules so you'll need to increase your limits or delete some schedules.`
+        );
+      }
+    }
+  }
+
+  static async getUsedSchedulesCount({
+    prisma,
+    projectId,
+  }: {
+    prisma: PrismaClientOrTransaction;
+    projectId: string;
+  }) {
+    return await prisma.taskScheduleInstance.count({
+      where: {
+        projectId,
+        active: true,
+        environment: {
+          type: {
+            not: "DEVELOPMENT",
+          },
+          archivedAt: null,
+        },
+        taskSchedule: {
+          active: true,
+        },
+      },
+    });
+  }
+}