wehub-resource-sync
80 lines
2.6 KiB
TypeScript
80 lines
2.6 KiB
TypeScript
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);
|
|
}
|