import { useMutation, useQueryClient } from '@tanstack/react-query';
import { apiClient } from '../api/client';
import type { Task, TaskCollection } from '../types/task';

interface ToggleTaskInput {
  taskId: number;
  done: boolean;
  projectId: number;
}

export function useToggleTaskDone() {
  const queryClient = useQueryClient();

  return useMutation({
    mutationFn: async ({ taskId, done }: ToggleTaskInput) => {
      const response = await apiClient.patch<Task>(
        `/tasks/${taskId}`,
        { done },
        { headers: { 'Content-Type': 'application/merge-patch+json' } },
      );

      return response.data;
    },
    onMutate: async ({ taskId, done, projectId }) => {
      const queryKey = ['projects', projectId, 'tasks'];

      await queryClient.cancelQueries({ queryKey });

      const previousTasks = queryClient.getQueryData<TaskCollection>(queryKey);

      queryClient.setQueryData<TaskCollection>(queryKey, (old) => {
        if (!old) {
          return old;
        }

        return {
          ...old,
          'hydra:member': old['hydra:member'].map((task) =>
            task.id === taskId ? { ...task, done } : task,
          ),
        };
      });

      return { previousTasks, queryKey };
    },
    onError: (_err, _variables, context) => {
      if (context) {
        queryClient.setQueryData(context.queryKey, context.previousTasks);
      }
    },
    onSettled: (_data, _error, variables) => {
      queryClient.invalidateQueries({
        queryKey: ['projects', variables.projectId, 'tasks'],
      });
    },
  });
}