From d704edb80d66ccfc5e4fbae5132e00d940713438 Mon Sep 17 00:00:00 2001
From: "Ray.Hao" <1490493387@qq.com>
Date: Sun, 5 Apr 2026 20:38:36 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E4=BB=A3=E7=A0=81?=
 =?UTF-8?q?=E7=94=9F=E6=88=90=E9=A2=84=E8=A7=88=E4=B8=AD=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E6=A0=91=E9=80=89=E4=B8=AD=E7=8A=B6=E6=80=81=E5=8F=8A=E7=BC=96?=
 =?UTF-8?q?=E8=BE=91=E5=99=A8=E5=88=B7=E6=96=B0=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/views/codegen/index.vue | 74 ++++++++++++++++++++++++++++++++++++-
 1 file changed, 72 insertions(+), 2 deletions(-)

diff --git a/src/views/codegen/index.vue b/src/views/codegen/index.vue
index 6963354d..0ca4c807 100644
--- a/src/views/codegen/index.vue
+++ b/src/views/codegen/index.vue
@@ -381,9 +381,12 @@
           <el-col :span="6">
             <el-scrollbar max-height="72vh">
               <el-tree
+                ref="fileTreeRef"
                 :data="filteredTreeData"
+                node-key="key"
                 default-expand-all
                 highlight-current
+                :current-node-key="currentFileKey"
                 @node-click="handleFileTreeNodeClick"
               >
                 <template #default="{ data }">
@@ -538,6 +541,7 @@ import MenuAPI from "@/api/system/menu";
 
 interface TreeNode {
   label: string;
+  key?: string;
   content?: string;
   children?: TreeNode[];
   scope?: "frontend" | "backend";
@@ -628,10 +632,18 @@ watch(
 const { copy, copied } = useClipboard();
 const code = ref<string>("");
 const cmRef = ref<CmComponentRef>();
+const fileTreeRef = ref();
+const currentFileKey = ref<string>("");
 const cmOptions: EditorConfiguration = {
   mode: "text/javascript",
 };
 
+function refreshPreviewEditor() {
+  const inst: any = cmRef.value as any;
+  const editor = inst?.cminstance || inst?.cm || inst?.editor;
+  editor?.refresh?.();
+}
+
 const prevBtnText = ref("");
 const nextBtnText = ref("下一步，字段配置");
 const active = ref(0);
@@ -679,6 +691,10 @@ watch(active, (val) => {
   } else if (val === 2) {
     prevBtnText.value = "上一步，字段配置";
     nextBtnText.value = "下载代码";
+
+    nextTick(() => {
+      refreshPreviewEditor();
+    });
   }
 });
 
@@ -930,6 +946,9 @@ async function handlePreview(tableName: string) {
     const firstLeafNode = findFirstLeafNode(tree);
     if (firstLeafNode) {
       code.value = firstLeafNode.content || "";
+      currentFileKey.value = firstLeafNode.key || "";
+      await nextTick();
+      fileTreeRef.value?.setCurrentKey?.(currentFileKey.value);
     }
   } catch {
     active.value = 0;
@@ -945,20 +964,24 @@ async function handlePreview(tableName: string) {
  */
 function buildTree(data: GeneratorPreviewItem[]): TreeNode {
   // 动态获取根节点
-  const root: TreeNode = { label: "前后端代码", children: [] };
+  const root: TreeNode = { label: "前后端代码", key: "root", children: [] };
 
   data.forEach((item) => {
     const normalizedPath = item.path.replace(/\\/g, "/");
     const parts = normalizedPath.split("/").filter(Boolean);
 
     let currentNode = root;
+    let currentKey = root.key || "root";
 
     parts.forEach((part) => {
       // 查找或创建当前部分的子节点
       let node = currentNode.children?.find((child) => child.label === part);
       if (!node) {
-        node = { label: part, children: [] };
+        currentKey = `${currentKey}/${part}`;
+        node = { label: part, key: currentKey, children: [] };
         currentNode.children?.push(node);
+      } else {
+        currentKey = node.key || `${currentKey}/${part}`;
       }
       currentNode = node;
     });
@@ -966,6 +989,7 @@ function buildTree(data: GeneratorPreviewItem[]): TreeNode {
     // 添加文件节点
     currentNode.children?.push({
       label: item.fileName,
+      key: `${item.scope || ""}:${normalizedPath}/${item.fileName}`,
       content: item?.content,
       scope: item.scope,
       language: item.language,
@@ -997,9 +1021,55 @@ function findFirstLeafNode(node: TreeNode): TreeNode | null {
 function handleFileTreeNodeClick(data: TreeNode) {
   if (!data.children || data.children.length === 0) {
     code.value = data.content || "";
+    currentFileKey.value = data.key || "";
+
+    nextTick(() => {
+      refreshPreviewEditor();
+    });
   }
 }
 
+function findLeafByKey(nodes: TreeNode[], key: string): TreeNode | null {
+  for (const node of nodes) {
+    if (!node.children || node.children.length === 0) {
+      if (node.key === key) return node;
+      continue;
+    }
+    const found = findLeafByKey(node.children || [], key);
+    if (found) return found;
+  }
+  return null;
+}
+
+watch(
+  () => filteredTreeData.value,
+  async (nodes) => {
+    if (!nodes.length) {
+      currentFileKey.value = "";
+      code.value = "";
+      return;
+    }
+
+    if (currentFileKey.value) {
+      const leaf = findLeafByKey(nodes, currentFileKey.value);
+      if (leaf) {
+        await nextTick();
+        fileTreeRef.value?.setCurrentKey?.(currentFileKey.value);
+        return;
+      }
+    }
+
+    const first = findFirstLeafNode({ label: "root", key: "root", children: nodes });
+    if (first) {
+      code.value = first.content || "";
+      currentFileKey.value = first.key || "";
+      await nextTick();
+      fileTreeRef.value?.setCurrentKey?.(currentFileKey.value);
+    }
+  },
+  { immediate: true }
+);
+
 /** 获取文件树节点图标 */
 function getFileTreeNodeIcon(node: TreeNode) {
   const ext = (node.language || node.label.split(".").pop() || "").toLowerCase();