Carlos Florêncio
68 lines
1.7 KiB
TypeScript
68 lines
1.7 KiB
TypeScript
"use client";
|
|
|
|
import { useCallback, useEffect, useRef, useState } from "react";
|
|
|
|
export type RequestStatus = "idle" | "loading" | "success" | "error";
|
|
|
|
export type RequestState<T> = {
|
|
status: RequestStatus;
|
|
data: T | null;
|
|
error: Error | null;
|
|
};
|
|
|
|
type UseRequestOptions = {
|
|
successDuration?: number;
|
|
};
|
|
|
|
export function useRequest<TArgs extends unknown[], TData>(
|
|
fn: (...args: TArgs) => Promise<TData>,
|
|
options: UseRequestOptions = {},
|
|
) {
|
|
const { successDuration = 1500 } = options;
|
|
const [state, setState] = useState<RequestState<TData>>({
|
|
status: "idle",
|
|
data: null,
|
|
error: null,
|
|
});
|
|
const timeoutRef = useRef<ReturnType<typeof setTimeout> | null>(null);
|
|
|
|
useEffect(() => {
|
|
return () => {
|
|
if (timeoutRef.current) {
|
|
clearTimeout(timeoutRef.current);
|
|
}
|
|
};
|
|
}, []);
|
|
|
|
const run = useCallback(
|
|
async (...args: TArgs) => {
|
|
if (timeoutRef.current) {
|
|
clearTimeout(timeoutRef.current);
|
|
}
|
|
setState({ status: "loading", data: null, error: null });
|
|
try {
|
|
const data = await fn(...args);
|
|
setState({ status: "success", data, error: null });
|
|
if (successDuration > 0) {
|
|
timeoutRef.current = setTimeout(() => {
|
|
setState((prev) => ({ ...prev, status: "idle" }));
|
|
}, successDuration);
|
|
}
|
|
return data;
|
|
} catch (error) {
|
|
const normalized = error instanceof Error ? error : new Error("Request failed");
|
|
setState({ status: "error", data: null, error: normalized });
|
|
throw normalized;
|
|
}
|
|
},
|
|
[fn, successDuration],
|
|
);
|
|
|
|
return {
|
|
...state,
|
|
isLoading: state.status === "loading",
|
|
run,
|
|
reset: () => setState({ status: "idle", data: null, error: null }),
|
|
};
|
|
}
|