Files
omnivore/packages/web/lib/hooks/useScrollWatcher.tsx
2023-02-27 10:34:12 +08:00

43 lines
1.2 KiB
TypeScript

import { useRef, useEffect, useState } from 'react'
type ScrollOffset = {
x: number
y: number
}
export type ScrollOffsetChangeset = {
current: ScrollOffset
previous: ScrollOffset
}
type Effect = (offset: ScrollOffsetChangeset) => void
export function useScrollWatcher(effect: Effect, delay: number): void {
const throttleTimeout = useRef<NodeJS.Timeout | undefined>(undefined)
const [currentOffset, setCurrentOffset] = useState<ScrollOffset>({
x: 0,
y: 0,
})
useEffect(() => {
const callback = () => {
const newOffset = {
x: window.document.documentElement.scrollLeft ?? window?.scrollX ?? 0,
y: window.document.documentElement.scrollTop ?? window?.scrollY ?? 0,
}
effect({ current: newOffset, previous: currentOffset })
setCurrentOffset(newOffset)
throttleTimeout.current = undefined
}
const handleScroll = () => {
if (typeof throttleTimeout.current === 'undefined') {
throttleTimeout.current = setTimeout(callback, delay)
}
}
window.addEventListener('scroll', handleScroll)
return () => window.removeEventListener('scroll', handleScroll)
}, [currentOffset, delay, effect])
}