1

video-gen-admin/node_modules/@rc-component/util/es/PortalWrapper.js

@@ -0,0 +1,192 @@
+/* eslint-disable no-underscore-dangle,react/require-default-props */
+import * as React from 'react';
+import raf from "./raf";
+import Portal from "./Portal";
+import canUseDom from "./Dom/canUseDom";
+import setStyle from "./setStyle";
+import ScrollLocker from "./Dom/scrollLocker";
+let openCount = 0;
+const supportDom = canUseDom();
+
+/** @private Test usage only */
+export function getOpenCount() {
+  return process.env.NODE_ENV === 'test' ? openCount : 0;
+}
+
+// https://github.com/ant-design/ant-design/issues/19340
+// https://github.com/ant-design/ant-design/issues/19332
+let cacheOverflow = {};
+const getParent = getContainer => {
+  if (!supportDom) {
+    return null;
+  }
+  if (getContainer) {
+    if (typeof getContainer === 'string') {
+      return document.querySelectorAll(getContainer)[0];
+    }
+    if (typeof getContainer === 'function') {
+      return getContainer();
+    }
+    if (typeof getContainer === 'object' && getContainer instanceof window.HTMLElement) {
+      return getContainer;
+    }
+  }
+  return document.body;
+};
+class PortalWrapper extends React.Component {
+  container;
+  componentRef = /*#__PURE__*/React.createRef();
+  rafId;
+  scrollLocker;
+  constructor(props) {
+    super(props);
+    this.scrollLocker = new ScrollLocker({
+      container: getParent(props.getContainer)
+    });
+  }
+  renderComponent;
+  componentDidMount() {
+    this.updateOpenCount();
+    if (!this.attachToParent()) {
+      this.rafId = raf(() => {
+        this.forceUpdate();
+      });
+    }
+  }
+  componentDidUpdate(prevProps) {
+    this.updateOpenCount(prevProps);
+    this.updateScrollLocker(prevProps);
+    this.setWrapperClassName();
+    this.attachToParent();
+  }
+  updateScrollLocker = prevProps => {
+    const {
+      visible: prevVisible
+    } = prevProps || {};
+    const {
+      getContainer,
+      visible
+    } = this.props;
+    if (visible && visible !== prevVisible && supportDom && getParent(getContainer) !== this.scrollLocker.getContainer()) {
+      this.scrollLocker.reLock({
+        container: getParent(getContainer)
+      });
+    }
+  };
+  updateOpenCount = prevProps => {
+    const {
+      visible: prevVisible,
+      getContainer: prevGetContainer
+    } = prevProps || {};
+    const {
+      visible,
+      getContainer
+    } = this.props;
+
+    // Update count
+    if (visible !== prevVisible && supportDom && getParent(getContainer) === document.body) {
+      if (visible && !prevVisible) {
+        openCount += 1;
+      } else if (prevProps) {
+        openCount -= 1;
+      }
+    }
+
+    // Clean up container if needed
+    const getContainerIsFunc = typeof getContainer === 'function' && typeof prevGetContainer === 'function';
+    if (getContainerIsFunc ? getContainer.toString() !== prevGetContainer.toString() : getContainer !== prevGetContainer) {
+      this.removeCurrentContainer();
+    }
+  };
+  componentWillUnmount() {
+    const {
+      visible,
+      getContainer
+    } = this.props;
+    if (supportDom && getParent(getContainer) === document.body) {
+      // 离开时不会 render， 导到离开时数值不变，改用 func 。。
+      openCount = visible && openCount ? openCount - 1 : openCount;
+    }
+    this.removeCurrentContainer();
+    raf.cancel(this.rafId);
+  }
+  attachToParent = (force = false) => {
+    if (force || this.container && !this.container.parentNode) {
+      const parent = getParent(this.props.getContainer);
+      if (parent) {
+        parent.appendChild(this.container);
+        return true;
+      }
+      return false;
+    }
+    return true;
+  };
+  getContainer = () => {
+    if (!supportDom) {
+      return null;
+    }
+    if (!this.container) {
+      this.container = document.createElement('div');
+      this.attachToParent(true);
+    }
+    this.setWrapperClassName();
+    return this.container;
+  };
+  setWrapperClassName = () => {
+    const {
+      wrapperClassName
+    } = this.props;
+    if (this.container && wrapperClassName && wrapperClassName !== this.container.className) {
+      this.container.className = wrapperClassName;
+    }
+  };
+  removeCurrentContainer = () => {
+    // Portal will remove from `parentNode`.
+    // Let's handle this again to avoid refactor issue.
+    this.container?.parentNode?.removeChild(this.container);
+  };
+
+  /**
+   * Enhance ./switchScrollingEffect
+   * 1. Simulate document body scroll bar with
+   * 2. Record body has overflow style and recover when all of PortalWrapper invisible
+   * 3. Disable body scroll when PortalWrapper has open
+   *
+   * @memberof PortalWrapper
+   */
+  switchScrollingEffect = () => {
+    if (openCount === 1 && !Object.keys(cacheOverflow).length) {
+      // Must be set after switchScrollingEffect
+      cacheOverflow = setStyle({
+        overflow: 'hidden',
+        overflowX: 'hidden',
+        overflowY: 'hidden'
+      });
+    } else if (!openCount) {
+      setStyle(cacheOverflow);
+      cacheOverflow = {};
+    }
+  };
+  render() {
+    const {
+      children,
+      forceRender,
+      visible
+    } = this.props;
+    let portal = null;
+    const childProps = {
+      getOpenCount: () => openCount,
+      getContainer: this.getContainer,
+      switchScrollingEffect: this.switchScrollingEffect,
+      scrollLocker: this.scrollLocker
+    };
+    if (forceRender || visible || this.componentRef.current) {
+      portal = /*#__PURE__*/React.createElement(Portal, {
+        getContainer: this.getContainer,
+        ref: this.componentRef
+      }, children(childProps));
+    }
+    return portal;
+  }
+}
+export default PortalWrapper;