Files
triggerdotdev--trigger.dev/apps/webapp/app/routes/admin.api.v1.orgs.$organizationId.billing-limit.hit.ts
2026-07-13 13:32:57 +08:00

68 lines
2.3 KiB
TypeScript

import { type ActionFunctionArgs, json } from "@remix-run/server-runtime";
import { z } from "zod";
import { prisma } from "~/db.server";
import {
BillingLimitHitWebhookBodySchema,
type BillingLimitHitWebhookBody,
} from "~/services/billingLimit.schemas";
import { logger } from "~/services/logger.server";
import { requireAdminApiRequest } from "~/services/personalAccessToken.server";
import { bustBillingLimitCaches } from "~/services/platform.v3.server";
import {
enqueueBillingLimitCancelInProgressRuns,
enqueueBillingLimitConverge,
} from "~/v3/billingLimitWorker.server";
import { BillingLimitConvergeEnvironmentsService } from "~/v3/services/billingLimit/billingLimitConvergeEnvironmentsService.server";
import { processBillingLimitHit } from "~/v3/services/billingLimit/billingLimitHit.server";
const ParamsSchema = z.object({
organizationId: z.string(),
});
/** Billing platform webhook: org entered billing limit grace. Idempotent — returns 202. */
export async function action({ request, params }: ActionFunctionArgs) {
await requireAdminApiRequest(request);
if (request.method.toLowerCase() !== "post") {
return json({ error: "Method not allowed" }, { status: 405 });
}
const { organizationId } = ParamsSchema.parse(params);
let body: BillingLimitHitWebhookBody;
try {
body = BillingLimitHitWebhookBodySchema.parse(await request.json());
} catch (error) {
logger.error("Invalid billing limit hit webhook payload", {
error,
organizationId,
});
return json({ error: "Invalid request body" }, { status: 400 });
}
const organization = await prisma.organization.findFirst({
where: { id: organizationId },
select: { id: true },
});
if (!organization) {
return json({ error: "Organization not found" }, { status: 404 });
}
await processBillingLimitHit(
{
organizationId,
hitAt: body.hitAt,
cancelInProgressRuns: body.cancelInProgressRuns,
},
{
bustCaches: bustBillingLimitCaches,
seedReconcileQueue: BillingLimitConvergeEnvironmentsService.seedReconcileQueue,
enqueueConverge: enqueueBillingLimitConverge,
enqueueCancelInProgressRuns: enqueueBillingLimitCancelInProgressRuns,
}
);
return json({ success: true, accepted: true }, { status: 202 });
}