first commit

calcom/packages/trpc/server/routers/viewer/admin/setSMSLockState.handler.ts

@@ -0,0 +1,90 @@
+import { prisma } from "@calcom/prisma";
+import { SMSLockState } from "@calcom/prisma/client";
+
+import { TRPCError } from "@trpc/server";
+
+import type { TrpcSessionUser } from "../../../trpc";
+import type { TSetSMSLockState } from "./setSMSLockState.schema";
+
+type GetOptions = {
+  ctx: {
+    user: NonNullable<TrpcSessionUser>;
+  };
+  input: TSetSMSLockState;
+};
+
+const setSMSLockState = async ({ input }: GetOptions) => {
+  const { userId, username, teamId, teamSlug, lock } = input;
+  if (userId) {
+    const updatedUser = await prisma.user.update({
+      where: {
+        id: userId,
+      },
+      data: {
+        smsLockState: lock ? SMSLockState.LOCKED : SMSLockState.UNLOCKED,
+      },
+    });
+    if (!updatedUser) {
+      throw new TRPCError({ code: "BAD_REQUEST", message: "User not found" });
+    }
+    return { name: updatedUser.username, locked: lock };
+  } else if (username) {
+    const userToUpdate = await prisma.user.findFirst({
+      where: {
+        username,
+        profiles: { none: {} },
+      },
+    });
+    if (!userToUpdate) {
+      throw new TRPCError({ code: "BAD_REQUEST", message: "User not found" });
+    }
+    if (userToUpdate) {
+      const updatedUser = await prisma.user.update({
+        where: {
+          id: userToUpdate.id,
+        },
+        data: {
+          smsLockState: lock ? SMSLockState.LOCKED : SMSLockState.UNLOCKED,
+        },
+      });
+      return { name: updatedUser.username, locked: lock };
+    }
+  } else if (teamId) {
+    const updatedTeam = await prisma.team.update({
+      where: {
+        id: teamId,
+      },
+      data: {
+        smsLockState: lock ? SMSLockState.LOCKED : SMSLockState.UNLOCKED,
+      },
+    });
+    if (!updatedTeam) {
+      throw new TRPCError({ code: "BAD_REQUEST", message: "Team not found" });
+    }
+    return { name: updatedTeam.slug, locked: lock };
+  } else if (teamSlug) {
+    const teamToUpdate = await prisma.team.findFirst({
+      where: {
+        slug: teamSlug,
+        parentId: null,
+      },
+    });
+    if (!teamToUpdate) {
+      throw new TRPCError({ code: "BAD_REQUEST", message: "Team not found" });
+    }
+    if (teamToUpdate) {
+      const updatedTeam = await prisma.team.update({
+        where: {
+          id: teamToUpdate.id,
+        },
+        data: {
+          smsLockState: lock ? SMSLockState.LOCKED : SMSLockState.UNLOCKED,
+        },
+      });
+      return { name: updatedTeam.slug, locked: lock };
+    }
+  }
+  throw new TRPCError({ code: "BAD_REQUEST", message: "Input data missing" });
+};
+
+export default setSMSLockState;