homeio

1"use client";
2
3import type {
4  CreateScheduledTaskInput,
5  ScheduledTaskRecord,
6  UpdateScheduledTaskInput,
7} from "@/lib/shared/contracts/scheduled-tasks";
8import { queryKeys } from "@/lib/shared/query-keys";
9import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
10
11async function fetchTasks(): Promise<ScheduledTaskRecord[]> {
12  const res = await fetch("/api/v1/scheduled-tasks");
13  if (!res.ok) throw new Error("Failed to fetch scheduled tasks");
14  const data = await res.json() as { tasks: ScheduledTaskRecord[] };
15  return data.tasks;
16}
17
18export function useScheduledTasks() {
19  const queryClient = useQueryClient();
20
21  const { data: tasks = [], isLoading } = useQuery({
22    queryKey: queryKeys.scheduledTasks,
23    queryFn: fetchTasks,
24    staleTime: 30_000,
25  });
26
27  const createMutation = useMutation({
28    mutationFn: async (input: CreateScheduledTaskInput) => {
29      const res = await fetch("/api/v1/scheduled-tasks", {
30        method: "POST",
31        headers: { "Content-Type": "application/json" },
32        body: JSON.stringify(input),
33      });
34      if (!res.ok) throw new Error("Failed to create task");
35      const data = await res.json() as { task: ScheduledTaskRecord };
36      return data.task;
37    },
38    onSuccess: () => {
39      queryClient.invalidateQueries({ queryKey: queryKeys.scheduledTasks });
40    },
41  });
42
43  const updateMutation = useMutation({
44    mutationFn: async ({ id, input }: { id: string; input: UpdateScheduledTaskInput }) => {
45      const res = await fetch(`/api/v1/scheduled-tasks/${id}`, {
46        method: "PATCH",
47        headers: { "Content-Type": "application/json" },
48        body: JSON.stringify(input),
49      });
50      if (!res.ok) throw new Error("Failed to update task");
51      const data = await res.json() as { task: ScheduledTaskRecord };
52      return data.task;
53    },
54    onSuccess: () => {
55      queryClient.invalidateQueries({ queryKey: queryKeys.scheduledTasks });
56    },
57  });
58
59  const deleteMutation = useMutation({
60    mutationFn: async (id: string) => {
61      const res = await fetch(`/api/v1/scheduled-tasks/${id}`, { method: "DELETE" });
62      if (!res.ok) throw new Error("Failed to delete task");
63    },
64    onSuccess: () => {
65      queryClient.invalidateQueries({ queryKey: queryKeys.scheduledTasks });
66    },
67  });
68
69  const runNowMutation = useMutation({
70    mutationFn: async (id: string) => {
71      const res = await fetch(`/api/v1/scheduled-tasks/${id}/run`, { method: "POST" });
72      if (!res.ok) throw new Error("Failed to run task");
73    },
74    onSuccess: () => {
75      queryClient.invalidateQueries({ queryKey: queryKeys.scheduledTasks });
76    },
77  });
78
79  return {
80    tasks,
81    isLoading,
82    createTask: createMutation.mutateAsync,
83    updateTask: (id: string, input: UpdateScheduledTaskInput) =>
84      updateMutation.mutateAsync({ id, input }),
85    deleteTask: deleteMutation.mutateAsync,
86    runNow: runNowMutation.mutateAsync,
87    isRunningNow: runNowMutation.isPending,
88  };
89}
90