Portal.js 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113
  1. "use strict";
  2. var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard").default;
  3. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
  4. Object.defineProperty(exports, "__esModule", {
  5. value: true
  6. });
  7. exports.default = void 0;
  8. var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
  9. var React = _interopRequireWildcard(require("react"));
  10. var _reactDom = require("react-dom");
  11. var _canUseDom = _interopRequireDefault(require("rc-util/lib/Dom/canUseDom"));
  12. var _warning = _interopRequireDefault(require("rc-util/lib/warning"));
  13. var _ref2 = require("rc-util/lib/ref");
  14. var _Context = _interopRequireDefault(require("./Context"));
  15. var _useDom3 = _interopRequireDefault(require("./useDom"));
  16. var _useScrollLocker = _interopRequireDefault(require("./useScrollLocker"));
  17. var _mock = require("./mock");
  18. var getPortalContainer = function getPortalContainer(getContainer) {
  19. if (getContainer === false) {
  20. return false;
  21. }
  22. if (!(0, _canUseDom.default)() || !getContainer) {
  23. return null;
  24. }
  25. if (typeof getContainer === 'string') {
  26. return document.querySelector(getContainer);
  27. }
  28. if (typeof getContainer === 'function') {
  29. return getContainer();
  30. }
  31. return getContainer;
  32. };
  33. var Portal = /*#__PURE__*/React.forwardRef(function (props, ref) {
  34. var open = props.open,
  35. autoLock = props.autoLock,
  36. getContainer = props.getContainer,
  37. debug = props.debug,
  38. _props$autoDestroy = props.autoDestroy,
  39. autoDestroy = _props$autoDestroy === void 0 ? true : _props$autoDestroy,
  40. children = props.children;
  41. var _React$useState = React.useState(open),
  42. _React$useState2 = (0, _slicedToArray2.default)(_React$useState, 2),
  43. shouldRender = _React$useState2[0],
  44. setShouldRender = _React$useState2[1];
  45. var mergedRender = shouldRender || open;
  46. // ========================= Warning =========================
  47. if (process.env.NODE_ENV !== 'production') {
  48. (0, _warning.default)((0, _canUseDom.default)() || !open, "Portal only work in client side. Please call 'useEffect' to show Portal instead default render in SSR.");
  49. }
  50. // ====================== Should Render ======================
  51. React.useEffect(function () {
  52. if (autoDestroy || open) {
  53. setShouldRender(open);
  54. }
  55. }, [open, autoDestroy]);
  56. // ======================== Container ========================
  57. var _React$useState3 = React.useState(function () {
  58. return getPortalContainer(getContainer);
  59. }),
  60. _React$useState4 = (0, _slicedToArray2.default)(_React$useState3, 2),
  61. innerContainer = _React$useState4[0],
  62. setInnerContainer = _React$useState4[1];
  63. React.useEffect(function () {
  64. var customizeContainer = getPortalContainer(getContainer);
  65. // Tell component that we check this in effect which is safe to be `null`
  66. setInnerContainer(customizeContainer !== null && customizeContainer !== void 0 ? customizeContainer : null);
  67. });
  68. var _useDom = (0, _useDom3.default)(mergedRender && !innerContainer, debug),
  69. _useDom2 = (0, _slicedToArray2.default)(_useDom, 2),
  70. defaultContainer = _useDom2[0],
  71. queueCreate = _useDom2[1];
  72. var mergedContainer = innerContainer !== null && innerContainer !== void 0 ? innerContainer : defaultContainer;
  73. // ========================= Locker ==========================
  74. (0, _useScrollLocker.default)(autoLock && open && (0, _canUseDom.default)() && (mergedContainer === defaultContainer || mergedContainer === document.body));
  75. // =========================== Ref ===========================
  76. var childRef = null;
  77. if (children && (0, _ref2.supportRef)(children) && ref) {
  78. var _ref = children;
  79. childRef = _ref.ref;
  80. }
  81. var mergedRef = (0, _ref2.useComposeRef)(childRef, ref);
  82. // ========================= Render ==========================
  83. // Do not render when nothing need render
  84. // When innerContainer is `undefined`, it may not ready since user use ref in the same render
  85. if (!mergedRender || !(0, _canUseDom.default)() || innerContainer === undefined) {
  86. return null;
  87. }
  88. // Render inline
  89. var renderInline = mergedContainer === false || (0, _mock.inlineMock)();
  90. var reffedChildren = children;
  91. if (ref) {
  92. reffedChildren = /*#__PURE__*/React.cloneElement(children, {
  93. ref: mergedRef
  94. });
  95. }
  96. return /*#__PURE__*/React.createElement(_Context.default.Provider, {
  97. value: queueCreate
  98. }, renderInline ? reffedChildren : /*#__PURE__*/(0, _reactDom.createPortal)(reffedChildren, mergedContainer));
  99. });
  100. if (process.env.NODE_ENV !== 'production') {
  101. Portal.displayName = 'Portal';
  102. }
  103. var _default = Portal;
  104. exports.default = _default;