import { EnvelopeType } from '@prisma/client';
import { match } from 'ts-pattern';

import { setFieldsForDocument } from '@documenso/lib/server-only/field/set-fields-for-document';
import { setFieldsForTemplate } from '@documenso/lib/server-only/field/set-fields-for-template';

import { authenticatedProcedure } from '../trpc';
import {
  ZSetEnvelopeFieldsRequestSchema,
  ZSetEnvelopeFieldsResponseSchema,
} from './set-envelope-fields.types';

// Note: This is intended to always be an internal route.
export const setEnvelopeFieldsRoute = authenticatedProcedure
  .input(ZSetEnvelopeFieldsRequestSchema)
  .output(ZSetEnvelopeFieldsResponseSchema)
  .mutation(async ({ input, ctx }) => {
    const { teamId } = ctx;
    const { envelopeId, envelopeType, fields } = input;

    ctx.logger.info({
      input: {
        envelopeId,
      },
    });

    const result = await match(envelopeType)
      .with(EnvelopeType.DOCUMENT, async () =>
        setFieldsForDocument({
          userId: ctx.user.id,
          teamId,
          id: {
            type: 'envelopeId',
            id: envelopeId,
          },
          fields: fields.map((field) => ({
            ...field,
            pageNumber: field.page,
            pageX: field.positionX,
            pageY: field.positionY,
            pageWidth: field.width,
            pageHeight: field.height,
          })),
          requestMetadata: ctx.metadata,
        }),
      )
      .with(EnvelopeType.TEMPLATE, async () =>
        setFieldsForTemplate({
          userId: ctx.user.id,
          teamId,
          id: {
            type: 'envelopeId',
            id: envelopeId,
          },
          fields: fields.map((field) => ({
            ...field,
            pageNumber: field.page,
            pageX: field.positionX,
            pageY: field.positionY,
            pageWidth: field.width,
            pageHeight: field.height,
          })),
        }),
      )
      .exhaustive();

    return {
      fields: result.fields.map((field) => ({
        id: field.id,
        formId: field.formId,
      })),
    };
  });