quivr/frontend/app/brains-management/[brainId]/components/BrainManagementTabs/hooks/useBrainFetcher.ts

import { useQuery } from "@tanstack/react-query";
import { UUID } from "crypto";
import { useRouter } from "next/navigation";

import { getBrainDataKey } from "@/lib/api/brain/config";
import { useBrainApi } from "@/lib/api/brain/useBrainApi";

type UseBrainFetcherProps = {
  brainId?: UUID;
};

// eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
export const useBrainFetcher = ({ brainId }: UseBrainFetcherProps) => {
  const { getBrain } = useBrainApi();
  const router = useRouter();

  const fetchBrain = async () => {
    try {
      if (brainId === undefined) {
        return undefined;
      }

      return await getBrain(brainId);
    } catch (error) {
      router.push("/brains-management");
    }
  };

  const { data: brain } = useQuery({
    queryKey: [getBrainDataKey(brainId!)],
    queryFn: fetchBrain,
    enabled: brainId !== undefined,
  });

  return {
    brain,
  };
};