useDom.js 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  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 = useDom;
  8. var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime/helpers/toConsumableArray"));
  9. var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
  10. var React = _interopRequireWildcard(require("react"));
  11. var _useLayoutEffect = _interopRequireDefault(require("rc-util/lib/hooks/useLayoutEffect"));
  12. var _canUseDom = _interopRequireDefault(require("rc-util/lib/Dom/canUseDom"));
  13. var _Context = _interopRequireDefault(require("./Context"));
  14. var EMPTY_LIST = [];
  15. /**
  16. * Will add `div` to document. Nest call will keep order
  17. * @param render Render DOM in document
  18. */
  19. function useDom(render, debug) {
  20. var _React$useState = React.useState(function () {
  21. if (!(0, _canUseDom.default)()) {
  22. return null;
  23. }
  24. var defaultEle = document.createElement('div');
  25. if (process.env.NODE_ENV !== 'production' && debug) {
  26. defaultEle.setAttribute('data-debug', debug);
  27. }
  28. return defaultEle;
  29. }),
  30. _React$useState2 = (0, _slicedToArray2.default)(_React$useState, 1),
  31. ele = _React$useState2[0];
  32. // ========================== Order ==========================
  33. var appendedRef = React.useRef(false);
  34. var queueCreate = React.useContext(_Context.default);
  35. var _React$useState3 = React.useState(EMPTY_LIST),
  36. _React$useState4 = (0, _slicedToArray2.default)(_React$useState3, 2),
  37. queue = _React$useState4[0],
  38. setQueue = _React$useState4[1];
  39. var mergedQueueCreate = queueCreate || (appendedRef.current ? undefined : function (appendFn) {
  40. setQueue(function (origin) {
  41. var newQueue = [appendFn].concat((0, _toConsumableArray2.default)(origin));
  42. return newQueue;
  43. });
  44. });
  45. // =========================== DOM ===========================
  46. function append() {
  47. if (!ele.parentElement) {
  48. document.body.appendChild(ele);
  49. }
  50. appendedRef.current = true;
  51. }
  52. function cleanup() {
  53. var _ele$parentElement;
  54. (_ele$parentElement = ele.parentElement) === null || _ele$parentElement === void 0 ? void 0 : _ele$parentElement.removeChild(ele);
  55. appendedRef.current = false;
  56. }
  57. (0, _useLayoutEffect.default)(function () {
  58. if (render) {
  59. if (queueCreate) {
  60. queueCreate(append);
  61. } else {
  62. append();
  63. }
  64. } else {
  65. cleanup();
  66. }
  67. return cleanup;
  68. }, [render]);
  69. (0, _useLayoutEffect.default)(function () {
  70. if (queue.length) {
  71. queue.forEach(function (appendFn) {
  72. return appendFn();
  73. });
  74. setQueue(EMPTY_LIST);
  75. }
  76. }, [queue]);
  77. return [ele, mergedQueueCreate];
  78. }