feat: embed authoring part two (#1768)

packages/trpc/server/embedding-router/update-embedding-document.ts

@@ -0,0 +1,118 @@
+import { AppError, AppErrorCode } from '@documenso/lib/errors/app-error';
+import { upsertDocumentMeta } from '@documenso/lib/server-only/document-meta/upsert-document-meta';
+import { updateDocument } from '@documenso/lib/server-only/document/update-document';
+import { verifyEmbeddingPresignToken } from '@documenso/lib/server-only/embedding-presign/verify-embedding-presign-token';
+import { setFieldsForDocument } from '@documenso/lib/server-only/field/set-fields-for-document';
+import { setDocumentRecipients } from '@documenso/lib/server-only/recipient/set-document-recipients';
+import { nanoid } from '@documenso/lib/universal/id';
+
+import { procedure } from '../trpc';
+import {
+  ZUpdateEmbeddingDocumentRequestSchema,
+  ZUpdateEmbeddingDocumentResponseSchema,
+} from './update-embedding-document.types';
+
+export const updateEmbeddingDocumentRoute = procedure
+  .input(ZUpdateEmbeddingDocumentRequestSchema)
+  .output(ZUpdateEmbeddingDocumentResponseSchema)
+  .mutation(async ({ input, ctx }) => {
+    try {
+      const authorizationHeader = ctx.req.headers.get('authorization');
+
+      const [presignToken] = (authorizationHeader || '')
+        .split('Bearer ')
+        .filter((s) => s.length > 0);
+
+      if (!presignToken) {
+        throw new AppError(AppErrorCode.UNAUTHORIZED, {
+          message: 'No presign token provided',
+        });
+      }
+
+      const apiToken = await verifyEmbeddingPresignToken({ token: presignToken });
+
+      const { documentId, title, externalId, recipients, meta } = input;
+
+      if (meta && Object.values(meta).length > 0) {
+        await upsertDocumentMeta({
+          documentId: documentId,
+          userId: apiToken.userId,
+          teamId: apiToken.teamId ?? undefined,
+          ...meta,
+          requestMetadata: ctx.metadata,
+        });
+      }
+
+      await updateDocument({
+        userId: apiToken.userId,
+        teamId: apiToken.teamId ?? undefined,
+        documentId: documentId,
+        data: {
+          title,
+          externalId,
+        },
+        requestMetadata: ctx.metadata,
+      });
+
+      const recipientsWithClientId = recipients.map((recipient) => ({
+        ...recipient,
+        clientId: nanoid(),
+      }));
+
+      const { recipients: updatedRecipients } = await setDocumentRecipients({
+        userId: apiToken.userId,
+        teamId: apiToken.teamId ?? undefined,
+        documentId: documentId,
+        recipients: recipientsWithClientId.map((recipient) => ({
+          id: recipient.id,
+          clientId: recipient.clientId,
+          email: recipient.email,
+          name: recipient.name ?? '',
+          role: recipient.role,
+          signingOrder: recipient.signingOrder,
+        })),
+        requestMetadata: ctx.metadata,
+      });
+
+      const fields = recipientsWithClientId.flatMap((recipient) => {
+        const recipientId = updatedRecipients.find((r) => r.clientId === recipient.clientId)?.id;
+
+        if (!recipientId) {
+          throw new AppError(AppErrorCode.UNKNOWN_ERROR, {
+            message: 'Recipient not found',
+          });
+        }
+
+        return (recipient.fields ?? []).map((field) => ({
+          ...field,
+          recipientId,
+          // !: Temp property to be removed once we don't link based on signer email
+          signerEmail: recipient.email,
+        }));
+      });
+
+      await setFieldsForDocument({
+        userId: apiToken.userId,
+        teamId: apiToken.teamId ?? undefined,
+        documentId,
+        fields: fields.map((field) => ({
+          ...field,
+          pageWidth: field.width,
+          pageHeight: field.height,
+        })),
+        requestMetadata: ctx.metadata,
+      });
+
+      return {
+        documentId,
+      };
+    } catch (error) {
+      if (error instanceof AppError) {
+        throw error;
+      }
+
+      throw new AppError(AppErrorCode.UNKNOWN_ERROR, {
+        message: 'Failed to update document',
+      });
+    }
+  });