'use client';

import { useCustomerQuery } from './customers';
import { createSlot, deleteSlot, getSlot, getSlots, updateSlot } from '@/actions/api/slots';
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';

export const useSlotsQuery = (variables: Parameters<typeof getSlots>[0]) => {
  const { data: { customer } = {} } = useCustomerQuery();

  const masterId = variables.filters?.master?.documentId?.eq || customer?.documentId;
  const date = variables.filters?.date?.eq;

  return useQuery({
    queryFn: () => getSlots(variables),
    queryKey: ['slots', { date: date?.toISOString(), masterId }],
  });
};

export const useSlotQuery = (variables: Parameters<typeof getSlot>[0]) => {
  const { documentId } = variables;

  return useQuery({
    queryFn: () => getSlot(variables),
    queryKey: ['slot', documentId],
  });
};

export const useSlotMutation = ({
  documentId,
}: Pick<Parameters<typeof updateSlot>[0], 'documentId'>) => {
  const queryClient = useQueryClient();

  return useMutation({
    mutationFn: ({ data }: Pick<Parameters<typeof updateSlot>[0], 'data'>) =>
      updateSlot({ data, documentId }),
    mutationKey: ['slot', 'update', documentId],
    onSuccess: () => {
      if (documentId) {
        queryClient.invalidateQueries({ queryKey: ['slot', documentId] });
      }
    },
  });
};

export const useSlotCreate = () => {
  const { data: { customer } = {} } = useCustomerQuery();
  const masterId = customer?.documentId;

  const queryClient = useQueryClient();

  return useMutation({
    mutationFn: createSlot,
    mutationKey: ['slot', 'create'],
    onSuccess: masterId
      ? (data) => {
          queryClient.invalidateQueries({
            queryKey: ['slots', { date: data?.createSlot?.date?.toISOString(), masterId }],
          });
        }
      : undefined,
  });
};

export const useSlotDelete = ({ documentId }: Parameters<typeof deleteSlot>[0]) => {
  const { data: { slot } = {} } = useSlotQuery({ documentId });

  const queryClient = useQueryClient();

  return useMutation({
    mutationFn: () => deleteSlot({ documentId }),
    mutationKey: ['slot', 'delete', documentId],
    onSuccess: () => {
      const date = slot?.date;
      const masterId = slot?.master;

      if (date && masterId) {
        queryClient.invalidateQueries({
          queryKey: ['slots', { date: date?.toISOString(), masterId }],
        });

        queryClient.invalidateQueries({
          queryKey: ['slot', documentId],
        });
      }
    },
  });
};