1
This commit is contained in:
+59
@@ -0,0 +1,59 @@
|
||||
import { getShadowRoot } from "@rc-component/util/es/Dom/shadow";
|
||||
import { warning } from "@rc-component/util/es/warning";
|
||||
import * as React from 'react';
|
||||
import { getWin } from "../util";
|
||||
|
||||
/**
|
||||
* Close if click on the window.
|
||||
* Return the function that click on the Popup element.
|
||||
*/
|
||||
export default function useWinClick(open, clickToHide, targetEle, popupEle, mask, maskClosable, inPopupOrChild, triggerOpen) {
|
||||
const openRef = React.useRef(open);
|
||||
openRef.current = open;
|
||||
const popupPointerDownRef = React.useRef(false);
|
||||
|
||||
// Click to hide is special action since click popup element should not hide
|
||||
React.useEffect(() => {
|
||||
if (clickToHide && popupEle && (!mask || maskClosable)) {
|
||||
const onPointerDown = () => {
|
||||
popupPointerDownRef.current = false;
|
||||
};
|
||||
const onTriggerClose = e => {
|
||||
if (openRef.current && !inPopupOrChild(e.composedPath?.()?.[0] || e.target) && !popupPointerDownRef.current) {
|
||||
triggerOpen(false);
|
||||
}
|
||||
};
|
||||
const win = getWin(popupEle);
|
||||
win.addEventListener('pointerdown', onPointerDown, true);
|
||||
win.addEventListener('mousedown', onTriggerClose, true);
|
||||
win.addEventListener('contextmenu', onTriggerClose, true);
|
||||
|
||||
// shadow root
|
||||
const targetShadowRoot = getShadowRoot(targetEle);
|
||||
if (targetShadowRoot) {
|
||||
targetShadowRoot.addEventListener('mousedown', onTriggerClose, true);
|
||||
targetShadowRoot.addEventListener('contextmenu', onTriggerClose, true);
|
||||
}
|
||||
|
||||
// Warning if target and popup not in same root
|
||||
if (process.env.NODE_ENV !== 'production' && targetEle) {
|
||||
const targetRoot = targetEle.getRootNode?.();
|
||||
const popupRoot = popupEle.getRootNode?.();
|
||||
warning(targetRoot === popupRoot, `trigger element and popup element should in same shadow root.`);
|
||||
}
|
||||
return () => {
|
||||
win.removeEventListener('pointerdown', onPointerDown, true);
|
||||
win.removeEventListener('mousedown', onTriggerClose, true);
|
||||
win.removeEventListener('contextmenu', onTriggerClose, true);
|
||||
if (targetShadowRoot) {
|
||||
targetShadowRoot.removeEventListener('mousedown', onTriggerClose, true);
|
||||
targetShadowRoot.removeEventListener('contextmenu', onTriggerClose, true);
|
||||
}
|
||||
};
|
||||
}
|
||||
}, [clickToHide, targetEle, popupEle, mask, maskClosable]);
|
||||
function onPopupPointerDown() {
|
||||
popupPointerDownRef.current = true;
|
||||
}
|
||||
return onPopupPointerDown;
|
||||
}
|
||||
Reference in New Issue
Block a user