22 lines
674 B
JavaScript
22 lines
674 B
JavaScript
|
import { useEvent } from 'rc-util';
|
||
|
import * as React from 'react';
|
||
|
function voidFunc() {}
|
||
|
const WatermarkContext = /*#__PURE__*/React.createContext({
|
||
|
add: voidFunc,
|
||
|
remove: voidFunc
|
||
|
});
|
||
|
export function usePanelRef(panelSelector) {
|
||
|
const watermark = React.useContext(WatermarkContext);
|
||
|
const panelEleRef = React.useRef();
|
||
|
const panelRef = useEvent(ele => {
|
||
|
if (ele) {
|
||
|
const innerContentEle = panelSelector ? ele.querySelector(panelSelector) : ele;
|
||
|
watermark.add(innerContentEle);
|
||
|
panelEleRef.current = innerContentEle;
|
||
|
} else {
|
||
|
watermark.remove(panelEleRef.current);
|
||
|
}
|
||
|
});
|
||
|
return panelRef;
|
||
|
}
|
||
|
export default WatermarkContext;
|