From 010918198a7daa97ed3bd3c797ac88c091ef90a1 Mon Sep 17 00:00:00 2001
From: parisma <parisma@tencent.com>
Date: Fri, 17 Feb 2023 15:06:21 +0800
Subject: [PATCH] =?UTF-8?q?fix(editor):=20=E4=BF=AE=E5=A4=8D=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96=E6=80=A7=E8=83=BD=E5=BC=95=E8=B5=B7=E7=9A=84=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98,=E6=94=AF=E6=8C=81=E5=9B=9E=E9=80=80=E6=92=A4?=
 =?UTF-8?q?=E9=94=80=E5=AF=B9=E4=BB=A3=E7=A0=81=E5=85=B3=E7=B3=BB=E7=9A=84?=
 =?UTF-8?q?=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../layouts/sidebar/code-block/CodeBlockList.vue    | 13 ++++++++-----
 packages/editor/src/services/codeBlock.ts           |  8 +++++++-
 packages/editor/src/services/editor.ts              |  1 +
 3 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/packages/editor/src/layouts/sidebar/code-block/CodeBlockList.vue b/packages/editor/src/layouts/sidebar/code-block/CodeBlockList.vue
index 9c970f6d..4f20f613 100644
--- a/packages/editor/src/layouts/sidebar/code-block/CodeBlockList.vue
+++ b/packages/editor/src/layouts/sidebar/code-block/CodeBlockList.vue
@@ -87,7 +87,7 @@
 </template>
 
 <script lang="ts" setup name="MEditorCodeBlockList">
-import { computed, inject, onMounted, reactive, ref, watch } from 'vue';
+import { computed, inject, reactive, ref, watch } from 'vue';
 import { Close, Edit, Link, View } from '@element-plus/icons-vue';
 import { cloneDeep, forIn, isEmpty } from 'lodash-es';
 
@@ -152,10 +152,13 @@ const refreshCodeList = async () => {
   });
 };
 
-onMounted(() => {
-  services?.codeBlockService.refreshAllRelations();
-  refreshCodeList();
-});
+watch(
+  () => services?.editorService.get('root'),
+  () => {
+    services?.codeBlockService.refreshAllRelations();
+    refreshCodeList();
+  },
+);
 
 watch(
   [() => services?.codeBlockService.getCodeDslSync(), () => services?.codeBlockService.getCombineInfo()],
diff --git a/packages/editor/src/services/codeBlock.ts b/packages/editor/src/services/codeBlock.ts
index bbbaac9b..021c5e66 100644
--- a/packages/editor/src/services/codeBlock.ts
+++ b/packages/editor/src/services/codeBlock.ts
@@ -19,7 +19,7 @@
 import { reactive } from 'vue';
 import { cloneDeep, forIn, isEmpty, keys, omit, pick, union } from 'lodash-es';
 
-import { CodeBlockContent, CodeBlockDSL, HookType, Id, MNode } from '@tmagic/schema';
+import { CodeBlockContent, CodeBlockDSL, HookType, Id, MNode, MPage } from '@tmagic/schema';
 
 import editorService from '../services/editor';
 import type { CodeRelation, CodeState, HookData } from '../type';
@@ -251,6 +251,12 @@ class CodeBlock extends BaseService {
         this.state.relations = this.deleteNodeRelation(node, this.state.relations);
       });
     });
+    // 监听历史记录，历史快照为页面整体，需要深层遍历更新
+    editorService.on('history-change', (page: MPage) => {
+      const relations: CodeRelation = {};
+      this.getNodeRelation(page, relations, true);
+      this.state.relations = relations;
+    });
   }
 
   /**
diff --git a/packages/editor/src/services/editor.ts b/packages/editor/src/services/editor.ts
index 1dc2aae5..228ce89b 100644
--- a/packages/editor/src/services/editor.ts
+++ b/packages/editor/src/services/editor.ts
@@ -886,6 +886,7 @@ class Editor extends BaseService {
       await this.select(value.nodeId);
       this.get('stage')?.select(value.nodeId);
     }, 0);
+    this.emit('history-change', value.data);
   }
 
   private async toggleFixedPosition(dist: MNode, src: MNode, root: MApp) {