feat(webapp): treat schedules as a purchasable add-on

Author: isshaddad

apps/webapp/app/presenters/v3/ScheduleListPresenter.server.ts

@@ -2,7 +2,7 @@ import { type RuntimeEnvironmentType, type ScheduleType } from "@trigger.dev/dat
 import { type ScheduleListFilters } from "~/components/runs/v3/ScheduleFilters";
 import { displayableEnvironment } from "~/models/runtimeEnvironment.server";
 import { getTaskIdentifiers } from "~/models/task.server";
-import { getLimit } from "~/services/platform.v3.server";
+import { getCurrentPlan, getLimit, getPlans } from "~/services/platform.v3.server";
 import { findCurrentWorkerFromEnvironment } from "~/v3/models/workerDeployment.server";
 import { ServiceValidationError } from "~/v3/services/baseService.server";
 import { CheckScheduleService } from "~/v3/services/checkSchedule.server";
@@ -103,7 +103,27 @@ export class ScheduleListPresenter extends BasePresenter {
       projectId,
     });
 
-    const limit = await getLimit(project.organizationId, "schedules", 100_000_000);
+    const baseLimit = await getLimit(project.organizationId, "schedules", 100_000_000);
+    const [currentPlan, plans] = await Promise.all([
+      getCurrentPlan(project.organizationId),
+      getPlans(),
+    ]);
+
+    const extraSchedules = currentPlan?.v3Subscription?.addOns?.schedules?.purchased ?? 0;
+    const limit = baseLimit + extraSchedules;
+    const canPurchaseSchedules =
+      currentPlan?.v3Subscription?.plan?.limits.schedules.canExceed === true;
+    const maxScheduleQuota = currentPlan?.v3Subscription?.addOns?.schedules?.quota ?? 0;
+    const planScheduleLimit = currentPlan?.v3Subscription?.plan?.limits.schedules.number ?? 0;
+    const schedulePricing = plans?.addOnPricing.schedules ?? null;
+
+    const purchaseInfo = {
+      canPurchaseSchedules,
+      extraSchedules,
+      maxScheduleQuota,
+      planScheduleLimit,
+      schedulePricing,
+    };
 
     //get the latest BackgroundWorker
     const latestWorker = await findCurrentWorkerFromEnvironment(environment, this._replica);
@@ -119,6 +139,7 @@ export class ScheduleListPresenter extends BasePresenter {
           used: schedulesCount,
           limit,
         },
+        ...purchaseInfo,
         filters: {
           tasks,
           search,
@@ -314,6 +335,7 @@ export class ScheduleListPresenter extends BasePresenter {
         used: schedulesCount,
         limit,
       },
+      ...purchaseInfo,
       filters: {
         tasks,
         search,

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

@@ -478,6 +478,22 @@ export async function setBranchesAddOn(organizationId: string, amount: number) {
   }
 }
 
+export async function setSchedulesAddOn(organizationId: string, amount: number) {
+  if (!client) return undefined;
+
+  try {
+    const result = await client.setAddOn(organizationId, { type: "schedules", amount });
+    if (!result.success) {
+      recordPlatformFailure("setSchedulesAddOn", "no_success");
+      return undefined;
+    }
+    return result;
+  } catch (e) {
+    recordPlatformFailure("setSchedulesAddOn", "caught");
+    return undefined;
+  }
+}
+
 export async function getUsage(organizationId: string, { from, to }: { from: Date; to: Date }) {
   if (!client) return undefined;
 

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

@@ -1,7 +1,7 @@
 import { ZodError } from "zod";
 import { CronPattern } from "../schedules";
 import { BaseService, ServiceValidationError } from "./baseService.server";
-import { getLimit } from "~/services/platform.v3.server";
+import { getCurrentPlan, getLimit } from "~/services/platform.v3.server";
 import { getTimezones } from "~/utils/timezones.server";
 import { env } from "~/env.server";
 import { type PrismaClientOrTransaction, type RuntimeEnvironmentType } from "@trigger.dev/database";
@@ -89,7 +89,11 @@ export class CheckScheduleService extends BaseService {
 
     //if creating a schedule, check they're under the limits
     if (!schedule.friendlyId) {
-      const limit = await getLimit(project.organizationId, "schedules", 100_000_000);
+      const baseLimit = await getLimit(project.organizationId, "schedules", 100_000_000);
+      const currentPlan = await getCurrentPlan(project.organizationId);
+      const purchasedSchedules =
+        currentPlan?.v3Subscription?.addOns?.schedules?.purchased ?? 0;
+      const limit = baseLimit + purchasedSchedules;
       const schedulesCount = await CheckScheduleService.getUsedSchedulesCount({
         prisma: this._prisma,
         projectId,

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

@@ -0,0 +1,100 @@
+import { BaseService } from "./baseService.server";
+import { tryCatch } from "@trigger.dev/core/utils";
+import { setSchedulesAddOn } from "~/services/platform.v3.server";
+import assertNever from "assert-never";
+import { sendToPlain } from "~/utils/plain.server";
+import { uiComponent } from "@team-plain/typescript-sdk";
+
+type Input = {
+  userId: string;
+  organizationId: string;
+  action: "purchase" | "quota-increase";
+  amount: number;
+};
+
+type Result =
+  | {
+      success: true;
+    }
+  | {
+      success: false;
+      error: string;
+    };
+
+export class SetSchedulesAddOnService extends BaseService {
+  async call({ userId, organizationId, action, amount }: Input): Promise<Result> {
+    switch (action) {
+      case "purchase": {
+        const result = await setSchedulesAddOn(organizationId, amount);
+        if (!result) {
+          return {
+            success: false,
+            error: "Failed to update schedules",
+          };
+        }
+
+        switch (result.result) {
+          case "success": {
+            return { success: true };
+          }
+          case "error": {
+            return { success: false, error: result.error };
+          }
+          case "max_quota_reached": {
+            return {
+              success: false,
+              error: `You can't purchase more than ${result.maxQuota} schedules without requesting an increase.`,
+            };
+          }
+          default: {
+            return {
+              success: false,
+              error: "Failed to update schedules, unknown result.",
+            };
+          }
+        }
+      }
+      case "quota-increase": {
+        const user = await this._replica.user.findFirst({
+          where: { id: userId },
+        });
+
+        if (!user) {
+          return { success: false, error: "No matching user found." };
+        }
+
+        const organization = await this._replica.organization.findFirst({
+          select: { title: true },
+          where: { id: organizationId },
+        });
+
+        const [error] = await tryCatch(
+          sendToPlain({
+            userId,
+            email: user.email,
+            name: user.name ?? user.displayName ?? user.email,
+            title: `Schedules quota request: ${amount}`,
+            components: [
+              uiComponent.text({
+                text: `Org: ${organization?.title} (${organizationId})`,
+              }),
+              uiComponent.divider({ spacingSize: "M" }),
+              uiComponent.text({
+                text: `Total schedules requested: ${amount}`,
+              }),
+            ],
+          })
+        );
+
+        if (error) {
+          return { success: false, error: error.message };
+        }
+
+        return { success: true };
+      }
+      default: {
+        assertNever(action);
+      }
+    }
+  }
+}