import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
import { ingestFolder, listRuns } from "@/services/ingestion";
import type { IngestFolderRequest } from "@/services/types";

export function useIngestionRuns() {
  return useQuery({
    queryKey: ["ingestion-runs"],
    queryFn: listRuns,
    refetchInterval: 15_000,
  });
}

export function useStartIngestion() {
  const qc = useQueryClient();
  return useMutation({
    mutationFn: (req: IngestFolderRequest) => ingestFolder(req),
    onSuccess: () => {
      qc.invalidateQueries({ queryKey: ["ingestion-runs"] });
      qc.invalidateQueries({ queryKey: ["documents"] });
      qc.invalidateQueries({ queryKey: ["dashboard"] });
    },
  });
}