From e117cdde4f2b878a84bd3c54c8634755a409aec5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=A5=9E=E4=BB=99=E9=83=BD=E6=B2=A1=E7=94=A8?=
 <615206459@qq.com>
Date: Fri, 28 Jun 2024 11:10:36 +0800
Subject: [PATCH] 1

---
 src/modules/helper/hooks/code.ts     | 34 ++++++++++++++++++++++++++--
 src/modules/helper/types/index.d.ts  |  2 ++
 src/modules/helper/views/ai-code.vue | 27 ++++++++++++++++------
 3 files changed, 54 insertions(+), 9 deletions(-)

diff --git a/src/modules/helper/hooks/code.ts b/src/modules/helper/hooks/code.ts
index f2e45f8..055ee90 100644
--- a/src/modules/helper/hooks/code.ts
+++ b/src/modules/helper/hooks/code.ts
@@ -176,7 +176,14 @@ export function useCode() {
 	}
 
 	// 创建 vue 代码
-	function createVue({ router = "", columns = [], prefix = "", api = [] }: EpsModule) {
+	function createVue({
+		router = "",
+		columns = [],
+		prefix = "",
+		api = [],
+		fieldEq = [],
+		keyWordLikeFields = []
+	}: EpsModule) {
 		// 新增、编辑
 		const upsert = {
 			items: [] as DeepPartial<ClForm.Item>[]
@@ -282,6 +289,23 @@ export function useCode() {
 			});
 		}
 
+		// 筛选
+		const clFilter = fieldEq.map((field) => {
+			const item = table.columns.find((e) => e.propertyName == field);
+
+			return item
+				? `<!-- 筛选${item.label} -->\n<cl-filter label="${item.label}"><cl-select options="${item.dict}" prop="${field}" /></cl-filter>`
+				: "";
+		});
+
+		// 关键字搜索
+		const clSearchKeyPlaceholder = keyWordLikeFields
+			.map((field) => {
+				return table.columns.find((e) => e.propertyName == field)?.label;
+			})
+			.filter((e) => !!e)
+			.join("、");
+
 		// 代码模板
 		return `<template>
             <cl-crud ref="Crud">
@@ -290,9 +314,10 @@ export function useCode() {
                     <cl-refresh-btn />
                     ${perms.add ? "<!-- 新增按钮 -->\n<cl-add-btn />" : ""}
                     ${perms.del ? "<!-- 删除按钮 -->\n<cl-multi-delete-btn />" : ""}
+					${clFilter}
                     <cl-flex1 />
                     <!-- 关键字搜索 -->
-                    <cl-search-key />
+                    <cl-search-key placeholder="搜索${clSearchKeyPlaceholder || "关键字"}" />
                 </cl-row>
         
                 <cl-row>
@@ -332,6 +357,11 @@ export function useCode() {
                 app.refresh();
             }
         );
+
+		// 刷新
+		function refresh(params?: any) {
+			Crud.value?.refresh(params);
+		}
         </script>`;
 	}
 
diff --git a/src/modules/helper/types/index.d.ts b/src/modules/helper/types/index.d.ts
index 3650a2f..d479403 100644
--- a/src/modules/helper/types/index.d.ts
+++ b/src/modules/helper/types/index.d.ts
@@ -23,6 +23,8 @@ export declare interface EpsModule {
 	prefix: string;
 	router: string;
 	module: string;
+	fieldEq?: string[];
+	keyWordLikeFields?: string[];
 	[key: string]: any;
 }
 
diff --git a/src/modules/helper/views/ai-code.vue b/src/modules/helper/views/ai-code.vue
index dafc71e..03d3731 100644
--- a/src/modules/helper/views/ai-code.vue
+++ b/src/modules/helper/views/ai-code.vue
@@ -212,17 +212,17 @@
 							</div>
 
 							<div class="op" v-if="!isEmpty(code.list) && !code.loading">
-								<el-tooltip content="创建文件">
-									<el-icon @click="createFile">
-										<download />
-									</el-icon>
-								</el-tooltip>
-
 								<el-tooltip content="复制代码">
 									<el-icon @click="code.copy()">
 										<copy-document />
 									</el-icon>
 								</el-tooltip>
+
+								<el-tooltip content="创建文件">
+									<el-icon @click="createFile">
+										<download />
+									</el-icon>
+								</el-tooltip>
 							</div>
 						</div>
 
@@ -442,6 +442,8 @@ const code = reactive({
 		// 下一步
 		step.next();
 
+		code.tips("AI 开始编码");
+
 		await sleep(300);
 
 		code.tips("Entity 代码生成中");
@@ -479,7 +481,8 @@ const code = reactive({
 		// controller 代码
 		await code.setContent("Controller 控制器", "node-controller", {
 			...serviceData,
-			...entityData
+			...entityData,
+			service
 		});
 
 		code.tips("Controller 生成成功");
@@ -489,6 +492,16 @@ const code = reactive({
 		code.tips("编码完成");
 
 		code.loading = false;
+
+		ElMessageBox.confirm("编码完成，是否创建文件？", "提示", {
+			type: "success",
+			confirmButtonText: "开始创建",
+			cancelButtonText: "稍后再看"
+		})
+			.then(() => {
+				createFile();
+			})
+			.catch(() => null);
 	},
 
 	// 创建vue