109 lines
3.9 KiB
JavaScript
109 lines
3.9 KiB
JavaScript
|
import _extends from "@babel/runtime/helpers/esm/extends";
|
||
|
|
import _objectDestructuringEmpty from "@babel/runtime/helpers/esm/objectDestructuringEmpty";
|
||
|
|
import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
|
||
|
|
import _objectWithoutProperties from "@babel/runtime/helpers/esm/objectWithoutProperties";
|
||
|
|
var _excluded = ["className", "style", "motion", "motionNodes", "motionType", "onMotionStart", "onMotionEnd", "active", "treeNodeRequiredProps"];
|
||
|
|
import classNames from 'classnames';
|
||
|
|
import CSSMotion from 'rc-motion';
|
||
|
|
import useLayoutEffect from "rc-util/es/hooks/useLayoutEffect";
|
||
|
|
import * as React from 'react';
|
||
|
|
import { TreeContext } from "./contextTypes";
|
||
|
|
import TreeNode from "./TreeNode";
|
||
|
|
import useUnmount from "./useUnmount";
|
||
|
|
import { getTreeNodeProps } from "./utils/treeUtil";
|
||
|
|
var MotionTreeNode = function MotionTreeNode(_ref, ref) {
|
||
|
|
var className = _ref.className,
|
||
|
|
style = _ref.style,
|
||
|
|
motion = _ref.motion,
|
||
|
|
motionNodes = _ref.motionNodes,
|
||
|
|
motionType = _ref.motionType,
|
||
|
|
onOriginMotionStart = _ref.onMotionStart,
|
||
|
|
onOriginMotionEnd = _ref.onMotionEnd,
|
||
|
|
active = _ref.active,
|
||
|
|
treeNodeRequiredProps = _ref.treeNodeRequiredProps,
|
||
|
|
props = _objectWithoutProperties(_ref, _excluded);
|
||
|
|
var _React$useState = React.useState(true),
|
||
|
|
_React$useState2 = _slicedToArray(_React$useState, 2),
|
||
|
|
visible = _React$useState2[0],
|
||
|
|
setVisible = _React$useState2[1];
|
||
|
|
var _React$useContext = React.useContext(TreeContext),
|
||
|
|
prefixCls = _React$useContext.prefixCls;
|
||
|
|
|
||
|
|
// Calculate target visible here.
|
||
|
|
// And apply in effect to make `leave` motion work.
|
||
|
|
var targetVisible = motionNodes && motionType !== 'hide';
|
||
|
|
useLayoutEffect(function () {
|
||
|
|
if (motionNodes) {
|
||
|
|
if (targetVisible !== visible) {
|
||
|
|
setVisible(targetVisible);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}, [motionNodes]);
|
||
|
|
var triggerMotionStart = function triggerMotionStart() {
|
||
|
|
if (motionNodes) {
|
||
|
|
onOriginMotionStart();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
// Should only trigger once
|
||
|
|
var triggerMotionEndRef = React.useRef(false);
|
||
|
|
var triggerMotionEnd = function triggerMotionEnd() {
|
||
|
|
if (motionNodes && !triggerMotionEndRef.current) {
|
||
|
|
triggerMotionEndRef.current = true;
|
||
|
|
onOriginMotionEnd();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
// Effect if unmount
|
||
|
|
useUnmount(triggerMotionStart, triggerMotionEnd);
|
||
|
|
|
||
|
|
// Motion end event
|
||
|
|
var onVisibleChanged = function onVisibleChanged(nextVisible) {
|
||
|
|
if (targetVisible === nextVisible) {
|
||
|
|
triggerMotionEnd();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
if (motionNodes) {
|
||
|
|
return /*#__PURE__*/React.createElement(CSSMotion, _extends({
|
||
|
|
ref: ref,
|
||
|
|
visible: visible
|
||
|
|
}, motion, {
|
||
|
|
motionAppear: motionType === 'show',
|
||
|
|
onVisibleChanged: onVisibleChanged
|
||
|
|
}), function (_ref2, motionRef) {
|
||
|
|
var motionClassName = _ref2.className,
|
||
|
|
motionStyle = _ref2.style;
|
||
|
|
return /*#__PURE__*/React.createElement("div", {
|
||
|
|
ref: motionRef,
|
||
|
|
className: classNames("".concat(prefixCls, "-treenode-motion"), motionClassName),
|
||
|
|
style: motionStyle
|
||
|
|
}, motionNodes.map(function (treeNode) {
|
||
|
|
var restProps = Object.assign({}, (_objectDestructuringEmpty(treeNode.data), treeNode.data)),
|
||
|
|
title = treeNode.title,
|
||
|
|
key = treeNode.key,
|
||
|
|
isStart = treeNode.isStart,
|
||
|
|
isEnd = treeNode.isEnd;
|
||
|
|
delete restProps.children;
|
||
|
|
var treeNodeProps = getTreeNodeProps(key, treeNodeRequiredProps);
|
||
|
|
return /*#__PURE__*/React.createElement(TreeNode, _extends({}, restProps, treeNodeProps, {
|
||
|
|
title: title,
|
||
|
|
active: active,
|
||
|
|
data: treeNode.data,
|
||
|
|
key: key,
|
||
|
|
isStart: isStart,
|
||
|
|
isEnd: isEnd
|
||
|
|
}));
|
||
|
|
}));
|
||
|
|
});
|
||
|
|
}
|
||
|
|
return /*#__PURE__*/React.createElement(TreeNode, _extends({
|
||
|
|
domRef: ref,
|
||
|
|
className: className,
|
||
|
|
style: style
|
||
|
|
}, props, {
|
||
|
|
active: active
|
||
|
|
}));
|
||
|
|
};
|
||
|
|
MotionTreeNode.displayName = 'MotionTreeNode';
|
||
|
|
var RefMotionTreeNode = /*#__PURE__*/React.forwardRef(MotionTreeNode);
|
||
|
|
export default RefMotionTreeNode;
|