SysCodeGenService.cs 31 KB

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