From e36b718666bf2e2c0ee4339c5227abf058212603 Mon Sep 17 00:00:00 2001
From: ray <1490493387@qq.com>
Date: Wed, 31 Jul 2024 21:36:25 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20:sparkles:=20=E9=87=8D=E7=BD=AE?=
 =?UTF-8?q?=E4=BB=A3=E7=A0=81=E9=85=8D=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/generator.ts          |  8 ++++++
 src/views/generator/index.vue | 52 +++++++++++++++++++++++++----------
 2 files changed, 45 insertions(+), 15 deletions(-)

diff --git a/src/api/generator.ts b/src/api/generator.ts
index 4686ad22..1f716a91 100644
--- a/src/api/generator.ts
+++ b/src/api/generator.ts
@@ -38,6 +38,14 @@ class GeneratorAPI {
       method: "get",
     });
   }
+
+  /** 重置代码配置 */
+  static resetGenConfig(tableName: string) {
+    return request({
+      url: `${GENERATOR_BASE_URL}/${tableName}/config`,
+      method: "delete",
+    });
+  }
 }
 
 export default GeneratorAPI;
diff --git a/src/views/generator/index.vue b/src/views/generator/index.vue
index fc4ae20c..016b9ab7 100644
--- a/src/views/generator/index.vue
+++ b/src/views/generator/index.vue
@@ -54,7 +54,17 @@
               @click="handleOpenDialog(scope.row.tableName)"
             >
               <i-ep-MagicStick />
-              生成
+              生成代码
+            </el-button>
+
+            <el-button
+              type="danger"
+              size="small"
+              link
+              @click="handleResetConfig(scope.row.tableName)"
+            >
+              <i-ep-RefreshLeft />
+              重置配置
             </el-button>
           </template>
         </el-table-column>
@@ -72,7 +82,7 @@
     <el-drawer
       v-model="dialog.visible"
       :title="dialog.title"
-      @close="handleCloseDialog"
+      @close="dialog.visible = false"
       size="80%"
     >
       <el-steps :active="active" align-center finish-status="success" simple>
@@ -452,10 +462,6 @@ function handleResetQuery() {
   handleQuery();
 }
 
-function handleCloseDialog() {
-  dialog.visible = false;
-}
-
 /** 打开弹窗 */
 function handleOpenDialog(tableName: string) {
   dialog.visible = true;
@@ -463,16 +469,32 @@ function handleOpenDialog(tableName: string) {
   // 获取字典数据
   DictAPI.getList().then((data) => {
     dictOptions.value = data;
+    loading.value = true;
 
-    GeneratorAPI.getGenConfig(tableName).then((data) => {
-      dialog.title = `${tableName} 代码生成`;
-      formData.value = data;
-      if (formData.value.id) {
-        active.value = 2;
-        handlePreview(tableName);
-      } else {
-        active.value = 0;
-      }
+    GeneratorAPI.getGenConfig(tableName)
+      .then((data) => {
+        dialog.title = `${tableName} 代码生成`;
+        formData.value = data;
+        if (formData.value.id) {
+          active.value = 2;
+          handlePreview(tableName);
+        } else {
+          active.value = 0;
+        }
+      })
+      .finally(() => {
+        loading.value = false;
+      });
+  });
+}
+
+/** 重置配置 */
+function handleResetConfig(tableName: string) {
+  ElMessageBox.confirm("确定要重置配置吗？", "提示", {
+    type: "warning",
+  }).then(() => {
+    GeneratorAPI.resetGenConfig(tableName).then(() => {
+      handleQuery();
     });
   });
 }