import { useMutation, useQueryClient, type UseMutationOptions } from '@tanstack/react-query';
import { api } from '@/lib/api-client';
import type { UpdateCmsInstanceStatusRequest, UpdateStatusResult } from './types';

type UpdateVariables = { id: string } & UpdateCmsInstanceStatusRequest;

export function useUpdateCmsInstanceStatus(
    options?: Pick<UseMutationOptions<UpdateStatusResult, Error, UpdateVariables>, 'onSuccess' | 'onError'>,
) {
    const queryClient = useQueryClient();
    return useMutation<UpdateStatusResult, Error, UpdateVariables>({
        mutationFn: ({ id, ...body }) =>
            api.put<UpdateStatusResult>(`/api/v1/CmsInstances/${id}/status`, body),
        onSuccess: (data, variables, ...rest) => {
            queryClient.invalidateQueries({ queryKey: ['cmsInstances'] });
            options?.onSuccess?.(data, variables, ...rest);
        },
        onError: options?.onError,
    });
}