36 lines
800 B
TypeScript
36 lines
800 B
TypeScript
import { useRevalidator } from 'react-router';
|
|
import { useEffect } from 'react';
|
|
import { useInterval } from 'usehooks-ts';
|
|
|
|
interface Props {
|
|
interval: number;
|
|
}
|
|
|
|
export function useLiveData({ interval }: Props) {
|
|
const revalidator = useRevalidator();
|
|
|
|
// Handle normal stale-while-revalidate behavior
|
|
useInterval(() => {
|
|
if (revalidator.state === 'idle') {
|
|
revalidator.revalidate();
|
|
}
|
|
}, interval);
|
|
|
|
useEffect(() => {
|
|
const handler = () => {
|
|
if (revalidator.state === 'idle') {
|
|
revalidator.revalidate();
|
|
}
|
|
};
|
|
|
|
window.addEventListener('online', handler);
|
|
document.addEventListener('focus', handler);
|
|
|
|
return () => {
|
|
window.removeEventListener('online', handler);
|
|
document.removeEventListener('focus', handler);
|
|
};
|
|
}, [revalidator]);
|
|
return revalidator;
|
|
}
|