SysCodeGenService.cs 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703
  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. }
  141. /// <summary>
  142. /// 根据表名获取列集合
  143. /// </summary>
  144. /// <returns></returns>
  145. [DisplayName("根据表名获取列集合")]
  146. public List<ColumnOuput> GetColumnListByTableName([Required] string tableName, string configId = SqlSugarConst.ConfigId)
  147. {
  148. // 切库---多库代码生成用
  149. var provider = _db.AsTenant().GetConnectionScope(configId);
  150. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId == configId);
  151. // 获取实体类型属性
  152. var entityType = provider.DbMaintenance.GetTableInfoList(false).FirstOrDefault(u => u.Name == tableName);
  153. if (entityType == null) return null;
  154. var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)];
  155. // 按原始类型的顺序获取所有实体类型属性(不包含导航属性,会返回null)
  156. return provider.DbMaintenance.GetColumnInfosByTableName(entityType.Name).Select(u => new ColumnOuput
  157. {
  158. ColumnName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
  159. ColumnKey = u.IsPrimarykey.ToString(),
  160. DataType = u.DataType.ToString(),
  161. NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  162. ColumnComment = u.ColumnDescription
  163. }).ToList();
  164. }
  165. /// <summary>
  166. /// 获取数据表列(实体属性)集合
  167. /// </summary>
  168. /// <returns></returns>
  169. private List<ColumnOuput> GetColumnList([FromQuery] AddCodeGenInput input)
  170. {
  171. var entityType = GetEntityInfos().GetAwaiter().GetResult().FirstOrDefault(m => m.EntityName == input.TableName);
  172. if (entityType == null)
  173. return null;
  174. // 切库---多库代码生成用
  175. var provider = _db.AsTenant().GetConnectionScope(!string.IsNullOrEmpty(input.ConfigId) ? input.ConfigId : SqlSugarConst.ConfigId);
  176. var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId == input.ConfigId);
  177. var dbTableName = config.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(entityType.DbTableName) : entityType.DbTableName;
  178. var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)];
  179. var columnInfos = provider.DbMaintenance.GetColumnInfosByTableName(dbTableName, false);
  180. var result = columnInfos.Select(u => new ColumnOuput
  181. {
  182. // 转下划线后的列名需要再转回来(暂时不转)
  183. //ColumnName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
  184. ColumnName = u.DbColumnName,
  185. ColumnLength = u.Length,
  186. IsPrimarykey = u.IsPrimarykey,
  187. IsNullable = u.IsNullable,
  188. ColumnKey = u.IsPrimarykey.ToString(),
  189. NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  190. DataType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
  191. ColumnComment = string.IsNullOrWhiteSpace(u.ColumnDescription) ? u.DbColumnName : u.ColumnDescription
  192. }).ToList();
  193. // 获取实体的属性信息,赋值给PropertyName属性(CodeFirst模式应以PropertyName为实际使用名称)
  194. var entityProperties = entityType.Type.GetProperties();
  195. for (int i = result.Count - 1; i >= 0; i--)
  196. {
  197. var columnOutput = result[i];
  198. var propertyName = entityProperties.FirstOrDefault(p =>
  199. p.Name == (config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(columnOutput.ColumnName, entityBasePropertyNames) : columnOutput.ColumnName) ||
  200. p.GetCustomAttribute<SugarColumn>()?.ColumnName?.ToLower() == columnOutput.ColumnName.ToLower()).Name;
  201. if (propertyName != null)
  202. columnOutput.PropertyName = propertyName;
  203. else
  204. result.RemoveAt(i); // 移除没有定义此属性的字段
  205. }
  206. return result;
  207. }
  208. /// <summary>
  209. /// 获取库表信息
  210. /// </summary>
  211. /// <returns></returns>
  212. private async Task<IEnumerable<EntityInfo>> GetEntityInfos()
  213. {
  214. var entityInfos = new List<EntityInfo>();
  215. var type = typeof(SugarTable);
  216. var types = new List<Type>();
  217. if (_codeGenOptions.EntityAssemblyNames != null)
  218. {
  219. foreach (var assemblyName in _codeGenOptions.EntityAssemblyNames)
  220. {
  221. Assembly asm = Assembly.Load(assemblyName);
  222. types.AddRange(asm.GetExportedTypes().ToList());
  223. }
  224. }
  225. bool IsMyAttribute(Attribute[] o)
  226. {
  227. foreach (Attribute a in o)
  228. {
  229. if (a.GetType() == type)
  230. return true;
  231. }
  232. return false;
  233. }
  234. Type[] cosType = types.Where(o =>
  235. {
  236. return IsMyAttribute(Attribute.GetCustomAttributes(o, true));
  237. }
  238. ).ToArray();
  239. foreach (var c in cosType)
  240. {
  241. var sugarAttribute = c.GetCustomAttributes(type, true)?.FirstOrDefault();
  242. var des = c.GetCustomAttributes(typeof(DescriptionAttribute), true);
  243. var description = "";
  244. if (des.Length > 0)
  245. {
  246. description = ((DescriptionAttribute)des[0]).Description;
  247. }
  248. entityInfos.Add(new EntityInfo()
  249. {
  250. EntityName = c.Name,
  251. DbTableName = sugarAttribute == null ? c.Name : ((SugarTable)sugarAttribute).TableName,
  252. TableDescription = description,
  253. Type = c
  254. });
  255. }
  256. return await Task.FromResult(entityInfos);
  257. }
  258. /// <summary>
  259. /// 代码生成到本地
  260. /// </summary>
  261. /// <returns></returns>
  262. [DisplayName("代码生成到本地")]
  263. public async Task<dynamic> RunLocal(SysCodeGen input)
  264. {
  265. if (string.IsNullOrEmpty(input.GenerateType))
  266. input.GenerateType = "200";
  267. // 先删除该表已生成的菜单列表
  268. var templatePathList = GetTemplatePathList(input);
  269. List<string> targetPathList;
  270. var zipPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName);
  271. if (input.GenerateType.StartsWith('1'))
  272. {
  273. targetPathList = GetZipPathList(input);
  274. if (Directory.Exists(zipPath))
  275. Directory.Delete(zipPath, true);
  276. }
  277. else
  278. targetPathList = GetTargetPathList(input);
  279. var tableFieldList = await _codeGenConfigService.GetList(new CodeGenConfig() { CodeGenId = input.Id }); // 字段集合
  280. var queryWhetherList = tableFieldList.Where(u => u.QueryWhether == YesNoEnum.Y.ToString()).ToList(); // 前端查询集合
  281. var joinTableList = tableFieldList.Where(u => u.EffectType == "Upload" || u.EffectType == "fk" || u.EffectType == "ApiTreeSelect").ToList(); // 需要连表查询的字段
  282. (string joinTableNames, string lowerJoinTableNames) = GetJoinTableStr(joinTableList); // 获取连表的实体名和别名
  283. var data = new CustomViewEngine(_db)
  284. {
  285. ConfigId = input.ConfigId,
  286. AuthorName = input.AuthorName,
  287. BusName = input.BusName,
  288. NameSpace = input.NameSpace,
  289. ClassName = input.TableName,
  290. ProjectLastName = input.NameSpace.Split('.').Last(),
  291. QueryWhetherList = queryWhetherList,
  292. TableField = tableFieldList,
  293. IsJoinTable = joinTableList.Count > 0,
  294. IsUpload = joinTableList.Where(u => u.EffectType == "Upload").Any(),
  295. };
  296. for (var i = 0; i < templatePathList.Count; i++)
  297. {
  298. var tContent = File.ReadAllText(templatePathList[i]);
  299. var tResult = _viewEngine.RunCompile<CustomViewEngine>(tContent, data, builderAction: builder =>
  300. {
  301. builder.AddAssemblyReferenceByName("System.Linq");
  302. builder.AddAssemblyReferenceByName("System.Collections");
  303. builder.AddUsing("System.Collections.Generic");
  304. builder.AddUsing("System.Linq");
  305. });
  306. var dirPath = new DirectoryInfo(targetPathList[i]).Parent.FullName;
  307. if (!Directory.Exists(dirPath))
  308. Directory.CreateDirectory(dirPath);
  309. File.WriteAllText(targetPathList[i], tResult, Encoding.UTF8);
  310. }
  311. await AddMenu(input.TableName, input.BusName, input.MenuPid, tableFieldList);
  312. // 非ZIP压缩返回空
  313. if (!input.GenerateType.StartsWith('1'))
  314. return null;
  315. else
  316. {
  317. string downloadPath = zipPath + ".zip";
  318. // 判断是否存在同名称文件
  319. if (File.Exists(downloadPath))
  320. File.Delete(downloadPath);
  321. ZipFile.CreateFromDirectory(zipPath, downloadPath);
  322. return new { url = $"{CommonUtil.GetLocalhost()}/CodeGen/{input.TableName}.zip" };
  323. }
  324. }
  325. /// <summary>
  326. /// 获取连表的实体名和别名
  327. /// </summary>
  328. /// <param name="configs"></param>
  329. /// <returns></returns>
  330. private (string, string) GetJoinTableStr(List<CodeGenConfig> configs)
  331. {
  332. var uploads = configs.Where(u => u.EffectType == "Upload").ToList();
  333. var fks = configs.Where(u => u.EffectType == "fk").ToList();
  334. string str = ""; // <Order, OrderItem, Custom>
  335. string lowerStr = ""; //(o, i, c)
  336. foreach (var item in uploads)
  337. {
  338. lowerStr += "sysFile_FK_" + item.LowerPropertyName + ",";
  339. str += "SysFile,";
  340. }
  341. foreach (var item in fks)
  342. {
  343. lowerStr += item.LowerFkEntityName + "_FK_" + item.LowerFkColumnName + ",";
  344. str += item.FkEntityName + ",";
  345. }
  346. return (str.TrimEnd(','), lowerStr.TrimEnd(','));
  347. }
  348. /// <summary>
  349. /// 增加菜单
  350. /// </summary>
  351. /// <param name="className"></param>
  352. /// <param name="busName"></param>
  353. /// <param name="pid"></param>
  354. /// <returns></returns>
  355. private async Task AddMenu(string className, string busName, long pid, List<CodeGenConfig> tableFieldList)
  356. {
  357. var pPath = string.Empty;
  358. // 若 pid=0 为顶级则创建菜单目录
  359. if (pid == 0)
  360. {
  361. // 目录
  362. var menuType0 = new SysMenu
  363. {
  364. Pid = 0,
  365. Title = busName + "管理",
  366. Type = MenuTypeEnum.Dir,
  367. Icon = "robot",
  368. Path = "/" + className.ToLower(),
  369. Component = "LAYOUT",
  370. };
  371. // 若先前存在则删除本级和下级
  372. var menuList0 = await _db.Queryable<SysMenu>().Where(e => e.Title == menuType0.Title && e.Type == menuType0.Type).ToListAsync();
  373. if (menuList0.Count > 0)
  374. {
  375. var listIds = menuList0.Select(f => f.Id).ToList();
  376. var childlistIds = new List<long>();
  377. foreach (var item in listIds)
  378. {
  379. var childlist = await _db.Queryable<SysMenu>().ToChildListAsync(u => u.Pid, item);
  380. childlistIds.AddRange(childlist.Select(f => f.Id).ToList());
  381. }
  382. listIds.AddRange(childlistIds);
  383. await _db.Deleteable<SysMenu>().Where(e => listIds.Contains(e.Id)).ExecuteCommandAsync();
  384. await _db.Deleteable<SysRoleMenu>().Where(e => listIds.Contains(e.MenuId)).ExecuteCommandAsync();
  385. }
  386. pid = (await _db.Insertable(menuType0).ExecuteReturnEntityAsync()).Id;
  387. }
  388. else
  389. {
  390. var pMenu = await _db.Queryable<SysMenu>().FirstAsync(e => e.Id == pid) ?? throw Oops.Oh(ErrorCodeEnum.D1505);
  391. pPath = pMenu.Path;
  392. }
  393. // 菜单
  394. var menuType1 = new SysMenu
  395. {
  396. Pid = pid,
  397. Title = busName + "管理",
  398. Name = className[..1].ToLower() + className[1..],
  399. Type = MenuTypeEnum.Menu,
  400. Path = pPath + "/" + className.ToLower(),
  401. Component = "/main/" + className[..1].ToLower() + className[1..] + "/index",
  402. };
  403. // 若先前存在则删除本级和下级
  404. var menuList1 = await _db.Queryable<SysMenu>().Where(e => e.Title == menuType1.Title && e.Type == menuType1.Type).ToListAsync();
  405. if (menuList1.Count > 0)
  406. {
  407. var listIds = menuList1.Select(f => f.Id).ToList();
  408. var childlistIds = new List<long>();
  409. foreach (var item in listIds)
  410. {
  411. var childlist = await _db.Queryable<SysMenu>().ToChildListAsync(u => u.Pid, item);
  412. childlistIds.AddRange(childlist.Select(f => f.Id).ToList());
  413. }
  414. listIds.AddRange(childlistIds);
  415. await _db.Deleteable<SysMenu>().Where(e => listIds.Contains(e.Id)).ExecuteCommandAsync();
  416. await _db.Deleteable<SysRoleMenu>().Where(e => listIds.Contains(e.MenuId)).ExecuteCommandAsync();
  417. }
  418. var pid1 = (await _db.Insertable(menuType1).ExecuteReturnEntityAsync()).Id;
  419. int menuOrder = 101;
  420. // 按钮-page
  421. var menuType2 = new SysMenu
  422. {
  423. Pid = pid1,
  424. Title = "查询",
  425. Type = MenuTypeEnum.Btn,
  426. Permission = className[..1].ToLower() + className[1..] + ":page",
  427. OrderNo = menuOrder
  428. };
  429. menuOrder += 3;
  430. // 按钮-detail
  431. var menuType2_1 = new SysMenu
  432. {
  433. Pid = pid1,
  434. Title = "详情",
  435. Type = MenuTypeEnum.Btn,
  436. Permission = className[..1].ToLower() + className[1..] + ":detail",
  437. OrderNo = menuOrder
  438. };
  439. menuOrder += 3;
  440. // 按钮-add
  441. var menuType2_2 = new SysMenu
  442. {
  443. Pid = pid1,
  444. Title = "增加",
  445. Type = MenuTypeEnum.Btn,
  446. Permission = className[..1].ToLower() + className[1..] + ":add",
  447. OrderNo = menuOrder
  448. };
  449. menuOrder += 3;
  450. // 按钮-delete
  451. var menuType2_3 = new SysMenu
  452. {
  453. Pid = pid1,
  454. Title = "删除",
  455. Type = MenuTypeEnum.Btn,
  456. Permission = className[..1].ToLower() + className[1..] + ":delete",
  457. OrderNo = menuOrder
  458. };
  459. menuOrder += 3;
  460. // 按钮-edit
  461. var menuType2_4 = new SysMenu
  462. {
  463. Pid = pid1,
  464. Title = "编辑",
  465. Type = MenuTypeEnum.Btn,
  466. Permission = className[..1].ToLower() + className[1..] + ":edit",
  467. OrderNo = menuOrder
  468. };
  469. menuOrder += 3;
  470. var menuList = new List<SysMenu>() { menuType2, menuType2_1, menuType2_2, menuType2_3, menuType2_4 };
  471. // 加入fk、Upload、ApiTreeSelect 等接口的权限
  472. var fkTableList = tableFieldList.Where(u => u.EffectType == "fk").ToList();
  473. foreach (var @column in fkTableList)
  474. {
  475. var menuType = new SysMenu
  476. {
  477. Pid = pid1,
  478. Title = "外键" + @column.ColumnName,
  479. Type = MenuTypeEnum.Btn,
  480. Permission = className[..1].ToLower() + className[1..] + ":" + column.FkEntityName + column.ColumnName + "Dropdown",
  481. OrderNo = menuOrder
  482. };
  483. menuOrder += 3;
  484. menuList.Add(menuType);
  485. }
  486. var treeSelectTableList = tableFieldList.Where(u => u.EffectType == "ApiTreeSelect").ToList();
  487. foreach (var @column in treeSelectTableList)
  488. {
  489. var menuType = new SysMenu
  490. {
  491. Pid = pid1,
  492. Title = "树型" + @column.ColumnName,
  493. Type = MenuTypeEnum.Btn,
  494. Permission = className[..1].ToLower() + className[1..] + ":" + column.FkEntityName + "Tree",
  495. OrderNo = menuOrder
  496. };
  497. menuOrder += 3;
  498. menuList.Add(menuType);
  499. }
  500. var uploadTableList = tableFieldList.Where(u => u.EffectType == "Upload").ToList();
  501. foreach (var @column in uploadTableList)
  502. {
  503. var menuType = new SysMenu
  504. {
  505. Pid = pid1,
  506. Title = "上传" + @column.ColumnName,
  507. Type = MenuTypeEnum.Btn,
  508. Permission = className[..1].ToLower() + className[1..] + ":Upload" + column.ColumnName,
  509. OrderNo = menuOrder
  510. };
  511. menuOrder += 3;
  512. menuList.Add(menuType);
  513. }
  514. await _db.Insertable(menuList).ExecuteCommandAsync();
  515. }
  516. /// <summary>
  517. /// 获取模板文件路径集合
  518. /// </summary>
  519. /// <returns></returns>
  520. private static List<string> GetTemplatePathList(SysCodeGen input)
  521. {
  522. var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "Template");
  523. if (input.GenerateType.Substring(1, 1).Contains('1'))
  524. {
  525. return new List<string>()
  526. {
  527. Path.Combine(templatePath , "index.vue.vm"),
  528. Path.Combine(templatePath , "editDialog.vue.vm"),
  529. Path.Combine(templatePath , "manage.js.vm"),
  530. };
  531. }
  532. else if (input.GenerateType.Substring(1, 1).Contains('2'))
  533. {
  534. return new List<string>()
  535. {
  536. Path.Combine(templatePath , "Service.cs.vm"),
  537. Path.Combine(templatePath , "Input.cs.vm"),
  538. Path.Combine(templatePath , "Output.cs.vm"),
  539. Path.Combine(templatePath , "Dto.cs.vm"),
  540. };
  541. }
  542. else
  543. {
  544. return new List<string>()
  545. {
  546. Path.Combine(templatePath , "Service.cs.vm"),
  547. Path.Combine(templatePath , "Input.cs.vm"),
  548. Path.Combine(templatePath , "Output.cs.vm"),
  549. Path.Combine(templatePath , "Dto.cs.vm"),
  550. Path.Combine(templatePath , "index.vue.vm"),
  551. Path.Combine(templatePath , "editDialog.vue.vm"),
  552. Path.Combine(templatePath , "manage.js.vm"),
  553. };
  554. }
  555. }
  556. /// <summary>
  557. /// 获取模板文件路径集合
  558. /// </summary>
  559. /// <returns></returns>
  560. private static List<string> GetTemplatePathList()
  561. {
  562. var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "Template");
  563. return new List<string>()
  564. {
  565. Path.Combine(templatePath , "Service.cs.vm"),
  566. Path.Combine(templatePath , "Input.cs.vm"),
  567. Path.Combine(templatePath , "Output.cs.vm"),
  568. Path.Combine(templatePath , "Dto.cs.vm"),
  569. Path.Combine(templatePath , "index.vue.vm"),
  570. Path.Combine(templatePath , "editDialog.vue.vm"),
  571. Path.Combine(templatePath , "manage.js.vm"),
  572. };
  573. }
  574. /// <summary>
  575. /// 设置生成文件路径
  576. /// </summary>
  577. /// <param name="input"></param>
  578. /// <returns></returns>
  579. private List<string> GetTargetPathList(SysCodeGen input)
  580. {
  581. //var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, _codeGenOptions.BackendApplicationNamespace, "Service", input.TableName);
  582. var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, input.NameSpace, "Service", input.TableName);
  583. var servicePath = Path.Combine(backendPath, input.TableName + "Service.cs");
  584. var inputPath = Path.Combine(backendPath, "Dto", input.TableName + "Input.cs");
  585. var outputPath = Path.Combine(backendPath, "Dto", input.TableName + "Output.cs");
  586. var viewPath = Path.Combine(backendPath, "Dto", input.TableName + "Dto.cs");
  587. var frontendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.Parent.FullName, _codeGenOptions.FrontRootPath, "src", "views", "main");
  588. var indexPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "index.vue");//
  589. var formModalPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "component", "editDialog.vue");
  590. 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");
  591. return new List<string>()
  592. {
  593. servicePath,
  594. inputPath,
  595. outputPath,
  596. viewPath,
  597. indexPath,
  598. formModalPath,
  599. apiJsPath
  600. };
  601. }
  602. /// <summary>
  603. /// 设置生成文件路径
  604. /// </summary>
  605. /// <param name="input"></param>
  606. /// <returns></returns>
  607. private List<string> GetZipPathList(SysCodeGen input)
  608. {
  609. var zipPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName);
  610. //var backendPath = Path.Combine(zipPath, _codeGenOptions.BackendApplicationNamespace, "Service", input.TableName);
  611. var backendPath = Path.Combine(zipPath, input.NameSpace, "Service", input.TableName);
  612. var servicePath = Path.Combine(backendPath, input.TableName + "Service.cs");
  613. var inputPath = Path.Combine(backendPath, "Dto", input.TableName + "Input.cs");
  614. var outputPath = Path.Combine(backendPath, "Dto", input.TableName + "Output.cs");
  615. var viewPath = Path.Combine(backendPath, "Dto", input.TableName + "Dto.cs");
  616. var frontendPath = Path.Combine(zipPath, _codeGenOptions.FrontRootPath, "src", "views", "main");
  617. var indexPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "index.vue");
  618. var formModalPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "component", "editDialog.vue");
  619. var apiJsPath = Path.Combine(zipPath, _codeGenOptions.FrontRootPath, "src", "api", "main", input.TableName[..1].ToLower() + input.TableName[1..] + ".ts");
  620. if (input.GenerateType.StartsWith("11"))
  621. {
  622. return new List<string>()
  623. {
  624. indexPath,
  625. formModalPath,
  626. apiJsPath
  627. };
  628. }
  629. else if (input.GenerateType.StartsWith("12"))
  630. {
  631. return new List<string>()
  632. {
  633. servicePath,
  634. inputPath,
  635. outputPath,
  636. viewPath
  637. };
  638. }
  639. else
  640. {
  641. return new List<string>()
  642. {
  643. servicePath,
  644. inputPath,
  645. outputPath,
  646. viewPath,
  647. indexPath,
  648. formModalPath,
  649. apiJsPath
  650. };
  651. }
  652. }
  653. }