import { z } from "zod"
import { CreateCustomerGroupSchema } from "../../routes/customer-groups/customer-group-create/components/create-customer-group-form"
import { EditCustomerGroupSchema } from "../../routes/customer-groups/customer-group-edit/components/edit-customer-group-form"
import { deleteRequest, getRequest, postRequest } from "./common"
import { HttpTypes, PaginatedResponse } from "@medusajs/types"

async function retrieveCustomerGroup(id: string, query?: Record<string, any>) {
  return getRequest<{ customer_group: HttpTypes.AdminCustomerGroup }>(
    `/admin/customer-groups/${id}`,
    query
  )
}

async function listCustomerGroups(query?: Record<string, any>) {
  return getRequest<
    PaginatedResponse<{ customer_groups: HttpTypes.AdminCustomerGroup[] }>
  >(`/admin/customer-groups`, query)
}

async function createCustomerGroup(
  payload: z.infer<typeof CreateCustomerGroupSchema>
) {
  return postRequest<{ customer_group: HttpTypes.AdminCustomerGroup }>(
    `/admin/customer-groups`,
    payload
  )
}

async function updateCustomerGroup(
  id: string,
  payload: z.infer<typeof EditCustomerGroupSchema>
) {
  return postRequest<{ customer_group: HttpTypes.AdminCustomerGroup }>(
    `/admin/customer-groups/${id}`,
    payload
  )
}

async function deleteCustomerGroup(id: string) {
  return deleteRequest<{
    id: string
    deleted: boolean
    object: "customer-group"
  }>(`/admin/customer-groups/${id}`)
}

async function batchAddCustomers(
  id: string,
  payload: { customer_ids: string[] }
) {
  return postRequest<{ customer_group: HttpTypes.AdminCustomerGroup }>(
    `/admin/customer-groups/${id}/customers`,
    {
      add: payload.customer_ids,
    }
  )
}

async function batchRemoveCustomers(
  id: string,
  payload: { customer_ids: string[] }
) {
  return postRequest<{ customer_group: HttpTypes.AdminCustomerGroup }>(
    `/admin/customer-groups/${id}/customers`,
    {
      remove: payload.customer_ids,
    }
  )
}

export const customerGroups = {
  retrieve: retrieveCustomerGroup,
  list: listCustomerGroups,
  create: createCustomerGroup,
  update: updateCustomerGroup,
  delete: deleteCustomerGroup,
  addCustomers: batchAddCustomers,
  removeCustomers: batchRemoveCustomers,
}