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,143 @@
+import { ManageConcurrencyPresenter } from "~/presenters/v3/ManageConcurrencyPresenter.server";
+import { BaseService } from "./baseService.server";
+import { tryCatch } from "@trigger.dev/core";
+import { setConcurrencyAddOn } 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;
+  projectId: string;
+  organizationId: string;
+  action: "purchase" | "quota-increase";
+  amount: number;
+};
+
+type Result =
+  | {
+      success: true;
+    }
+  | {
+      success: false;
+      error: string;
+    };
+
+export class SetConcurrencyAddOnService extends BaseService {
+  async call({ userId, projectId, organizationId, action, amount }: Input): Promise<Result> {
+    // fetch the current concurrency
+    const presenter = new ManageConcurrencyPresenter(this._prisma, this._replica);
+    const [error, result] = await tryCatch(
+      presenter.call({
+        userId,
+        projectId,
+        organizationId,
+      })
+    );
+
+    if (error) {
+      return {
+        success: false,
+        error: "Unknown error",
+      };
+    }
+
+    const currentConcurrency = result.extraConcurrency;
+    const totalExtraConcurrency = amount;
+
+    switch (action) {
+      case "purchase": {
+        const updatedConcurrency = await setConcurrencyAddOn(organizationId, totalExtraConcurrency);
+        if (!updatedConcurrency) {
+          return {
+            success: false,
+            error: "Failed to update concurrency",
+          };
+        }
+
+        switch (updatedConcurrency?.result) {
+          case "success": {
+            return {
+              success: true,
+            };
+          }
+          case "error": {
+            return {
+              success: false,
+              error: updatedConcurrency.error,
+            };
+          }
+          case "max_quota_reached": {
+            return {
+              success: false,
+              error: `You can't purchase more than ${updatedConcurrency.maxQuota} concurrency without requesting an increase.`,
+            };
+          }
+          default: {
+            return {
+              success: false,
+              error: "Failed to update concurrency, 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, _result] = await tryCatch(
+          sendToPlain({
+            userId,
+            email: user.email,
+            name: user.name ?? user.displayName ?? user.email,
+            title: `Concurrency quota request: ${totalExtraConcurrency}`,
+            components: [
+              uiComponent.text({
+                text: `Org: ${organization?.title} (${organizationId})`,
+              }),
+              uiComponent.divider({ spacingSize: "M" }),
+              uiComponent.text({
+                text: `Total concurrency (set this): ${totalExtraConcurrency}`,
+              }),
+              uiComponent.text({
+                text: `Current extra concurrency: ${currentConcurrency}`,
+              }),
+              uiComponent.text({
+                text: `Amount requested: ${amount}`,
+              }),
+            ],
+          })
+        );
+
+        if (error) {
+          return {
+            success: false,
+            error: error.message,
+          };
+        }
+
+        return {
+          success: true,
+        };
+      }
+      default: {
+        assertNever(action);
+      }
+    }
+  }
+}