feat: add folders (#1711)

packages/lib/server-only/folder/move-document-to-folder.ts

@@ -0,0 +1,130 @@
+import { TeamMemberRole } from '@prisma/client';
+import { match } from 'ts-pattern';
+
+import { AppError, AppErrorCode } from '@documenso/lib/errors/app-error';
+import { DocumentVisibility } from '@documenso/lib/types/document-visibility';
+import { FolderType } from '@documenso/lib/types/folder-type';
+import type { ApiRequestMetadata } from '@documenso/lib/universal/extract-request-metadata';
+import { prisma } from '@documenso/prisma';
+
+export interface MoveDocumentToFolderOptions {
+  userId: number;
+  teamId?: number;
+  documentId: number;
+  folderId?: string | null;
+  requestMetadata?: ApiRequestMetadata;
+}
+
+export const moveDocumentToFolder = async ({
+  userId,
+  teamId,
+  documentId,
+  folderId,
+}: MoveDocumentToFolderOptions) => {
+  let teamMemberRole = null;
+
+  if (teamId !== undefined) {
+    try {
+      const team = await prisma.team.findFirstOrThrow({
+        where: {
+          id: teamId,
+          members: {
+            some: {
+              userId,
+            },
+          },
+        },
+        include: {
+          members: {
+            where: {
+              userId,
+            },
+            select: {
+              role: true,
+            },
+          },
+        },
+      });
+
+      teamMemberRole = team.members[0].role;
+    } catch (error) {
+      console.error('Error finding team:', error);
+      throw new AppError(AppErrorCode.NOT_FOUND, {
+        message: 'Team not found',
+      });
+    }
+  }
+
+  const visibilityFilters = match(teamMemberRole)
+    .with(TeamMemberRole.ADMIN, () => ({
+      visibility: {
+        in: [
+          DocumentVisibility.EVERYONE,
+          DocumentVisibility.MANAGER_AND_ABOVE,
+          DocumentVisibility.ADMIN,
+        ],
+      },
+    }))
+    .with(TeamMemberRole.MANAGER, () => ({
+      visibility: {
+        in: [DocumentVisibility.EVERYONE, DocumentVisibility.MANAGER_AND_ABOVE],
+      },
+    }))
+    .otherwise(() => ({ visibility: DocumentVisibility.EVERYONE }));
+
+  const documentWhereClause = {
+    id: documentId,
+    ...(teamId
+      ? {
+          OR: [
+            { teamId, ...visibilityFilters },
+            { userId, teamId },
+          ],
+        }
+      : { userId, teamId: null }),
+  };
+
+  const document = await prisma.document.findFirst({
+    where: documentWhereClause,
+  });
+
+  if (!document) {
+    throw new AppError(AppErrorCode.NOT_FOUND, {
+      message: 'Document not found',
+    });
+  }
+
+  if (folderId) {
+    const folderWhereClause = {
+      id: folderId,
+      type: FolderType.DOCUMENT,
+      ...(teamId
+        ? {
+            OR: [
+              { teamId, ...visibilityFilters },
+              { userId, teamId },
+            ],
+          }
+        : { userId, teamId: null }),
+    };
+
+    const folder = await prisma.folder.findFirst({
+      where: folderWhereClause,
+    });
+
+    if (!folder) {
+      throw new AppError(AppErrorCode.NOT_FOUND, {
+        message: 'Folder not found',
+      });
+    }
+  }
+
+  return await prisma.document.update({
+    where: {
+      id: documentId,
+    },
+    data: {
+      folderId,
+    },
+  });
+};