Merge branch '5.x' of https://github.com/cool-team-official/cool-admin-vue into 5.x

build/cool/lib/eps/config.ts

@@ -0,0 +1,36 @@
+export default {
+	// 实体生成
+	entity: {
+		// 是否生成
+		enable: true,
+		mapping: [
+			{
+				// 自定义匹配
+				custom: ({ entityName, propertyName, type }) => {
+					// status原本是tinyint，如果是1的话，== true是可以的，但是不能 === true，请谨慎使用
+					if (propertyName === "status" && type == "tinyint") return "boolean";
+					//如果没有，返回null或者不返回，则继续遍历其他匹配规则
+					return null;
+				}
+			},
+			{
+				// 返回类型
+				type: "string",
+				// 匹配列类型
+				includes: ["varchar", "text"]
+			},
+			{
+				type: "Date",
+				includes: ["datetime", "date"]
+			},
+			{
+				type: "number",
+				includes: ["tinyint", "int", "decimal"]
+			},
+			{
+				type: "BigInt",
+				includes: ["bigint"]
+			}
+		]
+	}
+};

build/cool/lib/eps/index.ts

@@ -3,6 +3,8 @@ import { isEmpty, last } from "lodash";
 import { createDir, firstUpperCase, readFile, toCamel } from "../../utils";
 import { createWriteStream } from "fs";
 import { join } from "path";
+// import * as config from "/@/cool/config";
+import config from "./config";
 
 // 临时目录路径
 const tempPath = join(__dirname, "../../temp");
@@ -211,9 +213,70 @@ export async function createEps({ list, service }: any) {
 			})
 		)
 	);
+
+	if (config.entity.enable) createEntity(list);
 }
 
 // 获取描述
 export function getEps() {
 	return JSON.stringify(readFile(join(tempPath, "eps.json")));
 }
+
+function getType({ entityName, propertyName, type }) {
+	for (const map of config.entity.mapping) {
+		if (map.custom) {
+			const resType = map.custom({ entityName, propertyName, type });
+			if (resType) return resType;
+		}
+		if (map.includes?.includes(type)) return map.type;
+	}
+	return type;
+}
+
+// 创建Entity描述文件
+export function createEntity(list: any[]) {
+	const t2: any[] = [];
+
+	for (const item of list) {
+		if (!item.name) continue;
+		const t = [`declare interface ${item.name} {`];
+		for (const col of item.columns) {
+			// 描述
+			t.push("\n");
+			t.push("/**\n");
+			t.push(` * ${col.comment}\n`);
+			t.push(" */\n");
+			t.push(
+				`${col.propertyName}?: ${getType({
+					entityName: item.name,
+					propertyName: col.propertyName,
+					type: col.type
+				})};`
+			);
+		}
+		t.push("\n");
+		t.push("/**\n");
+		t.push(` * 任意键值\n`);
+		t.push(" */\n");
+		t.push(`[key: string]: any;`);
+		t.push("}");
+		t2.push(t);
+	}
+
+	// 文本内容
+	const content = prettier.format(t2.map((e) => e.join("")).join("\n\n"), {
+		parser: "typescript",
+		useTabs: true,
+		tabWidth: 4,
+		endOfLine: "lf",
+		semi: true,
+		singleQuote: false,
+		printWidth: 100,
+		trailingComma: "none"
+	});
+
+	// 创建 entity 描述文件
+	createWriteStream(join(tempPath, "entity.d.ts"), {
+		flags: "w"
+	}).write(content);
+}

build/svg/index.ts

@@ -1,5 +1,7 @@
 import { Plugin } from "vite";
-import { readFileSync, readdirSync } from "fs";
+import { readFileSync, readdirSync, accessSync } from "fs";
+import path from "path";
+import { isArray } from "lodash";
 
 let idPerfix = "";
 const svgTitle = /<svg([^>+].*?)>/;
@@ -9,16 +11,19 @@ const hasViewBox = /(viewBox="[^>+].*?")/g;
 
 const clearReturn = /(\r)|(\n)/g;
 
-function findSvgFile(dir: string): string[] {
+function findSvgFile(dir: string, uniqueNames: Record<string, boolean>): string[] {
 	const svgRes = [];
 	const dirents = readdirSync(dir, {
 		withFileTypes: true
 	});
 	for (const dirent of dirents) {
 		if (dirent.isDirectory()) {
-			svgRes.push(...findSvgFile(dir + dirent.name + "/"));
+			svgRes.push(...findSvgFile(path.join(dir, dirent.name), uniqueNames));
+		} else if (uniqueNames[dirent.name]) {
+			continue;
 		} else {
-			const svg = readFileSync(dir + dirent.name)
+			uniqueNames[dirent.name] = true;
+			const svg = readFileSync(path.join(dir, dirent.name))
 				.toString()
 				.replace(clearReturn, "")
 				.replace(svgTitle, (_: any, $2: any) => {
@@ -47,10 +52,16 @@ function findSvgFile(dir: string): string[] {
 	return svgRes;
 }
 
-export const svgBuilder = (path: string, perfix = "icon"): Plugin | null => {
+export const svgBuilder = (paths: string | string[], perfix = "icon"): Plugin | null => {
-	if (path !== "") {
+	if (paths) {
 		idPerfix = perfix;
-		const res = findSvgFile(path);
+		paths = isArray(paths) ? paths : [paths];
+		const uniqueNames: Record<string, boolean> = {};
+		const res = paths.reduce(
+			(previousValue, currentValue) =>
+				previousValue.concat(findSvgFile(currentValue, uniqueNames)),
+			[]
+		);
 		return {
 			name: "svg-transform",
 			transformIndexHtml(html): string {
@@ -69,3 +80,27 @@ export const svgBuilder = (path: string, perfix = "icon"): Plugin | null => {
 		return null;
 	}
 };
+
+export const findSvgFolders = (dir: string): string[] => {
+	const svgFolders = [];
+	const dirents = readdirSync(dir, {
+		withFileTypes: true
+	});
+
+	// 找到结构为icons/svg的文件夹
+	for (const dirent of dirents) {
+		if (dirent.isDirectory()) {
+			const testPath =
+				dirent.name === "icons"
+					? path.join(dir, "icons/svg")
+					: path.join(dir, dirent.name, "icons/svg");
+			try {
+				accessSync(testPath);
+				svgFolders.push(testPath);
+			} catch (e) {
+				continue;
+			}
+		}
+	}
+	return svgFolders;
+};

src/env.d.ts

@@ -1,4 +1,5 @@
 /// <reference types="@cool-vue/crud" />
 /// <reference types="../build/cool/temp/service" />
+/// <reference types="../build/cool/temp/entity" />
 
 declare const __EPS__: string;

vite.config.ts

@@ -8,7 +8,7 @@ import Unocss from "unocss/vite";
 import { presetUno } from "unocss";
 import { proxy } from "./src/cool/config/proxy";
 import { cool } from "./build/cool";
-import { svgBuilder } from "./build/svg";
+import { svgBuilder, findSvgFolders } from "./build/svg";
 
 function resolve(dir: string) {
 	return path.resolve(__dirname, ".", dir);
@@ -27,7 +27,7 @@ export default (): UserConfig => {
 			Unocss({
 				presets: [presetUno()]
 			}),
-			svgBuilder("./src/icons/svg/"),
+			svgBuilder(["./src/icons/svg/",...findSvgFolders("./src/modules/")]),
 			cool()
 		],
 		css: {