SysCodeGenService.cs 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642
  1. // Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
  2. //
  3. // 本项目主要遵循 MIT 许可证和 Apache 许可证(版本 2.0)进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
  4. //
  5. // 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动!任何基于本项目二次开发而产生的一切法律纠纷和责任,我们不承担任何责任!
  6. using System.IO.Compression;
  7. namespace Admin.NET.Core.Service;
  8. /// <summary>
  9. /// 系统代码生成器服务 🧩
  10. /// </summary>
  11. [ApiDescriptionSettings(Order = 270)]
  12. public class SysCodeGenService : IDynamicApiController, ITransient
  13. {
  14. private readonly ISqlSugarClient _db;
  15. private readonly SysCodeGenConfigService _codeGenConfigService;
  16. private readonly CodeGenOptions _codeGenOptions;
  17. private readonly IViewEngine _viewEngine;
  18. private readonly UserManager _userManager;
  19. public SysCodeGenService(ISqlSugarClient db,
  20. SysCodeGenConfigService codeGenConfigService,
  21. IOptions<CodeGenOptions> codeGenOptions,
  22. UserManager userManager,
  23. IViewEngine viewEngine)
  24. {
  25. _db = db;
  26. _viewEngine = viewEngine;
  27. _userManager = userManager;
  28. _codeGenOptions = codeGenOptions.Value;
  29. _codeGenConfigService = codeGenConfigService;
  30. }
  31. /// <summary>
  32. /// 获取代码生成分页列表 🔖
  33. /// </summary>
  34. /// <param name="input"></param>
  35. /// <returns></returns>
  36. [DisplayName("获取代码生成分页列表")]
  37. public async Task<SqlSugarPagedList<SysCodeGen>> Page(CodeGenInput input)
  38. {
  39. return await _db.Queryable<SysCodeGen>()
  40. .WhereIF(!string.IsNullOrWhiteSpace(input.TableName), u => u.TableName.Contains(input.TableName.Trim()))
  41. .WhereIF(!string.IsNullOrWhiteSpace(input.BusName), u => u.BusName.Contains(input.BusName.Trim()))
  42. .ToPagedListAsync(input.Page, input.PageSize);
  43. }
  44. /// <summary>
  45. /// 增加代码生成 🔖
  46. /// </summary>
  47. /// <param name="input"></param>
  48. /// <returns></returns>
  49. [ApiDescriptionSettings(Name = "Add"), HttpPost]
  50. [DisplayName("增加代码生成")]
  51. public async Task AddCodeGen(AddCodeGenInput input)
  52. {
  53. var isExist = await _db.Queryable<SysCodeGen>().Where(u => u.TableName == input.TableName).AnyAsync();
  54. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1400);
  55. if (input.TableUniqueList?.Count > 0) input.TableUniqueConfig = JSON.Serialize(input.TableUniqueList);
  56. var codeGen = input.Adapt<SysCodeGen>();
  57. var newCodeGen = await _db.Insertable(codeGen).ExecuteReturnEntityAsync();
  58. // 增加配置表
  59. _codeGenConfigService.AddList(GetColumnList(input), newCodeGen);
  60. }
  61. /// <summary>
  62. /// 更新代码生成 🔖
  63. /// </summary>
  64. /// <param name="input"></param>
  65. /// <returns></returns>
  66. [ApiDescriptionSettings(Name = "Update"), HttpPost]
  67. [DisplayName("更新代码生成")]
  68. public async Task UpdateCodeGen(UpdateCodeGenInput input)
  69. {
  70. var isExist = await _db.Queryable<SysCodeGen>().AnyAsync(u => u.TableName == input.TableName && u.Id != input.Id);
  71. if (isExist) throw Oops.Oh(ErrorCodeEnum.D1400);
  72. if (input.TableUniqueList?.Count > 0) input.TableUniqueConfig = JSON.Serialize(input.TableUniqueList);
  73. var codeGen = input.Adapt<SysCodeGen>();
  74. await _db.Updateable(codeGen).ExecuteCommandAsync();
  75. // 更新配置表
  76. await _codeGenConfigService.DeleteCodeGenConfig(codeGen.Id);
  77. _codeGenConfigService.AddList(GetColumnList(input.Adapt<AddCodeGenInput>()), codeGen);
  78. }
  79. /// <summary>
  80. /// 删除代码生成 🔖
  81. /// </summary>
  82. /// <param name="inputs"></param>
  83. /// <returns></returns>
  84. [ApiDescriptionSettings(Name = "Delete"), HttpPost]
  85. [DisplayName("删除代码生成")]
  86. public async Task DeleteCodeGen(List<DeleteCodeGenInput> inputs)
  87. {
  88. if (inputs == null || inputs.Count < 1) return;
  89. var codeGenConfigTaskList = new List<Task>();
  90. inputs.ForEach(u =>
  91. {
  92. _db.Deleteable<SysCodeGen>().In(u.Id).ExecuteCommand();
  93. // 删除配置表
  94. codeGenConfigTaskList.Add(_codeGenConfigService.DeleteCodeGenConfig(u.Id));
  95. });
  96. await Task.WhenAll(codeGenConfigTaskList);
  97. }
  98. /// <summary>
  99. /// 获取代码生成详情 🔖
  100. /// </summary>
  101. /// <param name="input"></param>
  102. /// <returns></returns>
  103. [DisplayName("获取代码生成详情")]
  104. public async Task<SysCodeGen> GetDetail([FromQuery] QueryCodeGenInput input)
  105. {
  106. return await _db.Queryable<SysCodeGen>().SingleAsync(u => u.Id == input.Id);
  107. }
  108. /// <summary>
  109. /// 获取数据库库集合 🔖
  110. /// </summary>
  111. /// <returns></returns>
  112. [DisplayName("获取数据库库集合")]
  113. public async Task<List<DatabaseOutput>> GetDatabaseList()
  114. {
  115. var dbConfigs = App.GetOptions<DbConnectionOptions>().ConnectionConfigs;
  116. return await Task.FromResult(dbConfigs.Adapt<List<DatabaseOutput>>());
  117. }
  118. /// <summary>
  119. /// 获取数据库表(实体)集合 🔖
  120. /// </summary>
  121. /// <returns></returns>
  122. [DisplayName("获取数据库表(实体)集合")]
  123. public async Task<List<TableOutput>> GetTableList(string configId = SqlSugarConst.MainConfigId)
  124. {
  125. var provider = _db.AsTenant().GetConnectionScope(configId);
  126. var dbTableInfos = provider.DbMaintenance.GetTableInfoList(false); // 不能走缓存,否则切库不起作用
  127. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => configId.Equals(u.ConfigId));
  128. // var dbTableNames = dbTableInfos.Select(u => u.Name.ToLower()).ToList();
  129. IEnumerable<EntityInfo> entityInfos = await GetEntityInfos();
  130. var tableOutputList = new List<TableOutput>();
  131. foreach (var item in entityInfos)
  132. {
  133. var table = dbTableInfos.FirstOrDefault(u => string.Equals(u.Name, (config!.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(item.DbTableName) : item.DbTableName), StringComparison.CurrentCultureIgnoreCase));
  134. if (table == null) continue;
  135. tableOutputList.Add(new TableOutput
  136. {
  137. ConfigId = configId,
  138. EntityName = item.EntityName,
  139. TableName = table.Name,
  140. TableComment = item.TableDescription
  141. });
  142. }
  143. return tableOutputList;
  144. }
  145. /// <summary>
  146. /// 根据表名获取列集合 🔖
  147. /// </summary>
  148. /// <returns></returns>
  149. [DisplayName("根据表名获取列集合")]
  150. public List<ColumnOuput> GetColumnListByTableName([Required] string tableName, string configId = SqlSugarConst.MainConfigId)
  151. {
  152. // 切库---多库代码生成用
  153. var provider = _db.AsTenant().GetConnectionScope(configId);
  154. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == configId);
  155. // 获取实体类型属性
  156. var entityType = provider.DbMaintenance.GetTableInfoList(false).FirstOrDefault(u => u.Name == tableName);
  157. if (entityType == null) return null;
  158. var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)];
  159. // 按原始类型的顺序获取所有实体类型属性(不包含导航属性,会返回null)
  160. return provider.DbMaintenance.GetColumnInfosByTableName(entityType.Name).Select(u => new ColumnOuput
  161. {
  162. ColumnName = config!.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
  163. ColumnKey = u.IsPrimarykey.ToString(),
  164. DataType = u.DataType.ToString(),
  165. NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  166. ColumnComment = u.ColumnDescription
  167. }).ToList();
  168. }
  169. /// <summary>
  170. /// 获取数据表列(实体属性)集合
  171. /// </summary>
  172. /// <returns></returns>
  173. private List<ColumnOuput> GetColumnList([FromQuery] AddCodeGenInput input)
  174. {
  175. var entityType = GetEntityInfos().GetAwaiter().GetResult().FirstOrDefault(u => u.EntityName == input.TableName);
  176. if (entityType == null)
  177. return null;
  178. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == input.ConfigId);
  179. var dbTableName = config!.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(entityType.DbTableName) : entityType.DbTableName;
  180. // 切库---多库代码生成用
  181. var provider = _db.AsTenant().GetConnectionScope(!string.IsNullOrEmpty(input.ConfigId) ? input.ConfigId : SqlSugarConst.MainConfigId);
  182. var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)];
  183. var columnInfos = provider.DbMaintenance.GetColumnInfosByTableName(dbTableName, false);
  184. var result = columnInfos.Select(u => new ColumnOuput
  185. {
  186. // 转下划线后的列名需要再转回来(暂时不转)
  187. //ColumnName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
  188. ColumnName = u.DbColumnName,
  189. ColumnLength = u.Length,
  190. IsPrimarykey = u.IsPrimarykey,
  191. IsNullable = u.IsNullable,
  192. ColumnKey = u.IsPrimarykey.ToString(),
  193. NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  194. DataType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  195. ColumnComment = string.IsNullOrWhiteSpace(u.ColumnDescription) ? u.DbColumnName : u.ColumnDescription
  196. }).ToList();
  197. // 获取实体的属性信息,赋值给PropertyName属性(CodeFirst模式应以PropertyName为实际使用名称)
  198. var entityProperties = entityType.Type.GetProperties();
  199. for (int i = result.Count - 1; i >= 0; i--)
  200. {
  201. var columnOutput = result[i];
  202. // 先找自定义字段名的,如果找不到就再找自动生成字段名的(并且过滤掉没有SugarColumn的属性)
  203. var propertyInfo = entityProperties.FirstOrDefault(u => string.Equals((u.GetCustomAttribute<SugarColumn>()?.ColumnName ?? ""), columnOutput.ColumnName, StringComparison.CurrentCultureIgnoreCase)) ??
  204. entityProperties.FirstOrDefault(u => u.GetCustomAttribute<SugarColumn>() != null && u.Name.ToLower() == (config.DbSettings.EnableUnderLine
  205. ? CodeGenUtil.CamelColumnName(columnOutput.ColumnName, entityBasePropertyNames).ToLower()
  206. : columnOutput.ColumnName.ToLower()));
  207. if (propertyInfo != null)
  208. {
  209. columnOutput.PropertyName = propertyInfo.Name;
  210. columnOutput.ColumnComment = propertyInfo.GetCustomAttribute<SugarColumn>()!.ColumnDescription;
  211. var propertyType = Nullable.GetUnderlyingType(propertyInfo.PropertyType);
  212. if (propertyInfo.PropertyType.IsEnum || (propertyType?.IsEnum ?? false))
  213. {
  214. columnOutput.DictTypeCode = (propertyType ?? propertyInfo.PropertyType).Name;
  215. }
  216. else
  217. {
  218. var dict = propertyInfo.GetCustomAttribute<DictAttribute>();
  219. if (dict != null) columnOutput.DictTypeCode = dict.DictTypeCode;
  220. }
  221. }
  222. else
  223. {
  224. result.RemoveAt(i); // 移除没有定义此属性的字段
  225. }
  226. }
  227. return result;
  228. }
  229. /// <summary>
  230. /// 获取库表信息
  231. /// </summary>
  232. /// <returns></returns>
  233. private async Task<IEnumerable<EntityInfo>> GetEntityInfos()
  234. {
  235. var entityInfos = new List<EntityInfo>();
  236. var type = typeof(SugarTable);
  237. var types = new List<Type>();
  238. if (_codeGenOptions.EntityAssemblyNames != null)
  239. {
  240. var assemblies = AppDomain.CurrentDomain.GetAssemblies();
  241. foreach (var assembly in assemblies)
  242. {
  243. var assemblyName = assembly.GetName().Name;
  244. if (!_codeGenOptions.EntityAssemblyNames.Contains(assemblyName) &&
  245. !_codeGenOptions.EntityAssemblyNames.Any(name => assemblyName!.Contains(name)))
  246. {
  247. continue;
  248. }
  249. Assembly asm = Assembly.Load(assemblyName!);
  250. types.AddRange(asm.GetExportedTypes().ToList());
  251. }
  252. }
  253. Type[] cosType = types.Where(o => IsMyAttribute(Attribute.GetCustomAttributes(o, true))).ToArray();
  254. foreach (var ct in cosType)
  255. {
  256. var sugarAttribute = ct.GetCustomAttributes(type, true).FirstOrDefault();
  257. var des = ct.GetCustomAttributes(typeof(DescriptionAttribute), true);
  258. var description = "";
  259. if (des.Length > 0)
  260. {
  261. description = ((DescriptionAttribute)des[0]).Description;
  262. }
  263. entityInfos.Add(new EntityInfo()
  264. {
  265. EntityName = ct.Name,
  266. DbTableName = sugarAttribute == null ? ct.Name : ((SugarTable)sugarAttribute).TableName,
  267. TableDescription = sugarAttribute == null ? description : ((SugarTable)sugarAttribute).TableDescription,
  268. Type = ct
  269. });
  270. }
  271. return await Task.FromResult(entityInfos);
  272. bool IsMyAttribute(Attribute[] o) => o.Any(a => a.GetType() == type);
  273. }
  274. /// <summary>
  275. /// 获取程序保存位置 🔖
  276. /// </summary>
  277. /// <returns></returns>
  278. [DisplayName("获取程序保存位置")]
  279. public List<string> GetApplicationNamespaces()
  280. {
  281. return _codeGenOptions.BackendApplicationNamespaces;
  282. }
  283. /// <summary>
  284. /// 代码生成到本地 🔖
  285. /// </summary>
  286. /// <returns></returns>
  287. [DisplayName("代码生成到本地")]
  288. public async Task<dynamic> RunLocal(SysCodeGen input)
  289. {
  290. if (string.IsNullOrEmpty(input.GenerateType))
  291. input.GenerateType = "200";
  292. // 先删除该表已生成的菜单列表
  293. List<string> targetPathList;
  294. var zipPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName!);
  295. if (input.GenerateType.StartsWith('1'))
  296. {
  297. targetPathList = GetZipPathList(input);
  298. if (Directory.Exists(zipPath)) Directory.Delete(zipPath, true);
  299. }
  300. else
  301. targetPathList = GetTargetPathList(input);
  302. var (tableFieldList, result) = await RenderTemplateAsync(input);
  303. var templatePathList = GetTemplatePathList(input);
  304. for (var i = 0; i < templatePathList.Count; i++)
  305. {
  306. var content = result.GetValueOrDefault(templatePathList[i]?.TrimEnd(".vm"));
  307. if (string.IsNullOrWhiteSpace(content)) continue;
  308. var dirPath = new DirectoryInfo(targetPathList[i]).Parent!.FullName;
  309. if (!Directory.Exists(dirPath)) Directory.CreateDirectory(dirPath);
  310. _ = File.WriteAllTextAsync(targetPathList[i], content, Encoding.UTF8);
  311. }
  312. if (input.GenerateMenu) await AddMenu(input.TableName, input.BusName, input.MenuPid ?? 0, input.MenuIcon, input.PagePath, tableFieldList);
  313. // 非ZIP压缩返回空
  314. if (!input.GenerateType.StartsWith('1')) return null;
  315. // 判断是否存在同名称文件
  316. string downloadPath = zipPath + ".zip";
  317. if (File.Exists(downloadPath)) File.Delete(downloadPath);
  318. // 创建zip文件并返回下载地址
  319. ZipFile.CreateFromDirectory(zipPath, downloadPath);
  320. return new { url = $"{App.HttpContext.Request.Scheme}://{App.HttpContext.Request.Host.Value}/codeGen/{input.TableName}.zip" };
  321. }
  322. /// <summary>
  323. /// 获取代码生成预览 🔖
  324. /// </summary>
  325. /// <returns></returns>
  326. [DisplayName("获取代码生成预览")]
  327. // ReSharper disable once MemberCanBePrivate.Global
  328. public async Task<Dictionary<string, string>> Preview(SysCodeGen input)
  329. {
  330. var (_, result) = await RenderTemplateAsync(input);
  331. return result;
  332. }
  333. /// <summary>
  334. /// 渲染模板
  335. /// </summary>
  336. /// <param name="input"></param>
  337. /// <returns></returns>
  338. private async Task<(List<CodeGenConfig> tableFieldList, Dictionary<string, string> result)> RenderTemplateAsync(SysCodeGen input)
  339. {
  340. var tableFieldList = await _codeGenConfigService.GetList(new CodeGenConfig { CodeGenId = input.Id }); // 字段集合
  341. var joinTableList = tableFieldList.Where(u => u.EffectType is "Upload" or "ForeignKey" or "ApiTreeSelector").ToList(); // 需要连表查询的字段
  342. var data = new CustomViewEngine
  343. {
  344. ConfigId = input.ConfigId,
  345. BusName = input.BusName,
  346. PagePath = input.PagePath,
  347. NameSpace = input.NameSpace,
  348. ClassName = input.TableName,
  349. PrintType = input.PrintType,
  350. PrintName = input.PrintName,
  351. AuthorName = input.AuthorName,
  352. ProjectLastName = input.NameSpace!.Split('.').Last(),
  353. LowerClassName = input.TableName![..1].ToLower() + input.TableName[1..],
  354. TableUniqueConfigList = input.TableUniqueList ?? new(),
  355. TableField = tableFieldList,
  356. QueryWhetherList = tableFieldList.Where(u => u.WhetherQuery == "Y").ToList(),
  357. ImportFieldList = tableFieldList.Where(u => u.WhetherImport == "Y").ToList(),
  358. UploadFieldList = tableFieldList.Where(u => u.EffectType == "Upload").ToList(),
  359. PrimaryKeyFieldList = tableFieldList.Where(c => c.ColumnKey == "True").ToList(),
  360. AddUpdateFieldList = tableFieldList.Where(u => u.WhetherAddUpdate == "Y").ToList(),
  361. ApiTreeFieldList = tableFieldList.Where(u => u.EffectType == "ApiTreeSelector").ToList(),
  362. DropdownFieldList = tableFieldList.Where(u => u.EffectType is "ForeignKey" or "ApiTreeSelector").ToList(),
  363. HasJoinTable = joinTableList.Count > 0,
  364. HasDictField = tableFieldList.Any(u => u.EffectType == "DictSelector"),
  365. HasEnumField = tableFieldList.Any(u => u.EffectType == "EnumSelector"),
  366. HasConstField = tableFieldList.Any(u => u.EffectType == "ConstSelector"),
  367. HasLikeQuery = tableFieldList.Any(c => c.WhetherQuery == "Y" && c.QueryType == "like")
  368. };
  369. // 获取模板文件并替换
  370. var templatePathList = GetTemplatePathList();
  371. var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "template");
  372. var result = new Dictionary<string, string>();
  373. foreach (var path in templatePathList)
  374. {
  375. var templateFilePath = Path.Combine(templatePath, path);
  376. if (!File.Exists(templateFilePath)) continue;
  377. var tContent = await File.ReadAllTextAsync(templateFilePath);
  378. var tResult = await _viewEngine.RunCompileFromCachedAsync(tContent, data, builderAction: builder =>
  379. {
  380. builder.AddAssemblyReferenceByName("System.Text.RegularExpressions");
  381. builder.AddAssemblyReferenceByName("System.Collections");
  382. builder.AddAssemblyReferenceByName("System.Linq");
  383. builder.AddUsing("System.Text.RegularExpressions");
  384. builder.AddUsing("System.Collections.Generic");
  385. builder.AddUsing("System.Linq");
  386. });
  387. result.Add(path?.TrimEnd(".vm"), tResult);
  388. }
  389. return (tableFieldList, result);
  390. }
  391. /// <summary>
  392. /// 增加菜单
  393. /// </summary>
  394. /// <param name="className"></param>
  395. /// <param name="busName"></param>
  396. /// <param name="pid"></param>
  397. /// <param name="menuIcon"></param>
  398. /// <param name="pagePath"></param>
  399. /// <param name="tableFieldList"></param>
  400. /// <returns></returns>
  401. private async Task AddMenu(string className, string busName, long pid, string menuIcon, string pagePath, List<CodeGenConfig> tableFieldList)
  402. {
  403. // 删除已存在的菜单
  404. var title = $"{busName}管理";
  405. await DeleteMenuTree(title, pid == 0 ? MenuTypeEnum.Dir : MenuTypeEnum.Menu);
  406. var parentMenuPath = "";
  407. var appId = _userManager.AppId;
  408. var appMenuList = new List<SysAppMenu>();
  409. var lowerClassName = className[..1].ToLower() + className[1..];
  410. if (pid == 0)
  411. {
  412. // 新增目录,并记录Id
  413. var dirMenu = new SysMenu { Pid=0, Title=title, Type=MenuTypeEnum.Dir, Icon="robot", Path="/" + className.ToLower(), Component="Layout" };
  414. pid = (await _db.Insertable(dirMenu).ExecuteReturnEntityAsync()).Id;
  415. appMenuList.Add(new SysAppMenu { AppId = appId, MenuId = pid });
  416. }
  417. else
  418. {
  419. var parentMenu = await _db.Queryable<SysMenu>().FirstAsync(u => u.Id == pid) ?? throw Oops.Oh(ErrorCodeEnum.D1505);
  420. parentMenuPath = parentMenu.Path;
  421. }
  422. // 新增菜单,并记录Id
  423. var rootMenu = new SysMenu { Pid=pid, Title=title, Type=MenuTypeEnum.Menu, Icon=menuIcon, Path=$"{parentMenuPath}/{className.ToLower()}", Component=$"/{pagePath}/{lowerClassName}/index" };
  424. pid = (await _db.Insertable(rootMenu).ExecuteReturnEntityAsync()).Id;
  425. appMenuList.Add(new SysAppMenu { AppId = appId, MenuId = pid });
  426. var orderNo = 100;
  427. var menuList = new List<SysMenu>
  428. {
  429. new() { Title="查询", Permission=$"{lowerClassName}:page", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10},
  430. new() { Title="详情", Permission=$"{lowerClassName}:detail", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10},
  431. new() { Title="增加", Permission=$"{lowerClassName}:add", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10},
  432. new() { Title="编辑", Permission=$"{lowerClassName}:update", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10},
  433. new() { Title="删除", Permission=$"{lowerClassName}:delete", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10},
  434. new() { Title="批量删除", Permission=$"{lowerClassName}:batchDelete", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10},
  435. new() { Title="设置状态", Permission=$"{lowerClassName}:setStatus", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10},
  436. new() { Title="打印", Permission=$"{lowerClassName}:print", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10},
  437. new() { Title="导入", Permission=$"{lowerClassName}:import", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10},
  438. new() { Title="导出", Permission=$"{lowerClassName}:export", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10}
  439. };
  440. if (tableFieldList.Any(u => u.EffectType is "ForeignKey" or "ApiTreeSelector" && (u.WhetherAddUpdate == "Y" || u.WhetherQuery == "Y")))
  441. menuList.Add(new SysMenu { Title="下拉列表数据", Permission=$"{lowerClassName}:dropdownData", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10});
  442. foreach (var column in tableFieldList.Where(u => u.EffectType == "Upload"))
  443. menuList.Add(new SysMenu { Title=$"上传{column.ColumnComment}", Permission=$"{lowerClassName}:upload{column.PropertyName}", Pid=pid, Type=MenuTypeEnum.Btn, OrderNo=orderNo+=10});
  444. await _db.Insertable(menuList).ExecuteCommandAsync();
  445. // 新增应用菜单关联
  446. appMenuList.AddRange(menuList.Select(u => new SysAppMenu { AppId = appId, MenuId = u.Id }));
  447. await _db.Insertable(appMenuList).ExecuteCommandAsync();
  448. }
  449. /// <summary>
  450. /// 根据菜单名称和类型删除关联的菜单树
  451. /// </summary>
  452. /// <param name="title"></param>
  453. /// <param name="type"></param>
  454. private async Task DeleteMenuTree(string title, MenuTypeEnum type)
  455. {
  456. var menuList = await _db.Queryable<SysMenu>().Where(u => u.Title == title && u.Type == type).ToListAsync() ?? new();
  457. foreach (var menu in menuList) await App.GetService<SysMenuService>().DeleteMenu(new DeleteMenuInput { Id = menu.Id });
  458. }
  459. /// <summary>
  460. /// 获取模板文件路径集合
  461. /// </summary>
  462. /// <returns></returns>
  463. private static List<string> GetTemplatePathList(SysCodeGen input)
  464. {
  465. if (input.GenerateType!.Substring(1, 1).Contains('1')) return new() { "index.vue.vm", "editDialog.vue.vm", "api.ts.vm" };
  466. if (input.GenerateType.Substring(1, 1).Contains('2')) return new() { "Service.cs.vm", "Input.cs.vm", "Output.cs.vm", "Dto.cs.vm" };
  467. return new() { "Service.cs.vm", "Input.cs.vm", "Output.cs.vm", "Dto.cs.vm", "index.vue.vm", "editDialog.vue.vm", "api.ts.vm" };
  468. }
  469. /// <summary>
  470. /// 获取模板文件路径集合
  471. /// </summary>
  472. /// <returns></returns>
  473. private static List<string> GetTemplatePathList() => new() { "Service.cs.vm", "Input.cs.vm", "Output.cs.vm", "Dto.cs.vm", "index.vue.vm", "editDialog.vue.vm", "api.ts.vm" };
  474. /// <summary>
  475. /// 设置生成文件路径
  476. /// </summary>
  477. /// <param name="input"></param>
  478. /// <returns></returns>
  479. private List<string> GetTargetPathList(SysCodeGen input)
  480. {
  481. //var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, _codeGenOptions.BackendApplicationNamespace, "Service", input.TableName);
  482. var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent!.FullName, input.NameSpace!, "Service", input.TableName!);
  483. var servicePath = Path.Combine(backendPath, input.TableName + "Service.cs");
  484. var inputPath = Path.Combine(backendPath, "Dto", input.TableName + "Input.cs");
  485. var outputPath = Path.Combine(backendPath, "Dto", input.TableName + "Output.cs");
  486. var viewPath = Path.Combine(backendPath, "Dto", input.TableName + "Dto.cs");
  487. var frontendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent!.Parent!.FullName, _codeGenOptions.FrontRootPath, "src", "views", input.PagePath!);
  488. var indexPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "index.vue");//
  489. var formModalPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "component", "editDialog.vue");
  490. var apiJsPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent!.Parent!.FullName, _codeGenOptions.FrontRootPath, "src", "api", input.PagePath, input.TableName[..1].ToLower() + input.TableName[1..] + ".ts");
  491. if (input.GenerateType!.Substring(1, 1).Contains('1'))
  492. {
  493. // 生成到本项目(前端)
  494. return new List<string>
  495. {
  496. indexPath,
  497. formModalPath,
  498. apiJsPath
  499. };
  500. }
  501. if (input.GenerateType.Substring(1, 1).Contains('2'))
  502. {
  503. // 生成到本项目(后端)
  504. return new List<string>
  505. {
  506. servicePath,
  507. inputPath,
  508. outputPath,
  509. viewPath,
  510. };
  511. }
  512. // 前后端同时生成到本项目
  513. return new List<string>
  514. {
  515. servicePath,
  516. inputPath,
  517. outputPath,
  518. viewPath,
  519. indexPath,
  520. formModalPath,
  521. apiJsPath
  522. };
  523. }
  524. /// <summary>
  525. /// 设置生成文件路径
  526. /// </summary>
  527. /// <param name="input"></param>
  528. /// <returns></returns>
  529. private List<string> GetZipPathList(SysCodeGen input)
  530. {
  531. var zipPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName!);
  532. //var backendPath = Path.Combine(zipPath, _codeGenOptions.BackendApplicationNamespace, "Service", input.TableName);
  533. var backendPath = Path.Combine(zipPath, input.NameSpace!, "Service", input.TableName);
  534. var servicePath = Path.Combine(backendPath, input.TableName + "Service.cs");
  535. var inputPath = Path.Combine(backendPath, "Dto", input.TableName + "Input.cs");
  536. var outputPath = Path.Combine(backendPath, "Dto", input.TableName + "Output.cs");
  537. var viewPath = Path.Combine(backendPath, "Dto", input.TableName + "Dto.cs");
  538. var frontendPath = Path.Combine(zipPath, _codeGenOptions.FrontRootPath, "src", "views", input.PagePath!);
  539. var indexPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "index.vue");
  540. var formModalPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "component", "editDialog.vue");
  541. var apiJsPath = Path.Combine(zipPath, _codeGenOptions.FrontRootPath, "src", "api", input.PagePath, input.TableName[..1].ToLower() + input.TableName[1..] + ".ts");
  542. if (input.GenerateType!.StartsWith("11"))
  543. {
  544. return new List<string>
  545. {
  546. indexPath,
  547. formModalPath,
  548. apiJsPath
  549. };
  550. }
  551. if (input.GenerateType.StartsWith("12"))
  552. {
  553. return new List<string>
  554. {
  555. servicePath,
  556. inputPath,
  557. outputPath,
  558. viewPath
  559. };
  560. }
  561. return new List<string>
  562. {
  563. servicePath,
  564. inputPath,
  565. outputPath,
  566. viewPath,
  567. indexPath,
  568. formModalPath,
  569. apiJsPath
  570. };
  571. }
  572. }