From ea37ebbdca0f92dea6a39756d4647fa39dafc9e5 Mon Sep 17 00:00:00 2001
From: liujuping <liujup@foxmail.com>
Date: Thu, 13 Apr 2023 14:58:08 +0800
Subject: [PATCH] fix: fix that the outline tree does not respond to modal
 addition/deletion related operations

---
 .../src/views/tree-node.tsx                   | 24 +++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/packages/plugin-outline-pane/src/views/tree-node.tsx b/packages/plugin-outline-pane/src/views/tree-node.tsx
index 712dc20fa..331397a0b 100644
--- a/packages/plugin-outline-pane/src/views/tree-node.tsx
+++ b/packages/plugin-outline-pane/src/views/tree-node.tsx
@@ -9,6 +9,8 @@ import { IPublicModelPluginContext, IPublicModelModalNodesManager, IPublicTypeDi
 class ModalTreeNodeView extends PureComponent<{
   treeNode: TreeNode;
   pluginContext: IPublicModelPluginContext;
+}, {
+  treeChildren: TreeNode[] | null;
 }> {
   private modalNodesManager: IPublicModelModalNodesManager | undefined | null;
   readonly pluginContext: IPublicModelPluginContext;
@@ -20,18 +22,36 @@ class ModalTreeNodeView extends PureComponent<{
     this.pluginContext = props.pluginContext;
     const { project } = this.pluginContext;
     this.modalNodesManager = project.currentDocument?.modalNodesManager;
+    this.state = {
+      treeChildren: this.rootTreeNode.children,
+    };
   }
 
   hideAllNodes() {
     this.modalNodesManager?.hideModalNodes();
   }
 
-  render() {
+  componentDidMount(): void {
+    const rootTreeNode = this.rootTreeNode;
+    rootTreeNode.onExpandableChanged(() => {
+      this.setState({
+        treeChildren: rootTreeNode.children,
+      });
+    });
+  }
+
+  get rootTreeNode() {
     const { treeNode } = this.props;
     // 当指定了新的根节点时，要从原始的根节点去获取模态节点
     const { project } = this.pluginContext;
     const rootNode = project.currentDocument?.root;
     const rootTreeNode = treeNode.tree.getTreeNode(rootNode!);
+
+    return rootTreeNode;
+  }
+
+  render() {
+    const rootTreeNode = this.rootTreeNode;
     const { expanded } = rootTreeNode;
 
     const hasVisibleModalNode = !!this.modalNodesManager?.getVisibleModalNode();
@@ -49,7 +69,7 @@ class ModalTreeNodeView extends PureComponent<{
         <div className="tree-pane-modal-content">
           <TreeBranches
             treeNode={rootTreeNode}
-            treeChildren={rootTreeNode.children}
+            treeChildren={this.state.treeChildren}
             expanded={expanded}
             isModal
             pluginContext={this.pluginContext}