import { useMutation, useQueryClient } from 'react-query';

import useHuSnackbar from '@material-hu/components/design-system/Snackbar';

import { useLokaliseTranslation } from 'src/utils/i18n';

import { knowledgeBaseKeys } from '../queries';
import { editKnowledgeBaseAudience } from '../services';
import { type EditKnowledgeBaseAudiencePayload } from '../types';

export const useEditKnowledgeBaseAudience = () => {
  const { t } = useLokaliseTranslation('agents');
  const { enqueueSnackbar } = useHuSnackbar();
  const queryClient = useQueryClient();

  return useMutation({
    mutationFn: (payload: EditKnowledgeBaseAudiencePayload) =>
      editKnowledgeBaseAudience(payload),
    onSuccess: () => {
      enqueueSnackbar({
        title: t('knowledge_base.edit_audience_drawer.success'),
        variant: 'success',
      });
      queryClient.invalidateQueries(knowledgeBaseKeys.all);
    },
    onError: () => {
      enqueueSnackbar({
        title: t('knowledge_base.edit_audience_drawer.error'),
        variant: 'error',
      });
    },
  });
};