mirror of
https://github.com/documenso/documenso.git
synced 2025-11-14 08:42:12 +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>
87 lines
2.2 KiB
TypeScript
87 lines
2.2 KiB
TypeScript
import { DateTime } from 'luxon';
|
|
|
|
import { getFlag } from '@documenso/lib/universal/get-feature-flag';
|
|
import { prisma } from '@documenso/prisma';
|
|
import { SubscriptionStatus } from '@documenso/prisma/client';
|
|
|
|
import { getPricesByType } from '../stripe/get-prices-by-type';
|
|
import { FREE_PLAN_LIMITS, SELFHOSTED_PLAN_LIMITS } from './constants';
|
|
import { ERROR_CODES } from './errors';
|
|
import { ZLimitsSchema } from './schema';
|
|
|
|
export type GetServerLimitsOptions = {
|
|
email?: string | null;
|
|
};
|
|
|
|
export const getServerLimits = async ({ email }: GetServerLimitsOptions) => {
|
|
const isBillingEnabled = await getFlag('app_billing');
|
|
|
|
if (!isBillingEnabled) {
|
|
return {
|
|
quota: SELFHOSTED_PLAN_LIMITS,
|
|
remaining: SELFHOSTED_PLAN_LIMITS,
|
|
};
|
|
}
|
|
|
|
if (!email) {
|
|
throw new Error(ERROR_CODES.UNAUTHORIZED);
|
|
}
|
|
|
|
const user = await prisma.user.findFirst({
|
|
where: {
|
|
email,
|
|
},
|
|
include: {
|
|
Subscription: true,
|
|
},
|
|
});
|
|
|
|
if (!user) {
|
|
throw new Error(ERROR_CODES.USER_FETCH_FAILED);
|
|
}
|
|
|
|
let quota = structuredClone(FREE_PLAN_LIMITS);
|
|
let remaining = structuredClone(FREE_PLAN_LIMITS);
|
|
|
|
const activeSubscriptions = user.Subscription.filter(
|
|
({ status }) => status === SubscriptionStatus.ACTIVE,
|
|
);
|
|
|
|
if (activeSubscriptions.length > 0) {
|
|
const individualPrices = await getPricesByType('individual');
|
|
|
|
for (const subscription of activeSubscriptions) {
|
|
const price = individualPrices.find((price) => price.id === subscription.priceId);
|
|
if (!price || typeof price.product === 'string' || price.product.deleted) {
|
|
continue;
|
|
}
|
|
|
|
const currentQuota = ZLimitsSchema.parse(
|
|
'metadata' in price.product ? price.product.metadata : {},
|
|
);
|
|
|
|
// Use the subscription with the highest quota.
|
|
if (currentQuota.documents > quota.documents && currentQuota.recipients > quota.recipients) {
|
|
quota = currentQuota;
|
|
remaining = structuredClone(quota);
|
|
}
|
|
}
|
|
}
|
|
|
|
const documents = await prisma.document.count({
|
|
where: {
|
|
userId: user.id,
|
|
createdAt: {
|
|
gte: DateTime.utc().startOf('month').toJSDate(),
|
|
},
|
|
},
|
|
});
|
|
|
|
remaining.documents = Math.max(remaining.documents - documents, 0);
|
|
|
|
return {
|
|
quota,
|
|
remaining,
|
|
};
|
|
};
|