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 |
|
19 | var defaultEle = document.createElement('div');
|
20 |
|
21 | if (process.env.NODE_ENV !== 'production' && debug) {
|
22 | defaultEle.setAttribute('data-debug', debug);
|
23 | }
|
24 |
|
25 | return defaultEle;
|
26 | }),
|
27 | _React$useState2 = _slicedToArray(_React$useState, 1),
|
28 | ele = _React$useState2[0];
|
29 |
|
30 |
|
31 | var queueCreate = React.useContext(OrderContext);
|
32 |
|
33 | var _React$useState3 = React.useState(EMPTY_LIST),
|
34 | _React$useState4 = _slicedToArray(_React$useState3, 2),
|
35 | queue = _React$useState4[0],
|
36 | setQueue = _React$useState4[1];
|
37 |
|
38 | var mergedQueueCreate = queueCreate || function (appendFn) {
|
39 | setQueue(function (origin) {
|
40 | var newQueue = [appendFn].concat(_toConsumableArray(origin));
|
41 | return newQueue;
|
42 | });
|
43 | };
|
44 |
|
45 |
|
46 | function append() {
|
47 | if (!ele.parentElement) {
|
48 | document.body.appendChild(ele);
|
49 | }
|
50 | }
|
51 |
|
52 | function cleanup() {
|
53 | var _ele$parentElement;
|
54 |
|
55 | (_ele$parentElement = ele.parentElement) === null || _ele$parentElement === void 0 ? void 0 : _ele$parentElement.removeChild(ele);
|
56 | }
|
57 |
|
58 | useLayoutEffect(function () {
|
59 | if (render) {
|
60 | if (queueCreate) {
|
61 | queueCreate(append);
|
62 | } else {
|
63 | append();
|
64 | }
|
65 | } else {
|
66 | cleanup();
|
67 | }
|
68 |
|
69 | return cleanup;
|
70 | }, [render]);
|
71 | useLayoutEffect(function () {
|
72 | if (queue.length) {
|
73 | queue.forEach(function (appendFn) {
|
74 | return appendFn();
|
75 | });
|
76 | setQueue(EMPTY_LIST);
|
77 | }
|
78 | }, [queue]);
|
79 | return [ele, mergedQueueCreate];
|
80 | } |
\ | No newline at end of file |