import { useMutation, useQueryClient } from "@tanstack/react-query";
import * as api from "../service/TaskPhaseService";
import { CreateTaskPhaseType } from "../types/TaskPhaseTypes";

export const useCreateTaskPhase = () => {
  const queryClient = useQueryClient();

  return useMutation({
    mutationFn: (variables: CreateTaskPhaseType) => api.createTaskPhase(variables),
    onSuccess: (_data, variables) => {
      queryClient.invalidateQueries({
        queryKey: ["task-phases", variables.projectId],
      });
    },
  });
};

export const useDeleteTaskPhase = () => {
  const queryClient = useQueryClient();

  return useMutation({
    mutationFn: (id: string) => api.deleteTaskPhase(id),
    onSuccess: () => {
      queryClient.invalidateQueries({ queryKey: ["task-phases"] });
    },
  });
};