46 lines
1.5 KiB
TypeScript
46 lines
1.5 KiB
TypeScript
import { IS_TEAM_BILLING_ENABLED } from "@calcom/lib/constants";
|
|
import { prisma } from "@calcom/prisma";
|
|
import { MembershipRole } from "@calcom/prisma/enums";
|
|
import { teamMetadataSchema } from "@calcom/prisma/zod-utils";
|
|
import type { TrpcSessionUser } from "@calcom/trpc/server/trpc";
|
|
|
|
type GetUpgradeableOptions = {
|
|
ctx: {
|
|
user: NonNullable<TrpcSessionUser>;
|
|
};
|
|
};
|
|
|
|
export const getUpgradeableHandler = async ({ ctx }: GetUpgradeableOptions) => {
|
|
if (!IS_TEAM_BILLING_ENABLED) return [];
|
|
|
|
// Get all teams/orgs where the user is an owner
|
|
let teams = await prisma.membership.findMany({
|
|
where: {
|
|
user: {
|
|
id: ctx.user.id,
|
|
},
|
|
role: MembershipRole.OWNER,
|
|
team: {
|
|
parentId: null, // Since ORGS relay on their parent's subscription, we don't need to return them
|
|
},
|
|
},
|
|
include: {
|
|
team: {
|
|
include: {
|
|
children: true,
|
|
},
|
|
},
|
|
},
|
|
});
|
|
|
|
/** We only need to return teams that don't have a `subscriptionId` on their metadata */
|
|
teams = teams.filter((m) => {
|
|
const metadata = teamMetadataSchema.safeParse(m.team.metadata);
|
|
if (metadata.success && metadata.data?.subscriptionId) return false;
|
|
if (metadata.success && metadata.data?.isOrganization) return false; // We also dont return ORGs as it will be handled in OrgUpgradeBanner
|
|
if (m.team.children.length > 0) return false; // We also dont return ORGs as it will be handled in OrgUpgradeBanner
|
|
return true;
|
|
});
|
|
return teams;
|
|
};
|