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; }; }; 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 (m.team.children.length > 0) return false; // We also dont return ORGs as it will be handled in OrgUpgradeBanner return true; }); return teams; };