86 lines
2.5 KiB
TypeScript
86 lines
2.5 KiB
TypeScript
'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 useSlots = (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', 'master', masterId, 'list', date.toISOString()],
|
|
});
|
|
};
|
|
|
|
export const useSlotQuery = (variables: Parameters<typeof getSlot>[0]) => {
|
|
const { documentId } = variables;
|
|
|
|
return useQuery({
|
|
queryFn: () => getSlot(variables),
|
|
queryKey: ['slots', 'get', documentId],
|
|
});
|
|
};
|
|
|
|
export const useSlotMutation = ({
|
|
documentId,
|
|
}: Pick<Parameters<typeof updateSlot>[0], 'documentId'>) => {
|
|
const queryClient = useQueryClient();
|
|
|
|
function handleOnSuccess() {
|
|
queryClient.invalidateQueries({
|
|
queryKey: ['slots', 'get', documentId],
|
|
});
|
|
}
|
|
|
|
return useMutation({
|
|
mutationFn: ({ data }: Pick<Parameters<typeof updateSlot>[0], 'data'>) =>
|
|
updateSlot({ data, documentId }),
|
|
mutationKey: ['slots', 'update', documentId],
|
|
onSuccess: handleOnSuccess,
|
|
});
|
|
};
|
|
|
|
export const useSlotCreate = ({ date }: { date: Date }) => {
|
|
const { data: { customer } = {} } = useCustomerQuery();
|
|
const masterId = customer?.documentId;
|
|
|
|
const queryClient = useQueryClient();
|
|
|
|
function handleOnSuccess() {
|
|
queryClient.invalidateQueries({
|
|
queryKey: ['slots', 'master', masterId, 'list', date.toISOString()],
|
|
});
|
|
}
|
|
|
|
return useMutation({
|
|
mutationFn: ({ input }: { input: Omit<Parameters<typeof createSlot>[0]['input'], 'date'> }) =>
|
|
createSlot({ input: { ...input, date } }),
|
|
mutationKey: ['slots', 'create', 'date', date.toISOString(), 'master', masterId],
|
|
onSuccess: handleOnSuccess,
|
|
});
|
|
};
|
|
|
|
export const useSlotDelete = ({ documentId }: Parameters<typeof deleteSlot>[0]) => {
|
|
const { data: { slot } = {} } = useSlotQuery({ documentId });
|
|
|
|
const queryClient = useQueryClient();
|
|
|
|
function handleOnSuccess() {
|
|
const date = slot?.date;
|
|
const masterId = slot?.master;
|
|
|
|
queryClient.invalidateQueries({
|
|
queryKey: ['slots', 'master', masterId, 'list', date.toISOString()],
|
|
});
|
|
}
|
|
|
|
return useMutation({
|
|
mutationFn: () => deleteSlot({ documentId }),
|
|
mutationKey: ['slots', 'delete', documentId],
|
|
onSuccess: handleOnSuccess,
|
|
});
|
|
};
|