import { useMutation } from "@tanstack/react-query";
import type {
  MutationFunction,
  UseMutationOptions,
  UseMutationResult,
  QueryClient,
} from "@tanstack/react-query";
import { customInstance } from "@/config/axios";

// Types
export interface CreateLeadDTO {
  name: string;
  email: string;
  phone?: string;
  subject: string;
  message: string;
  source?: string;
}

export interface ContactSubmitResponse {
  message: string;
  data: {
    lead_number: string;
  };
}

export interface ErrorResponse {
  message: string;
  statusCode?: number;
}

// API function
export const submitContactForm = (
  createLeadDTO: CreateLeadDTO,
  options?: Parameters<typeof customInstance>[1]
) => {
  return customInstance<ContactSubmitResponse>(
    {
      url: `/v1/contact`,
      method: "POST",
      headers: { "Content-Type": "application/json" },
      data: createLeadDTO,
    },
    options
  );
};

// Mutation options
export const getSubmitContactFormMutationOptions = <
  TError = ErrorResponse,
  TContext = unknown
>(options?: {
  mutation?: UseMutationOptions<
    Awaited<ReturnType<typeof submitContactForm>>,
    TError,
    { data: CreateLeadDTO },
    TContext
  >;
  request?: Parameters<typeof customInstance>[1];
}): UseMutationOptions<
  Awaited<ReturnType<typeof submitContactForm>>,
  TError,
  { data: CreateLeadDTO },
  TContext
> => {
  const mutationKey = ["submitContactForm"];
  const { mutation: mutationOptions, request: requestOptions } = options
    ? options.mutation &&
      "mutationKey" in options.mutation &&
      options.mutation.mutationKey
      ? options
      : { ...options, mutation: { ...options.mutation, mutationKey } }
    : { mutation: { mutationKey }, request: undefined };

  const mutationFn: MutationFunction<
    Awaited<ReturnType<typeof submitContactForm>>,
    { data: CreateLeadDTO }
  > = (props) => {
    const { data } = props ?? {};
    return submitContactForm(data, requestOptions);
  };

  return { mutationFn, ...mutationOptions };
};

// React Query hook
export const useSubmitContactForm = <
  TError = ErrorResponse,
  TContext = unknown
>(
  options?: {
    mutation?: UseMutationOptions<
      Awaited<ReturnType<typeof submitContactForm>>,
      TError,
      { data: CreateLeadDTO },
      TContext
    >;
    request?: Parameters<typeof customInstance>[1];
  },
  queryClient?: QueryClient
): UseMutationResult<
  Awaited<ReturnType<typeof submitContactForm>>,
  TError,
  { data: CreateLeadDTO },
  TContext
> => {
  const mutationOptions = getSubmitContactFormMutationOptions(options);
  return useMutation(mutationOptions, queryClient);
};