SysCodeGenService.cs 31 KB

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