UNPKG

@rc-component/trigger

Version:

base abstract trigger component for react

35 lines (34 loc) 1.18 kB
import useLayoutEffect from "@rc-component/util/es/hooks/useLayoutEffect"; import { collectScroller, getWin } from "../util"; export default function useWatch(open, target, popup, onAlign, onScroll) { useLayoutEffect(() => { if (open && target && popup) { const targetElement = target; const popupElement = popup; const targetScrollList = collectScroller(targetElement); const popupScrollList = collectScroller(popupElement); const win = getWin(popupElement); const mergedList = new Set([win, ...targetScrollList, ...popupScrollList]); function notifyScroll() { onAlign(); onScroll(); } mergedList.forEach(scroller => { scroller.addEventListener('scroll', notifyScroll, { passive: true }); }); win.addEventListener('resize', notifyScroll, { passive: true }); // First time always do align onAlign(); return () => { mergedList.forEach(scroller => { scroller.removeEventListener('scroll', notifyScroll); win.removeEventListener('resize', notifyScroll); }); }; } }, [open, target, popup]); }