usePoller.ts

usePoller.ts

import { useCallback, useEffect, useMemo } from 'react'

export class Poller<T> {
	private interval: ReturnType<typeof setInterval> | null = null
	constructor(
		private readonly callback: (arg: { didCancelRef: { current: boolean } }) => T,
		private readonly intervalMs: number
	) {}
	readonly didCancelRef = { current: false }
	start() {
		if (this.interval) clearInterval(this.interval)
		this.didCancelRef.current = false
		this.interval = setInterval(() => {
			this.callback({ didCancelRef: this.didCancelRef })
		}, this.intervalMs)
	}
	stop() {
		this.didCancelRef.current = true
		if (this.interval) clearInterval(this.interval)
	}
	force() {
		this.start()
		return this.callback({ didCancelRef: this.didCancelRef })
	}
}

export function usePoller<T>(
	callback: (arg: { didCancelRef: { current: boolean } }) => T,
	duration: number,
	disabled = false
): () => T {
	const poller = useMemo(() => new Poller(callback, duration), [callback, duration])

	useEffect(() => {
		if (disabled) return
		poller.start()
		return () => {
			poller.stop()
		}
	}, [poller, disabled])

	return useCallback(() => poller.force(), [poller])
}