import { ConsumerSelect, ConsumerWhereInput } from '@/generated/prisma/models'

const now = new Date()

export const infoSelect: ConsumerSelect = {
  type: true,
  legal: {
    select: {
      name: true,
      registration_code: true,
      partner: {
        select: {
          id: true,
          name: true,
          code: true,
        },
      },
    },
  },
  individual: {
    select: {
      first_name: true,
      last_name: true,
      mobile_number: true,
      partner: {
        select: {
          id: true,
          name: true,
          code: true,
        },
      },
    },
  },
}

export const activeBusinessCount: ConsumerSelect = {
  _count: {
    select: {
      business_activities: {
        where: {
          license_activation: {
            OR: [
              {
                expires_at: {
                  gt: now,
                },
              },
              {
                license_renews: {
                  some: {
                    expires_at: {
                      gt: now,
                    },
                  },
                },
              },
            ],
          },
        },
      },
    },
  },
}

export const consumerRelatedPartner = (partner_id: string): ConsumerWhereInput => ({
  OR: [
    {
      legal: {
        is: {
          partner_id,
        },
      },
    },
    {
      individual: {
        is: {
          partner_id,
        },
      },
    },
  ],
})