using iWare.Wms.Core; using iWare.Wms.Core.Util.LowCode.Front.Code; using iWare.Wms.Core.Util.LowCode.Front.Model; using Furion; using Furion.DatabaseAccessor; using Furion.DatabaseAccessor.Extensions; using Furion.DependencyInjection; using Furion.DynamicApiController; using Furion.Extras.iWare.Wms.Entity; using Furion.Extras.iWare.Wms.Util.LowCode.Front.Code; using Furion.FriendlyException; using Furion.ViewEngine; using Mapster; using Microsoft.AspNetCore.Mvc; using Microsoft.EntityFrameworkCore; using Microsoft.EntityFrameworkCore.Infrastructure; using Microsoft.EntityFrameworkCore.Metadata; using Newtonsoft.Json; using System.Text; namespace iWare.Wms.Application.CodeGen { /// /// 代码生成器服务 /// [Route("api/[Controller]")] [ApiDescriptionSettings(Name = "CodeGenerate", Order = 100)] public class CodeGenerateService : ICodeGenService, IDynamicApiController, ITransient { private readonly IRepository _sysCodeGenRep; // 代码生成器仓储 private readonly IRepository _sysLowCodeRep; // 代码生成器仓储 private readonly ICodeGenConfigService _codeGenConfigService; private readonly IViewEngine _viewEngine; private readonly IRepository _sysMenuRep; // 菜单表仓储 public CodeGenerateService(IRepository sysCodeGenRep, IRepository sysLowCodeRep, ICodeGenConfigService codeGenConfigService, IViewEngine viewEngine, IRepository sysMenuRep) { _sysCodeGenRep = sysCodeGenRep; _sysLowCodeRep = sysLowCodeRep; _codeGenConfigService = codeGenConfigService; _viewEngine = viewEngine; _sysMenuRep = sysMenuRep; } /// /// 分页查询 /// /// /// [HttpGet("page")] public async Task> QueryCodeGenPageList([FromQuery] CodeGenPageInput input) { //if(input.) //// 加入配置表中 //_codeGenConfigService.AddList(GetColumnList(input.Adapt()), codeGen); var tableName = !string.IsNullOrEmpty(input.TableName?.Trim()); var codeGens = await _sysCodeGenRep.DetachedEntities .Where((tableName, u => EF.Functions.Like(u.TableName, $"%{input.TableName.Trim()}%"))) .ToADPagedListAsync(input.PageNo, input.PageSize); return codeGens; } /// /// 增加 /// /// /// [HttpPost("add")] public async Task AddCodeGen(AddCodeGenInput input) { var isExist = await _sysCodeGenRep.DetachedEntities.AnyAsync(u => u.TableName == input.TableName); if (isExist) throw Oops.Oh(ErrorCode.D1400); if (input.LowCodeId != null && input.LowCodeId > 0) { isExist = await _sysCodeGenRep.DetachedEntities.AnyAsync(u => u.LowCodeId == input.LowCodeId); } if (!isExist) { var codeGen = input.Adapt(); var newCodeGen = await codeGen.InsertNowAsync(); // 加入配置表中 await _codeGenConfigService.DelAndAddList(GetColumnList(input), newCodeGen.Entity); } } /// /// 删除 /// /// /// [HttpPost("delete")] public Task DeleteCodeGen(List inputs) { if (inputs == null || inputs.Count < 1) return Task.Run(() => { }); var taskList = new List(); inputs.ForEach(u => { taskList.Add(_sysCodeGenRep.DeleteAsync(u.Id)); // 删除配置表中 taskList.Add(_codeGenConfigService.Delete(u.Id)); }); return Task.WhenAll(taskList);//等待所有任务完成 } /// /// 更新 /// /// /// [HttpPost("edit")] public async Task UpdateCodeGen(UpdateCodeGenInput input) { var isExist = await _sysCodeGenRep.DetachedEntities.AnyAsync(u => u.TableName == input.TableName && u.Id != input.Id); if (isExist) throw Oops.Oh(ErrorCode.D1400); var codeGen = input.Adapt(); // 加入配置表中 _codeGenConfigService.DelAndAddList(GetColumnList(input.Adapt()), codeGen); } /// /// 刷新配置表 /// /// [HttpGet("refresh/{id}")] public void Refresh(long id) { var item = _sysCodeGenRep.Where(x => x.Id == id).FirstOrDefault(); // 加入配置表中 _codeGenConfigService.DelAndAddList(GetColumnList(item.Adapt()), item); } /// /// 详情 /// /// /// [HttpGet("detail")] public async Task GetCodeGen([FromQuery] QueryCodeGenInput input) { return await _sysCodeGenRep.DetachedEntities.FirstOrDefaultAsync(u => u.Id == input.Id); } /// /// 获取数据库库集合 /// /// [HttpGet("DatabaseList")] public List GetDatabaseList() { var DbContextLocators = AppDomain.CurrentDomain.GetAssemblies() .SelectMany( a => a.GetTypes().Where(t => t.GetInterfaces().Contains(typeof(IDbContextLocator))) ) .Select(x => new DatabaseOutput { DatabaseName = x.Name, DatabaseComment = x.FullName }) .ToList(); return DbContextLocators; } /// /// 获取数据库表(实体)集合 /// /// [HttpGet("InformationList")] public List GetTableList(string dbContextLocatorName) { var dbContext = Db.GetDbContext();//默认数据库 if (!string.IsNullOrEmpty(dbContextLocatorName)) { var dbContentLocator = AppDomain.CurrentDomain.GetAssemblies() .SelectMany(a => a.GetTypes().Where(t => t.GetInterfaces().Contains(typeof(IDbContextLocator)))).Where(x => x.Name == dbContextLocatorName).FirstOrDefault(); dbContext = Db.GetDbContext(dbContentLocator); } // 获取实体类型属性 //var entityType = dbContext.Model.GetEntityTypes()要改成 var entityType = dbContext.GetService().Model.GetEntityTypes() return dbContext.GetService().Model.GetEntityTypes().Select(u => new TableOutput { DatabaseName = dbContextLocatorName, TableName = u.GetDefaultTableName(), TableComment = u.GetComment() }).ToList(); } /// /// 根据表名获取列 /// /// [HttpGet("ColumnList/{databaseName}/{tableName}")] public List GetColumnListByTableName(string databaseName, string tableName) { var dbContext = Db.GetDbContext();//默认数据库 if (!string.IsNullOrEmpty(databaseName)) { var dbContentLocator = AppDomain.CurrentDomain.GetAssemblies() .SelectMany(a => a.GetTypes().Where(t => t.GetInterfaces().Contains(typeof(IDbContextLocator)))).Where(x => x.Name == databaseName).FirstOrDefault(); dbContext = Db.GetDbContext(dbContentLocator); } // 获取实体类型属性 var entityType = dbContext.GetService().Model.GetEntityTypes().FirstOrDefault(u => u.ClrType.Name == tableName); if (entityType == null) return null; // 获取原始类型属性 var type = entityType.ClrType; if (type == null) return null; // 按原始类型的顺序获取所有实体类型属性(不包含导航属性,会返回null) return type.GetProperties().Select(propertyInfo => entityType.FindProperty(propertyInfo.Name)) .Where(p => p != null).Select(p => new TableColumnOuput { ColumnName = p.Name, ColumnKey = p.IsKey().ToString(), DataType = p.PropertyInfo.PropertyType.ToString(), NetType = CodeGenUtil.ConvertDataType(p.PropertyInfo.PropertyType.ToString()), ColumnComment = p.GetComment() }).ToList(); } /// /// 获取数据表列(实体属性)集合 /// /// [NonAction] public List GetColumnList([FromQuery] AddCodeGenInput input) { var dbContext = Db.GetDbContext();//默认数据库 if (!string.IsNullOrEmpty(input.DatabaseName)) { var dbContentLocator = AppDomain.CurrentDomain.GetAssemblies() .SelectMany(a => a.GetTypes().Where(t => t.GetInterfaces().Contains(typeof(IDbContextLocator)))).Where(x => x.Name == input.DatabaseName).FirstOrDefault(); dbContext = Db.GetDbContext(dbContentLocator); } // 获取实体类型属性 var entityType = dbContext.GetService().Model.GetEntityTypes() .FirstOrDefault(u => u.ClrType.Name == input.TableName); if (entityType == null) return null; // 获取原始类型属性 var type = entityType.ClrType; if (type == null) return null; // 按原始类型的顺序获取所有实体类型属性(不包含导航属性,会返回null) return type.GetProperties().Select(propertyInfo => entityType.FindProperty(propertyInfo.Name)) .Where(p => p != null) .Select(p => new TableColumnOuput { ColumnName = p.Name, ColumnKey = p.IsKey().ToString(), DataType = p.PropertyInfo.PropertyType.ToString(), ColumnComment = p.GetComment() }).ToList(); } /// /// 代码生成_本地项目 /// /// [HttpPost("runLocal")] public async Task RunLocal(SysCodeGen input) { var templatePathList = GetTemplatePathList(); var targetPathList = GetTargetPathList(input); for (var i = 0; i < templatePathList.Count; i++) { var tContent = File.ReadAllText(templatePathList[i]); var tableFieldList = await _codeGenConfigService.List(new CodeGenConfig() { CodeGenId = input.Id }); // 字段集合 tableFieldList.ForEach(u => { switch (u.NetType.ToLower()) { case "int": case "int32": case "long": case "guid": case "decimal": case "datetime": case "datetimeoffset": u.NetTypeIsNullLable = "?"; break; } u.OriginalColumnName = u.ColumnName; }); if (i >= 6) // 适应前端首字母小写 { tableFieldList.ForEach(u => { u.ColumnName = u.ColumnName.Substring(0, 1).ToLower() + u.ColumnName[1..]; }); } var queryWhetherList = tableFieldList.Where(u => u.QueryWhether == YesOrNot.Y.ToString()).ToList(); // 前端查询集合 string FormDesign = ""; if (input.LowCodeId != null && input.LowCodeId > 0) { FormDesign = _sysLowCodeRep.Where(x => x.Id == input.LowCodeId).Select(x => x.FormDesign).FirstOrDefault(); } List dynamicLoad_dict = new List(); Dictionary> dynamicData = new Dictionary>(); if (!string.IsNullOrEmpty(FormDesign)) { try { var AllDynamic = FormDesign.ConvertToFront().AllFront().AllDynamic(); AllDynamic.Where(x => x.Dynamic).Select(x => x.DynamicKey).ToList().ForEach(item => { dynamicData.Add(item, new List()); var d = item.GetDynamic(); if (d != null) { if (d.Head == "dict") { dynamicLoad_dict.Add(d); } } }); } catch { } } try { var tResult = _viewEngine.RunCompileFromCached(tContent, new { input.AuthorName, input.BusName, input.NameSpace, input.ProName, input.DatabaseName, ClassName = input.TableName, CamelizeClassName = input.TableName.Substring(0, 1).ToLower() + input.TableName[1..], //首字母小写 QueryWhetherList = queryWhetherList, TableField = tableFieldList, input.LowCodeId, FormDesign, DynamicData = JsonConvert.SerializeObject(dynamicData), DynamicLoad_Dict = dynamicLoad_dict, IsFile = tableFieldList.Where(x => x.DtoNetType.Contains("Front_FileDto")).Any(), FileTableField = tableFieldList.Where(x => x.DtoNetType.Contains("Front_FileDto")).ToList() }); var dirPath = new DirectoryInfo(targetPathList[i]).Parent.FullName; if (!Directory.Exists(dirPath)) Directory.CreateDirectory(dirPath); File.WriteAllText(targetPathList[i], tResult, Encoding.UTF8); } catch (Exception ex) { throw Oops.Oh($"错误模板:{templatePathList[i]}。错误信息:{ex.Message}。"); } } await AddMenu(input.DatabaseName.Substring(0, 5), input.TableName, input.BusName, input.MenuApplication, input.MenuPid); } private async Task AddMenu(string menucodePre, string className, string busName, string application, long pid) { // 定义菜单编码前缀 var codePrefix = menucodePre + "_" + className.ToLower();//改为取数据库定位器的前五个字母方便区分业务 //"dilon_" + className.ToLower(); // 先删除该表已生成的菜单列表 var menus = await _sysMenuRep.DetachedEntities.Where(u => u.Code == codePrefix || u.Code.StartsWith(codePrefix + "_")).ToListAsync(); await _sysMenuRep.DeleteAsync(menus); // 如果 pid 为 0 说明为顶级菜单, 需要创建顶级目录 if (pid == 0) { // 目录 var menuType0 = new SysMenu { Pid = 0, Pids = "[0],", Name = busName + "管理", Code = codePrefix, Type = MenuType.DIR, Icon = "robot", Router = "/" + className.ToLower(), Component = "PageView", Application = application }; pid = _sysMenuRep.InsertNowAsync(menuType0).GetAwaiter().GetResult().Entity.Id; } // 由于后续菜单会有修改, 需要判断下 pid 是否存在, 不存在报错 else if (!await _sysMenuRep.DetachedEntities.AnyAsync(e => e.Id == pid)) throw Oops.Oh(ErrorCode.D1505); // 菜单 var menuType1 = new SysMenu { Pid = pid, Pids = "[0],[" + pid + "],", Name = busName + "管理", Code = codePrefix + "_mgr", Type = MenuType.MENU, Router = "/" + className.ToLower(), Component = "main/" + className + "/index", Application = application, OpenType = MenuOpenType.COMPONENT }; var pid1 = _sysMenuRep.InsertNowAsync(menuType1).GetAwaiter().GetResult().Entity.Id; // 按钮-page var menuType2 = new SysMenu { Pid = pid1, Pids = "[0],[" + pid + "],[" + pid1 + "],", Name = busName + "查询", Code = codePrefix + "_mgr_page", Type = MenuType.BTN, Permission = className + ":page", Application = application, }.InsertAsync(); // 按钮-detail var menuType2_1 = new SysMenu { Pid = pid1, Pids = "[0],[" + pid + "],[" + pid1 + "],", Name = busName + "详情", Code = codePrefix + "_mgr_detail", Type = MenuType.BTN, Permission = className + ":detail", Application = application, }.InsertAsync(); // 按钮-add var menuType2_2 = new SysMenu { Pid = pid1, Pids = "[0],[" + pid + "],[" + pid1 + "],", Name = busName + "增加", Code = codePrefix + "_mgr_add", Type = MenuType.BTN, Permission = className + ":add", Application = application, }.InsertAsync(); // 按钮-delete var menuType2_3 = new SysMenu { Pid = pid1, Pids = "[0],[" + pid + "],[" + pid1 + "],", Name = busName + "删除", Code = codePrefix + "_mgr_delete", Type = MenuType.BTN, Permission = className + ":delete", Application = application, }.InsertAsync(); // 按钮-edit var menuType2_4 = new SysMenu { Pid = pid1, Pids = "[0],[" + pid + "],[" + pid1 + "],", Name = busName + "编辑", Code = codePrefix + "_mgr_edit", Type = MenuType.BTN, Permission = className + ":edit", Application = application, }.InsertAsync(); } /// /// 获取模板文件路径集合 /// /// private List GetTemplatePathList() { var templatePath = App.WebHostEnvironment.WebRootPath + @"\Template\"; return new List() { templatePath + "Service.cs.vm", templatePath + "IService.cs.vm", templatePath + "Input.cs.vm", templatePath + "Output.cs.vm", templatePath + "Dto.cs.vm", templatePath + "Mapper.cs.vm", templatePath + "index.vue.vm", templatePath + "addForm.vue.vm", templatePath + "editForm.vue.vm", templatePath + "Manage.js.vm", }; } /// /// 设置生成文件路径 /// /// /// private List GetTargetPathList(SysCodeGen input) { var backendPath = new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName + @"\" + input.NameSpace + @"\Service\" + input.TableName + @"\"; var servicePath = backendPath + input.TableName + "Service.cs"; var iservicePath = backendPath + "I" + input.TableName + "Service.cs"; var inputPath = backendPath + @"Dto\" + input.TableName + "Input.cs"; var outputPath = backendPath + @"Dto\" + input.TableName + "Output.cs"; var viewPath = backendPath + @"Dto\" + input.TableName + "Dto.cs"; var mapperPath = backendPath + @"Map\" + input.TableName + "Mapper.cs"; var frontendPath = new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.Parent.FullName + @"\iwara-wms-web\src\views\main\"; var indexPath = frontendPath + input.TableName + @"\index.vue"; var addFormPath = frontendPath + input.TableName + @"\addForm.vue"; var editFormPath = frontendPath + input.TableName + @"\editForm.vue"; var apiJsPath = new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.Parent.FullName + @"\iwara-wms-web\src\api\modular\main\" + input.TableName + "Manage.js"; return new List() { servicePath, iservicePath, inputPath, outputPath, viewPath, mapperPath, indexPath, addFormPath, editFormPath, apiJsPath }; } } }