From 5c8e0483b1f4729c48ae44e3ca589e12769740dc Mon Sep 17 00:00:00 2001
From: Somay Chauhan <somaychauhan98@gmail.com>
Date: Thu, 7 Mar 2024 21:48:45 +0530
Subject: [PATCH] fix: add future team members toggle (#14008)

* fix: add future team members toggle

fix: add future team members toggle

* code refactor

* removed updateNewTeamMemberEventTypes for regularUsers
---
 .../loggedInViewer/updateProfile.handler.ts   | 22 +++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/packages/trpc/server/routers/loggedInViewer/updateProfile.handler.ts b/packages/trpc/server/routers/loggedInViewer/updateProfile.handler.ts
index f9ddb11518..4b0d3eb4d0 100644
--- a/packages/trpc/server/routers/loggedInViewer/updateProfile.handler.ts
+++ b/packages/trpc/server/routers/loggedInViewer/updateProfile.handler.ts
@@ -14,6 +14,7 @@ import { HttpError } from "@calcom/lib/http-error";
 import logger from "@calcom/lib/logger";
 import { getTranslation } from "@calcom/lib/server";
 import { checkUsername } from "@calcom/lib/server/checkUsername";
+import { updateNewTeamMemberEventTypes } from "@calcom/lib/server/queries";
 import { resizeBase64Image } from "@calcom/lib/server/resizeBase64Image";
 import slugify from "@calcom/lib/slugify";
 import { updateWebUser as syncServicesUpdateWebUser } from "@calcom/lib/sync/SyncServiceManager";
@@ -202,6 +203,27 @@ export const updateProfileHandler = async ({ ctx, input }: UpdateProfileOptions)
     data.avatarUrl = null;
     data.avatar = null;
   }
+  if (input.completedOnboarding) {
+    const userTeams = await prisma.user.findFirst({
+      where: {
+        id: user.id,
+      },
+      select: {
+        teams: {
+          select: {
+            id: true,
+          },
+        },
+      },
+    });
+    if (userTeams && userTeams.teams.length > 0) {
+      await Promise.all(
+        userTeams.teams.map(async (team) => {
+          await updateNewTeamMemberEventTypes(user.id, team.id);
+        })
+      );
+    }
+  }
 
   const updatedUserSelect = Prisma.validator<Prisma.UserDefaultArgs>()({
     select: {