File tree Expand file tree Collapse file tree 1 file changed +21
-4
lines changed Expand file tree Collapse file tree 1 file changed +21
-4
lines changed Original file line number Diff line number Diff line change @@ -17,7 +17,10 @@ interface MotionTreeNodeProps extends Omit<TreeNodeProps, 'domRef'> {
17
17
treeNodeRequiredProps : TreeNodeRequiredProps ;
18
18
}
19
19
20
- const MotionTreeNode : React . FC < MotionTreeNodeProps > = (
20
+ const MotionTreeNode : React . ForwardRefRenderFunction <
21
+ CSSMotion ,
22
+ MotionTreeNodeProps
23
+ > = (
21
24
{
22
25
className,
23
26
style,
@@ -53,7 +56,10 @@ const MotionTreeNode: React.FC<MotionTreeNodeProps> = (
53
56
{ ( { className : motionClassName , style : motionStyle } , motionRef ) => (
54
57
< div
55
58
ref = { motionRef }
56
- className = { classNames ( `${ prefixCls } -treenode-motion` , motionClassName ) }
59
+ className = { classNames (
60
+ `${ prefixCls } -treenode-motion` ,
61
+ motionClassName ,
62
+ ) }
57
63
style = { motionStyle }
58
64
>
59
65
{ motionNodes . map ( ( treeNode : FlattenNode ) => {
@@ -64,7 +70,10 @@ const MotionTreeNode: React.FC<MotionTreeNodeProps> = (
64
70
} = treeNode ;
65
71
delete restProps . children ;
66
72
67
- const treeNodeProps = getTreeNodeProps ( key , treeNodeRequiredProps ) ;
73
+ const treeNodeProps = getTreeNodeProps (
74
+ key ,
75
+ treeNodeRequiredProps ,
76
+ ) ;
68
77
69
78
return (
70
79
< TreeNode
@@ -83,7 +92,15 @@ const MotionTreeNode: React.FC<MotionTreeNodeProps> = (
83
92
</ CSSMotion >
84
93
) ;
85
94
}
86
- return < TreeNode domRef = { ref } className = { className } style = { style } { ...props } active = { active } /> ;
95
+ return (
96
+ < TreeNode
97
+ domRef = { ref }
98
+ className = { className }
99
+ style = { style }
100
+ { ...props }
101
+ active = { active }
102
+ />
103
+ ) ;
87
104
} ;
88
105
89
106
MotionTreeNode . displayName = 'MotionTreeNode' ;
You can’t perform that action at this time.
0 commit comments