From 8b44ed44295aa66105c9799fcd41f360c411e324 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A9=99=E6=9E=97?= <chenglin@dian.so>
Date: Wed, 17 May 2023 19:57:15 +0800
Subject: [PATCH] fix: fix outlinePane treeView render only once

---
 .../plugin-outline-pane/src/views/pane.tsx    | 22 +++++++++++++------
 1 file changed, 15 insertions(+), 7 deletions(-)

diff --git a/packages/plugin-outline-pane/src/views/pane.tsx b/packages/plugin-outline-pane/src/views/pane.tsx
index 36a2bfa0b..4b807ca18 100644
--- a/packages/plugin-outline-pane/src/views/pane.tsx
+++ b/packages/plugin-outline-pane/src/views/pane.tsx
@@ -17,7 +17,9 @@ export class Pane extends PureComponent<{
 }> {
   private controller;
 
-  private dispose: IPublicTypeDisposable;
+  private simulatorRendererReadyDispose: IPublicTypeDisposable;
+  private changeDocumentDispose: IPublicTypeDisposable;
+  private removeDocumentDispose: IPublicTypeDisposable;
 
   constructor(props: any) {
     super(props);
@@ -26,16 +28,22 @@ export class Pane extends PureComponent<{
     this.state = {
       tree: treeMaster.currentTree,
     };
-    this.dispose = this.props.treeMaster.pluginContext?.project?.onSimulatorRendererReady(() => {
-      this.setState({
-        tree: this.props.treeMaster.currentTree,
-      });
-    });
+    this.simulatorRendererReadyDispose = this.props.treeMaster.pluginContext?.project?.onSimulatorRendererReady(this.changeTree);
+    this.changeDocumentDispose = this.props.treeMaster.pluginContext?.project?.onChangeDocument(this.changeTree);
+    this.removeDocumentDispose = this.props.treeMaster.pluginContext?.project?.onRemoveDocument(this.changeTree);
   }
 
+  changeTree = () => {
+    this.setState({
+      tree: this.props.treeMaster.currentTree,
+    });
+  };
+
   componentWillUnmount() {
     this.controller.purge();
-    this.dispose && this.dispose();
+    this.simulatorRendererReadyDispose?.();
+    this.changeDocumentDispose?.();
+    this.removeDocumentDispose?.();
   }
 
   render() {