Open-Assistant/website/src/hooks/tasks/useGenericTaskAPI.tsx

import { useState } from "react";
import { get, post } from "src/lib/api";
import { BaseTask, TaskResponse } from "src/types/Task";
import useSWRImmutable from "swr/immutable";
import useSWRMutation from "swr/mutation";

export const useGenericTaskAPI = <TaskType extends BaseTask>(taskApiEndpoint: string) => {
  type ConcreteTaskResponse = TaskResponse<TaskType>;

  const [tasks, setTasks] = useState<ConcreteTaskResponse[]>([]);

  const { isLoading, mutate, error } = useSWRImmutable<ConcreteTaskResponse>("/api/new_task/" + taskApiEndpoint, get, {
    onSuccess: (data) => setTasks([data]),
    revalidateOnMount: true,
    dedupingInterval: 500,
  });

  const { trigger } = useSWRMutation("/api/update_task", post, {
    onSuccess: async (response) => {
      const newTask: ConcreteTaskResponse = response;
      setTasks((oldTasks) => [...oldTasks, newTask]);
      mutate();
    },
  });

  return { tasks, isLoading, trigger, error, reset: mutate };
};