@rc-component/trigger
Version:
base abstract trigger component for react
35 lines (34 loc) • 1.18 kB
JavaScript
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]);
}