SysCodeGenService.cs 35 KB

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