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,79 @@
+import { parseWithZod } from "@conform-to/zod";
+import { type ActionFunctionArgs, json } from "@remix-run/server-runtime";
+import { tryCatch } from "@trigger.dev/core/v3";
+import { z } from "zod";
+import { prisma } from "~/db.server";
+import { getCurrentPlan, getSelfServePurchaseBlockReason } from "~/services/platform.v3.server";
+import { requireUserId } from "~/services/session.server";
+import { SetSchedulesAddOnService } from "~/v3/services/setSchedulesAddOn.server";
+
+export const PurchaseSchema = z.discriminatedUnion("action", [
+  z.object({
+    action: z.literal("purchase"),
+    amount: z.coerce.number().int("Must be a whole number").min(0, "Amount must be 0 or more"),
+  }),
+  z.object({
+    action: z.literal("quota-increase"),
+    amount: z.coerce.number().int("Must be a whole number").min(1, "Amount must be greater than 0"),
+  }),
+]);
+
+const ParamsSchema = z.object({ organizationSlug: z.string() });
+
+export async function action({ request, params }: ActionFunctionArgs) {
+  const userId = await requireUserId(request);
+  const { organizationSlug } = ParamsSchema.parse(params);
+
+  const organization = await prisma.organization.findFirst({
+    where: { slug: organizationSlug, members: { some: { userId } } },
+    select: { id: true },
+  });
+  if (!organization) {
+    return json({ error: "Organization not found" }, { status: 404 });
+  }
+
+  const currentPlan = await getCurrentPlan(organization.id);
+  const purchaseBlockReason = getSelfServePurchaseBlockReason(currentPlan);
+  if (purchaseBlockReason === "plan_unavailable") {
+    return json(
+      { ok: false, error: "Unable to verify billing status. Please try again." } as const,
+      { status: 503 }
+    );
+  }
+  if (purchaseBlockReason === "managed_billing") {
+    return json({ ok: false, error: "Contact us to request more schedules." } as const, {
+      status: 403,
+    });
+  }
+
+  const formData = await request.formData();
+  const submission = parseWithZod(formData, { schema: PurchaseSchema });
+
+  if (submission.status !== "success") {
+    return json(submission.reply());
+  }
+
+  const service = new SetSchedulesAddOnService();
+  const [error, result] = await tryCatch(
+    service.call({
+      userId,
+      organizationId: organization.id,
+      action: submission.value.action,
+      amount: submission.value.amount,
+    })
+  );
+
+  if (error) {
+    return json(
+      submission.reply({
+        fieldErrors: { amount: [error instanceof Error ? error.message : "Unknown error"] },
+      })
+    );
+  }
+
+  if (!result.success) {
+    return json(submission.reply({ fieldErrors: { amount: [result.error] } }));
+  }
+
+  return json({ ok: true } as const);
+}