@@ -4,7 +4,7 @@ import TreeNode from '../controllers/tree-node';
44import TreeTitle from './tree-title' ;
55import TreeBranches from './tree-branches' ;
66import { IconEyeClose } from '../icons/eye-close' ;
7- import { IPublicModelPluginContext , IPublicModelModalNodesManager , IPublicModelDocumentModel , IPublicTypeDisposable } from '@alilc/lowcode-types' ;
7+ import { IPublicModelPluginContext , IPublicModelModalNodesManager , IPublicTypeDisposable } from '@alilc/lowcode-types' ;
88
99class ModalTreeNodeView extends PureComponent < {
1010 treeNode : TreeNode ;
@@ -49,6 +49,7 @@ class ModalTreeNodeView extends PureComponent<{
4949 < div className = "tree-pane-modal-content" >
5050 < TreeBranches
5151 treeNode = { rootTreeNode }
52+ treeChildren = { rootTreeNode . children }
5253 expanded = { expanded }
5354 isModal
5455 pluginContext = { this . pluginContext }
@@ -65,7 +66,19 @@ export default class TreeNodeView extends PureComponent<{
6566 pluginContext : IPublicModelPluginContext ;
6667 isRootNode ?: boolean ;
6768} > {
68- state = {
69+ state : {
70+ expanded : boolean ;
71+ selected : boolean ;
72+ hidden : boolean ;
73+ locked : boolean ;
74+ detecting : boolean ;
75+ isRoot : boolean ;
76+ highlight : boolean ;
77+ dropping : boolean ;
78+ conditionFlow : boolean ;
79+ expandable : boolean ;
80+ treeChildren : TreeNode [ ] | null ;
81+ } = {
6982 expanded : false ,
7083 selected : false ,
7184 hidden : false ,
@@ -76,6 +89,7 @@ export default class TreeNodeView extends PureComponent<{
7689 dropping : false ,
7790 conditionFlow : false ,
7891 expandable : false ,
92+ treeChildren : [ ] ,
7993 } ;
8094
8195 eventOffCallbacks : Array < IPublicTypeDisposable | undefined > = [ ] ;
@@ -95,6 +109,7 @@ export default class TreeNodeView extends PureComponent<{
95109 conditionFlow : treeNode . node . conditionGroup != null ,
96110 highlight : treeNode . isFocusingNode ( ) ,
97111 expandable : treeNode . expandable ,
112+ treeChildren : treeNode . children ,
98113 } ;
99114 }
100115
@@ -114,11 +129,13 @@ export default class TreeNodeView extends PureComponent<{
114129 this . setState ( { locked } ) ;
115130 } ) ;
116131 treeNode . onExpandableChanged ( ( expandable : boolean ) => {
117- this . setState ( { expandable } ) ;
132+ this . setState ( {
133+ expandable,
134+ treeChildren : treeNode . children ,
135+ } ) ;
118136 } ) ;
119-
120137 this . eventOffCallbacks . push (
121- doc ?. onDropLocationChanged ( ( document : IPublicModelDocumentModel ) => {
138+ doc ?. onDropLocationChanged ( ( ) => {
122139 this . setState ( {
123140 dropping : treeNode . dropDetail ?. index != null ,
124141 } ) ;
@@ -210,6 +227,7 @@ export default class TreeNodeView extends PureComponent<{
210227 isModal = { false }
211228 expanded = { this . state . expanded }
212229 pluginContext = { this . props . pluginContext }
230+ treeChildren = { this . state . treeChildren }
213231 />
214232 </ div >
215233 ) ;
0 commit comments