107 lines
3.8 KiB
TypeScript
107 lines
3.8 KiB
TypeScript
import { z } from "zod";
|
|
|
|
import { getStripeCustomerIdFromUserId } from "@calcom/app-store/stripepayment/lib/customer";
|
|
import stripe from "@calcom/app-store/stripepayment/lib/server";
|
|
import { WEBAPP_URL } from "@calcom/lib/constants";
|
|
import prisma from "@calcom/prisma";
|
|
import { teamMetadataSchema } from "@calcom/prisma/zod-utils";
|
|
|
|
const teamPaymentMetadataSchema = z.object({
|
|
paymentId: z.string(),
|
|
subscriptionId: z.string(),
|
|
subscriptionItemId: z.string(),
|
|
});
|
|
|
|
/** Used to prevent double charges for the same team */
|
|
export const checkIfTeamPaymentRequired = async ({ teamId = -1 }) => {
|
|
const team = await prisma.team.findUniqueOrThrow({
|
|
where: { id: teamId },
|
|
select: { metadata: true },
|
|
});
|
|
const metadata = teamMetadataSchema.parse(team.metadata);
|
|
/** If there's no paymentId, we need to pay this team */
|
|
if (!metadata?.paymentId) return { url: null };
|
|
const checkoutSession = await stripe.checkout.sessions.retrieve(metadata.paymentId);
|
|
/** If there's a pending session but it isn't paid, we need to pay this team */
|
|
if (checkoutSession.payment_status === "paid") return { url: null };
|
|
/** If the session is already paid we return the upgrade URL so team is updated. */
|
|
return { url: `${WEBAPP_URL}/api/teams/${teamId}/upgrade?session_id=${metadata.paymentId}` };
|
|
};
|
|
|
|
export const purchaseTeamSubscription = async (input: { teamId: number; seats: number; userId: number }) => {
|
|
const { teamId, seats, userId } = input;
|
|
const { url } = await checkIfTeamPaymentRequired({ teamId });
|
|
if (url) return { url };
|
|
const customer = await getStripeCustomerIdFromUserId(userId);
|
|
const session = await stripe.checkout.sessions.create({
|
|
customer,
|
|
mode: "subscription",
|
|
allow_promotion_codes: true,
|
|
success_url: `${WEBAPP_URL}/api/teams/${teamId}/upgrade?session_id={CHECKOUT_SESSION_ID}`,
|
|
cancel_url: `${WEBAPP_URL}/settings/profile`,
|
|
line_items: [
|
|
{
|
|
/** We only need to set the base price and we can upsell it directly on Stripe's checkout */
|
|
price: process.env.STRIPE_TEAM_MONTHLY_PRICE_ID,
|
|
quantity: seats,
|
|
},
|
|
],
|
|
customer_update: {
|
|
address: "auto",
|
|
},
|
|
automatic_tax: {
|
|
enabled: true,
|
|
},
|
|
metadata: {
|
|
teamId,
|
|
},
|
|
subscription_data: {
|
|
metadata: {
|
|
teamId,
|
|
},
|
|
},
|
|
});
|
|
return { url: session.url };
|
|
};
|
|
|
|
const getTeamWithPaymentMetadata = async (teamId: number) => {
|
|
const team = await prisma.team.findUniqueOrThrow({
|
|
where: { id: teamId },
|
|
select: { metadata: true, members: true },
|
|
});
|
|
const metadata = teamPaymentMetadataSchema.parse(team.metadata);
|
|
return { ...team, metadata };
|
|
};
|
|
|
|
export const cancelTeamSubscriptionFromStripe = async (teamId: number) => {
|
|
try {
|
|
const team = await getTeamWithPaymentMetadata(teamId);
|
|
const { subscriptionId } = team.metadata;
|
|
return await stripe.subscriptions.cancel(subscriptionId);
|
|
} catch (error) {
|
|
let message = "Unknown error on cancelTeamSubscriptionFromStripe";
|
|
if (error instanceof Error) message = error.message;
|
|
console.error(message);
|
|
}
|
|
};
|
|
|
|
export const updateQuantitySubscriptionFromStripe = async (teamId: number) => {
|
|
try {
|
|
const { url } = await checkIfTeamPaymentRequired({ teamId });
|
|
/**
|
|
* If there's no pending checkout URL it means that this team has not been paid.
|
|
* We cannot update the subscription yet, this will be handled on publish/checkout.
|
|
**/
|
|
if (!url) return;
|
|
const team = await getTeamWithPaymentMetadata(teamId);
|
|
const { subscriptionId, subscriptionItemId } = team.metadata;
|
|
await stripe.subscriptions.update(subscriptionId, {
|
|
items: [{ quantity: team.members.length, id: subscriptionItemId }],
|
|
});
|
|
} catch (error) {
|
|
let message = "Unknown error on updateQuantitySubscriptionFromStripe";
|
|
if (error instanceof Error) message = error.message;
|
|
console.error(message);
|
|
}
|
|
};
|