mirror of
https://github.com/documenso/documenso.git
synced 2025-11-13 08:13:56 +10:00
88534fa1c6
## Description Previously we assumed that there can only be 1 subscription per user. However, that will soon no longer the case with the introduction of the Teams subscription. This PR will apply the required migrations to support multiple subscriptions. ## Changes Made - Updated the Prisma schema to allow for multiple `Subscriptions` per `User` - Added a Stripe `customerId` field to the `User` model - Updated relevant billing sections to support multiple subscriptions ## Testing Performed - Tested running the Prisma migration on a demo database created on the main branch Will require a lot of additional testing. ## Checklist - [ ] I have tested these changes locally and they work as expected. - [ ] I have added/updated tests that prove the effectiveness of these changes. - [X] I have followed the project's coding style guidelines. ## Additional Notes Added the following custom SQL statement to the migration: > DELETE FROM "Subscription" WHERE "planId" IS NULL OR "priceId" IS NULL; Prior to deployment this will require changes to Stripe products: - Adding `type` meta attribute --------- Co-authored-by: Lucas Smith <me@lucasjamessmith.me>
95 lines
2.3 KiB
TypeScript
95 lines
2.3 KiB
TypeScript
import { stripe } from '@documenso/lib/server-only/stripe';
|
|
import { prisma } from '@documenso/prisma';
|
|
import type { User } from '@documenso/prisma/client';
|
|
|
|
import { onSubscriptionUpdated } from './webhook/on-subscription-updated';
|
|
|
|
export const getStripeCustomerByEmail = async (email: string) => {
|
|
const foundStripeCustomers = await stripe.customers.list({
|
|
email,
|
|
});
|
|
|
|
return foundStripeCustomers.data[0] ?? null;
|
|
};
|
|
|
|
export const getStripeCustomerById = async (stripeCustomerId: string) => {
|
|
try {
|
|
const stripeCustomer = await stripe.customers.retrieve(stripeCustomerId);
|
|
|
|
return !stripeCustomer.deleted ? stripeCustomer : null;
|
|
} catch {
|
|
return null;
|
|
}
|
|
};
|
|
|
|
/**
|
|
* Get a stripe customer by user.
|
|
*
|
|
* Will create a Stripe customer and update the relevant user if one does not exist.
|
|
*/
|
|
export const getStripeCustomerByUser = async (user: User) => {
|
|
if (user.customerId) {
|
|
const stripeCustomer = await getStripeCustomerById(user.customerId);
|
|
|
|
if (!stripeCustomer) {
|
|
throw new Error('Missing Stripe customer');
|
|
}
|
|
|
|
return {
|
|
user,
|
|
stripeCustomer,
|
|
};
|
|
}
|
|
|
|
let stripeCustomer = await getStripeCustomerByEmail(user.email);
|
|
|
|
const isSyncRequired = Boolean(stripeCustomer && !stripeCustomer.deleted);
|
|
|
|
if (!stripeCustomer) {
|
|
stripeCustomer = await stripe.customers.create({
|
|
name: user.name ?? undefined,
|
|
email: user.email,
|
|
metadata: {
|
|
userId: user.id,
|
|
},
|
|
});
|
|
}
|
|
|
|
const updatedUser = await prisma.user.update({
|
|
where: {
|
|
id: user.id,
|
|
},
|
|
data: {
|
|
customerId: stripeCustomer.id,
|
|
},
|
|
});
|
|
|
|
// Sync subscriptions if the customer already exists for back filling the DB
|
|
// and local development.
|
|
if (isSyncRequired) {
|
|
await syncStripeCustomerSubscriptions(user.id, stripeCustomer.id).catch((e) => {
|
|
console.error(e);
|
|
});
|
|
}
|
|
|
|
return {
|
|
user: updatedUser,
|
|
stripeCustomer,
|
|
};
|
|
};
|
|
|
|
const syncStripeCustomerSubscriptions = async (userId: number, stripeCustomerId: string) => {
|
|
const stripeSubscriptions = await stripe.subscriptions.list({
|
|
customer: stripeCustomerId,
|
|
});
|
|
|
|
await Promise.all(
|
|
stripeSubscriptions.data.map(async (subscription) =>
|
|
onSubscriptionUpdated({
|
|
userId,
|
|
subscription,
|
|
}),
|
|
),
|
|
);
|
|
};
|