| 1 |
import { useEffect, useState } from "@wordpress/element"; |
| 2 |
|
| 3 |
const getData = async (path, parse = true) => { |
| 4 |
const response = await wp.apiFetch({ path, parse }); |
| 5 |
|
| 6 |
if (!parse) { |
| 7 |
const data = await response.json(); |
| 8 |
const totalPages = response.headers.get("x-wp-totalpages"); |
| 9 |
const totalItems = response.headers.get("x-wp-total"); |
| 10 |
return { data, totalPages, totalItems }; |
| 11 |
} |
| 12 |
|
| 13 |
return { data: response }; |
| 14 |
}; |
| 15 |
|
| 16 |
const useFetch = (props, dependency = []) => { |
| 17 |
const { path, parse = true, currentPage } = props; |
| 18 |
const [data, setData] = useState([]); |
| 19 |
const [isLoading, setIsLoading] = useState(true); |
| 20 |
const [error, setError] = useState(null); |
| 21 |
const [totalPages, setTotalPages] = useState(); |
| 22 |
const [totalItems, setTotalItems] = useState(); |
| 23 |
const [refetch, setRefetch] = useState(false); |
| 24 |
|
| 25 |
useEffect(() => { |
| 26 |
getData(path, parse) |
| 27 |
.then(({ data, totalPages, totalItems }) => { |
| 28 |
setData(data); |
| 29 |
setTotalPages(totalPages); |
| 30 |
setTotalItems(totalItems); |
| 31 |
setIsLoading(false); |
| 32 |
setRefetch(false); |
| 33 |
}) |
| 34 |
.finally(() => { |
| 35 |
setIsLoading(false); |
| 36 |
setRefetch(false) |
| 37 |
}); |
| 38 |
}, [path, currentPage, ...dependency, refetch]); |
| 39 |
|
| 40 |
return { data, isLoading, error, totalPages, totalItems, setRefetch, setData, setIsLoading }; |
| 41 |
}; |
| 42 |
|
| 43 |
export default useFetch; |
| 44 |
|
