SysCodeGenService.cs 31 KB

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