1 | import _toConsumableArray from "@babel/runtime/helpers/esm/toConsumableArray";
|
2 | import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
|
3 | import * as React from 'react';
|
4 | import useLayoutEffect from "rc-util/es/hooks/useLayoutEffect";
|
5 | import canUseDom from "rc-util/es/Dom/canUseDom";
|
6 | import OrderContext from "./Context";
|
7 | var EMPTY_LIST = [];
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 | export default function useDom(render, debug) {
|
14 | var _React$useState = React.useState(function () {
|
15 | if (!canUseDom()) {
|
16 | return null;
|
17 | }
|
18 | var defaultEle = document.createElement('div');
|
19 | if (process.env.NODE_ENV !== 'production' && debug) {
|
20 | defaultEle.setAttribute('data-debug', debug);
|
21 | }
|
22 | return defaultEle;
|
23 | }),
|
24 | _React$useState2 = _slicedToArray(_React$useState, 1),
|
25 | ele = _React$useState2[0];
|
26 |
|
27 |
|
28 | var appendedRef = React.useRef(false);
|
29 | var queueCreate = React.useContext(OrderContext);
|
30 | var _React$useState3 = React.useState(EMPTY_LIST),
|
31 | _React$useState4 = _slicedToArray(_React$useState3, 2),
|
32 | queue = _React$useState4[0],
|
33 | setQueue = _React$useState4[1];
|
34 | var mergedQueueCreate = queueCreate || (appendedRef.current ? undefined : function (appendFn) {
|
35 | setQueue(function (origin) {
|
36 | var newQueue = [appendFn].concat(_toConsumableArray(origin));
|
37 | return newQueue;
|
38 | });
|
39 | });
|
40 |
|
41 |
|
42 | function append() {
|
43 | if (!ele.parentElement) {
|
44 | document.body.appendChild(ele);
|
45 | }
|
46 | appendedRef.current = true;
|
47 | }
|
48 | function cleanup() {
|
49 | var _ele$parentElement;
|
50 | (_ele$parentElement = ele.parentElement) === null || _ele$parentElement === void 0 ? void 0 : _ele$parentElement.removeChild(ele);
|
51 | appendedRef.current = false;
|
52 | }
|
53 | useLayoutEffect(function () {
|
54 | if (render) {
|
55 | if (queueCreate) {
|
56 | queueCreate(append);
|
57 | } else {
|
58 | append();
|
59 | }
|
60 | } else {
|
61 | cleanup();
|
62 | }
|
63 | return cleanup;
|
64 | }, [render]);
|
65 | useLayoutEffect(function () {
|
66 | if (queue.length) {
|
67 | queue.forEach(function (appendFn) {
|
68 | return appendFn();
|
69 | });
|
70 | setQueue(EMPTY_LIST);
|
71 | }
|
72 | }, [queue]);
|
73 | return [ele, mergedQueueCreate];
|
74 | } |
\ | No newline at end of file |