import { useMutation, useQueryClient } from "@tanstack/react-query";
import axios, { type Method } from "axios";

type MutationDataOptions<TInput, TOutput> = {
  url: string;
  method?: Method;
  onSuccess?: (data: TOutput) => void;
  onError?: (error: any) => void;
  invalidate?: string[][];
};

export function useMutationData<TInput = any, TOutput = any>({
  url,
  method = "POST",
  onSuccess,
  onError,
  invalidate = [],
}: MutationDataOptions<TInput, TOutput>) {
  const queryClient = useQueryClient();

  return useMutation<TOutput, any, { data: TInput; config?: any }>({
    mutationFn: async ({ data, config }) => {
       const isFormData = data instanceof FormData;

      const response = await axios.request({
        url,
        method,
        data,
        headers: {
          ...(isFormData
            ? {} 
            : { "Content-Type": "application/json" }),
        },
        ...config,
      });
      return response.data;
    },
    onSuccess: (data) => {
      invalidate.forEach((key) =>
        queryClient.invalidateQueries({ queryKey: key })
      );
      onSuccess?.(data);
    },
    onError,
  });
}