From c9198b315c25cb3e8fd7f769b543e98e131f878c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=BD=97=E6=95=AC?= <1126184155@qq.com>
Date: Thu, 9 May 2024 12:05:37 +0000
Subject: [PATCH] =?UTF-8?q?refactor:=20=E4=BB=A3=E7=A0=81=E7=94=9F?=
 =?UTF-8?q?=E6=88=90=E9=A2=84=E8=A7=88=E8=B0=83=E6=95=B4=E4=B8=BA=E4=BB=A5?=
 =?UTF-8?q?=E6=96=87=E4=BB=B6=E6=A0=91=E7=BB=93=E6=9E=84=E5=BD=A2=E5=BC=8F?=
 =?UTF-8?q?=E6=98=BE=E7=A4=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/apis/tool/type.ts                        |   1 +
 src/views/tool/generator/GenPreviewModal.vue | 151 ++++++++++++++++---
 src/views/tool/generator/index.vue           |  35 +----
 3 files changed, 129 insertions(+), 58 deletions(-)

diff --git a/src/apis/tool/type.ts b/src/apis/tool/type.ts
index 1ba075f..5937be2 100644
--- a/src/apis/tool/type.ts
+++ b/src/apis/tool/type.ts
@@ -44,6 +44,7 @@ export interface GeneratorConfigResp {
   fieldConfigs: FieldConfigResp[]
 }
 export interface GeneratePreviewResp {
+  path: string
   fileName: string
   content: string
 }
diff --git a/src/views/tool/generator/GenPreviewModal.vue b/src/views/tool/generator/GenPreviewModal.vue
index 4f36264..07f7bda 100644
--- a/src/views/tool/generator/GenPreviewModal.vue
+++ b/src/views/tool/generator/GenPreviewModal.vue
@@ -7,53 +7,122 @@
     width="90%"
     :footer="false"
   >
-    <div>
-      <a-scrollbar style="height: 700px; overflow: auto">
-        <a-link style="position: absolute; right: 20px; top: 50px; z-index: 999" @click="onCopy">
-          <template #icon>
-            <icon-copy size="large" />
-          </template>
-          复制
-        </a-link>
-        <a-tabs size="large" @tab-click="onTabClick">
-          <a-tab-pane v-for="item in genPreviewList" :key="item.fileName" :title="item.fileName">
-            <GiCodeView
-              :type="'vue' === item.fileName.split('.')[1] ? 'vue' : 'javascript'"
-              :code-json="item.content"
-            />
-          </a-tab-pane>
-        </a-tabs>
-      </a-scrollbar>
+    <template #title>
+      <a-button type="primary" status="success" size="medium" title="生成预览" @click="onGenerate()">
+        <template #icon><icon-download /></template>生成 {{ previewTableName }} 表
+      </a-button>
+    </template>
+    <div class="preview-content">
+      <a-layout>
+        <a-layout-sider theme="dark" :resize-directions="['right']" style="height: 700px">
+          <a-tree class="selectPreview" :data="treeData" :show-line="true" @select="onSelectPreview" />
+        </a-layout-sider>
+        <a-layout-content>
+          <a-scrollbar style="height: 700px; overflow: auto">
+            <a-link style="position: absolute; right: 20px; top: 50px; z-index: 999" @click="onCopy">
+              <template #icon>
+                <icon-copy size="large" />
+              </template>
+              复制
+            </a-link>
+            <a-card :title="currentPreview?.fileName">
+              <GiCodeView
+                :type="'vue' === currentPreview?.fileName.split('.')[1] ? 'vue' : 'javascript'"
+                :code-json="currentPreview?.content"
+              />
+            </a-card>
+          </a-scrollbar>
+        </a-layout-content>
+      </a-layout>
     </div>
   </a-modal>
 </template>
 
 <script setup lang="ts">
-import { genPreview, type GeneratePreviewResp } from '@/apis'
-import { Message } from '@arco-design/web-vue'
+import { genPreview, type GeneratePreviewResp, generate } from '@/apis'
+import { Message, type TreeNodeData } from '@arco-design/web-vue'
 import { useClipboard } from '@vueuse/core'
 
 const { copy, copied } = useClipboard()
-const copyCodeContent = ref()
+const currentPreview = ref<GeneratePreviewResp>()
 const genPreviewList = ref<GeneratePreviewResp[]>([])
 
 const visible = ref(false)
+const treeData = ref<TreeNodeData[]>([])
+const previewTableName = ref<string>('')
+
 // 打开
 const onPreview = async (tableName: string) => {
+  treeData.value = []
+  previewTableName.value = tableName
   const res = await genPreview(tableName)
   genPreviewList.value = res.data
-  copyCodeContent.value = genPreviewList.value[0].content
+  for (const genPreview of genPreviewList.value) {
+    assembleTree(genPreview)
+  }
+  for (const valueElement of treeData.value) {
+    mergeDir(valueElement)
+  }
+  currentPreview.value = genPreviewList.value[0]
   visible.value = true
 }
 
