48 lines
1.3 KiB
TypeScript
48 lines
1.3 KiB
TypeScript
|
|
import { useState, useEffect, useCallback } from 'react';
|
|
import { type ApiResponse } from '@/types/apiResponse';
|
|
|
|
// Define options for the hook
|
|
interface UseApiOptions {
|
|
// If true, the API call will be executed immediately on mount
|
|
immediate?: boolean;
|
|
}
|
|
|
|
export function useApi<T extends ApiResponse, P extends any[]>(
|
|
apiFunction: (...args: P) => Promise<{ data: T }>,
|
|
options: UseApiOptions = {},
|
|
) {
|
|
const [data, setData] = useState<T | null>(null);
|
|
const [loading, setLoading] = useState(false);
|
|
const [error, setError] = useState<unknown>(null);
|
|
|
|
const execute = useCallback(
|
|
async (...args: P) => {
|
|
setLoading(true);
|
|
setError(null);
|
|
|
|
try {
|
|
const response = await apiFunction(...args);
|
|
|
|
setData(response.data);
|
|
} catch (err) {
|
|
// TODO: can handle some common errors here, 400 and 500 errors
|
|
setError(err);
|
|
} finally {
|
|
setLoading(false);
|
|
}
|
|
},
|
|
[apiFunction],
|
|
);
|
|
|
|
// If the 'immediate' option is true, execute the function on mount
|
|
useEffect(() => {
|
|
if (options.immediate) {
|
|
// We pass undefined as params for the initial call.
|
|
execute(...(undefined as unknown as P));
|
|
}
|
|
}, [execute, options.immediate]);
|
|
|
|
return { data, loading, error, execute };
|
|
}
|