// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。 // // 本项目主要遵循 MIT 许可证和 Apache 许可证(版本 2.0)进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。 // // 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动!任何基于本项目二次开发而产生的一切法律纠纷和责任,我们不承担任何责任! using System.IO.Compression; namespace Admin.NET.Core.Service; /// /// 系统代码生成器服务 🧩 /// [ApiDescriptionSettings(Order = 270, Description = "代码生成器")] public class SysCodeGenService : IDynamicApiController, ITransient { private readonly ISqlSugarClient _db; private readonly SqlSugarRepository _codeGetTemplateRep; private readonly SysCodeGenConfigService _codeGenConfigService; private readonly IViewEngine _viewEngine; private readonly CodeGenOptions _codeGenOptions; public SysCodeGenService(ISqlSugarClient db, SqlSugarRepository codeGetTemplateRep, SysCodeGenConfigService codeGenConfigService, IViewEngine viewEngine, IOptions codeGenOptions) { _db = db; this._codeGetTemplateRep = codeGetTemplateRep; _codeGenConfigService = codeGenConfigService; _viewEngine = viewEngine; _codeGenOptions = codeGenOptions.Value; } /// /// 获取代码生成分页列表 🔖 /// /// /// [DisplayName("获取代码生成分页列表")] public async Task> Page(PageCodeGenInput input) { return await _db.Queryable() .Includes(u => u.CodeGenTemplateRelations) .WhereIF(!string.IsNullOrWhiteSpace(input.TableName), u => u.TableName!.Contains(input.TableName.Trim())) .WhereIF(!string.IsNullOrWhiteSpace(input.BusName), u => u.BusName!.Contains(input.BusName.Trim())) .OrderBy(u => u.Id, OrderByType.Desc) .ToPagedListAsync(input.Page, input.PageSize); } /// /// 增加代码生成 🔖 /// /// /// [ApiDescriptionSettings(Name = "Add"), HttpPost] [DisplayName("增加代码生成")] public async Task AddCodeGen(AddCodeGenInput input) { var isExist = await _db.Queryable().Where(u => u.TableName == input.TableName).AnyAsync(); if (isExist) throw Oops.Oh(ErrorCodeEnum.D1400); var codeGen = input.Adapt(); long id = YitIdHelper.NextId(); var templateRelations = GetCodeGenTemplateRelation(id, input.CodeGenTemplateIds); codeGen.Id = id; codeGen.CodeGenTemplateRelations = templateRelations; //var newCodeGen = await _db.Insertable(codeGen).ExecuteReturnEntityAsync(); var newCodeGen = await _db.InsertNav(codeGen) .Include(t => t.CodeGenTemplateRelations) .ExecuteReturnEntityAsync(); // 增加配置表 _codeGenConfigService.AddList(GetColumnList(input), newCodeGen); } /// /// 获取代码生成模板关系集合 🔖 /// /// /// /// private List GetCodeGenTemplateRelation(long codeGenId, List templateIds) { List list = new List(); foreach (var item in templateIds) { SysCodeGenTemplateRelation relation = new SysCodeGenTemplateRelation(); relation.CodeGenId = codeGenId; relation.TemplateId = item; list.Add(relation); } return list; } /// /// 更新代码生成 🔖 /// /// /// [ApiDescriptionSettings(Name = "Update"), HttpPost] [DisplayName("更新代码生成")] public async Task UpdateCodeGen(UpdateCodeGenInput input) { var isExist = await _db.Queryable().AnyAsync(u => u.TableName == input.TableName && u.Id != input.Id); if (isExist) throw Oops.Oh(ErrorCodeEnum.D1400); var codeGen = input.Adapt(); var templateRelations = GetCodeGenTemplateRelation(codeGen.Id, input.CodeGenTemplateIds); codeGen.CodeGenTemplateRelations = templateRelations; //await _db.Updateable(codeGen).ExecuteCommandAsync(); await _db.UpdateNav(codeGen) .Include(t => t.CodeGenTemplateRelations) .ExecuteCommandAsync(); // 更新配置表 _codeGenConfigService.AddList(GetColumnList(input.Adapt()), codeGen); } /// /// 删除代码生成 🔖 /// /// /// [ApiDescriptionSettings(Name = "Delete"), HttpPost] [DisplayName("删除代码生成")] public async Task DeleteCodeGen(List inputs) { if (inputs == null || inputs.Count < 1) return; var codeGenConfigTaskList = new List(); inputs.ForEach(u => { //_db.Deleteable().In(u.Id).ExecuteCommand(); _db.DeleteNav(t => t.Id == u.Id) .Include(t => t.CodeGenTemplateRelations) .ExecuteCommand(); // 删除配置表 codeGenConfigTaskList.Add(_codeGenConfigService.DeleteCodeGenConfig(u.Id)); }); await Task.WhenAll(codeGenConfigTaskList); } /// /// 获取代码生成详情 🔖 /// /// /// [DisplayName("获取代码生成详情")] public async Task GetDetail([FromQuery] QueryCodeGenInput input) { return await _db.Queryable().SingleAsync(u => u.Id == input.Id); } /// /// 获取数据库库集合 🔖 /// /// [DisplayName("获取数据库库集合")] public async Task> GetDatabaseList() { var dbConfigs = App.GetOptions().ConnectionConfigs; return await Task.FromResult(dbConfigs.Adapt>()); } /// /// 获取数据库表(实体)集合 🔖 /// /// [DisplayName("获取数据库表(实体)集合")] public async Task> GetTableList(string configId = SqlSugarConst.MainConfigId) { var provider = _db.AsTenant().GetConnectionScope(configId); var dbTableInfos = provider.DbMaintenance.GetTableInfoList(false); // 不能走缓存,否则切库不起作用 var config = App.GetOptions().ConnectionConfigs.FirstOrDefault(u => configId.Equals(u.ConfigId)); var dbTableNames = dbTableInfos.Select(u => u.Name.ToLower()).ToList(); IEnumerable entityInfos = await GetEntityInfos(); entityInfos = entityInfos.OrderBy(u => u.EntityName.StartsWith("Sys") ? 1 : 0).ThenBy(u => u.EntityName); var tableOutputList = new List(); foreach (var item in entityInfos) { string dbTableName = item.DbTableName; int bracketIndex = dbTableName.IndexOf('{'); if (bracketIndex != -1) { dbTableName = dbTableName.Substring(0, bracketIndex); } var table = dbTableInfos.FirstOrDefault(x => x.Name.ToLower().StartsWith((config != null && config.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(dbTableName) : dbTableName).ToLower())); if (table == null) continue; tableOutputList.Add(new TableOutput { ConfigId = configId, EntityName = item.EntityName, TableName = table.Name, TableComment = item.TableDescription }); } return tableOutputList; } /// /// 根据表名获取列集合 🔖 /// /// [DisplayName("根据表名获取列集合")] public List GetColumnListByTableName([Required] string tableName, string configId = SqlSugarConst.MainConfigId) { // 切库---多库代码生成用 var provider = _db.AsTenant().GetConnectionScope(configId); var config = App.GetOptions().ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == configId); // 获取实体类型属性 var entityType = provider.DbMaintenance.GetTableInfoList(false).FirstOrDefault(u => u.Name == tableName); if (entityType == null) return null; var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)]; // 按原始类型的顺序获取所有实体类型属性(不包含导航属性,会返回null) return provider.DbMaintenance.GetColumnInfosByTableName(entityType.Name).Select(u => new ColumnOuput { ColumnName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName, ColumnKey = u.IsPrimarykey.ToString(), DataType = u.DataType.ToString(), NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType), ColumnComment = u.ColumnDescription }).ToList(); } /// /// 获取数据表列(实体属性)集合 /// /// private List GetColumnList([FromQuery] AddCodeGenInput input) { var entityType = GetEntityInfos().GetAwaiter().GetResult().FirstOrDefault(u => u.EntityName == input.TableName); if (entityType == null) return null; var config = App.GetOptions().ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == input.ConfigId); var dbTableName = config.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(entityType.DbTableName) : entityType.DbTableName; int bracketIndex = dbTableName.IndexOf('{'); if (bracketIndex != -1) { dbTableName = dbTableName.Substring(0, bracketIndex); var dbTableInfos = _db.AsTenant().GetConnectionScope(input.ConfigId).DbMaintenance.GetTableInfoList(false); var table = dbTableInfos.FirstOrDefault(x => x.Name.ToLower().StartsWith(config != null && config.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(dbTableName) : dbTableName)); if (table != null) dbTableName = table.Name; } // 切库---多库代码生成用 var provider = _db.AsTenant().GetConnectionScope(!string.IsNullOrEmpty(input.ConfigId) ? input.ConfigId : SqlSugarConst.MainConfigId); var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)]; var columnInfos = provider.DbMaintenance.GetColumnInfosByTableName(dbTableName, false); var result = columnInfos.Select(u => new ColumnOuput { // 转下划线后的列名需要再转回来(暂时不转) //ColumnName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName, ColumnName = u.DbColumnName, ColumnLength = u.Length, IsPrimarykey = u.IsPrimarykey, IsNullable = u.IsNullable, ColumnKey = u.IsPrimarykey.ToString(), NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType), DataType = u.DataType, ColumnComment = string.IsNullOrWhiteSpace(u.ColumnDescription) ? u.DbColumnName : u.ColumnDescription, DefaultValue = u.DefaultValue, }).ToList(); // 获取实体的属性信息,赋值给PropertyName属性(CodeFirst模式应以PropertyName为实际使用名称) var entityProperties = entityType.Type.GetProperties(); for (int i = result.Count - 1; i >= 0; i--) { var columnOutput = result[i]; // 先找自定义字段名的,如果找不到就再找自动生成字段名的(并且过滤掉没有SugarColumn的属性) var propertyInfo = entityProperties.FirstOrDefault(u => (u.GetCustomAttribute()?.ColumnName ?? "").ToLower() == columnOutput.ColumnName.ToLower()) ?? entityProperties.FirstOrDefault(u => u.GetCustomAttribute() != null && u.Name.ToLower() == (config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(columnOutput.ColumnName, entityBasePropertyNames).ToLower() : columnOutput.ColumnName.ToLower())); if (propertyInfo != null) { columnOutput.PropertyName = propertyInfo.Name; columnOutput.ColumnComment = propertyInfo.GetCustomAttribute()!.ColumnDescription; } else { result.RemoveAt(i); // 移除没有定义此属性的字段 } } return result; } /// /// 获取库表信息 /// /// private async Task> GetEntityInfos() { var entityInfos = new List(); var type = typeof(SugarTable); var types = new List(); if (_codeGenOptions.EntityAssemblyNames != null) { var assemblies = AppDomain.CurrentDomain.GetAssemblies(); foreach (var assembly in assemblies) { var assemblyName = assembly.GetName().Name!; if (_codeGenOptions.EntityAssemblyNames.Contains(assemblyName) || _codeGenOptions.EntityAssemblyNames.Any(name => assemblyName.Contains(name))) { Assembly asm = Assembly.Load(assemblyName); types.AddRange(asm.GetExportedTypes().ToList()); } } } bool IsMyAttribute(Attribute[] o) { foreach (Attribute a in o) { if (a.GetType() == type) return true; } return false; } Type[] cosType = types.Where(o => { return IsMyAttribute(Attribute.GetCustomAttributes(o, true)); } ).ToArray(); foreach (var ct in cosType) { var sugarAttribute = ct.GetCustomAttributes(type, true)?.FirstOrDefault(); var des = ct.GetCustomAttributes(typeof(DescriptionAttribute), true); var description = ""; if (des.Length > 0) { description = ((DescriptionAttribute)des[0]).Description; } entityInfos.Add(new EntityInfo() { EntityName = ct.Name, DbTableName = sugarAttribute == null ? ct.Name : ((SugarTable)sugarAttribute).TableName, TableDescription = sugarAttribute == null ? description : ((SugarTable)sugarAttribute).TableDescription, Type = ct }); } return await Task.FromResult(entityInfos); } /// /// 获取程序保存位置 🔖 /// /// [DisplayName("获取程序保存位置")] public List GetApplicationNamespaces() { return _codeGenOptions.BackendApplicationNamespaces; } /// /// 执行代码生成 🔖 /// /// [DisplayName("执行代码生成")] public async Task RunLocal(SysCodeGen input) { if (string.IsNullOrEmpty(input.GenerateType)) input.GenerateType = "200"; string outputPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName!); if (Directory.Exists(outputPath)) Directory.Delete(outputPath, true); var tableFieldList = await _codeGenConfigService.GetList(new CodeGenConfig() { CodeGenId = input.Id }); // 字段集合 #region 构造前端需要验证的数据 foreach (var item in tableFieldList) { List list = new List(); if (!string.IsNullOrWhiteSpace(item.Rules)) { if (item.Rules != "[]") { list = JSON.Deserialize>(item.Rules); } } else { item.Rules = "[]"; } item.RuleItems = list; item.WhetherRequired = list.Any(t => t.Type == "required") ? YesNoEnum.Y.ToString() : YesNoEnum.N.ToString(); item.AnyRule = list.Count > 0; item.RemoteVerify = list.Any(t => t.Type == "remote"); } #endregion 构造前端需要验证的数据 var queryWhetherList = tableFieldList.Where(u => u.QueryWhether == YesNoEnum.Y.ToString()).ToList(); // 前端查询集合 var joinTableList = tableFieldList.Where(u => u.EffectType == "Upload" || u.EffectType == "fk" || u.EffectType == "ApiTreeSelect").ToList(); // 需要连表查询的字段 (string joinTableNames, string lowerJoinTableNames) = GetJoinTableStr(joinTableList); // 获取连表的实体名和别名 var data = new CustomViewEngine(_db) { ConfigId = input.ConfigId!,//库定位器名 AuthorName = input.AuthorName!,//作者 BusName = input.BusName!,//业务名称 NameSpace = input.NameSpace!,//命名空间 ClassName = input.TableName!,//类名称 PagePath = input.PagePath!,//页面目录 ProjectLastName = input.NameSpace!.Split('.').Last(),//项目最后个名称,生成的时候赋值 QueryWhetherList = queryWhetherList,//查询条件 TableField = tableFieldList,//表字段配置信息 IsJoinTable = joinTableList.Count > 0,//是否联表 IsUpload = joinTableList.Where(u => u.EffectType == "Upload").Any(),//是否上传 PrintType = input.PrintType!,//支持打印类型 PrintName = input.PrintName!,//打印模板名称 IsApiService = input.IsApiService, RemoteVerify = tableFieldList.Any(t => t.RemoteVerify == true),//远程验证 }; // 模板 var templateAllList = GetTemplateList(input); var templateList = templateAllList.Where(t => t.Type != CodeGenTypeEnum.SeedData).ToList(); var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "template"); for (var i = 0; i < templateList.Count; i++) { var templateFilePath = Path.Combine(templatePath, templateList[i].Name); if (!File.Exists(templateFilePath)) continue; var tContent = File.ReadAllText(templateFilePath); var tResult = await _viewEngine.RunCompileFromCachedAsync(tContent, data, builderAction: builder => { builder.AddAssemblyReferenceByName("System.Linq"); builder.AddAssemblyReferenceByName("System.Collections"); builder.AddUsing("System.Collections.Generic"); builder.AddUsing("System.Linq"); }); string targetFile = templateList[i].OutputFile .Replace("{PagePath}", input.PagePath) .Replace("{TableName}", input.TableName) .Replace("{TableNameLower}", ToFirstLetterLowerCase(input.TableName!)); string tmpPath; if (!input.GenerateType.StartsWith('1')) { if (templateList[i].Type == CodeGenTypeEnum.Frontend) tmpPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent!.Parent!.FullName, _codeGenOptions.FrontRootPath, "src"); else tmpPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent!.FullName, input.NameSpace); } else { tmpPath = templateList[i].Type == CodeGenTypeEnum.Frontend ? Path.Combine(outputPath, _codeGenOptions.FrontRootPath, "src") : Path.Combine(outputPath, input!.NameSpace!); } targetFile = Path.Combine(tmpPath, targetFile); var dirPath = new DirectoryInfo(targetFile).Parent!.FullName; if (!Directory.Exists(dirPath)) Directory.CreateDirectory(dirPath); File.WriteAllText(targetFile, tResult, Encoding.UTF8); } if (input.GenerateMenu) { var menus = await AddMenu(input.TableName!, input.BusName!, input.MenuPid ?? 0, input.MenuIcon!, input.PagePath!, tableFieldList); //获得种子模板 var seedDataTemplates = templateAllList.Where(t => t.Type == CodeGenTypeEnum.SeedData).ToList(); for (var i = 0; i < seedDataTemplates.Count; i++) { var templateFilePath = Path.Combine(templatePath, seedDataTemplates[i].Name); if (!File.Exists(templateFilePath)) continue; var tContent = File.ReadAllText(templateFilePath); var seedData = new { AuthorName = input.AuthorName!,//作者 BusName = input.BusName!,//业务名称 NameSpace = input.NameSpace!,//命名空间 ClassName = input.TableName!,//类名称 input.ConfigId, MenuList = menus,//菜单集合 }; var tResult = await _viewEngine.RunCompileAsync(tContent, seedData, builderAction: builder => { builder.AddAssemblyReferenceByName("System.Linq"); builder.AddAssemblyReferenceByName("System.Collections"); builder.AddUsing("System.Collections.Generic"); builder.AddUsing("System.Linq"); }); string targetFile = seedDataTemplates[i].OutputFile .Replace("{PagePath}", input.PagePath) .Replace("{TableName}", input.TableName) .Replace("{TableNameLower}", ToFirstLetterLowerCase(input.TableName!)); string tmpPath; if (!input.GenerateType.StartsWith('1')) { tmpPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent!.FullName, input.NameSpace); } else { tmpPath = Path.Combine(outputPath, input!.NameSpace!); } targetFile = Path.Combine(tmpPath, targetFile); var dirPath = new DirectoryInfo(targetFile).Parent!.FullName; if (!Directory.Exists(dirPath)) Directory.CreateDirectory(dirPath); File.WriteAllText(targetFile, tResult, Encoding.UTF8); } } // 非ZIP压缩返回空 if (!input.GenerateType.StartsWith('1')) return null; else { string downloadPath = outputPath + ".zip"; // 判断是否存在同名称文件 if (File.Exists(downloadPath)) File.Delete(downloadPath); ZipFile.CreateFromDirectory(outputPath, downloadPath); return new { url = $"{App.HttpContext.Request.Scheme}://{App.HttpContext.Request.Host.Value}/CodeGen/{input.TableName}.zip" }; } } /// /// 获取代码生成预览 🔖 /// /// [DisplayName("获取代码生成预览")] public async Task> Preview(SysCodeGen input) { var tableFieldList = await _codeGenConfigService.GetList(new CodeGenConfig() { CodeGenId = input.Id }); // 字段集合 #region 构造前端需要验证的数据 foreach (var item in tableFieldList) { List list = new List(); if (!string.IsNullOrWhiteSpace(item.Rules)) { if (item.Rules != "[]") { list = JSON.Deserialize>(item.Rules); } } else { item.Rules = "[]"; } item.RuleItems = list; item.WhetherRequired = list.Any(t => t.Type == "required") ? YesNoEnum.Y.ToString() : YesNoEnum.N.ToString(); item.AnyRule = list.Count > 0; item.RemoteVerify = list.Any(t => t.Type == "remote"); } #endregion 构造前端需要验证的数据 var queryWhetherList = tableFieldList.Where(u => u.QueryWhether == YesNoEnum.Y.ToString()).ToList(); // 前端查询集合 var joinTableList = tableFieldList.Where(u => u.EffectType == "Upload" || u.EffectType == "fk" || u.EffectType == "ApiTreeSelect").ToList(); // 需要连表查询的字段 (string joinTableNames, string lowerJoinTableNames) = GetJoinTableStr(joinTableList); // 获取连表的实体名和别名 var data = new CustomViewEngine(_db) { ConfigId = input.ConfigId!,//库定位器名 AuthorName = input.AuthorName!,//作者 BusName = input.BusName!,//业务名称 NameSpace = input.NameSpace!,//命名空间 ClassName = input.TableName!,//类名称 PagePath = input.PagePath!,//页面目录 ProjectLastName = input.NameSpace!.Split('.').Last(),//项目最后个名称,生成的时候赋值 QueryWhetherList = queryWhetherList,//查询条件 TableField = tableFieldList,//表字段配置信息 IsJoinTable = joinTableList.Count > 0,//是否联表 IsUpload = joinTableList.Where(u => u.EffectType == "Upload").Any(),//是否上传 PrintType = input.PrintType!,//支持打印类型 PrintName = input.PrintName!,//打印模板名称 IsApiService = input.IsApiService, RemoteVerify = tableFieldList.Any(t => t.RemoteVerify == true)//远程验证 }; // 获取模板文件并替换 var templateAllList = GetTemplateList(input); var templateList = templateAllList.Where(t => t.Type != CodeGenTypeEnum.SeedData).ToList(); var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "template"); var result = new Dictionary(); for (var i = 0; i < templateList.Count; i++) { var templateFilePath = Path.Combine(templatePath, templateList[i].Name); if (!File.Exists(templateFilePath)) continue; var tContent = File.ReadAllText(templateFilePath); var tResult = await _viewEngine.RunCompileFromCachedAsync(tContent, data, builderAction: builder => { builder.AddAssemblyReferenceByName("System.Linq"); builder.AddAssemblyReferenceByName("System.Collections"); builder.AddUsing("System.Collections.Generic"); builder.AddUsing("System.Linq"); }); result.Add(templateList[i].Name?.TrimEnd(".vm")!, tResult); } return result; } /// /// 获取连表的实体名和别名 /// /// /// private static (string, string) GetJoinTableStr(List configs) { var uploads = configs.Where(u => u.EffectType == "Upload").ToList(); var fks = configs.Where(u => u.EffectType == "fk").ToList(); string str = ""; // string lowerStr = ""; // (o, i, c) foreach (var item in uploads) { lowerStr += "sysFile_FK_" + item.LowerPropertyName + ","; str += "SysFile,"; } foreach (var item in fks) { lowerStr += item.LowerFkEntityName + "_FK_" + item.LowerFkColumnName + ","; str += item.FkEntityName + ","; } return (str.TrimEnd(','), lowerStr.TrimEnd(',')); } /// /// 增加菜单 /// /// /// /// /// /// /// /// private async Task> AddMenu(string className, string busName, long pid, string menuIcon, string pagePath, List tableFieldList) { var pPath = string.Empty; // 若 pid=0 为顶级则创建菜单目录 if (pid == 0) { // 目录 var menuType0 = new SysMenu { Pid = 0, Title = busName + "管理", Type = MenuTypeEnum.Dir, Icon = "robot", Path = "/" + className.ToLower(), Component = "Layout", }; // 若先前存在则删除本级和下级 var menuList0 = await _db.Queryable().Where(u => u.Title == menuType0.Title && u.Type == menuType0.Type).ToListAsync(); if (menuList0.Count > 0) { var listIds = menuList0.Select(u => u.Id).ToList(); var childlistIds = new List(); foreach (var item in listIds) { var childlist = await _db.Queryable().ToChildListAsync(u => u.Pid, item); childlistIds.AddRange(childlist.Select(u => u.Id).ToList()); } listIds.AddRange(childlistIds); await _db.Deleteable().Where(u => listIds.Contains(u.Id)).ExecuteCommandAsync(); await _db.Deleteable().Where(u => listIds.Contains(u.MenuId)).ExecuteCommandAsync(); } pid = (await _db.Insertable(menuType0).ExecuteReturnEntityAsync()).Id; } else { var pMenu = await _db.Queryable().SingleAsync(u => u.Id == pid) ?? throw Oops.Oh(ErrorCodeEnum.D1505); pPath = pMenu.Path; } // 菜单 var menuType = new SysMenu { Pid = pid, Title = busName + "管理", Name = className[..1].ToLower() + className[1..], Type = MenuTypeEnum.Menu, Icon = menuIcon, Path = pPath + "/" + className.ToLower(), Component = "/" + pagePath + "/" + className[..1].ToLower() + className[1..] + "/index", }; // 若先前存在则删除本级和下级 var menuListCurrent = await _db.Queryable().Where(u => u.Title == menuType.Title && u.Type == menuType.Type).ToListAsync(); if (menuListCurrent.Count > 0) { var listIds = menuListCurrent.Select(u => u.Id).ToList(); var childListIds = new List(); foreach (var item in listIds) { var childList = await _db.Queryable().ToChildListAsync(u => u.Pid, item); childListIds.AddRange(childList.Select(u => u.Id).ToList()); } listIds.AddRange(childListIds); await _db.Deleteable().Where(u => listIds.Contains(u.Id)).ExecuteCommandAsync(); await _db.Deleteable().Where(u => listIds.Contains(u.MenuId)).ExecuteCommandAsync(); } var menuPid = (await _db.Insertable(menuType).ExecuteReturnEntityAsync()).Id; int menuOrder = 100; // 按钮-page var menuTypePage = new SysMenu { Pid = menuPid, Title = "查询", Type = MenuTypeEnum.Btn, Permission = className[..1].ToLower() + className[1..] + "/page", OrderNo = menuOrder }; menuOrder += 10; // 按钮-detail var menuTypeDetail = new SysMenu { Pid = menuPid, Title = "详情", Type = MenuTypeEnum.Btn, Permission = className[..1].ToLower() + className[1..] + "/detail", OrderNo = menuOrder }; menuOrder += 10; // 按钮-add var menuTypeAdd = new SysMenu { Pid = menuPid, Title = "增加", Type = MenuTypeEnum.Btn, Permission = className[..1].ToLower() + className[1..] + "/add", OrderNo = menuOrder }; menuOrder += 10; // 按钮-delete var menuTypeDelete = new SysMenu { Pid = menuPid, Title = "删除", Type = MenuTypeEnum.Btn, Permission = className[..1].ToLower() + className[1..] + "/delete", OrderNo = menuOrder }; menuOrder += 10; // 按钮-update var menuTypeUpdate = new SysMenu { Pid = menuPid, Title = "编辑", Type = MenuTypeEnum.Btn, Permission = className[..1].ToLower() + className[1..] + "/update", OrderNo = menuOrder }; menuOrder += 10; // 按钮-print var menuTypePrint = new SysMenu { Pid = menuPid, Title = "打印", Type = MenuTypeEnum.Btn, Permission = className[..1].ToLower() + className[1..] + "/print", OrderNo = menuOrder }; menuOrder += 10; // 按钮-import var menuTypeImport = new SysMenu { Pid = menuPid, Title = "导入", Type = MenuTypeEnum.Btn, Permission = className[..1].ToLower() + className[1..] + "/import", OrderNo = menuOrder }; menuOrder += 10; // 按钮-export var menuTypeExport = new SysMenu { Pid = menuPid, Title = "导出", Type = MenuTypeEnum.Btn, Permission = className[..1].ToLower() + className[1..] + "/export", OrderNo = menuOrder }; menuOrder += 10; var menuList = new List() { menuTypePage, menuTypeDetail, menuTypeAdd, menuTypeDelete, menuTypeUpdate, menuTypePrint, menuTypeImport, menuTypeExport }; // 加入fk、Upload、ApiTreeSelect 等接口的权限 // 在生成表格时,有些字段只是查询时显示,不需要填写(WhetherAddUpdate),所以这些字段没必要生成相应接口 var fkTableList = tableFieldList.Where(u => u.EffectType == "fk" && (u.WhetherAddUpdate == "Y" || u.QueryWhether == "Y")).ToList(); foreach (var @column in fkTableList) { var menuType1 = new SysMenu { Pid = menuPid, Title = "外键" + @column.ColumnName, Type = MenuTypeEnum.Btn, Permission = className[..1].ToLower() + className[1..] + "/" + column.FkEntityName + column.ColumnName + "Dropdown", OrderNo = menuOrder }; menuOrder += 10; menuList.Add(menuType1); } var treeSelectTableList = tableFieldList.Where(u => u.EffectType == "ApiTreeSelect").ToList(); foreach (var @column in treeSelectTableList) { var menuType1 = new SysMenu { Pid = menuPid, Title = "树型" + @column.ColumnName, Type = MenuTypeEnum.Btn, Permission = className[..1].ToLower() + className[1..] + "/" + column.FkEntityName + "Tree", OrderNo = menuOrder }; menuOrder += 10; menuList.Add(menuType1); } var uploadTableList = tableFieldList.Where(u => u.EffectType == "Upload").ToList(); foreach (var @column in uploadTableList) { var menuType1 = new SysMenu { Pid = menuPid, Title = "上传" + @column.ColumnName, Type = MenuTypeEnum.Btn, Permission = className[..1].ToLower() + className[1..] + "/Upload" + column.ColumnName, OrderNo = menuOrder }; menuOrder += 10; menuList.Add(menuType1); } await _db.Insertable(menuList).ExecuteCommandAsync(); menuList.Insert(0, menuType); return menuList; } /// /// 获取模板文件集合 /// /// private List GetTemplateList(SysCodeGen input) { //var codeGen= _codeGenRep.AsQueryable() //TODO: 只获取选中的模板 if (input.GenerateType!.Substring(1, 1).Contains('1')) { return _codeGetTemplateRep.AsQueryable() .Where(u => u.Type == CodeGenTypeEnum.Frontend) .Where(u => u.Id == SqlFunc.Subqueryable().Where(s => s.CodeGenId == input.Id).GroupBy(s => s.TemplateId).Select(s => s.TemplateId)) .ToList(); } else if (input.GenerateType.Substring(1, 1).Contains('2')) { return _codeGetTemplateRep.AsQueryable() .Where(u => u.Type == CodeGenTypeEnum.Backend) .Where(u => u.Id == SqlFunc.Subqueryable().Where(s => s.CodeGenId == input.Id).GroupBy(s => s.TemplateId).Select(s => s.TemplateId)) .ToList(); } else { return _codeGetTemplateRep.AsQueryable() .Where(u => u.Id == SqlFunc.Subqueryable().Where(s => s.CodeGenId == input.Id).GroupBy(s => s.TemplateId).Select(s => s.TemplateId)) .ToList(); } } /// /// 小驼峰命名法(camelCase):首字母小写 /// /// /// public static string ToFirstLetterLowerCase(string input) { // 检查字符串是否为空或空格 if (string.IsNullOrWhiteSpace(input)) { return input; } // 只转换首字母为小写 return char.ToLower(input[0]) + input.Substring(1); } }