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;
}