-// 点击 Tab
-const onTabClick = (key: any) => {
-  copyCodeContent.value = genPreviewList.value.filter((p) => p.fileName === key)[0].content
+// 自增的一个key 因为key相同的节点会出现一些问题
+let autoIncrementKey = 0
+// 将生成的目录组装成树结构
+const assembleTree = (genPreview: GeneratePreviewResp) => {
+  const paths: string[] = genPreview.path.split('\\')
+  let tempChildren: TreeNodeData[] | undefined = treeData.value
+  for (const path of paths) {
+    // 向treeData中推送目录,如果该级目录有那么不推送进行下级的合并
+    tempChildren = pushDir(tempChildren, { title: path, key: autoIncrementKey++, children: new Array<TreeNodeData>() })
+  }
+  tempChildren?.push({ title: genPreview.fileName, key: genPreview.fileName, children: new Array<TreeNodeData>() })
+}
+
+const pushDir = (children: TreeNodeData[] | undefined, treeNode: TreeNodeData) => {
+  if (children) {
+    for (const child of children) {
+      if (child.title === treeNode.title) {
+        return child.children
+      }
+    }
+  }
+  children?.push(treeNode)
+  return treeNode.children
+}
+
+const mergeDir = (parent: TreeNodeData) => {
+  // 合并目录
+  if (parent.children?.length == 1 && typeof parent.children[0].key === 'number') {
+    const mergeTitle = mergeDir(parent.children[0])
+    if (mergeTitle != '') {
+      parent.title = parent.title + '/' + mergeTitle
+    }
+    parent.children = parent.children[0].children
+    return parent.title
+  }
+  // 合并子目录
+  if (parent?.children) {
+    for (const child of parent.children) {
+      mergeDir(child)
+    }
+  }
+  return ''
+}
+
+// 选择文件预览
+const onSelectPreview = (selectedKeys) => {
+  if (typeof selectedKeys[0] === 'string') {
+    currentPreview.value = genPreviewList.value.filter((p) => p.fileName === selectedKeys[0])[0]
+  }
 }
 
 // 复制
 const onCopy = () => {
-  copy(copyCodeContent.value)
+  if (currentPreview.value) {
+    copy(currentPreview.value?.content)
+  }
 }
 watch(copied, () => {
   if (copied.value) {
@@ -61,5 +130,37 @@ watch(copied, () => {
   }
 })
 
+// 生成
+const onGenerate = async () => {
+  const res = await generate([previewTableName.value])
+  const contentDisposition = res.headers['content-disposition']
+  const pattern = new RegExp('filename=([^;]+\\.[^\.;]+);*')
+  const result = pattern.exec(contentDisposition) || ''
+  // 对名字进行解码
+  const fileName = window.decodeURI(result[1])
+  // 创建下载的链接
+  const blob = new Blob([res.data])
+  const downloadElement = document.createElement('a')
+  const href = window.URL.createObjectURL(blob)
+  downloadElement.style.display = 'none'
+  downloadElement.href = href
+  // 下载后文件名
+  downloadElement.download = fileName
+  document.body.appendChild(downloadElement)
+  // 点击下载
+  downloadElement.click()
+  // 下载完成，移除元素
+  document.body.removeChild(downloadElement)
+  // 释放掉 blob 对象
+  window.URL.revokeObjectURL(href)
+}
+
 defineExpose({ onPreview })
 </script>
+
+<style scoped>
+.preview-content :deep(.arco-layout-sider) {
+  min-width: 200px;
+  white-space: nowrap;
+}
+</style>
diff --git a/src/views/tool/generator/index.vue b/src/views/tool/generator/index.vue
index 7357c3e..2855737 100644
--- a/src/views/tool/generator/index.vue
+++ b/src/views/tool/generator/index.vue
@@ -21,16 +21,10 @@
         <template #action="{ record }">
           <a-space>
             <a-link @click="onConfig(record.tableName, record.comment)">配置</a-link>
-            <a-link
-              :title="record.isConfiged ? '生成预览' : '请先进行生成配置'"
-              :disabled="!record.isConfiged"
-              @click="onPreview(record.tableName)"
-              >预览</a-link
-            >
             <a-link
               :title="record.isConfiged ? '生成' : '请先进行生成配置'"
               :disabled="!record.isConfiged"
-              @click="onGenerate([record.tableName])"
+              @click="onPreview(record.tableName)"
               >生成</a-link
             >
           </a-space>
@@ -44,7 +38,7 @@
 </template>
 
 <script setup lang="ts">
-import { listGenerator, generate } from '@/apis'
+import { listGenerator } from '@/apis'
 import GenConfigDrawer from './GenConfigDrawer.vue'
 import GenPreviewModal from './GenPreviewModal.vue'
 import type { TableInstanceColumns } from '@/components/GiTable/type'
@@ -97,31 +91,6 @@ const GenPreviewModalRef = ref<InstanceType<typeof GenPreviewModal>>()
 const onPreview = (tableName: string) => {
   GenPreviewModalRef.value?.onPreview(tableName)
 }
-
-// 生成
-const onGenerate = async (tableNames: Array<string>) => {
-  const res = await generate(tableNames)
-  const contentDisposition = res.headers['content-disposition']
-  const pattern = new RegExp('filename=([^;]+\\.[^\\.;]+);*')
-  const result = pattern.exec(contentDisposition) || ''
-  // 对名字进行解码
-  const fileName = window.decodeURI(result[1])
-  // 创建下载的链接
-  const blob = new Blob([res.data])
-  const downloadElement = document.createElement('a')
-  const href = window.URL.createObjectURL(blob)
-  downloadElement.style.display = 'none'
-  downloadElement.href = href
-  // 下载后文件名
-  downloadElement.download = fileName
-  document.body.appendChild(downloadElement)
-  // 点击下载
-  downloadElement.click()
-  // 下载完成，移除元素
-  document.body.removeChild(downloadElement)
-  // 释放掉 blob 对象
-  window.URL.revokeObjectURL(href)
-}
 </script>
 
 <style></style>