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,100 @@
+import { BaseService } from "./baseService.server";
+import { tryCatch } from "@trigger.dev/core/utils";
+import { setBranchesAddOn } 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 SetBranchesAddOnService extends BaseService {
+  async call({ userId, organizationId, action, amount }: Input): Promise<Result> {
+    switch (action) {
+      case "purchase": {
+        const result = await setBranchesAddOn(organizationId, amount);
+        if (!result) {
+          return {
+            success: false,
+            error: "Failed to update preview branches",
+          };
+        }
+
+        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} preview branches without requesting an increase.`,
+            };
+          }
+          default: {
+            return {
+              success: false,
+              error: "Failed to update preview branches, 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: `Preview branches quota request: ${amount}`,
+            components: [
+              uiComponent.text({
+                text: `Org: ${organization?.title} (${organizationId})`,
+              }),
+              uiComponent.divider({ spacingSize: "M" }),
+              uiComponent.text({
+                text: `Total preview branches requested: ${amount}`,
+              }),
+            ],
+          })
+        );
+
+        if (error) {
+          return { success: false, error: error.message };
+        }
+
+        return { success: true };
+      }
+      default: {
+        assertNever(action);
+      }
+    }
+  }
+}