import { Prisma } from "@prisma/client";
import { cache as reactCache } from "react";
import { prisma } from "@formbricks/database";
import { TContactAttributes } from "@formbricks/types/contact-attribute";
import { DatabaseError } from "@formbricks/types/errors";

export const getContact = reactCache(async (contactId: string) => {
  try {
    const contact = await prisma.contact.findUnique({
      where: { id: contactId },
      select: { id: true },
    });

    return contact;
  } catch (error) {
    if (error instanceof Prisma.PrismaClientKnownRequestError) {
      throw new DatabaseError(error.message);
    }
  }
});

export const getContactByUserId = reactCache(
  async (
    environmentId: string,
    userId: string
  ): Promise<{
    id: string;
    attributes: TContactAttributes;
  } | null> => {
    const contact = await prisma.contact.findFirst({
      where: {
        attributes: {
          some: {
            attributeKey: {
              key: "userId",
              environmentId,
            },
            value: userId,
          },
        },
      },
      select: {
        id: true,
        attributes: {
          select: {
            attributeKey: { select: { key: true } },
            value: true,
          },
        },
      },
    });

    if (!contact) {
      return null;
    }

    const contactAttributes = contact.attributes.reduce<TContactAttributes>((acc, attr) => {
      acc[attr.attributeKey.key] = attr.value;
      return acc;
    }, {});

    return {
      id: contact.id,
      attributes: contactAttributes,
    };
  }
);
