import type { z } from 'zod';

import { prisma } from '@documenso/prisma';
import { SubscriptionSchema } from '@documenso/prisma/generated/zod/modelSchema/SubscriptionSchema';
import { TeamMemberSchema } from '@documenso/prisma/generated/zod/modelSchema/TeamMemberSchema';
import { TeamSchema } from '@documenso/prisma/generated/zod/modelSchema/TeamSchema';

export type GetTeamsOptions = {
  userId: number;
};

export const ZGetTeamsResponseSchema = TeamSchema.extend({
  currentTeamMember: TeamMemberSchema.pick({
    role: true,
  }),
  subscription: SubscriptionSchema.pick({
    status: true,
  }).nullable(),
}).array();

export type TGetTeamsResponse = z.infer<typeof ZGetTeamsResponseSchema>;

export const getTeams = async ({ userId }: GetTeamsOptions): Promise<TGetTeamsResponse> => {
  const teams = await prisma.team.findMany({
    where: {
      members: {
        some: {
          userId,
        },
      },
    },
    include: {
      subscription: {
        select: {
          status: true,
        },
      },
      members: {
        where: {
          userId,
        },
        select: {
          role: true,
        },
      },
    },
  });

  return teams.map(({ members, ...team }) => ({
    ...team,
    currentTeamMember: members[0],
  }));
};