refactor: Team Creation Flow [CAL-2751] (#12501)

* Create new endpoint for creating a team * Generate a team checkout session * Create team navigate to checkout * Clean up * UI changes * Add comments * Fix * Type fix * Type fix * Type fix * Type fixes * Set telemetry * Import fix * Type fix * Update tests * Type fix * fix: e2e * fix: e2e * fix: e2e * fix: e2e * Update teams.e2e.ts * fix: e2e --------- Co-authored-by: Omar López <zomars@me.com>
2023-11-29 09:39:21 -07:00
parent bae3bd76e5
commit 877cd4cdff
13 changed files with 353 additions and 169 deletions
@@ -1,3 +1,5 @@
+import { generateTeamCheckoutSession } from "@calcom/features/ee/teams/lib/payments";
+import { IS_TEAM_BILLING_ENABLED, WEBAPP_URL } from "@calcom/lib/constants";
 import { closeComUpsertTeamUser } from "@calcom/lib/sync/SyncServiceManager";
 import { prisma } from "@calcom/prisma";
 import { MembershipRole } from "@calcom/prisma/enums";
@@ -14,6 +16,34 @@ type CreateOptions = {
  input: TCreateInputSchema;
 };

+const generateCheckoutSession = async ({
+  teamSlug,
+  teamName,
+  userId,
+}: {
+  teamSlug: string;
+  teamName: string;
+  userId: number;
+}) => {
+  if (!IS_TEAM_BILLING_ENABLED) {
+    console.info("Team billing is disabled, not generating a checkout session.");
+    return;
+  }
+
+  const checkoutSession = await generateTeamCheckoutSession({
+    teamSlug,
+    teamName,
+    userId,
+  });
+
+  if (!checkoutSession.url)
+    throw new TRPCError({
+      code: "INTERNAL_SERVER_ERROR",
+      message: "Failed retrieving a checkout session URL.",
+    });
+  return { url: checkoutSession.url, message: "Payment required to publish team" };
+};
+
 export const createHandler = async ({ ctx, input }: CreateOptions) => {
  const { user } = ctx;
  const { slug, name, logo } = input;
@@ -45,28 +75,26 @@ export const createHandler = async ({ ctx, input }: CreateOptions) => {
    if (nameCollisions) throw new TRPCError({ code: "BAD_REQUEST", message: "team_slug_exists_as_user" });
  }

-  // Ensure that the user is not duplicating a requested team
-  const duplicatedRequest = await prisma.team.findFirst({
-    where: {
-      members: {
-        some: {
-          userId: ctx.user.id,
-        },
-      },
-      metadata: {
-        path: ["requestedSlug"],
-        equals: slug,
-      },
-    },
-  });
+  // If the user is not a part of an org, then make them pay before creating the team
+  if (!isOrgChildTeam) {
+    const checkoutSession = await generateCheckoutSession({
+      teamSlug: slug,
+      teamName: name,
+      userId: user.id,
+    });

-  if (duplicatedRequest) {
-    return duplicatedRequest;
+    // If there is a checkout session, return it. Otherwise, it means it's disabled.
+    if (checkoutSession)
+      return {
+        url: checkoutSession.url,
+        message: checkoutSession.message,
+        team: null,
+      };
  }

-  const createTeam = await prisma.team.create({
+  const createdTeam = await prisma.team.create({
    data: {
-      ...(isOrgChildTeam ? { slug } : {}),
+      slug,
      name,
      logo,
      members: {
@@ -76,17 +104,16 @@ export const createHandler = async ({ ctx, input }: CreateOptions) => {
          accepted: true,
        },
      },
-      metadata: !isOrgChildTeam
-        ? {
-            requestedSlug: slug,
-          }
-        : undefined,
      ...(isOrgChildTeam && { parentId: user.organizationId }),
    },
  });

  // Sync Services: Close.com
-  closeComUpsertTeamUser(createTeam, ctx.user, MembershipRole.OWNER);
+  closeComUpsertTeamUser(createdTeam, ctx.user, MembershipRole.OWNER);

-  return createTeam;
+  return {
+    url: `${WEBAPP_URL}/settings/teams/${createdTeam.id}/onboard-members`,
+    message: "Team billing is disabled, not generating a checkout session.",
+    team: createdTeam,
+  };
 };