SysCodeGenService.cs 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853
  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 IViewEngine _viewEngine;
  17. private readonly CodeGenOptions _codeGenOptions;
  18. public SysCodeGenService(ISqlSugarClient db,
  19. SysCodeGenConfigService codeGenConfigService,
  20. IViewEngine viewEngine,
  21. IOptions<CodeGenOptions> codeGenOptions)
  22. {
  23. _db = db;
  24. _codeGenConfigService = codeGenConfigService;
  25. _viewEngine = viewEngine;
  26. _codeGenOptions = codeGenOptions.Value;
  27. }
  28. /// <summary>
  29. /// 获取代码生成分页列表 🔖
  30. /// </summary>
  31. /// <param name="input"></param>
  32. /// <returns></returns>
  33. [DisplayName("获取代码生成分页列表")]
  34. public async Task<SqlSugarPagedList<SysCodeGen>> Page(CodeGenInput input)
  35. {
  36. return await _db.Queryable<SysCodeGen>()
  37. .WhereIF(!string.IsNullOrWhiteSpace(input.TableName), u => u.TableName.Contains(input.TableName.Trim()))
  38. .WhereIF(!string.IsNullOrWhiteSpace(input.BusName), u => u.BusName.Contains(input.BusName.Trim()))
  39. .ToPagedListAsync(input.Page, input.PageSize);
  40. }
  41. /// <summary>
  42. /// 增加代码生成 🔖
  43. /// </summary>
  44. /// <param name="input"></param>
  45. /// <returns></returns>
  46. [ApiDescriptionSettings(Name = "Add"), HttpPost]
  47. [DisplayName("增加代码生成")]
  48. public async Task AddCodeGen(AddCodeGenInput input)
  49. {
  50. var isExist = await _db.Queryable<SysCodeGen>().Where(u => u.TableName == input.TableName).AnyAsync();
  51. if (isExist)
  52. throw Oops.Oh(ErrorCodeEnum.D1400);
  53. var codeGen = input.Adapt<SysCodeGen>();
  54. var newCodeGen = await _db.Insertable(codeGen).ExecuteReturnEntityAsync();
  55. // 增加配置表
  56. _codeGenConfigService.AddList(GetColumnList(input), newCodeGen);
  57. }
  58. /// <summary>
  59. /// 更新代码生成 🔖
  60. /// </summary>
  61. /// <param name="input"></param>
  62. /// <returns></returns>
  63. [ApiDescriptionSettings(Name = "Update"), HttpPost]
  64. [DisplayName("更新代码生成")]
  65. public async Task UpdateCodeGen(UpdateCodeGenInput input)
  66. {
  67. var isExist = await _db.Queryable<SysCodeGen>().AnyAsync(u => u.TableName == input.TableName && u.Id != input.Id);
  68. if (isExist)
  69. throw Oops.Oh(ErrorCodeEnum.D1400);
  70. var codeGen = input.Adapt<SysCodeGen>();
  71. await _db.Updateable(codeGen).ExecuteCommandAsync();
  72. // 更新配置表
  73. await _codeGenConfigService.DeleteCodeGenConfig(codeGen.Id);
  74. _codeGenConfigService.AddList(GetColumnList(input.Adapt<AddCodeGenInput>()), codeGen);
  75. }
  76. /// <summary>
  77. /// 删除代码生成 🔖
  78. /// </summary>
  79. /// <param name="inputs"></param>
  80. /// <returns></returns>
  81. [ApiDescriptionSettings(Name = "Delete"), HttpPost]
  82. [DisplayName("删除代码生成")]
  83. public async Task DeleteCodeGen(List<DeleteCodeGenInput> inputs)
  84. {
  85. if (inputs == null || inputs.Count < 1) return;
  86. var codeGenConfigTaskList = new List<Task>();
  87. inputs.ForEach(u =>
  88. {
  89. _db.Deleteable<SysCodeGen>().In(u.Id).ExecuteCommand();
  90. // 删除配置表
  91. codeGenConfigTaskList.Add(_codeGenConfigService.DeleteCodeGenConfig(u.Id));
  92. });
  93. await Task.WhenAll(codeGenConfigTaskList);
  94. }
  95. /// <summary>
  96. /// 获取代码生成详情 🔖
  97. /// </summary>
  98. /// <param name="input"></param>
  99. /// <returns></returns>
  100. [DisplayName("获取代码生成详情")]
  101. public async Task<SysCodeGen> GetDetail([FromQuery] QueryCodeGenInput input)
  102. {
  103. return await _db.Queryable<SysCodeGen>().SingleAsync(u => u.Id == input.Id);
  104. }
  105. /// <summary>
  106. /// 获取数据库库集合 🔖
  107. /// </summary>
  108. /// <returns></returns>
  109. [DisplayName("获取数据库库集合")]
  110. public async Task<List<DatabaseOutput>> GetDatabaseList()
  111. {
  112. var dbConfigs = App.GetOptions<DbConnectionOptions>().ConnectionConfigs;
  113. return await Task.FromResult(dbConfigs.Adapt<List<DatabaseOutput>>());
  114. }
  115. /// <summary>
  116. /// 获取数据库表(实体)集合 🔖
  117. /// </summary>
  118. /// <returns></returns>
  119. [DisplayName("获取数据库表(实体)集合")]
  120. public async Task<List<TableOutput>> GetTableList(string configId = SqlSugarConst.MainConfigId)
  121. {
  122. var provider = _db.AsTenant().GetConnectionScope(configId);
  123. var dbTableInfos = provider.DbMaintenance.GetTableInfoList(false); // 不能走缓存,否则切库不起作用
  124. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => configId.Equals(u.ConfigId));
  125. var dbTableNames = dbTableInfos.Select(u => u.Name.ToLower()).ToList();
  126. IEnumerable<EntityInfo> entityInfos = await GetEntityInfos();
  127. var tableOutputList = new List<TableOutput>();
  128. foreach (var item in entityInfos)
  129. {
  130. var table = dbTableInfos.FirstOrDefault(u => u.Name.ToLower() == (config.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(item.DbTableName) : item.DbTableName).ToLower());
  131. if (table == null) continue;
  132. tableOutputList.Add(new TableOutput
  133. {
  134. ConfigId = configId,
  135. EntityName = item.EntityName,
  136. TableName = table.Name,
  137. TableComment = item.TableDescription
  138. });
  139. }
  140. return tableOutputList;
  141. }
  142. /// <summary>
  143. /// 根据表名获取列集合 🔖
  144. /// </summary>
  145. /// <returns></returns>
  146. [DisplayName("根据表名获取列集合")]
  147. public List<ColumnOuput> GetColumnListByTableName([Required] string tableName, string configId = SqlSugarConst.MainConfigId)
  148. {
  149. // 切库---多库代码生成用
  150. var provider = _db.AsTenant().GetConnectionScope(configId);
  151. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == configId);
  152. // 获取实体类型属性
  153. var entityType = provider.DbMaintenance.GetTableInfoList(false).FirstOrDefault(u => u.Name == tableName);
  154. if (entityType == null) return null;
  155. var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)];
  156. // 按原始类型的顺序获取所有实体类型属性(不包含导航属性,会返回null)
  157. return provider.DbMaintenance.GetColumnInfosByTableName(entityType.Name).Select(u => new ColumnOuput
  158. {
  159. ColumnName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
  160. ColumnKey = u.IsPrimarykey.ToString(),
  161. DataType = u.DataType.ToString(),
  162. NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  163. ColumnComment = u.ColumnDescription
  164. }).ToList();
  165. }
  166. /// <summary>
  167. /// 获取数据表列(实体属性)集合
  168. /// </summary>
  169. /// <returns></returns>
  170. private List<ColumnOuput> GetColumnList([FromQuery] AddCodeGenInput input)
  171. {
  172. var entityType = GetEntityInfos().GetAwaiter().GetResult().FirstOrDefault(u => u.EntityName == input.TableName);
  173. if (entityType == null)
  174. return null;
  175. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == input.ConfigId);
  176. var dbTableName = config.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(entityType.DbTableName) : entityType.DbTableName;
  177. // 切库---多库代码生成用
  178. var provider = _db.AsTenant().GetConnectionScope(!string.IsNullOrEmpty(input.ConfigId) ? input.ConfigId : SqlSugarConst.MainConfigId);
  179. var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)];
  180. var columnInfos = provider.DbMaintenance.GetColumnInfosByTableName(dbTableName, false);
  181. var result = columnInfos.Select(u => new ColumnOuput
  182. {
  183. // 转下划线后的列名需要再转回来(暂时不转)
  184. //ColumnName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
  185. ColumnName = u.DbColumnName,
  186. ColumnLength = u.Length,
  187. IsPrimarykey = u.IsPrimarykey,
  188. IsNullable = u.IsNullable,
  189. ColumnKey = u.IsPrimarykey.ToString(),
  190. NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  191. DataType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  192. ColumnComment = string.IsNullOrWhiteSpace(u.ColumnDescription) ? u.DbColumnName : u.ColumnDescription
  193. }).ToList();
  194. // 获取实体的属性信息,赋值给PropertyName属性(CodeFirst模式应以PropertyName为实际使用名称)
  195. var entityProperties = entityType.Type.GetProperties();
  196. for (int i = result.Count - 1; i >= 0; i--)
  197. {
  198. var columnOutput = result[i];
  199. // 先找自定义字段名的,如果找不到就再找自动生成字段名的(并且过滤掉没有SugarColumn的属性)
  200. var propertyInfo = entityProperties.FirstOrDefault(u => (u.GetCustomAttribute<SugarColumn>()?.ColumnName ?? "").ToLower() == columnOutput.ColumnName.ToLower()) ??
  201. entityProperties.FirstOrDefault(u => u.GetCustomAttribute<SugarColumn>() != null && u.Name.ToLower() == (config.DbSettings.EnableUnderLine
  202. ? CodeGenUtil.CamelColumnName(columnOutput.ColumnName, entityBasePropertyNames).ToLower()
  203. : columnOutput.ColumnName.ToLower()));
  204. if (propertyInfo != null)
  205. {
  206. columnOutput.PropertyName = propertyInfo.Name;
  207. columnOutput.ColumnComment = propertyInfo.GetCustomAttribute<SugarColumn>().ColumnDescription;
  208. var propertyType = Nullable.GetUnderlyingType(propertyInfo.PropertyType);
  209. if (propertyInfo.PropertyType.IsEnum || propertyType != null)
  210. {
  211. columnOutput.DictTypeCode = (propertyType ?? propertyInfo.PropertyType).Name;
  212. }
  213. else
  214. {
  215. var dict = propertyInfo.GetCustomAttribute<DictAttribute>();
  216. if (dict != null) columnOutput.DictTypeCode = dict.DictTypeCode;
  217. }
  218. }
  219. else
  220. {
  221. result.RemoveAt(i); // 移除没有定义此属性的字段
  222. }
  223. }
  224. return result;
  225. }
  226. /// <summary>
  227. /// 获取库表信息
  228. /// </summary>
  229. /// <returns></returns>
  230. private async Task<IEnumerable<EntityInfo>> GetEntityInfos()
  231. {
  232. var entityInfos = new List<EntityInfo>();
  233. var type = typeof(SugarTable);
  234. var types = new List<Type>();
  235. if (_codeGenOptions.EntityAssemblyNames != null)
  236. {
  237. var assemblies = AppDomain.CurrentDomain.GetAssemblies();
  238. foreach (var assembly in assemblies)
  239. {
  240. var assemblyName = assembly.GetName().Name;
  241. if (_codeGenOptions.EntityAssemblyNames.Contains(assemblyName) || _codeGenOptions.EntityAssemblyNames.Any(name => assemblyName.Contains(name)))
  242. {
  243. Assembly asm = Assembly.Load(assemblyName);
  244. types.AddRange(asm.GetExportedTypes().ToList());
  245. }
  246. }
  247. }
  248. bool IsMyAttribute(Attribute[] o)
  249. {
  250. foreach (Attribute a in o)
  251. {
  252. if (a.GetType() == type)
  253. return true;
  254. }
  255. return false;
  256. }
  257. Type[] cosType = types.Where(o =>
  258. {
  259. return IsMyAttribute(Attribute.GetCustomAttributes(o, true));
  260. }
  261. ).ToArray();
  262. foreach (var ct in cosType)
  263. {
  264. var sugarAttribute = ct.GetCustomAttributes(type, true)?.FirstOrDefault();
  265. var des = ct.GetCustomAttributes(typeof(DescriptionAttribute), true);
  266. var description = "";
  267. if (des.Length > 0)
  268. {
  269. description = ((DescriptionAttribute)des[0]).Description;
  270. }
  271. entityInfos.Add(new EntityInfo()
  272. {
  273. EntityName = ct.Name,
  274. DbTableName = sugarAttribute == null ? ct.Name : ((SugarTable)sugarAttribute).TableName,
  275. TableDescription = sugarAttribute == null ? description : ((SugarTable)sugarAttribute).TableDescription,
  276. Type = ct
  277. });
  278. }
  279. return await Task.FromResult(entityInfos);
  280. }
  281. /// <summary>
  282. /// 获取程序保存位置 🔖
  283. /// </summary>
  284. /// <returns></returns>
  285. [DisplayName("获取程序保存位置")]
  286. public List<string> GetApplicationNamespaces()
  287. {
  288. return _codeGenOptions.BackendApplicationNamespaces;
  289. }
  290. /// <summary>
  291. /// 代码生成到本地 🔖
  292. /// </summary>
  293. /// <returns></returns>
  294. [DisplayName("代码生成到本地")]
  295. public async Task<dynamic> RunLocal(SysCodeGen input)
  296. {
  297. if (string.IsNullOrEmpty(input.GenerateType))
  298. input.GenerateType = "200";
  299. // 先删除该表已生成的菜单列表
  300. List<string> targetPathList;
  301. var zipPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName);
  302. if (input.GenerateType.StartsWith('1'))
  303. {
  304. targetPathList = GetZipPathList(input);
  305. if (Directory.Exists(zipPath))
  306. Directory.Delete(zipPath, true);
  307. }
  308. else
  309. targetPathList = GetTargetPathList(input);
  310. var tableFieldList = await _codeGenConfigService.GetList(new CodeGenConfig() { CodeGenId = input.Id }); // 字段集合
  311. var queryWhetherList = tableFieldList.Where(u => u.QueryWhether == YesNoEnum.Y.ToString()).ToList(); // 前端查询集合
  312. var joinTableList = tableFieldList.Where(u => u.EffectType == "Upload" || u.EffectType == "fk" || u.EffectType == "ApiTreeSelect").ToList(); // 需要连表查询的字段
  313. (string joinTableNames, string lowerJoinTableNames) = GetJoinTableStr(joinTableList); // 获取连表的实体名和别名
  314. var data = new CustomViewEngine(_db)
  315. {
  316. ConfigId = input.ConfigId,
  317. AuthorName = input.AuthorName,
  318. BusName = input.BusName,
  319. NameSpace = input.NameSpace,
  320. ClassName = input.TableName,
  321. PagePath = input.PagePath,
  322. ProjectLastName = input.NameSpace.Split('.').Last(),
  323. QueryWhetherList = queryWhetherList,
  324. TableField = tableFieldList,
  325. IsJoinTable = joinTableList.Count > 0,
  326. IsUpload = joinTableList.Where(u => u.EffectType == "Upload").Any(),
  327. PrintType = input.PrintType,
  328. PrintName = input.PrintName,
  329. };
  330. // 模板目录
  331. var templatePathList = GetTemplatePathList(input);
  332. var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "template");
  333. for (var i = 0; i < templatePathList.Count; i++)
  334. {
  335. var templateFilePath = Path.Combine(templatePath, templatePathList[i]);
  336. if (!File.Exists(templateFilePath)) continue;
  337. var tContent = File.ReadAllText(templateFilePath);
  338. var tResult = await _viewEngine.RunCompileFromCachedAsync(tContent, data, builderAction: builder =>
  339. {
  340. builder.AddAssemblyReferenceByName("System.Linq");
  341. builder.AddAssemblyReferenceByName("System.Collections");
  342. builder.AddAssemblyReferenceByName("System.Text.RegularExpressions");
  343. builder.AddUsing("System.Text.RegularExpressions");
  344. builder.AddUsing("System.Collections.Generic");
  345. builder.AddUsing("System.Linq");
  346. });
  347. var dirPath = new DirectoryInfo(targetPathList[i]).Parent.FullName;
  348. if (!Directory.Exists(dirPath))
  349. Directory.CreateDirectory(dirPath);
  350. File.WriteAllText(targetPathList[i], tResult, Encoding.UTF8);
  351. }
  352. if (input.GenerateMenu)
  353. await AddMenu(input.TableName, input.BusName, input.MenuPid ?? 0, input.MenuIcon, input.PagePath, tableFieldList);
  354. // 非ZIP压缩返回空
  355. if (!input.GenerateType.StartsWith('1'))
  356. return null;
  357. else
  358. {
  359. string downloadPath = zipPath + ".zip";
  360. // 判断是否存在同名称文件
  361. if (File.Exists(downloadPath))
  362. File.Delete(downloadPath);
  363. ZipFile.CreateFromDirectory(zipPath, downloadPath);
  364. return new { url = $"{App.HttpContext.Request.Scheme}://{App.HttpContext.Request.Host.Value}/codeGen/{input.TableName}.zip" };
  365. }
  366. }
  367. /// <summary>
  368. /// 获取代码生成预览 🔖
  369. /// </summary>
  370. /// <returns></returns>
  371. [DisplayName("获取代码生成预览")]
  372. public async Task<Dictionary<string, string>> Preview(SysCodeGen input)
  373. {
  374. var tableFieldList = await _codeGenConfigService.GetList(new CodeGenConfig() { CodeGenId = input.Id }); // 字段集合
  375. var queryWhetherList = tableFieldList.Where(u => u.QueryWhether == YesNoEnum.Y.ToString()).ToList(); // 前端查询集合
  376. var joinTableList = tableFieldList.Where(u => u.EffectType == "Upload" || u.EffectType == "fk" || u.EffectType == "ApiTreeSelect").ToList(); // 需要连表查询的字段
  377. (string joinTableNames, string lowerJoinTableNames) = GetJoinTableStr(joinTableList); // 获取连表的实体名和别名
  378. var data = new CustomViewEngine(_db)
  379. {
  380. ConfigId = input.ConfigId,
  381. AuthorName = input.AuthorName,
  382. BusName = input.BusName,
  383. NameSpace = input.NameSpace,
  384. ClassName = input.TableName,
  385. PagePath = input.PagePath,
  386. ProjectLastName = input.NameSpace.Split('.').Last(),
  387. QueryWhetherList = queryWhetherList,
  388. TableField = tableFieldList,
  389. IsJoinTable = joinTableList.Count > 0,
  390. IsUpload = joinTableList.Where(u => u.EffectType == "Upload").Any(),
  391. PrintType = input.PrintType,
  392. PrintName = input.PrintName,
  393. };
  394. // 获取模板文件并替换
  395. var templatePathList = GetTemplatePathList();
  396. var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "template");
  397. var result = new Dictionary<string, string>();
  398. for (var i = 0; i < templatePathList.Count; i++)
  399. {
  400. var templateFilePath = Path.Combine(templatePath, templatePathList[i]);
  401. if (!File.Exists(templateFilePath)) continue;
  402. var tContent = File.ReadAllText(templateFilePath);
  403. var tResult = await _viewEngine.RunCompileFromCachedAsync(tContent, data, builderAction: builder =>
  404. {
  405. builder.AddAssemblyReferenceByName("System.Linq");
  406. builder.AddAssemblyReferenceByName("System.Collections");
  407. builder.AddAssemblyReferenceByName("System.Text.RegularExpressions");
  408. builder.AddUsing("System.Text.RegularExpressions");
  409. builder.AddUsing("System.Collections.Generic");
  410. builder.AddUsing("System.Linq");
  411. });
  412. result.Add(templatePathList[i]?.TrimEnd(".vm"), tResult);
  413. }
  414. return result;
  415. }
  416. /// <summary>
  417. /// 获取连表的实体名和别名
  418. /// </summary>
  419. /// <param name="configs"></param>
  420. /// <returns></returns>
  421. private static (string, string) GetJoinTableStr(List<CodeGenConfig> configs)
  422. {
  423. var uploads = configs.Where(u => u.EffectType == "Upload").ToList();
  424. var fks = configs.Where(u => u.EffectType == "fk").ToList();
  425. string str = ""; // <Order, OrderItem, Custom>
  426. string lowerStr = ""; // (o, i, c)
  427. foreach (var item in uploads)
  428. {
  429. lowerStr += "sysFile_FK_" + item.LowerPropertyName + ",";
  430. str += "SysFile,";
  431. }
  432. foreach (var item in fks)
  433. {
  434. lowerStr += item.LowerFkEntityName + "_FK_" + item.LowerFkColumnName + ",";
  435. str += item.FkEntityName + ",";
  436. }
  437. return (str.TrimEnd(','), lowerStr.TrimEnd(','));
  438. }
  439. /// <summary>
  440. /// 增加菜单
  441. /// </summary>
  442. /// <param name="className"></param>
  443. /// <param name="busName"></param>
  444. /// <param name="pid"></param>
  445. /// <param name="menuIcon"></param>
  446. /// <param name="pagePath"></param>
  447. /// <param name="tableFieldList"></param>
  448. /// <returns></returns>
  449. private async Task AddMenu(string className, string busName, long pid, string menuIcon, string pagePath, List<CodeGenConfig> tableFieldList)
  450. {
  451. var pPath = string.Empty;
  452. // 若 pid=0 为顶级则创建菜单目录
  453. if (pid == 0)
  454. {
  455. // 目录
  456. var menuType0 = new SysMenu
  457. {
  458. Pid = 0,
  459. Title = busName + "管理",
  460. Type = MenuTypeEnum.Dir,
  461. Icon = "robot",
  462. Path = "/" + className.ToLower(),
  463. Component = "Layout",
  464. };
  465. // 若先前存在则删除本级和下级
  466. var menuList0 = await _db.Queryable<SysMenu>().Where(u => u.Title == menuType0.Title && u.Type == menuType0.Type).ToListAsync();
  467. if (menuList0.Count > 0)
  468. {
  469. var listIds = menuList0.Select(u => u.Id).ToList();
  470. var childlistIds = new List<long>();
  471. foreach (var item in listIds)
  472. {
  473. var childlist = await _db.Queryable<SysMenu>().ToChildListAsync(u => u.Pid, item);
  474. childlistIds.AddRange(childlist.Select(u => u.Id).ToList());
  475. }
  476. listIds.AddRange(childlistIds);
  477. await _db.Deleteable<SysMenu>().Where(u => listIds.Contains(u.Id)).ExecuteCommandAsync();
  478. await _db.Deleteable<SysRoleMenu>().Where(u => listIds.Contains(u.MenuId)).ExecuteCommandAsync();
  479. }
  480. pid = (await _db.Insertable(menuType0).ExecuteReturnEntityAsync()).Id;
  481. }
  482. else
  483. {
  484. var pMenu = await _db.Queryable<SysMenu>().FirstAsync(u => u.Id == pid) ?? throw Oops.Oh(ErrorCodeEnum.D1505);
  485. pPath = pMenu.Path;
  486. }
  487. // 菜单
  488. var menuType = new SysMenu
  489. {
  490. Pid = pid,
  491. Title = busName + "管理",
  492. Name = className[..1].ToLower() + className[1..],
  493. Type = MenuTypeEnum.Menu,
  494. Icon = menuIcon,
  495. Path = pPath + "/" + className.ToLower(),
  496. Component = "/" + pagePath + "/" + className[..1].ToLower() + className[1..] + "/index",
  497. };
  498. // 若先前存在则删除本级和下级
  499. var menuListCurrent = await _db.Queryable<SysMenu>().Where(u => u.Title == menuType.Title && u.Type == menuType.Type).ToListAsync();
  500. if (menuListCurrent.Count > 0)
  501. {
  502. var listIds = menuListCurrent.Select(u => u.Id).ToList();
  503. var childListIds = new List<long>();
  504. foreach (var item in listIds)
  505. {
  506. var childList = await _db.Queryable<SysMenu>().ToChildListAsync(u => u.Pid, item);
  507. childListIds.AddRange(childList.Select(u => u.Id).ToList());
  508. }
  509. listIds.AddRange(childListIds);
  510. await _db.Deleteable<SysMenu>().Where(u => listIds.Contains(u.Id)).ExecuteCommandAsync();
  511. await _db.Deleteable<SysRoleMenu>().Where(u => listIds.Contains(u.MenuId)).ExecuteCommandAsync();
  512. }
  513. var menuPid = (await _db.Insertable(menuType).ExecuteReturnEntityAsync()).Id;
  514. int menuOrder = 100;
  515. // 按钮-page
  516. var menuTypePage = new SysMenu
  517. {
  518. Pid = menuPid,
  519. Title = "查询",
  520. Type = MenuTypeEnum.Btn,
  521. Permission = className[..1].ToLower() + className[1..] + ":page",
  522. OrderNo = menuOrder
  523. };
  524. menuOrder += 10;
  525. // 按钮-detail
  526. var menuTypeDetail = new SysMenu
  527. {
  528. Pid = menuPid,
  529. Title = "详情",
  530. Type = MenuTypeEnum.Btn,
  531. Permission = className[..1].ToLower() + className[1..] + ":detail",
  532. OrderNo = menuOrder
  533. };
  534. menuOrder += 10;
  535. // 按钮-add
  536. var menuTypeAdd = new SysMenu
  537. {
  538. Pid = menuPid,
  539. Title = "增加",
  540. Type = MenuTypeEnum.Btn,
  541. Permission = className[..1].ToLower() + className[1..] + ":add",
  542. OrderNo = menuOrder
  543. };
  544. menuOrder += 10;
  545. // 按钮-delete
  546. var menuTypeDelete = new SysMenu
  547. {
  548. Pid = menuPid,
  549. Title = "删除",
  550. Type = MenuTypeEnum.Btn,
  551. Permission = className[..1].ToLower() + className[1..] + ":delete",
  552. OrderNo = menuOrder
  553. };
  554. menuOrder += 10;
  555. // 按钮-update
  556. var menuTypeUpdate = new SysMenu
  557. {
  558. Pid = menuPid,
  559. Title = "编辑",
  560. Type = MenuTypeEnum.Btn,
  561. Permission = className[..1].ToLower() + className[1..] + ":update",
  562. OrderNo = menuOrder
  563. };
  564. menuOrder += 10;
  565. // 按钮-Status
  566. var menuTypeStatus = new SysMenu
  567. {
  568. Pid = menuPid,
  569. Title = "状态",
  570. Type = MenuTypeEnum.Btn,
  571. Permission = className[..1].ToLower() + className[1..] + ":setStatus",
  572. OrderNo = menuOrder
  573. };
  574. menuOrder += 10;
  575. // 按钮-batchDelete
  576. var menuTypeBatchDelete = new SysMenu
  577. {
  578. Pid = menuPid,
  579. Title = "批量删除",
  580. Type = MenuTypeEnum.Btn,
  581. Permission = className[..1].ToLower() + className[1..] + ":batchDelete",
  582. OrderNo = menuOrder
  583. };
  584. menuOrder += 10;
  585. // 按钮-print
  586. var menuTypePrint = new SysMenu
  587. {
  588. Pid = menuPid,
  589. Title = "打印",
  590. Type = MenuTypeEnum.Btn,
  591. Permission = className[..1].ToLower() + className[1..] + ":print",
  592. OrderNo = menuOrder
  593. };
  594. menuOrder += 10;
  595. // 按钮-import
  596. var menuTypeImport = new SysMenu
  597. {
  598. Pid = menuPid,
  599. Title = "导入",
  600. Type = MenuTypeEnum.Btn,
  601. Permission = className[..1].ToLower() + className[1..] + ":import",
  602. OrderNo = menuOrder
  603. };
  604. menuOrder += 10;
  605. // 按钮-export
  606. var menuTypeExport = new SysMenu
  607. {
  608. Pid = menuPid,
  609. Title = "导出",
  610. Type = MenuTypeEnum.Btn,
  611. Permission = className[..1].ToLower() + className[1..] + ":export",
  612. OrderNo = menuOrder
  613. };
  614. menuOrder += 10;
  615. var menuList = new List<SysMenu> { menuTypePage, menuTypeDetail, menuTypeAdd, menuTypeStatus, menuTypeDelete, menuTypeBatchDelete, menuTypeUpdate, menuTypePrint, menuTypeImport, menuTypeExport };
  616. // 加入fk、Upload、ApiTreeSelect 等接口的权限
  617. // 在生成表格时,有些字段只是查询时显示,不需要填写(WhetherAddUpdate),所以这些字段没必要生成相应接口
  618. var fkTableList = tableFieldList.Where(u => u.EffectType == "fk" && (u.WhetherAddUpdate == "Y" || u.QueryWhether == "Y")).ToList();
  619. foreach (var @column in fkTableList)
  620. {
  621. var menuType1 = new SysMenu
  622. {
  623. Pid = menuPid,
  624. Title = "外键" + @column.ColumnName,
  625. Type = MenuTypeEnum.Btn,
  626. Permission = className[..1].ToLower() + className[1..] + ":" + column.FkEntityName + column.ColumnName + "Dropdown",
  627. OrderNo = menuOrder
  628. };
  629. menuOrder += 10;
  630. menuList.Add(menuType1);
  631. }
  632. var treeSelectTableList = tableFieldList.Where(u => u.EffectType == "ApiTreeSelect").ToList();
  633. foreach (var @column in treeSelectTableList)
  634. {
  635. var menuType1 = new SysMenu
  636. {
  637. Pid = menuPid,
  638. Title = "树型" + @column.ColumnName,
  639. Type = MenuTypeEnum.Btn,
  640. Permission = className[..1].ToLower() + className[1..] + ":" + column.FkEntityName + "Tree",
  641. OrderNo = menuOrder
  642. };
  643. menuOrder += 10;
  644. menuList.Add(menuType1);
  645. }
  646. var uploadTableList = tableFieldList.Where(u => u.EffectType == "Upload").ToList();
  647. foreach (var @column in uploadTableList)
  648. {
  649. var menuType1 = new SysMenu
  650. {
  651. Pid = menuPid,
  652. Title = "上传" + @column.ColumnName,
  653. Type = MenuTypeEnum.Btn,
  654. Permission = className[..1].ToLower() + className[1..] + ":Upload" + column.ColumnName,
  655. OrderNo = menuOrder
  656. };
  657. menuOrder += 10;
  658. menuList.Add(menuType1);
  659. }
  660. await _db.Insertable(menuList).ExecuteCommandAsync();
  661. }
  662. /// <summary>
  663. /// 获取模板文件路径集合
  664. /// </summary>
  665. /// <returns></returns>
  666. private static List<string> GetTemplatePathList(SysCodeGen input)
  667. {
  668. if (input.GenerateType.Substring(1, 1).Contains('1'))
  669. {
  670. return new() { "index.vue.vm", "editDialog.vue.vm", "manage.js.vm" };
  671. }
  672. else if (input.GenerateType.Substring(1, 1).Contains('2'))
  673. {
  674. return new() { "Service.cs.vm", "Input.cs.vm", "Output.cs.vm", "Dto.cs.vm" };
  675. }
  676. else
  677. {
  678. return new() { "Service.cs.vm", "Input.cs.vm", "Output.cs.vm", "Dto.cs.vm", "index.vue.vm", "editDialog.vue.vm", "manage.js.vm" };
  679. }
  680. }
  681. /// <summary>
  682. /// 获取模板文件路径集合
  683. /// </summary>
  684. /// <returns></returns>
  685. private static List<string> GetTemplatePathList() => new() { "Service.cs.vm", "Input.cs.vm", "Output.cs.vm", "Dto.cs.vm", "index.vue.vm", "editDialog.vue.vm", "manage.js.vm" };
  686. /// <summary>
  687. /// 设置生成文件路径
  688. /// </summary>
  689. /// <param name="input"></param>
  690. /// <returns></returns>
  691. private List<string> GetTargetPathList(SysCodeGen input)
  692. {
  693. //var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, _codeGenOptions.BackendApplicationNamespace, "Service", input.TableName);
  694. var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, input.NameSpace, "Service", input.TableName);
  695. var servicePath = Path.Combine(backendPath, input.TableName + "Service.cs");
  696. var inputPath = Path.Combine(backendPath, "Dto", input.TableName + "Input.cs");
  697. var outputPath = Path.Combine(backendPath, "Dto", input.TableName + "Output.cs");
  698. var viewPath = Path.Combine(backendPath, "Dto", input.TableName + "Dto.cs");
  699. var frontendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.Parent.FullName, _codeGenOptions.FrontRootPath, "src", "views", input.PagePath);
  700. var indexPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "index.vue");//
  701. var formModalPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "component", "editDialog.vue");
  702. 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");
  703. if (input.GenerateType.Substring(1, 1).Contains('1'))
  704. {
  705. // 生成到本项目(前端)
  706. return new List<string>()
  707. {
  708. indexPath,
  709. formModalPath,
  710. apiJsPath
  711. };
  712. }
  713. else if (input.GenerateType.Substring(1, 1).Contains('2'))
  714. {
  715. // 生成到本项目(后端)
  716. return new List<string>()
  717. {
  718. servicePath,
  719. inputPath,
  720. outputPath,
  721. viewPath,
  722. };
  723. }
  724. else
  725. {
  726. // 前后端同时生成到本项目
  727. return new List<string>()
  728. {
  729. servicePath,
  730. inputPath,
  731. outputPath,
  732. viewPath,
  733. indexPath,
  734. formModalPath,
  735. apiJsPath
  736. };
  737. }
  738. }
  739. /// <summary>
  740. /// 设置生成文件路径
  741. /// </summary>
  742. /// <param name="input"></param>
  743. /// <returns></returns>
  744. private List<string> GetZipPathList(SysCodeGen input)
  745. {
  746. var zipPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName);
  747. //var backendPath = Path.Combine(zipPath, _codeGenOptions.BackendApplicationNamespace, "Service", input.TableName);
  748. var backendPath = Path.Combine(zipPath, input.NameSpace, "Service", input.TableName);
  749. var servicePath = Path.Combine(backendPath, input.TableName + "Service.cs");
  750. var inputPath = Path.Combine(backendPath, "Dto", input.TableName + "Input.cs");
  751. var outputPath = Path.Combine(backendPath, "Dto", input.TableName + "Output.cs");
  752. var viewPath = Path.Combine(backendPath, "Dto", input.TableName + "Dto.cs");
  753. var frontendPath = Path.Combine(zipPath, _codeGenOptions.FrontRootPath, "src", "views", input.PagePath);
  754. var indexPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "index.vue");
  755. var formModalPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "component", "editDialog.vue");
  756. var apiJsPath = Path.Combine(zipPath, _codeGenOptions.FrontRootPath, "src", "api", input.PagePath, input.TableName[..1].ToLower() + input.TableName[1..] + ".ts");
  757. if (input.GenerateType.StartsWith("11"))
  758. {
  759. return new List<string>()
  760. {
  761. indexPath,
  762. formModalPath,
  763. apiJsPath
  764. };
  765. }
  766. else if (input.GenerateType.StartsWith("12"))
  767. {
  768. return new List<string>()
  769. {
  770. servicePath,
  771. inputPath,
  772. outputPath,
  773. viewPath
  774. };
  775. }
  776. else
  777. {
  778. return new List<string>()
  779. {
  780. servicePath,
  781. inputPath,
  782. outputPath,
  783. viewPath,
  784. indexPath,
  785. formModalPath,
  786. apiJsPath
  787. };
  788. }
  789. }
  790. }