feat: add organisations (#1820)

packages/lib/server-only/profile/set-avatar-image.ts

@@ -2,51 +2,49 @@ import sharp from 'sharp';
 
 import { prisma } from '@documenso/prisma';
 
-import { AppError, AppErrorCode } from '../../errors/app-error';
+import { ORGANISATION_MEMBER_ROLE_PERMISSIONS_MAP } from '../../constants/organisations';
+import { TEAM_MEMBER_ROLE_PERMISSIONS_MAP } from '../../constants/teams';
+import { AppError } from '../../errors/app-error';
 import type { ApiRequestMetadata } from '../../universal/extract-request-metadata';
+import { buildOrganisationWhereQuery } from '../../utils/organisations';
+import { buildTeamWhereQuery } from '../../utils/teams';
 
 export type SetAvatarImageOptions = {
   userId: number;
-  teamId?: number | null;
+  target:
+    | {
+        type: 'user';
+      }
+    | {
+        type: 'team';
+        teamId: number;
+      }
+    | {
+        type: 'organisation';
+        organisationId: string;
+      };
   bytes?: string | null;
   requestMetadata: ApiRequestMetadata;
 };
 
+/**
+ * Pretty nasty but will do for now.
+ */
 export const setAvatarImage = async ({
   userId,
-  teamId,
+  target,
   bytes,
   requestMetadata,
 }: SetAvatarImageOptions) => {
   let oldAvatarImageId: string | null = null;
 
-  const user = await prisma.user.findUnique({
-    where: {
-      id: userId,
-    },
-    include: {
-      avatarImage: true,
-    },
-  });
-
-  if (!user) {
-    throw new AppError(AppErrorCode.NOT_FOUND, {
-      message: 'User not found',
-    });
-  }
-
-  oldAvatarImageId = user.avatarImageId;
-
-  if (teamId) {
+  if (target.type === 'team') {
     const team = await prisma.team.findFirst({
-      where: {
-        id: teamId,
-        members: {
-          some: {
-            userId,
-          },
-        },
-      },
+      where: buildTeamWhereQuery({
+        teamId: target.teamId,
+        userId,
+        roles: TEAM_MEMBER_ROLE_PERMISSIONS_MAP['MANAGE_TEAM'],
+      }),
     });
 
     if (!team) {
@@ -56,6 +54,39 @@ export const setAvatarImage = async ({
     }
 
     oldAvatarImageId = team.avatarImageId;
+  } else if (target.type === 'organisation') {
+    const organisation = await prisma.organisation.findFirst({
+      where: buildOrganisationWhereQuery({
+        organisationId: target.organisationId,
+        userId,
+        roles: ORGANISATION_MEMBER_ROLE_PERMISSIONS_MAP['MANAGE_ORGANISATION'],
+      }),
+    });
+
+    if (!organisation) {
+      throw new AppError('ORGANISATION_NOT_FOUND', {
+        statusCode: 404,
+      });
+    }
+
+    oldAvatarImageId = organisation.avatarImageId;
+  } else {
+    const user = await prisma.user.findUnique({
+      where: {
+        id: userId,
+      },
+      include: {
+        avatarImage: true,
+      },
+    });
+
+    if (!user) {
+      throw new AppError('USER_NOT_FOUND', {
+        statusCode: 404,
+      });
+    }
+
+    oldAvatarImageId = user.avatarImageId;
   }
 
   if (oldAvatarImageId) {
@@ -83,17 +114,26 @@ export const setAvatarImage = async ({
     newAvatarImageId = avatarImage.id;
   }
 
-  if (teamId) {
+  // TODO: Audit Logs
+
+  if (target.type === 'team') {
     await prisma.team.update({
       where: {
-        id: teamId,
+        id: target.teamId,
+      },
+      data: {
+        avatarImageId: newAvatarImageId,
+      },
+    });
+  } else if (target.type === 'organisation') {
+    await prisma.organisation.update({
+      where: {
+        id: target.organisationId,
       },
       data: {
         avatarImageId: newAvatarImageId,
       },
     });
-
-    // TODO: Audit Logs
   } else {
     await prisma.user.update({
       where: {
@@ -103,8 +143,6 @@ export const setAvatarImage = async ({
         avatarImageId: newAvatarImageId,
       },
     });
-
-    // TODO: Audit Logs
   }
 
   return newAvatarImageId;