front-hub/src/utils/hooks/useTariffFetcher.ts

import { useEffect, useLayoutEffect, useRef, useState } from "react";

import { getTariffs } from "@root/api/tariff";

import type { Tariff } from "@frontend/kitui";

export function useTariffFetcher({
  tariffsPerPage,
  apiPage,
  onSuccess,
  onError,
}: {
  baseUrl?: string;
  tariffsPerPage: number;
  apiPage: number;
  onSuccess: (response: Tariff[]) => void;
  onError?: (error: Error) => void;
}) {
  const [fetchState, setFetchState] = useState<"fetching" | "idle" | "all fetched">("idle");
  const onSuccessRef = useRef(onSuccess);
  const onErrorRef = useRef(onError);

  useLayoutEffect(() => {
    onSuccessRef.current = onSuccess;
    onErrorRef.current = onError;
  }, [onError, onSuccess]);

  useEffect(() => {
    const controller = new AbortController();

    setFetchState("fetching");
    getTariffs(apiPage, tariffsPerPage, controller.signal)
      .then(([result]) => {
        if (result && result.tariffs.length > 0) {
          onSuccessRef.current(result.tariffs);
          setFetchState("idle");
        } else setFetchState("all fetched");
      })
      .catch(([_, error]) => {
        onErrorRef.current?.(error);
      });

    return () => controller.abort();
  }, [apiPage, tariffsPerPage]);

  return fetchState;
}