From 3d18404fd6b61bff7ff5d5757206b0972c4feede Mon Sep 17 00:00:00 2001
From: wanglin2 <1013335014@qq.com>
Date: Wed, 26 Apr 2023 15:49:52 +0800
Subject: [PATCH] =?UTF-8?q?Fix=EF=BC=9A=E4=BF=AE=E5=A4=8D=E5=AF=8C?=
 =?UTF-8?q?=E6=96=87=E6=9C=AC=E6=A8=A1=E5=BC=8F=E4=B8=8B=EF=BC=8C=E5=88=87?=
 =?UTF-8?q?=E6=8D=A2=E4=B8=BB=E9=A2=98=E3=80=81=E5=AF=BC=E5=85=A5=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E5=90=8E=E6=B2=A1=E6=9C=89=E8=A7=A6=E5=8F=91=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E6=94=B9=E5=8F=98=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 simple-mind-map/index.js      | 6 +++---
 simple-mind-map/src/Render.js | 5 +++++
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/simple-mind-map/index.js b/simple-mind-map/index.js
index a015deeb..b60c3417 100644
--- a/simple-mind-map/index.js
+++ b/simple-mind-map/index.js
@@ -210,12 +210,12 @@ class MindMap {
   }
 
   //  重新渲染
-  reRender(callback) {
+  reRender(callback, source = '') {
     this.batchExecution.push('render', () => {
       this.draw.clear()
       this.initTheme()
       this.renderer.reRender = true
-      this.renderer.render(callback)
+      this.renderer.render(callback, source)
     })
   }
 
@@ -320,7 +320,7 @@ class MindMap {
     } else {
       this.renderer.renderTree = data
     }
-    this.reRender()
+    this.reRender(() => {}, CONSTANTS.SET_DATA)
   }
 
   //  动态设置思维导图数据，包括节点数据、布局、主题、视图
diff --git a/simple-mind-map/src/Render.js b/simple-mind-map/src/Render.js
index 7f46c596..92ea717e 100644
--- a/simple-mind-map/src/Render.js
+++ b/simple-mind-map/src/Render.js
@@ -288,6 +288,11 @@ class Render {
         if (this.hasWaitRendering) {
           this.hasWaitRendering = false
           this.render(callback, source)
+        } else {
+          // 触发一次保存，因为修改了渲染树的数据
+          if (this.mindMap.richText && [CONSTANTS.CHANGE_THEME, CONSTANTS.SET_DATA].includes(source)) {
+            this.mindMap.command.addHistory()
+          }
         }
       }
       let { enableNodeTransitionMove, nodeTransitionMoveDuration } =