59 lines
2.4 KiB
JavaScript
59 lines
2.4 KiB
JavaScript
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;
|
|
} |