feat: wip

packages/lib/server-only/user/create-user.ts

@@ -1,11 +1,13 @@
 import { hash } from 'bcrypt';
 
 import { getStripeCustomerByUser } from '@documenso/ee/server-only/stripe/get-customer';
+import { updateSubscriptionItemQuantity } from '@documenso/ee/server-only/stripe/update-subscription-item-quantity';
 import { prisma } from '@documenso/prisma';
-import { IdentityProvider } from '@documenso/prisma/client';
+import { IdentityProvider, Prisma, TeamMemberInviteStatus } from '@documenso/prisma/client';
 
+import { IS_BILLING_ENABLED } from '../../constants/app';
 import { SALT_ROUNDS } from '../../constants/auth';
-import { getFlag } from '../../universal/get-feature-flag';
+import { getTeamSeatPriceId } from '../../utils/billing';
 
 export interface CreateUserOptions {
   name: string;
@@ -15,8 +17,6 @@ export interface CreateUserOptions {
 }
 
 export const createUser = async ({ name, email, password, signature }: CreateUserOptions) => {
-  const isBillingEnabled = await getFlag('app_billing');
-
   const hashedPassword = await hash(password, SALT_ROUNDS);
 
   const userExists = await prisma.user.findFirst({
@@ -29,24 +29,77 @@ export const createUser = async ({ name, email, password, signature }: CreateUse
     throw new Error('User already exists');
   }
 
-  let user = await prisma.user.create({
-    data: {
-      name,
-      email: email.toLowerCase(),
-      password: hashedPassword,
-      signature,
-      identityProvider: IdentityProvider.DOCUMENSO,
-    },
-  });
+  return prisma.$transaction(async (tx) => {
+    const user = await tx.user.create({
+      data: {
+        name,
+        email: email.toLowerCase(),
+        password: hashedPassword,
+        signature,
+        identityProvider: IdentityProvider.DOCUMENSO,
+      },
+    });
 
-  if (isBillingEnabled) {
-    try {
-      const stripeSession = await getStripeCustomerByUser(user);
-      user = stripeSession.user;
-    } catch (e) {
-      console.error(e);
+    const acceptedTeamInvites = await tx.teamMemberInvite.findMany({
+      where: {
+        email: {
+          equals: email,
+          mode: Prisma.QueryMode.insensitive,
+        },
+        status: TeamMemberInviteStatus.ACCEPTED,
+      },
+    });
+
+    // For each team invite, add the user to the team and delete the team invite.
+    await Promise.all(
+      acceptedTeamInvites.map(async (invite) => {
+        await tx.teamMember.create({
+          data: {
+            teamId: invite.teamId,
+            userId: user.id,
+            role: invite.role,
+          },
+        });
+
+        await tx.teamMemberInvite.delete({
+          where: {
+            id: invite.id,
+          },
+        });
+
+        if (IS_BILLING_ENABLED) {
+          const team = await tx.team.findFirstOrThrow({
+            where: {
+              id: invite.teamId,
+            },
+            include: {
+              members: {
+                select: {
+                  id: true,
+                },
+              },
+            },
+          });
+
+          if (team.subscriptionId) {
+            await updateSubscriptionItemQuantity({
+              priceId: getTeamSeatPriceId(),
+              subscriptionId: team.subscriptionId,
+              quantity: team.members.length,
+            });
+          }
+        }
+      }),
+    );
+
+    if (IS_BILLING_ENABLED) {
+      try {
+        return await getStripeCustomerByUser(user).then((session) => session.user);
+      } catch (err) {
+        console.error(err);
+      }
     }
-  }
 
-  return user;
+    return user;
+  });
 };