From f292db55af7c7ee21949c7109537a96dbf6443cc Mon Sep 17 00:00:00 2001
From: Charles7c <charles7c@126.com>
Date: Thu, 12 Dec 2024 20:52:22 +0800
Subject: [PATCH] =?UTF-8?q?feat(generator):=20=E4=BB=A3=E7=A0=81=E7=94=9F?=
 =?UTF-8?q?=E6=88=90=E3=80=81=E9=A2=84=E8=A7=88=E6=94=AF=E6=8C=81=E6=89=B9?=
 =?UTF-8?q?=E9=87=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/apis/code/generator.ts                   |  4 +--
 src/components/GiTable/index.vue             |  8 +++++
 src/views/code/generator/GenPreviewModal.vue | 14 ++++----
 src/views/code/generator/index.vue           | 37 +++++++++++++++++---
 4 files changed, 50 insertions(+), 13 deletions(-)

diff --git a/src/apis/code/generator.ts b/src/apis/code/generator.ts
index 06147f8..d8c0a2a 100644
--- a/src/apis/code/generator.ts
+++ b/src/apis/code/generator.ts
@@ -27,8 +27,8 @@ export function saveGenConfig(tableName: string, req: T.GeneratorConfigResp) {
 }
 
 /** @desc 生成预览 */
