import {useMutation} from '@tanstack/react-query';
import {apiClient} from '@common/http/query-client';
import {toast} from '@common/ui/toast/toast';
import {useTrans} from '@common/i18n/use-trans';
import {message} from '@common/i18n/message';
import {showHttpErrorToast} from '@common/utils/http/show-http-error-toast';

interface Payload {
  model: string;
  driver: string;
}

export function useImportSearchModels() {
  const {trans} = useTrans();
  return useMutation({
    mutationFn: (payload: Payload) => importModels(payload),
    onSuccess: () => {
      toast(trans(message('Imported search models')));
    },
    onError: err => showHttpErrorToast(err),
  });
}

function importModels(payload: Payload): Promise<Response> {
  return apiClient.post('admin/search/import', payload).then(r => r.data);
}