import { updateSubscriptionItemQuantity } from '@documenso/ee/server-only/stripe/update-subscription-item-quantity';
import { IS_BILLING_ENABLED } from '@documenso/lib/constants/app';
import { prisma } from '@documenso/prisma';

export type LeaveTeamOptions = {
  /**
   * The ID of the user who is leaving the team.
   */
  userId: number;

  /**
   * The ID of the team the user is leaving.
   */
  teamId: number;
};

export const leaveTeam = async ({ userId, teamId }: LeaveTeamOptions) => {
  await prisma.$transaction(async (tx) => {
    const team = await tx.team.findFirstOrThrow({
      where: {
        id: teamId,
        ownerUserId: {
          not: userId,
        },
      },
      include: {
        subscription: true,
      },
    });

    await tx.teamMember.delete({
      where: {
        userId_teamId: {
          userId,
          teamId,
        },
        team: {
          ownerUserId: {
            not: userId,
          },
        },
      },
    });

    if (IS_BILLING_ENABLED() && team.subscription) {
      const numberOfSeats = await tx.teamMember.count({
        where: {
          teamId,
        },
      });

      await updateSubscriptionItemQuantity({
        priceId: team.subscription.priceId,
        subscriptionId: team.subscription.planId,
        quantity: numberOfSeats,
      });
    }
  });
};