From 117ba1832e9fcdb4a7eaf36bef15ad31e4f51a01 Mon Sep 17 00:00:00 2001
From: lbw <wangiegie@163.com>
Date: Thu, 9 Feb 2023 21:30:05 +0800
Subject: [PATCH] =?UTF-8?q?feat=20=E4=BB=A3=E7=A0=81=E7=94=9F=E6=88=90?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81=E7=9B=AE=E5=BD=95=E5=86=99=E5=85=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/gen/table.ts              |  8 ++++++++
 src/views/gen/table/generator.vue | 18 ++++++++++++------
 2 files changed, 20 insertions(+), 6 deletions(-)

diff --git a/src/api/gen/table.ts b/src/api/gen/table.ts
index 7db9199d..26b00012 100644
--- a/src/api/gen/table.ts
+++ b/src/api/gen/table.ts
@@ -52,6 +52,14 @@ export const useTableFieldSubmitApi = (dsName: string, tableName: string, fieldL
   return request.put('/gen/table/field/' + dsName + '/' + tableName, fieldList)
 }
 
+export const useGeneratorCodeApi = (tableIds: any) => {
+  return request({
+    url: '/gen/generator/code',
+    method: 'get',
+    params: { tableIds: tableIds }
+  })
+}
+
 export const useGeneratorPreviewApi = (tableId: any) => {
   return request({
     url: '/gen/generator/preview',
diff --git a/src/views/gen/table/generator.vue b/src/views/gen/table/generator.vue
index 75ffd4d3..c1ba1455 100644
--- a/src/views/gen/table/generator.vue
+++ b/src/views/gen/table/generator.vue
@@ -56,8 +56,8 @@
 				<el-col :span="18" class="mb20">
 					<el-form-item label="生成方式" prop="generatorType">
 						<el-radio-group v-model="dataForm.generatorType">
-							<el-radio :label="0">zip压缩包</el-radio>
-							<el-radio :label="1">自定义路径</el-radio>
+							<el-radio-button :label="0">ZIP 压缩包</el-radio-button>
+							<el-radio-button :label="1">自定义路径</el-radio-button>
 						</el-radio-group>
 					</el-form-item>
 				</el-col>
@@ -83,7 +83,7 @@
 
 <script setup lang="ts">
 import { useI18n } from 'vue-i18n';
-import { putObj, useTableApi } from '/@/api/gen/table'
+import { putObj, useTableApi, useGeneratorCodeApi } from '/@/api/gen/table'
 import { useMessage } from '/@/hooks/message';
 import { downBlobFile } from '/@/utils/other';
 
@@ -158,7 +158,7 @@ const submitHandle = () => {
 		putObj(dataForm).then(() => {
 			visible.value = false
 			emit('refreshDataList')
-			useMessage().success(t('common.editSuccessText'))
+			useMessage().success(t('common.optSuccessText'))
 		})
 	})
 }
@@ -174,11 +174,17 @@ const generatorHandle = () => {
 		await submitHandle()
 
 		// 生成代码，zip压缩包
-
 		if (dataForm.generatorType === 0) {
 			downBlobFile('/gen/generator/download?tableIds=' + [dataForm.id].join(','), {}, `${dataForm.tableName}.zip`)
 			visible.value = false
-			return
+		}
+
+		// 写入到指定目录
+		if (dataForm.generatorType === 1) {
+			useGeneratorCodeApi([dataForm.id].join(',')).then(() => {
+				useMessage().success(t('common.optSuccessText'))
+				visible.value = false
+			})
 		}
 	})
 }