-export function genPreview(tableName: string) {
-  return http.get<T.GeneratePreviewResp[]>(`${BASE_URL}/preview/${tableName}`)
+export function genPreview(tableNames: Array<string>) {
+  return http.get<T.GeneratePreviewResp[]>(`${BASE_URL}/preview/${tableNames}`)
 }
 
 /** @desc 生成代码 */
diff --git a/src/components/GiTable/index.vue b/src/components/GiTable/index.vue
index e7312c4..713592a 100644
--- a/src/components/GiTable/index.vue
+++ b/src/components/GiTable/index.vue
@@ -72,6 +72,9 @@
         </a-tooltip>
       </a-space>
     </a-row>
+    <a-row class="gi-table__toolbar-bottom">
+      <slot name="toolbar-bottom"></slot>
+    </a-row>
     <div class="gi-table__body" :class="`gi-table__body-pagination-${attrs['page-position']}`">
       <div class="gi-table__container">
         <a-table
@@ -128,6 +131,7 @@ defineSlots<{
   'top': () => void
   'toolbar-left': () => void
   'toolbar-right': () => void
+  'toolbar-bottom': () => void
   [propsName: string]: (props: { key: string, record: T, column: TableColumnData, rowIndex: number }) => void
 }>()
 
@@ -364,6 +368,10 @@ defineExpose({ tableRef })
     :deep(.arco-form-layout-inline .arco-form-item) {
       margin-bottom: 0;
     }
+
+    &-bottom {
+      margin-bottom: 8px;
+    }
   }
 
   &__draggable {
diff --git a/src/views/code/generator/GenPreviewModal.vue b/src/views/code/generator/GenPreviewModal.vue
index 28cce5e..d6e9ac6 100644
--- a/src/views/code/generator/GenPreviewModal.vue
+++ b/src/views/code/generator/GenPreviewModal.vue
@@ -1,7 +1,7 @@
 <template>
   <a-modal v-model:visible="visible" width="90%" :footer="false">
     <template #title>
-      {{ `生成 ${previewTableName} 表预览` }}
+      {{ previewTableNames.length === 1 ? `生成 ${previewTableNames[0]} 表预览` : '批量生成预览' }}
       <a-link v-permission="['code:generator:generate']" style="margin-left: 10px" icon @click="onDownload">下载源码</a-link>
     </template>
     <div class="preview-content">
@@ -71,13 +71,13 @@ import { Message, type TreeNodeData } from '@arco-design/web-vue'
 import { useClipboard } from '@vueuse/core'
 import { type GeneratePreviewResp, genPreview } from '@/apis/code/generator'
 
-const emit = defineEmits<{ (e: 'generate', tableNames: string[]): void }>()
+const emit = defineEmits<{ (e: 'generate', previewTableNames: string[]): void }>()
 const { copy, copied } = useClipboard()
 
 const genPreviewList = ref<GeneratePreviewResp[]>([])
 const currentPreview = ref<GeneratePreviewResp>()
 const visible = ref(false)
-const previewTableName = ref<string>('')
+const previewTableNames = ref<string[]>([])
 const treeRef = ref()
 const treeData = ref<TreeNodeData[]>([])
 // 合并目录
@@ -128,7 +128,7 @@ const assembleTree = (genPreview: GeneratePreviewResp) => {
 
 // 下载
 const onDownload = () => {
-  emit('generate', [previewTableName.value])
+  emit('generate', [previewTableNames.value])
 }
 // 校验文件类型
 const checkFileType = (title: string, type: string) => {
@@ -160,10 +160,10 @@ const onSelectPreview = (keys: (string | number)[]) => {
 }
 
 // 打开
-const onOpen = async (tableName: string) => {
+const onOpen = async (tableNames: Array<string>) => {
   treeData.value = []
-  previewTableName.value = tableName
-  const { data } = await genPreview(tableName)
+  previewTableNames.value = tableNames
+  const { data } = await genPreview(tableNames)
   genPreviewList.value = data
   for (const genPreview of genPreviewList.value) {
     assembleTree(genPreview)
diff --git a/src/views/code/generator/index.vue b/src/views/code/generator/index.vue
index 47471bd..d14dc1a 100644
--- a/src/views/code/generator/index.vue
+++ b/src/views/code/generator/index.vue
@@ -1,6 +1,7 @@
 <template>
   <div class="table-page">
     <GiTable
+      v-model:selectedKeys="selectedKeys"
       title="代码生成"
       row-key="tableName"
       :data="dataList"
@@ -10,6 +11,9 @@
       :pagination="pagination"
       :disabled-tools="['size', 'setting']"
       :disabled-column-keys="['tableName']"
+      :row-selection="{ type: 'checkbox', showCheckedAll: true }"
+      @select="select"
+      @select-all="selectAll"
       @refresh="search"
     >
       <template #toolbar-left>
@@ -19,6 +23,23 @@
           <template #default>重置</template>
         </a-button>
       </template>
+      <template #toolbar-right>
+        <a-button type="primary" :disabled="!selectedKeys.length" :title="!selectedKeys.length ? '请选择' : ''" @click="onPreview(selectedKeys)">
+          <template #icon><icon-code-sandbox /></template>
+          <template #default>批量生成</template>
+        </a-button>
+      </template>
+      <template #toolbar-bottom>
+        <a-alert>
+          <template v-if="selectedKeys.length > 0">
+            已选中 {{ selectedKeys.length }} 条记录(可跨页)
+          </template>
+          <template v-else>未选中任何记录</template>
+          <template v-if="selectedKeys.length > 0" #action>
+            <a-link @click="onClearSelected">清空</a-link>
+          </template>
+        </a-alert>
+      </template>
       <template #action="{ record }">
         <a-space>
           <a-link v-permission="['code:generator:config']" title="配置" @click="onConfig(record.tableName, record.comment)">配置</a-link>
@@ -26,7 +47,7 @@
             v-permission="['code:generator:preview']"
             :disabled="!record.createTime"
             :title="record.createTime ? '生成' : '请先进行生成配置'"
-            @click="onPreview(record.tableName)"
+            @click="onPreview([record.tableName])"
           >
             生成
           </a-link>
@@ -57,8 +78,11 @@ const {
   tableData: dataList,
   loading,
   pagination,
+  selectedKeys,
+  select,
+  selectAll,
   search,
-} = useTable((page) => listGenConfig({ ...queryForm, ...page }), { immediate: true })
+} = useTable((page) => listGenConfig({ ...queryForm, ...page }), { immediate: true, formatResult: (data) => data.map((i) => ({ ...i, disabled: !i.createTime })) })
 const columns: TableInstanceColumns[] = [
   {
     title: '序号',
@@ -83,6 +107,11 @@ const reset = () => {
   search()
 }
 
+// 清空所有选中数据
+const onClearSelected = () => {
+  selectedKeys.value = []
+}
+
 const GenConfigDrawerRef = ref<InstanceType<typeof GenConfigDrawer>>()
 // 配置
 const onConfig = (tableName: string, comment: string) => {
@@ -91,8 +120,8 @@ const onConfig = (tableName: string, comment: string) => {
 
 const GenPreviewModalRef = ref<InstanceType<typeof GenPreviewModal>>()
 // 预览
-const onPreview = (tableName: string) => {
-  GenPreviewModalRef.value?.onOpen(tableName)
+const onPreview = (tableNames: Array<string>) => {
+  GenPreviewModalRef.value?.onOpen(tableNames)
 }
 
 // 生成