dow-prod-tracker/src/hooks/use-comments.ts

"use client";

import { useQuery, useMutation, useQueryClient } from "@tanstack/react-query";
import type { CreateCommentInput } from "@/lib/validators/comment";

async function fetchJson<T>(url: string, init?: RequestInit): Promise<T> {
  const res = await fetch(url, init);
  if (!res.ok) {
    const body = await res.json().catch(() => ({}));
    throw new Error(body.error || `Request failed: ${res.status}`);
  }
  return res.json();
}

export function useComments(stageId: string) {
  return useQuery({
    queryKey: ["comments", stageId],
    queryFn: () => fetchJson(`/api/stages/${stageId}/comments`),
    enabled: !!stageId,
  });
}

export function useCreateComment(stageId: string) {
  const queryClient = useQueryClient();

  return useMutation({
    mutationFn: (data: CreateCommentInput) =>
      fetchJson(`/api/stages/${stageId}/comments`, {
        method: "POST",
        headers: { "Content-Type": "application/json" },
        body: JSON.stringify(data),
      }),
    onSuccess: () => {
      queryClient.invalidateQueries({ queryKey: ["comments", stageId] });
    },
  });
}

export function useDeleteComment(stageId: string) {
  const queryClient = useQueryClient();

  return useMutation({
    mutationFn: (commentId: string) =>
      fetchJson(`/api/stages/${stageId}/comments/${commentId}`, {
        method: "DELETE",
      }),
    onSuccess: () => {
      queryClient.invalidateQueries({ queryKey: ["comments", stageId] });
    },
  });
}