chore: template attachments

packages/lib/server-only/attachment/find-document-attachments.ts

@@ -2,13 +2,17 @@ import { prisma } from '@documenso/prisma';
 
 import { buildTeamWhereQuery } from '../../utils/teams';
 
-export type FindAttachmentsOptions = {
-  documentId: number;
+export type FindDocumentAttachmentsOptions = {
+  documentId?: number;
   userId: number;
   teamId: number;
 };
 
-export const findAttachments = async ({ documentId, userId, teamId }: FindAttachmentsOptions) => {
+export const findDocumentAttachments = async ({
+  documentId,
+  userId,
+  teamId,
+}: FindDocumentAttachmentsOptions) => {
   const attachments = await prisma.attachment.findMany({
     where: {
       document: {

packages/lib/server-only/attachment/find-template-attachments.ts

@@ -0,0 +1,28 @@
+import { prisma } from '@documenso/prisma';
+
+import { buildTeamWhereQuery } from '../../utils/teams';
+
+export type FindTemplateAttachmentsOptions = {
+  templateId: number;
+  userId: number;
+  teamId: number;
+};
+
+export const findTemplateAttachments = async ({
+  templateId,
+  userId,
+  teamId,
+}: FindTemplateAttachmentsOptions) => {
+  const attachments = await prisma.attachment.findMany({
+    where: {
+      template: {
+        id: templateId,
+        team: buildTeamWhereQuery({ teamId, userId }),
+      },
+    },
+  });
+
+  console.log('attachments', attachments);
+
+  return attachments;
+};

packages/lib/server-only/attachment/set-document-attachments.ts

@@ -6,14 +6,12 @@ import { AppError } from '../../errors/app-error';
 import { AppErrorCode } from '../../errors/app-error';
 
 export type CreateAttachmentsOptions = {
-  documentId?: number;
-  templateId?: number;
+  documentId: number;
   attachments: Pick<Attachment, 'id' | 'label' | 'url' | 'type'>[];
 };
 
 export const setDocumentAttachments = async ({
   documentId,
-  templateId,
   attachments,
 }: CreateAttachmentsOptions) => {
   const document = await prisma.document.findUnique({
@@ -55,13 +53,11 @@ export const setDocumentAttachments = async ({
           label: attachment.label,
           url: attachment.url,
           type: attachment.type,
-          templateId,
         },
         create: {
           label: attachment.label,
           url: attachment.url,
           type: attachment.type,
-          templateId,
           documentId,
         },
       });
@@ -72,7 +68,6 @@ export const setDocumentAttachments = async ({
           label: attachment.label,
           url: attachment.url,
           type: attachment.type,
-          templateId,
           documentId,
         },
       });

packages/lib/server-only/attachment/set-template-attachments.ts

@@ -0,0 +1,80 @@
+import type { Attachment } from '@prisma/client';
+
+import { prisma } from '@documenso/prisma';
+
+import { AppError } from '../../errors/app-error';
+import { AppErrorCode } from '../../errors/app-error';
+
+export type CreateAttachmentsOptions = {
+  templateId: number;
+  attachments: Pick<Attachment, 'id' | 'label' | 'url' | 'type'>[];
+};
+
+export const setTemplateAttachments = async ({
+  templateId,
+  attachments,
+}: CreateAttachmentsOptions) => {
+  const template = await prisma.template.findUnique({
+    where: {
+      id: templateId,
+    },
+  });
+
+  if (!template) {
+    throw new AppError(AppErrorCode.NOT_FOUND, {
+      message: 'Template not found',
+    });
+  }
+
+  const existingAttachments = await prisma.attachment.findMany({
+    where: {
+      templateId,
+    },
+  });
+
+  const newIds = attachments.map((a) => a.id).filter(Boolean);
+  const toDelete = existingAttachments.filter((existing) => !newIds.includes(existing.id));
+
+  if (toDelete.length > 0) {
+    await prisma.attachment.deleteMany({
+      where: {
+        id: { in: toDelete.map((a) => a.id) },
+      },
+    });
+  }
+
+  const upsertedAttachments: Attachment[] = [];
+
+  for (const attachment of attachments) {
+    if (attachment.id) {
+      const updated = await prisma.attachment.upsert({
+        where: { id: attachment.id },
+        update: {
+          label: attachment.label,
+          url: attachment.url,
+          type: attachment.type,
+          templateId,
+        },
+        create: {
+          label: attachment.label,
+          url: attachment.url,
+          type: attachment.type,
+          templateId,
+        },
+      });
+      upsertedAttachments.push(updated);
+    } else {
+      const created = await prisma.attachment.create({
+        data: {
+          label: attachment.label,
+          url: attachment.url,
+          type: attachment.type,
+          templateId,
+        },
+      });
+      upsertedAttachments.push(created);
+    }
+  }
+
+  return upsertedAttachments;
+};