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
}