SysCodeGenService.cs 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710
  1. // 麻省理工学院许可证
  2. //
  3. // 版权所有 (c) 2021-2023 zuohuaijun,大名科技(天津)有限公司 联系电话/微信:18020030720 QQ:515096995
  4. //
  5. // 特此免费授予获得本软件的任何人以处理本软件的权利,但须遵守以下条件:在所有副本或重要部分的软件中必须包括上述版权声明和本许可声明。
  6. //
  7. // 软件按“原样”提供,不提供任何形式的明示或暗示的保证,包括但不限于对适销性、适用性和非侵权的保证。
  8. // 在任何情况下,作者或版权持有人均不对任何索赔、损害或其他责任负责,无论是因合同、侵权或其他方式引起的,与软件或其使用或其他交易有关。
  9. using System.IO.Compression;
  10. namespace Admin.NET.Core.Service;
  11. /// <summary>
  12. /// 系统代码生成器服务
  13. /// </summary>
  14. [ApiDescriptionSettings(Order = 270)]
  15. public class SysCodeGenService : IDynamicApiController, ITransient
  16. {
  17. private readonly ISqlSugarClient _db;
  18. private readonly SysCodeGenConfigService _codeGenConfigService;
  19. private readonly IViewEngine _viewEngine;
  20. private readonly CodeGenOptions _codeGenOptions;
  21. public SysCodeGenService(ISqlSugarClient db,
  22. SysCodeGenConfigService codeGenConfigService,
  23. IViewEngine viewEngine,
  24. IOptions<CodeGenOptions> codeGenOptions)
  25. {
  26. _db = db;
  27. _codeGenConfigService = codeGenConfigService;
  28. _viewEngine = viewEngine;
  29. _codeGenOptions = codeGenOptions.Value;
  30. }
  31. /// <summary>
  32. /// 获取代码生成分页列表
  33. /// </summary>
  34. /// <param name="input"></param>
  35. /// <returns></returns>
  36. [DisplayName("获取代码生成分页列表")]
  37. public async Task<SqlSugarPagedList<SysCodeGen>> Page(CodeGenInput input)
  38. {
  39. return await _db.Queryable<SysCodeGen>()
  40. .WhereIF(!string.IsNullOrWhiteSpace(input.TableName), u => u.TableName.Contains(input.TableName.Trim()))
  41. .WhereIF(!string.IsNullOrWhiteSpace(input.BusName), u => u.BusName.Contains(input.BusName.Trim()))
  42. .ToPagedListAsync(input.Page, input.PageSize);
  43. }
  44. /// <summary>
  45. /// 增加代码生成
  46. /// </summary>
  47. /// <param name="input"></param>
  48. /// <returns></returns>
  49. [ApiDescriptionSettings(Name = "Add"), HttpPost]
  50. [DisplayName("增加代码生成")]
  51. public async Task AddCodeGen(AddCodeGenInput input)
  52. {
  53. var isExist = await _db.Queryable<SysCodeGen>().Where(u => u.TableName == input.TableName).AnyAsync();
  54. if (isExist)
  55. throw Oops.Oh(ErrorCodeEnum.D1400);
  56. var codeGen = input.Adapt<SysCodeGen>();
  57. var newCodeGen = await _db.Insertable(codeGen).ExecuteReturnEntityAsync();
  58. // 加入配置表中
  59. _codeGenConfigService.AddList(GetColumnList(input), newCodeGen);
  60. }
  61. /// <summary>
  62. /// 更新代码生成
  63. /// </summary>
  64. /// <param name="input"></param>
  65. /// <returns></returns>
  66. [ApiDescriptionSettings(Name = "Update"), HttpPost]
  67. [DisplayName("更新代码生成")]
  68. public async Task UpdateCodeGen(UpdateCodeGenInput input)
  69. {
  70. var isExist = await _db.Queryable<SysCodeGen>().AnyAsync(u => u.TableName == input.TableName && u.Id != input.Id);
  71. if (isExist)
  72. throw Oops.Oh(ErrorCodeEnum.D1400);
  73. await _db.Updateable(input.Adapt<SysCodeGen>()).ExecuteCommandAsync();
  74. }
  75. /// <summary>
  76. /// 删除代码生成
  77. /// </summary>
  78. /// <param name="inputs"></param>
  79. /// <returns></returns>
  80. [ApiDescriptionSettings(Name = "Delete"), HttpPost]
  81. [DisplayName("删除代码生成")]
  82. public async Task DeleteCodeGen(List<DeleteCodeGenInput> inputs)
  83. {
  84. if (inputs == null || inputs.Count < 1) return;
  85. var codeGenConfigTaskList = new List<Task>();
  86. inputs.ForEach(u =>
  87. {
  88. _db.Deleteable<SysCodeGen>().In(u.Id).ExecuteCommand();
  89. // 删除配置表中
  90. codeGenConfigTaskList.Add(_codeGenConfigService.DeleteCodeGenConfig(u.Id));
  91. });
  92. await Task.WhenAll(codeGenConfigTaskList);
  93. }
  94. /// <summary>
  95. /// 获取代码生成详情
  96. /// </summary>
  97. /// <param name="input"></param>
  98. /// <returns></returns>
  99. [DisplayName("获取代码生成详情")]
  100. public async Task<SysCodeGen> GetDetail([FromQuery] QueryCodeGenInput input)
  101. {
  102. return await _db.Queryable<SysCodeGen>().SingleAsync(m => m.Id == input.Id);
  103. }
  104. /// <summary>
  105. /// 获取数据库库集合
  106. /// </summary>
  107. /// <returns></returns>
  108. [DisplayName("获取数据库库集合")]
  109. public async Task<List<DatabaseOutput>> GetDatabaseList()
  110. {
  111. var dbCongigs = App.GetOptions<DbConnectionOptions>().ConnectionConfigs;
  112. return await Task.FromResult(dbCongigs.Adapt<List<DatabaseOutput>>());
  113. }
  114. /// <summary>
  115. /// 获取数据库表(实体)集合
  116. /// </summary>
  117. /// <returns></returns>
  118. [DisplayName("获取数据库表(实体)集合")]
  119. public async Task<List<TableOutput>> GetTableList(string configId = SqlSugarConst.ConfigId)
  120. {
  121. var provider = _db.AsTenant().GetConnectionScope(configId);
  122. var dbTableInfos = provider.DbMaintenance.GetTableInfoList(false); // 不能走缓存,否则切库不起作用
  123. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId == configId);
  124. var dbTableNames = dbTableInfos.Select(u => u.Name.ToLower()).ToList();
  125. IEnumerable<EntityInfo> entityInfos = await GetEntityInfos();
  126. var tableOutputList = new List<TableOutput>();
  127. foreach (var item in entityInfos)
  128. {
  129. var table = dbTableInfos.FirstOrDefault(x => x.Name.ToLower() == (config.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(item.DbTableName) : item.DbTableName).ToLower());
  130. if (table == null) continue;
  131. tableOutputList.Add(new TableOutput
  132. {
  133. ConfigId = configId,
  134. EntityName = item.EntityName,
  135. TableName = table.Name,
  136. TableComment = item.TableDescription
  137. });
  138. }
  139. return tableOutputList;
  140. //return entityInfos.Where(u => dbTableNames.Contains(config.EnableUnderLine ? UtilMethods.ToUnderLine(u.DbTableName) : u.DbTableName.ToLower())).Select(u => new TableOutput()
  141. //{
  142. // ConfigId = configId,
  143. // EntityName = u.EntityName,
  144. // TableName = config.EnableUnderLine ? UtilMethods.ToUnderLine(u.DbTableName) : u.DbTableName,
  145. // TableComment = u.TableDescription
  146. //}).ToList();
  147. }
  148. /// <summary>
  149. /// 根据表名获取列集合
  150. /// </summary>
  151. /// <returns></returns>
  152. [DisplayName("根据表名获取列集合")]
  153. public List<ColumnOuput> GetColumnListByTableName([Required] string tableName, string configId = SqlSugarConst.ConfigId)
  154. {
  155. // 切库---多库代码生成用
  156. var provider = _db.AsTenant().GetConnectionScope(configId);
  157. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId == configId);
  158. // 获取实体类型属性
  159. var entityType = provider.DbMaintenance.GetTableInfoList(false).FirstOrDefault(u => u.Name == tableName);
  160. if (entityType == null) return null;
  161. var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)];
  162. // 按原始类型的顺序获取所有实体类型属性(不包含导航属性,会返回null)
  163. return provider.DbMaintenance.GetColumnInfosByTableName(entityType.Name).Select(u => new ColumnOuput
  164. {
  165. ColumnName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
  166. ColumnKey = u.IsPrimarykey.ToString(),
  167. DataType = u.DataType.ToString(),
  168. NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  169. ColumnComment = u.ColumnDescription
  170. }).ToList();
  171. }
  172. /// <summary>
  173. /// 获取数据表列(实体属性)集合
  174. /// </summary>
  175. /// <returns></returns>
  176. private List<ColumnOuput> GetColumnList([FromQuery] AddCodeGenInput input)
  177. {
  178. var entityType = GetEntityInfos().GetAwaiter().GetResult().FirstOrDefault(m => m.EntityName == input.TableName);
  179. if (entityType == null)
  180. return null;
  181. // 切库---多库代码生成用
  182. var provider = _db.AsTenant().GetConnectionScope(!string.IsNullOrEmpty(input.ConfigId) ? input.ConfigId : SqlSugarConst.ConfigId);
  183. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId == input.ConfigId);
  184. var dbTableName = config.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(entityType.DbTableName) : entityType.DbTableName;
  185. var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)];
  186. var columnInfos = provider.DbMaintenance.GetColumnInfosByTableName(dbTableName, false);
  187. var result = columnInfos.Select(u => new ColumnOuput
  188. {
  189. // 转下划线后的列名需要再转回来,2023/9/11,后面会根据字段名获取对应Property,所以这里不用转换了,以后面转换。
  190. //ColumnName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
  191. ColumnName = u.DbColumnName,
  192. ColumnLength = u.Length,
  193. IsPrimarykey = u.IsPrimarykey,
  194. IsNullable = u.IsNullable,
  195. ColumnKey = u.IsPrimarykey.ToString(),
  196. NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  197. DataType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  198. ColumnComment = string.IsNullOrWhiteSpace(u.ColumnDescription) ? u.DbColumnName : u.ColumnDescription
  199. }).ToList();
  200. // 获取实体的属性信息,并赋值给PropertyName属性(因为CodeFirst模式,应该以PropertyName为实际使用名称)
  201. var entityProperties = entityType.Type.GetProperties();
  202. for (int i= result.Count-1;i>=0;i--)
  203. {
  204. var columnOutput = result[i];
  205. var propertyName = entityProperties.FirstOrDefault(p =>
  206. p.Name == (config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(columnOutput.ColumnName, entityBasePropertyNames) : columnOutput.ColumnName) ||
  207. p.GetCustomAttribute<SugarColumn>()?.ColumnName?.ToLower() == columnOutput.ColumnName.ToLower()).Name;
  208. if (propertyName != null)
  209. columnOutput.PropertyName = propertyName;
  210. else
  211. result.RemoveAt(i);// 如果实体中都没有定义这个属性,那么显示字段也没有意义,所以这里移除它
  212. }
  213. return result;
  214. }
  215. /// <summary>
  216. /// 获取库表信息
  217. /// </summary>
  218. /// <returns></returns>
  219. private async Task<IEnumerable<EntityInfo>> GetEntityInfos()
  220. {
  221. var entityInfos = new List<EntityInfo>();
  222. var type = typeof(SugarTable);
  223. var types = new List<Type>();
  224. if (_codeGenOptions.EntityAssemblyNames != null)
  225. {
  226. foreach (var assemblyName in _codeGenOptions.EntityAssemblyNames)
  227. {
  228. Assembly asm = Assembly.Load(assemblyName);
  229. types.AddRange(asm.GetExportedTypes().ToList());
  230. }
  231. }
  232. bool IsMyAttribute(Attribute[] o)
  233. {
  234. foreach (Attribute a in o)
  235. {
  236. if (a.GetType() == type)
  237. return true;
  238. }
  239. return false;
  240. }
  241. Type[] cosType = types.Where(o =>
  242. {
  243. return IsMyAttribute(Attribute.GetCustomAttributes(o, true));
  244. }
  245. ).ToArray();
  246. foreach (var c in cosType)
  247. {
  248. var sugarAttribute = c.GetCustomAttributes(type, true)?.FirstOrDefault();
  249. var des = c.GetCustomAttributes(typeof(DescriptionAttribute), true);
  250. var description = "";
  251. if (des.Length > 0)
  252. {
  253. description = ((DescriptionAttribute)des[0]).Description;
  254. }
  255. entityInfos.Add(new EntityInfo()
  256. {
  257. EntityName = c.Name,
  258. DbTableName = sugarAttribute == null ? c.Name : ((SugarTable)sugarAttribute).TableName,
  259. TableDescription = description,
  260. Type = c
  261. });
  262. }
  263. return await Task.FromResult(entityInfos);
  264. }
  265. /// <summary>
  266. /// 代码生成到本地
  267. /// </summary>
  268. /// <returns></returns>
  269. [DisplayName("代码生成到本地")]
  270. public async Task<dynamic> RunLocal(SysCodeGen input)
  271. {
  272. if (string.IsNullOrEmpty(input.GenerateType))
  273. input.GenerateType = "200";
  274. // 先删除该表已生成的菜单列表
  275. var templatePathList = GetTemplatePathList(input);
  276. List<string> targetPathList;
  277. var zipPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName);
  278. if (input.GenerateType.StartsWith('1'))
  279. {
  280. targetPathList = GetZipPathList(input);
  281. if (Directory.Exists(zipPath))
  282. Directory.Delete(zipPath, true);
  283. }
  284. else
  285. targetPathList = GetTargetPathList(input);
  286. var tableFieldList = await _codeGenConfigService.GetList(new CodeGenConfig() { CodeGenId = input.Id }); // 字段集合
  287. var queryWhetherList = tableFieldList.Where(u => u.QueryWhether == YesNoEnum.Y.ToString()).ToList(); // 前端查询集合
  288. var joinTableList = tableFieldList.Where(u => u.EffectType == "Upload" || u.EffectType == "fk" || u.EffectType == "ApiTreeSelect").ToList(); // 需要连表查询的字段
  289. (string joinTableNames, string lowerJoinTableNames) = GetJoinTableStr(joinTableList); // 获取连表的实体名和别名
  290. var data = new CustomViewEngine(_db)
  291. {
  292. ConfigId = input.ConfigId,
  293. AuthorName = input.AuthorName,
  294. BusName = input.BusName,
  295. NameSpace = input.NameSpace,
  296. ClassName = input.TableName,
  297. ProjectLastName = input.NameSpace.Split('.').Last(),
  298. QueryWhetherList = queryWhetherList,
  299. TableField = tableFieldList,
  300. IsJoinTable = joinTableList.Count > 0,
  301. IsUpload = joinTableList.Where(u => u.EffectType == "Upload").Any(),
  302. };
  303. for (var i = 0; i < templatePathList.Count; i++)
  304. {
  305. var tContent = File.ReadAllText(templatePathList[i]);
  306. var tResult = _viewEngine.RunCompile<CustomViewEngine>(tContent, data, builderAction: builder =>
  307. {
  308. builder.AddAssemblyReferenceByName("System.Linq");
  309. builder.AddAssemblyReferenceByName("System.Collections");
  310. builder.AddUsing("System.Collections.Generic");
  311. builder.AddUsing("System.Linq");
  312. });
  313. var dirPath = new DirectoryInfo(targetPathList[i]).Parent.FullName;
  314. if (!Directory.Exists(dirPath))
  315. Directory.CreateDirectory(dirPath);
  316. File.WriteAllText(targetPathList[i], tResult, Encoding.UTF8);
  317. }
  318. await AddMenu(input.TableName, input.BusName, input.MenuPid, tableFieldList);
  319. // 非ZIP压缩返回空
  320. if (!input.GenerateType.StartsWith('1'))
  321. return null;
  322. else
  323. {
  324. string downloadPath = zipPath + ".zip";
  325. // 判断是否存在同名称文件
  326. if (File.Exists(downloadPath))
  327. File.Delete(downloadPath);
  328. ZipFile.CreateFromDirectory(zipPath, downloadPath);
  329. return new { url = $"{CommonUtil.GetLocalhost()}/CodeGen/{input.TableName}.zip" };
  330. }
  331. }
  332. /// <summary>
  333. /// 获取连表的实体名和别名
  334. /// </summary>
  335. /// <param name="configs"></param>
  336. /// <returns></returns>
  337. private (string, string) GetJoinTableStr(List<CodeGenConfig> configs)
  338. {
  339. var uploads = configs.Where(u => u.EffectType == "Upload").ToList();
  340. var fks = configs.Where(u => u.EffectType == "fk").ToList();
  341. string str = ""; // <Order, OrderItem, Custom>
  342. string lowerStr = ""; //(o, i, c)
  343. foreach (var item in uploads)
  344. {
  345. lowerStr += "sysFile_FK_" + item.LowerPropertyName + ",";
  346. str += "SysFile,";
  347. }
  348. foreach (var item in fks)
  349. {
  350. lowerStr += item.LowerFkEntityName + "_FK_" + item.LowerFkColumnName + ",";
  351. str += item.FkEntityName + ",";
  352. }
  353. return (str.TrimEnd(','), lowerStr.TrimEnd(','));
  354. }
  355. /// <summary>
  356. /// 增加菜单
  357. /// </summary>
  358. /// <param name="className"></param>
  359. /// <param name="busName"></param>
  360. /// <param name="pid"></param>
  361. /// <returns></returns>
  362. private async Task AddMenu(string className, string busName, long pid, List<CodeGenConfig> tableFieldList)
  363. {
  364. var pPath = string.Empty;
  365. // 若 pid=0 为顶级则创建菜单目录
  366. if (pid == 0)
  367. {
  368. // 目录
  369. var menuType0 = new SysMenu
  370. {
  371. Pid = 0,
  372. Title = busName + "管理",
  373. Type = MenuTypeEnum.Dir,
  374. Icon = "robot",
  375. Path = "/" + className.ToLower(),
  376. Component = "LAYOUT",
  377. };
  378. // 若先前存在则删除本级和下级
  379. var menuList0 = await _db.Queryable<SysMenu>().Where(e => e.Title == menuType0.Title && e.Type == menuType0.Type).ToListAsync();
  380. if (menuList0.Count > 0)
  381. {
  382. var listIds = menuList0.Select(f => f.Id).ToList();
  383. var childlistIds = new List<long>();
  384. foreach (var item in listIds)
  385. {
  386. var childlist = await _db.Queryable<SysMenu>().ToChildListAsync(u => u.Pid, item);
  387. childlistIds.AddRange(childlist.Select(f => f.Id).ToList());
  388. }
  389. listIds.AddRange(childlistIds);
  390. await _db.Deleteable<SysMenu>().Where(e => listIds.Contains(e.Id)).ExecuteCommandAsync();
  391. await _db.Deleteable<SysRoleMenu>().Where(e => listIds.Contains(e.MenuId)).ExecuteCommandAsync();
  392. }
  393. pid = (await _db.Insertable(menuType0).ExecuteReturnEntityAsync()).Id;
  394. }
  395. else
  396. {
  397. var pMenu = await _db.Queryable<SysMenu>().FirstAsync(e => e.Id == pid) ?? throw Oops.Oh(ErrorCodeEnum.D1505);
  398. pPath = pMenu.Path;
  399. }
  400. // 菜单
  401. var menuType1 = new SysMenu
  402. {
  403. Pid = pid,
  404. Title = busName + "管理",
  405. Name = className[..1].ToLower() + className[1..],
  406. Type = MenuTypeEnum.Menu,
  407. Path = pPath + "/" + className.ToLower(),
  408. Component = "/main/" + className[..1].ToLower() + className[1..] + "/index",
  409. };
  410. // 若先前存在则删除本级和下级
  411. var menuList1 = await _db.Queryable<SysMenu>().Where(e => e.Title == menuType1.Title && e.Type == menuType1.Type).ToListAsync();
  412. if (menuList1.Count > 0)
  413. {
  414. var listIds = menuList1.Select(f => f.Id).ToList();
  415. var childlistIds = new List<long>();
  416. foreach (var item in listIds)
  417. {
  418. var childlist = await _db.Queryable<SysMenu>().ToChildListAsync(u => u.Pid, item);
  419. childlistIds.AddRange(childlist.Select(f => f.Id).ToList());
  420. }
  421. listIds.AddRange(childlistIds);
  422. await _db.Deleteable<SysMenu>().Where(e => listIds.Contains(e.Id)).ExecuteCommandAsync();
  423. await _db.Deleteable<SysRoleMenu>().Where(e => listIds.Contains(e.MenuId)).ExecuteCommandAsync();
  424. }
  425. var pid1 = (await _db.Insertable(menuType1).ExecuteReturnEntityAsync()).Id;
  426. int menuOrder = 101;
  427. // 按钮-page
  428. var menuType2 = new SysMenu
  429. {
  430. Pid = pid1,
  431. Title = "查询",
  432. Type = MenuTypeEnum.Btn,
  433. Permission = className[..1].ToLower() + className[1..] + ":page",
  434. OrderNo = menuOrder
  435. };
  436. menuOrder += 3;
  437. // 按钮-detail
  438. var menuType2_1 = new SysMenu
  439. {
  440. Pid = pid1,
  441. Title = "详情",
  442. Type = MenuTypeEnum.Btn,
  443. Permission = className[..1].ToLower() + className[1..] + ":detail",
  444. OrderNo = menuOrder
  445. };
  446. menuOrder += 3;
  447. // 按钮-add
  448. var menuType2_2 = new SysMenu
  449. {
  450. Pid = pid1,
  451. Title = "增加",
  452. Type = MenuTypeEnum.Btn,
  453. Permission = className[..1].ToLower() + className[1..] + ":add",
  454. OrderNo = menuOrder
  455. };
  456. menuOrder += 3;
  457. // 按钮-delete
  458. var menuType2_3 = new SysMenu
  459. {
  460. Pid = pid1,
  461. Title = "删除",
  462. Type = MenuTypeEnum.Btn,
  463. Permission = className[..1].ToLower() + className[1..] + ":delete",
  464. OrderNo = menuOrder
  465. };
  466. menuOrder += 3;
  467. // 按钮-edit
  468. var menuType2_4 = new SysMenu
  469. {
  470. Pid = pid1,
  471. Title = "编辑",
  472. Type = MenuTypeEnum.Btn,
  473. Permission = className[..1].ToLower() + className[1..] + ":edit",
  474. OrderNo = menuOrder
  475. };
  476. menuOrder += 3;
  477. var menuList = new List<SysMenu>() { menuType2, menuType2_1, menuType2_2, menuType2_3, menuType2_4 };
  478. // 加入fk、Upload、ApiTreeSelect 等接口的权限
  479. var fkTableList = tableFieldList.Where(u => u.EffectType == "fk").ToList();
  480. foreach (var @column in fkTableList)
  481. {
  482. var menuType = new SysMenu
  483. {
  484. Pid = pid1,
  485. Title = "外键" + @column.ColumnName,
  486. Type = MenuTypeEnum.Btn,
  487. Permission = className[..1].ToLower() + className[1..] + ":" + column.FkEntityName + column.ColumnName + "Dropdown",
  488. OrderNo = menuOrder
  489. };
  490. menuOrder += 3;
  491. menuList.Add(menuType);
  492. }
  493. var treeSelectTableList = tableFieldList.Where(u => u.EffectType == "ApiTreeSelect").ToList();
  494. foreach (var @column in treeSelectTableList)
  495. {
  496. var menuType = new SysMenu
  497. {
  498. Pid = pid1,
  499. Title = "树型" + @column.ColumnName,
  500. Type = MenuTypeEnum.Btn,
  501. Permission = className[..1].ToLower() + className[1..] + ":" + column.FkEntityName + "Tree",
  502. OrderNo = menuOrder
  503. };
  504. menuOrder += 3;
  505. menuList.Add(menuType);
  506. }
  507. var uploadTableList = tableFieldList.Where(u => u.EffectType == "Upload").ToList();
  508. foreach (var @column in uploadTableList)
  509. {
  510. var menuType = new SysMenu
  511. {
  512. Pid = pid1,
  513. Title = "上传" + @column.ColumnName,
  514. Type = MenuTypeEnum.Btn,
  515. Permission = className[..1].ToLower() + className[1..] + ":Upload" + column.ColumnName,
  516. OrderNo = menuOrder
  517. };
  518. menuOrder += 3;
  519. menuList.Add(menuType);
  520. }
  521. await _db.Insertable(menuList).ExecuteCommandAsync();
  522. }
  523. /// <summary>
  524. /// 获取模板文件路径集合
  525. /// </summary>
  526. /// <returns></returns>
  527. private static List<string> GetTemplatePathList(SysCodeGen input)
  528. {
  529. var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "Template");
  530. if (input.GenerateType.Substring(1, 1).Contains('1'))
  531. {
  532. return new List<string>()
  533. {
  534. Path.Combine(templatePath , "index.vue.vm"),
  535. Path.Combine(templatePath , "editDialog.vue.vm"),
  536. Path.Combine(templatePath , "manage.js.vm"),
  537. };
  538. }
  539. else if (input.GenerateType.Substring(1, 1).Contains('2'))
  540. {
  541. return new List<string>()
  542. {
  543. Path.Combine(templatePath , "Service.cs.vm"),
  544. Path.Combine(templatePath , "Input.cs.vm"),
  545. Path.Combine(templatePath , "Output.cs.vm"),
  546. Path.Combine(templatePath , "Dto.cs.vm"),
  547. };
  548. }
  549. else
  550. {
  551. return new List<string>()
  552. {
  553. Path.Combine(templatePath , "Service.cs.vm"),
  554. Path.Combine(templatePath , "Input.cs.vm"),
  555. Path.Combine(templatePath , "Output.cs.vm"),
  556. Path.Combine(templatePath , "Dto.cs.vm"),
  557. Path.Combine(templatePath , "index.vue.vm"),
  558. Path.Combine(templatePath , "editDialog.vue.vm"),
  559. Path.Combine(templatePath , "manage.js.vm"),
  560. };
  561. }
  562. }
  563. /// <summary>
  564. /// 获取模板文件路径集合
  565. /// </summary>
  566. /// <returns></returns>
  567. private static List<string> GetTemplatePathList()
  568. {
  569. var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "Template");
  570. return new List<string>()
  571. {
  572. Path.Combine(templatePath , "Service.cs.vm"),
  573. Path.Combine(templatePath , "Input.cs.vm"),
  574. Path.Combine(templatePath , "Output.cs.vm"),
  575. Path.Combine(templatePath , "Dto.cs.vm"),
  576. Path.Combine(templatePath , "index.vue.vm"),
  577. Path.Combine(templatePath , "editDialog.vue.vm"),
  578. Path.Combine(templatePath , "manage.js.vm"),
  579. };
  580. }
  581. /// <summary>
  582. /// 设置生成文件路径
  583. /// </summary>
  584. /// <param name="input"></param>
  585. /// <returns></returns>
  586. private List<string> GetTargetPathList(SysCodeGen input)
  587. {
  588. //var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, _codeGenOptions.BackendApplicationNamespace, "Service", input.TableName);
  589. var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, input.NameSpace, "Service", input.TableName);
  590. var servicePath = Path.Combine(backendPath, input.TableName + "Service.cs");
  591. var inputPath = Path.Combine(backendPath, "Dto", input.TableName + "Input.cs");
  592. var outputPath = Path.Combine(backendPath, "Dto", input.TableName + "Output.cs");
  593. var viewPath = Path.Combine(backendPath, "Dto", input.TableName + "Dto.cs");
  594. var frontendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.Parent.FullName, _codeGenOptions.FrontRootPath, "src", "views", "main");
  595. var indexPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "index.vue");//
  596. var formModalPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "component", "editDialog.vue");
  597. var apiJsPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.Parent.FullName, _codeGenOptions.FrontRootPath, "src", "api", "main", input.TableName[..1].ToLower() + input.TableName[1..] + ".ts");
  598. return new List<string>()
  599. {
  600. servicePath,
  601. inputPath,
  602. outputPath,
  603. viewPath,
  604. indexPath,
  605. formModalPath,
  606. apiJsPath
  607. };
  608. }
  609. /// <summary>
  610. /// 设置生成文件路径
  611. /// </summary>
  612. /// <param name="input"></param>
  613. /// <returns></returns>
  614. private List<string> GetZipPathList(SysCodeGen input)
  615. {
  616. var zipPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName);
  617. //var backendPath = Path.Combine(zipPath, _codeGenOptions.BackendApplicationNamespace, "Service", input.TableName);
  618. var backendPath = Path.Combine(zipPath, input.NameSpace, "Service", input.TableName);
  619. var servicePath = Path.Combine(backendPath, input.TableName + "Service.cs");
  620. var inputPath = Path.Combine(backendPath, "Dto", input.TableName + "Input.cs");
  621. var outputPath = Path.Combine(backendPath, "Dto", input.TableName + "Output.cs");
  622. var viewPath = Path.Combine(backendPath, "Dto", input.TableName + "Dto.cs");
  623. var frontendPath = Path.Combine(zipPath, _codeGenOptions.FrontRootPath, "src", "views", "main");
  624. var indexPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "index.vue");
  625. var formModalPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "component", "editDialog.vue");
  626. var apiJsPath = Path.Combine(zipPath, _codeGenOptions.FrontRootPath, "src", "api", "main", input.TableName[..1].ToLower() + input.TableName[1..] + ".ts");
  627. if (input.GenerateType.StartsWith("11"))
  628. {
  629. return new List<string>()
  630. {
  631. indexPath,
  632. formModalPath,
  633. apiJsPath
  634. };
  635. }
  636. else if (input.GenerateType.StartsWith("12"))
  637. {
  638. return new List<string>()
  639. {
  640. servicePath,
  641. inputPath,
  642. outputPath,
  643. viewPath
  644. };
  645. }
  646. else
  647. {
  648. return new List<string>()
  649. {
  650. servicePath,
  651. inputPath,
  652. outputPath,
  653. viewPath,
  654. indexPath,
  655. formModalPath,
  656. apiJsPath
  657. };
  658. }
  659. }
  660. }