// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
|
//
|
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
|
//
|
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！
|
|
using OfficeOpenXml;
|
using System.IO.Compression;
|
|
namespace Admin.NET.Core.Service;
|
|
/// <summary>
|
/// 系统代码生成器服务 🧩
|
/// </summary>
|
[ApiDescriptionSettings(Order = 270)]
|
public class SysCodeGenService : IDynamicApiController, ITransient
|
{
|
private readonly ISqlSugarClient _db;
|
|
private readonly SysCodeGenConfigService _codeGenConfigService;
|
private readonly SysEnumService _sysEnumService;
|
private readonly IViewEngine _viewEngine;
|
private readonly CodeGenOptions _codeGenOptions;
|
|
public SysCodeGenService(ISqlSugarClient db,
|
SysCodeGenConfigService codeGenConfigService,
|
IViewEngine viewEngine,
|
SysEnumService sysEnumService,
|
IOptions<CodeGenOptions> codeGenOptions)
|
{
|
_db = db;
|
_codeGenConfigService = codeGenConfigService;
|
_sysEnumService = sysEnumService;
|
_viewEngine = viewEngine;
|
_codeGenOptions = codeGenOptions.Value;
|
}
|
|
/// <summary>
|
/// 获取代码生成分页列表 🔖
|
/// </summary>
|
/// <param name="input"></param>
|
/// <returns></returns>
|
[DisplayName("获取代码生成分页列表")]
|
public async Task<SqlSugarPagedList<CodeGenOutput>> Page(CodeGenInput input)
|
{
|
var ret = await _db.Queryable<SysCodeGen>()
|
.WhereIF(!string.IsNullOrWhiteSpace(input.TableName), u => u.TableName.Contains(input.TableName.Trim()))
|
.WhereIF(!string.IsNullOrWhiteSpace(input.BusName), u => u.BusName.Contains(input.BusName.Trim()))
|
.Select<CodeGenOutput>()
|
.OrderByDescending(x => x.UpdateTime)
|
.ToPagedListAsync(input.Page, input.PageSize);
|
foreach (var item in ret.Items)
|
{
|
//单独处理按钮 【Editby shaocx,2024-05-27】
|
if (!string.IsNullOrEmpty(item.Buttons))
|
{
|
item.ButtonsList = item.Buttons.Split(",").ToList();
|
}
|
}
|
return ret;
|
}
|
|
/// <summary>
|
/// 增加代码生成 🔖
|
/// </summary>
|
/// <param name="input"></param>
|
/// <returns></returns>
|
[ApiDescriptionSettings(Name = "Add"), HttpPost]
|
[DisplayName("增加代码生成")]
|
public async Task AddCodeGen(AddCodeGenInput input)
|
{
|
var isExist = await _db.Queryable<SysCodeGen>().Where(u => u.TableName == input.TableName).AnyAsync();
|
if (isExist)
|
throw Oops.Oh(ErrorCodeEnum.D1400);
|
|
var codeGen = input.Adapt<SysCodeGen>();
|
|
//单独处理按钮 【Editby shaocx,2024-05-27】
|
codeGen.Buttons = string.Join(",", input.ButtonsList);
|
|
var newCodeGen = await _db.Insertable(codeGen).ExecuteReturnEntityAsync();
|
// 加入配置表中
|
_codeGenConfigService.AddList(GetColumnList(input), newCodeGen);
|
}
|
|
/// <summary>
|
/// 更新代码生成 🔖
|
/// </summary>
|
/// <param name="input"></param>
|
/// <returns></returns>
|
[ApiDescriptionSettings(Name = "Update"), HttpPost]
|
[DisplayName("更新代码生成")]
|
public async Task UpdateCodeGen(UpdateCodeGenInput input)
|
{
|
var isExist = await _db.Queryable<SysCodeGen>().AnyAsync(u => u.TableName == input.TableName && u.Id != input.Id);
|
if (isExist)
|
throw Oops.Oh(ErrorCodeEnum.D1400);
|
|
var isExistObj = await _db.Queryable<SysCodeGen>().FirstAsync(u => u.Id == input.Id);
|
if (isExistObj == null)
|
throw Oops.Oh(ErrorCodeEnum.D1002);
|
|
//判断如果生成表变化了，就需要重置配置表 【Editby shaocx,2024-04-13】
|
var isChangeTableName = false;
|
if (input.TableName != isExistObj.TableName)
|
{
|
isChangeTableName = true;
|
}
|
|
var codeGen = input.Adapt<SysCodeGen>();
|
if (input.ButtonsList?.Count > 0)
|
{
|
//单独处理按钮 【Editby shaocx,2024-05-27】
|
codeGen.Buttons = string.Join(",", input.ButtonsList);
|
}
|
|
await _db.Updateable(codeGen).ExecuteCommandAsync();
|
|
if (isChangeTableName)
|
{
|
// 刷新配置表
|
await Refresh(input.Id);
|
}
|
}
|
|
/// <summary>
|
/// 删除代码生成 🔖
|
/// </summary>
|
/// <param name="inputs"></param>
|
/// <returns></returns>
|
[ApiDescriptionSettings(Name = "Delete"), HttpPost]
|
[DisplayName("删除代码生成")]
|
public async Task DeleteCodeGen(List<DeleteCodeGenInput> inputs)
|
{
|
if (inputs == null || inputs.Count < 1) return;
|
|
int flag = inputs.First().DelFlag;
|
if (flag == 2)
|
{//重置配置
|
await Refresh(inputs.First().Id);
|
}
|
else
|
{//删除
|
var codeGenConfigTaskList = new List<Task>();
|
inputs.ForEach(u =>
|
{
|
_db.Deleteable<SysCodeGen>().In(u.Id).ExecuteCommand();
|
|
// 删除配置表中
|
codeGenConfigTaskList.Add(_codeGenConfigService.DeleteCodeGenConfig(u.Id));
|
});
|
await Task.WhenAll(codeGenConfigTaskList);
|
}
|
}
|
|
/// <summary>
|
/// 刷新配置表
|
/// </summary>
|
/// <returns></returns>
|
[HttpGet("refresh/{id}")]
|
[DisplayName("刷新配置表")]
|
public async Task Refresh(long id)
|
{
|
var item = await _db.Queryable<SysCodeGen>().Where(u => u.Id == id).FirstAsync();
|
// 删除配置表中
|
await _codeGenConfigService.DeleteCodeGenConfig(id);
|
// 加入配置表中
|
_codeGenConfigService.AddList(GetColumnList(item.Adapt<AddCodeGenInput>()), item);
|
}
|
|
/// <summary>
|
/// 获取代码生成详情 🔖
|
/// </summary>
|
/// <param name="input"></param>
|
/// <returns></returns>
|
[DisplayName("获取代码生成详情")]
|
public async Task<SysCodeGen> GetDetail([FromQuery] QueryCodeGenInput input)
|
{
|
return await _db.Queryable<SysCodeGen>().SingleAsync(u => u.Id == input.Id);
|
}
|
|
/// <summary>
|
/// 获取数据库库集合 🔖
|
/// </summary>
|
/// <returns></returns>
|
[DisplayName("获取数据库库集合")]
|
public async Task<List<DatabaseOutput>> GetDatabaseList()
|
{
|
var dbConfigs = App.GetOptions<DbConnectionOptions>().ConnectionConfigs;
|
return await Task.FromResult(dbConfigs.Adapt<List<DatabaseOutput>>());
|
}
|
|
/// <summary>
|
/// 获取数据库表(实体)集合 🔖
|
/// </summary>
|
/// <returns></returns>
|
[DisplayName("获取数据库表(实体)集合")]
|
public async Task<List<TableOutput>> GetTableList(string configId = SqlSugarConst.MainConfigId)
|
{
|
var provider = _db.AsTenant().GetConnectionScope(configId);
|
var dbTableInfos = provider.DbMaintenance.GetTableInfoList(false); // 不能走缓存,否则切库不起作用
|
var dbViewInfos = provider.DbMaintenance.GetViewInfoList(false);// 不能走缓存,否则切库不起作用,增加视图 【Editby shaocx,2024-05-15】
|
|
var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => configId.Equals(u.ConfigId));
|
|
var dbTableNames = dbTableInfos.Select(u => u.Name.ToLower()).ToList();
|
IEnumerable<EntityInfo> entityInfos = await GetEntityInfos();
|
|
var tableOutputList = new List<TableOutput>();
|
foreach (var item in entityInfos)
|
{
|
var table = dbTableInfos.FirstOrDefault(u => u.Name.ToLower() == (config.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(item.DbTableName) : item.DbTableName).ToLower());
|
if (table != null)
|
{
|
tableOutputList.Add(new TableOutput
|
{
|
ConfigId = configId,
|
EntityName = item.EntityName,
|
TableName = table.Name,
|
TableComment = item.TableDescription
|
});
|
}
|
else
|
{//增加视图 【Editby shaocx,2024-05-15】
|
var view = dbViewInfos.FirstOrDefault(u => u.Name.ToLower() == (config.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(item.DbTableName) : item.DbTableName).ToLower());
|
if (view != null)
|
{
|
tableOutputList.Add(new TableOutput
|
{
|
ConfigId = configId,
|
EntityName = item.EntityName,
|
TableName = view.Name,
|
TableComment = item.TableDescription
|
});
|
}
|
}
|
}
|
return tableOutputList;
|
}
|
|
/// <summary>
|
/// 根据表名获取列集合 🔖
|
/// </summary>
|
/// <returns></returns>
|
[DisplayName("根据表名获取列集合")]
|
public List<ColumnOuput> GetColumnListByTableName([Required] string tableName, string configId = SqlSugarConst.MainConfigId)
|
{
|
// 切库---多库代码生成用
|
var provider = _db.AsTenant().GetConnectionScope(configId);
|
|
var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == configId);
|
// 获取实体类型属性
|
var entityType = provider.DbMaintenance.GetTableInfoList(false).FirstOrDefault(u => u.Name == tableName);
|
if (entityType == null) return null;
|
var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)];
|
// 按原始类型的顺序获取所有实体类型属性（不包含导航属性，会返回null）
|
return provider.DbMaintenance.GetColumnInfosByTableName(entityType.Name).Select(u => new ColumnOuput
|
{
|
ColumnName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
|
ColumnKey = u.IsPrimarykey.ToString(),
|
DataType = u.DataType.ToString(),
|
NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
|
ColumnComment = u.ColumnDescription
|
}).ToList();
|
}
|
|
/// <summary>
|
/// 获取数据表列（实体属性）集合
|
/// </summary>
|
/// <returns></returns>
|
private List<ColumnOuput> GetColumnList([FromQuery] AddCodeGenInput input)
|
{
|
var entityType = GetEntityInfos().GetAwaiter().GetResult().FirstOrDefault(u => u.EntityName == input.TableName);
|
if (entityType == null)
|
return null;
|
var config = App.GetOptions<DbConnectionOptions>().ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == input.ConfigId);
|
var dbTableName = config.DbSettings.EnableUnderLine ? UtilMethods.ToUnderLine(entityType.DbTableName) : entityType.DbTableName;
|
|
// 切库---多库代码生成用
|
var provider = _db.AsTenant().GetConnectionScope(!string.IsNullOrEmpty(input.ConfigId) ? input.ConfigId : SqlSugarConst.MainConfigId);
|
|
var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)];
|
var columnInfos = provider.DbMaintenance.GetColumnInfosByTableName(dbTableName, false);
|
var result = columnInfos.Select(u => new ColumnOuput
|
{
|
// 转下划线后的列名需要再转回来（暂时不转）
|
//ColumnName = config.DbSettings.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
|
ColumnName = u.DbColumnName,
|
ColumnLength = u.Length,
|
IsPrimarykey = u.IsPrimarykey,
|
IsNullable = u.IsNullable,
|
ColumnKey = u.IsPrimarykey.ToString(),
|
NetType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
|
DataType = CodeGenUtil.ConvertDataType(u, provider.CurrentConnectionConfig.DbType),
|
ColumnComment = string.IsNullOrWhiteSpace(u.ColumnDescription) ? u.DbColumnName : u.ColumnDescription
|
}).ToList();
|
|
// 获取实体的属性信息，赋值给PropertyName属性(CodeFirst模式应以PropertyName为实际使用名称)
|
var entityProperties = entityType.Type.GetProperties();
|
|
for (int i = result.Count - 1; i >= 0; i--)
|
{
|
var columnOutput = result[i];
|
// 先找自定义字段名的，如果找不到就再找自动生成字段名的(并且过滤掉没有SugarColumn的属性)
|
var propertyInfo = entityProperties.FirstOrDefault(p => (p.GetCustomAttribute<SugarColumn>()?.ColumnName ?? "").ToLower() == columnOutput.ColumnName.ToLower()) ??
|
entityProperties.FirstOrDefault(p => p.GetCustomAttribute<SugarColumn>() != null && p.Name.ToLower() == (config.DbSettings.EnableUnderLine
|
? CodeGenUtil.CamelColumnName(columnOutput.ColumnName, entityBasePropertyNames).ToLower()
|
: columnOutput.ColumnName.ToLower()));
|
if (propertyInfo != null)
|
{
|
columnOutput.PropertyName = propertyInfo.Name;
|
columnOutput.ColumnComment = propertyInfo.GetCustomAttribute<SugarColumn>().ColumnDescription;
|
//修改赋值类型 【Editby shaocx,2024-05-10】
|
columnOutput.NetType = propertyInfo.PropertyType.ToString();
|
//判断是否是枚举类型 【Editby shaocx,2024-05-10】
|
if (propertyInfo.PropertyType.IsEnum || propertyInfo.PropertyType.FullName.IndexOf("Enum") > -1)
|
{
|
columnOutput.IsEnum = true;
|
}
|
}
|
else
|
{
|
result.RemoveAt(i); // 移除没有定义此属性的字段
|
}
|
}
|
return result;
|
}
|
|
/// <summary>
|
/// 获取库表信息
|
/// </summary>
|
/// <returns></returns>
|
private async Task<IEnumerable<EntityInfo>> GetEntityInfos()
|
{
|
var entityInfos = new List<EntityInfo>();
|
|
var type = typeof(SugarTable);
|
var type_view = typeof(MySugarTableViewAttribute);//增加视图 【Editby shaocx,2024-05-15】
|
var types = new List<Type>();
|
if (_codeGenOptions.EntityAssemblyNames != null)
|
{
|
foreach (var assemblyName in _codeGenOptions.EntityAssemblyNames)
|
{
|
Assembly asm = Assembly.Load(assemblyName);
|
types.AddRange(asm.GetExportedTypes().ToList());
|
}
|
}
|
bool IsMyAttribute(Attribute[] o)
|
{
|
foreach (Attribute a in o)
|
{
|
if (a.GetType() == type || a.GetType() == type_view)
|
return true;
|
}
|
return false;
|
}
|
Type[] cosType = types.Where(o =>
|
{
|
return IsMyAttribute(Attribute.GetCustomAttributes(o, true));
|
}
|
).ToArray();
|
|
var _TableName = "";
|
var _TableDescription = "";
|
foreach (var c in cosType)
|
{
|
var sugarAttribute = c.GetCustomAttributes(type, true)?.FirstOrDefault();
|
|
if (sugarAttribute == null)
|
{//增加视图 【Editby shaocx,2024-05-15】
|
sugarAttribute = c.GetCustomAttributes(type_view, true)?.FirstOrDefault();
|
_TableName = ((MySugarTableViewAttribute)sugarAttribute).TableName;
|
_TableDescription = ((MySugarTableViewAttribute)sugarAttribute).TableDescription;
|
}
|
else
|
{
|
_TableName = ((SugarTable)sugarAttribute).TableName;
|
_TableDescription = ((SugarTable)sugarAttribute).TableDescription;
|
}
|
|
|
var des = c.GetCustomAttributes(typeof(DescriptionAttribute), true);
|
var description = "";
|
if (des.Length > 0)
|
{
|
description = ((DescriptionAttribute)des[0]).Description;
|
}
|
entityInfos.Add(new EntityInfo()
|
{
|
EntityName = c.Name,
|
DbTableName = sugarAttribute == null ? c.Name : _TableName,
|
TableDescription = sugarAttribute == null ? description : _TableDescription,
|
Type = c
|
});
|
}
|
return await Task.FromResult(entityInfos);
|
}
|
|
/// <summary>
|
/// 获取程序保存位置 🔖
|
/// </summary>
|
/// <returns></returns>
|
[DisplayName("获取程序保存位置")]
|
public List<string> GetApplicationNamespaces()
|
{
|
return _codeGenOptions.BackendApplicationNamespaces;
|
}
|
|
/// <summary>
|
/// 代码生成到本地 🔖
|
/// </summary>
|
/// <returns></returns>
|
[DisplayName("代码生成到本地")]
|
public async Task<dynamic> RunLocal(SysCodeGen input)
|
{
|
if (string.IsNullOrEmpty(input.GenerateType))
|
input.GenerateType = "200";
|
|
// 先删除该表已生成的菜单列表
|
var templatePathList = GetTemplatePathList(input);
|
List<string> targetPathList;
|
var zipPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName);
|
if (input.GenerateType.StartsWith('1'))
|
{
|
targetPathList = GetZipPathList(input);
|
if (Directory.Exists(zipPath))
|
Directory.Delete(zipPath, true);
|
}
|
else
|
targetPathList = GetTargetPathList(input);
|
|
var tableFieldList = await _codeGenConfigService.GetList(new CodeGenConfig() { CodeGenId = input.Id }); // 字段集合
|
var queryWhetherList = tableFieldList.Where(u => u.QueryWhether == YesNoEnum.Y.ToString()).ToList(); // 前端查询集合
|
var joinTableList = tableFieldList.Where(u => u.EffectType == "Upload" || u.EffectType == "fk" || u.EffectType == "ApiTreeSelect").ToList(); // 需要连表查询的字段
|
(string joinTableNames, string lowerJoinTableNames) = GetJoinTableStr(joinTableList); // 获取连表的实体名和别名
|
|
if (input.GenerateType.StartsWith('2'))
|
{
|
CodeGenUtil.ValidateCodeGenConfig(tableFieldList);
|
}
|
|
//处理可空类型 【Editby shaocx,2024-04-07】
|
tableFieldList.ForEach(u =>
|
{
|
u.NetTypeIsNullLableForQueryInput = "?";
|
u.NetTypeIsNullLableForAddEditOutParam = "";
|
switch (u.NetType.ToLower())
|
{
|
case "int":
|
case "int32":
|
case "long":
|
case "guid":
|
case "decimal":
|
case "datetime":
|
case "datetimeoffset":
|
case "bool": //增加布尔类型的条件 【Editby shaocx,2024-04-07】
|
u.NetTypeIsNullLable = "?";
|
if (u.WhetherRequired != "Y")
|
{
|
u.NetTypeIsNullLableForAddEditOutParam = "?";
|
}
|
break;
|
case "string":
|
u.NetTypeIsNullLableForQueryInput = "";
|
break;
|
default://其他，比如枚举 【Editby shaocx,2024-04-07】
|
if (u.WhetherRequired != "Y")
|
{
|
u.NetTypeIsNullLableForAddEditOutParam = "?";
|
}
|
break;
|
}
|
//u.OriginalColumnName = u.ColumnName;
|
});
|
|
|
//获取导入功能自定义模板代码
|
StringBuilder templateContent = null;
|
templateContent = CreateImportExcelTemplteCode(input, tableFieldList);
|
|
var data = new CustomViewEngine(_db)
|
{
|
ConfigId = input.ConfigId,
|
AuthorName = input.AuthorName,
|
BusName = input.BusName,
|
NameSpace = input.NameSpace,
|
ClassName = input.TableName,
|
ModuleName = input.ModuleName,
|
Buttons = input.Buttons,
|
ImportExcelCustomizationContent = templateContent.ToString(),
|
ProjectLastName = input.NameSpace.Split('.').Last(),
|
QueryWhetherList = queryWhetherList,
|
TableField = tableFieldList,
|
IsJoinTable = joinTableList.Count > 0,
|
IsUpload = joinTableList.Where(u => u.EffectType == "Upload").Any(),
|
PrintType = input.PrintType,
|
PrintName = input.PrintName,
|
TableType = input.TableType,
|
KeyQueryStr = string.Join(",", tableFieldList.Where(x => x.WhetherKeyQuery == "Y").Select(x => x.ColumnComment).ToList())
|
};
|
|
for (var i = 0; i < templatePathList.Count; i++)
|
{
|
if (data.TableType == "视图" && templatePathList[i].IndexOf("editDialog.vue.vm") > -1) continue;
|
|
if (!File.Exists(templatePathList[i])) continue;
|
var tContent = File.ReadAllText(templatePathList[i]);
|
var tResult = await _viewEngine.RunCompileFromCachedAsync(tContent, data, builderAction: builder =>
|
{
|
builder.AddAssemblyReferenceByName("System.Linq");
|
builder.AddAssemblyReferenceByName("System.Collections");
|
builder.AddUsing("System.Collections.Generic");
|
builder.AddUsing("System.Linq");
|
});
|
var dirPath = new DirectoryInfo(targetPathList[i]).Parent.FullName;
|
if (!Directory.Exists(dirPath))
|
Directory.CreateDirectory(dirPath);
|
File.WriteAllText(targetPathList[i], tResult, Encoding.UTF8);
|
}
|
|
if (!string.IsNullOrEmpty(input.Buttons))
|
{//如果没有任何按钮，就不生成菜单 【Editby shaocx,2024-07-31】
|
await AddMenu(input, input.TableName, input.BusName, input.MenuPid, tableFieldList, input.ModuleName);
|
}
|
|
if (input.Buttons.Contains("导入"))
|
{
|
//创建导入模版
|
await CreateImportTemplte(input);
|
}
|
|
// 非ZIP压缩返回空
|
if (!input.GenerateType.StartsWith('1'))
|
return null;
|
else
|
{
|
string downloadPath = zipPath + ".zip";
|
// 判断是否存在同名称文件
|
if (File.Exists(downloadPath))
|
File.Delete(downloadPath);
|
ZipFile.CreateFromDirectory(zipPath, downloadPath);
|
return new { url = $"{App.HttpContext.Request.Scheme}://{App.HttpContext.Request.Host.Value}/CodeGen/{input.TableName}.zip" };
|
}
|
}
|
|
/// <summary>
|
/// 获取连表的实体名和别名
|
/// </summary>
|
/// <param name="configs"></param>
|
/// <returns></returns>
|
private static (string, string) GetJoinTableStr(List<CodeGenConfig> configs)
|
{
|
var uploads = configs.Where(u => u.EffectType == "Upload").ToList();
|
var fks = configs.Where(u => u.EffectType == "fk").ToList();
|
string str = ""; // <Order, OrderItem, Custom>
|
string lowerStr = ""; // (o, i, c)
|
foreach (var item in uploads)
|
{
|
lowerStr += "sysFile_FK_" + item.LowerPropertyName + ",";
|
str += "SysFile,";
|
}
|
foreach (var item in fks)
|
{
|
lowerStr += item.LowerFkEntityName + "_FK_" + item.LowerFkColumnName + ",";
|
str += item.FkEntityName + ",";
|
}
|
return (str.TrimEnd(','), lowerStr.TrimEnd(','));
|
}
|
|
/// <summary>
|
/// 增加菜单
|
/// </summary>
|
/// <param name="input"></param>
|
/// <param name="className"></param>
|
/// <param name="busName"></param>
|
/// <param name="pid"></param>
|
/// <param name="tableFieldList"></param>
|
/// <param name="moduleName"></param>
|
/// <returns></returns>
|
private async Task AddMenu(SysCodeGen input, string className, string busName, long pid, List<CodeGenConfig> tableFieldList, string moduleName)
|
{
|
var pPath = string.Empty;
|
// 若 pid=0 为顶级则创建菜单目录
|
if (pid == 0)
|
{
|
// 目录
|
var menuType0 = new SysMenu
|
{
|
Pid = 0,
|
Title = busName,
|
Type = MenuTypeEnum.Dir,
|
Icon = "robot",
|
Path = "/" + className.ToLower(),
|
Component = "Layout",
|
};
|
// 若先前存在则删除本级和下级
|
var menuList0 = await _db.Queryable<SysMenu>().Where(u => u.Title == menuType0.Title && u.Type == menuType0.Type).ToListAsync();
|
if (menuList0.Count > 0)
|
{
|
var listIds = menuList0.Select(u => u.Id).ToList();
|
var childlistIds = new List<long>();
|
foreach (var item in listIds)
|
{
|
var childlist = await _db.Queryable<SysMenu>().ToChildListAsync(u => u.Pid, item);
|
childlistIds.AddRange(childlist.Select(u => u.Id).ToList());
|
}
|
listIds.AddRange(childlistIds);
|
await _db.Deleteable<SysMenu>().Where(u => listIds.Contains(u.Id)).ExecuteCommandAsync();
|
await _db.Deleteable<SysRoleMenu>().Where(u => listIds.Contains(u.MenuId)).ExecuteCommandAsync();
|
}
|
pid = (await _db.Insertable(menuType0).ExecuteReturnEntityAsync()).Id;
|
}
|
else
|
{
|
var pMenu = await _db.Queryable<SysMenu>().FirstAsync(u => u.Id == pid) ?? throw Oops.Oh(ErrorCodeEnum.D1505);
|
pPath = pMenu.Path;
|
}
|
|
// 菜单
|
var menuType1 = new SysMenu
|
{
|
Pid = pid,
|
Title = busName,
|
Name = className[..1].ToLower() + className[1..],
|
Type = MenuTypeEnum.Menu,
|
Path = pPath + "/" + className.ToLower(),
|
Component = "/main/" + moduleName + @"/" + className[..1].ToLower() + className[1..] + "/index",
|
};
|
// 若先前存在则删除本级和下级
|
var menuList1 = await _db.Queryable<SysMenu>().Where(u => u.Title == menuType1.Title && u.Type == menuType1.Type).ToListAsync();
|
if (menuList1.Count > 0)
|
{
|
var listIds = menuList1.Select(u => u.Id).ToList();
|
var childlistIds = new List<long>();
|
foreach (var item in listIds)
|
{
|
var childlist = await _db.Queryable<SysMenu>().ToChildListAsync(u => u.Pid, item);
|
childlistIds.AddRange(childlist.Select(u => u.Id).ToList());
|
}
|
listIds.AddRange(childlistIds);
|
await _db.Deleteable<SysMenu>().Where(u => listIds.Contains(u.Id)).ExecuteCommandAsync();
|
await _db.Deleteable<SysRoleMenu>().Where(u => listIds.Contains(u.MenuId)).ExecuteCommandAsync();
|
}
|
|
var pid1 = (await _db.Insertable(menuType1).ExecuteReturnEntityAsync()).Id;
|
int menuOrder = 100;
|
|
var menuList = new List<SysMenu>();
|
|
if (input.Buttons.Contains("查询"))
|
{
|
// 按钮-page
|
var menuType2 = new SysMenu
|
{
|
Pid = pid1,
|
Title = "查询",
|
Type = MenuTypeEnum.Btn,
|
Permission = className[..1].ToLower() + className[1..] + ":page",
|
OrderNo = menuOrder
|
};
|
menuOrder += 10;
|
menuList.Add(menuType2);
|
}
|
|
if (input.Buttons.Contains("详情"))
|
{
|
// 按钮-detail
|
var menuType2_1 = new SysMenu
|
{
|
Pid = pid1,
|
Title = "详情",
|
Type = MenuTypeEnum.Btn,
|
Permission = className[..1].ToLower() + className[1..] + ":detail",
|
OrderNo = menuOrder
|
};
|
menuOrder += 10;
|
menuList.Add(menuType2_1);
|
}
|
|
if (input.Buttons.Contains("新增"))
|
{
|
// 按钮-add
|
var menuType2_2 = new SysMenu
|
{
|
Pid = pid1,
|
Title = "新增",
|
Type = MenuTypeEnum.Btn,
|
Permission = className[..1].ToLower() + className[1..] + ":add",
|
OrderNo = menuOrder
|
};
|
menuOrder += 10;
|
menuList.Add(menuType2_2);
|
}
|
|
if (input.Buttons.Contains("删除"))
|
{
|
// 按钮-delete
|
var menuType2_3 = new SysMenu
|
{
|
Pid = pid1,
|
Title = "删除",
|
Type = MenuTypeEnum.Btn,
|
Permission = className[..1].ToLower() + className[1..] + ":delete",
|
OrderNo = menuOrder
|
};
|
menuOrder += 10;
|
menuList.Add(menuType2_3);
|
}
|
|
if (input.Buttons.Contains("编辑"))
|
{
|
// 按钮-update
|
var menuType2_4 = new SysMenu
|
{
|
Pid = pid1,
|
Title = "编辑",
|
Type = MenuTypeEnum.Btn,
|
Permission = className[..1].ToLower() + className[1..] + ":update",
|
OrderNo = menuOrder
|
};
|
menuOrder += 10;
|
menuList.Add(menuType2_4);
|
}
|
|
if (input.Buttons.Contains("导入"))
|
{
|
// 按钮-import 【Editby shaocx,2024-05-27】
|
var menuType2_5 = new SysMenu
|
{
|
Pid = pid1,
|
Title = "导入",
|
Type = MenuTypeEnum.Btn,
|
Permission = className[..1].ToLower() + className[1..] + ":importExcel",
|
OrderNo = menuOrder
|
};
|
menuOrder += 10;
|
menuList.Add(menuType2_5);
|
}
|
|
if (input.Buttons.Contains("导出"))
|
{
|
// 按钮-import 【Editby shaocx,2024-05-27】
|
var menuType2_6 = new SysMenu
|
{
|
Pid = pid1,
|
Title = "导出",
|
Type = MenuTypeEnum.Btn,
|
Permission = className[..1].ToLower() + className[1..] + ":exportExcel",
|
OrderNo = menuOrder
|
};
|
menuOrder += 10;
|
menuList.Add(menuType2_6);
|
}
|
|
|
// 加入fk、Upload、ApiTreeSelect 等接口的权限
|
// 在生成表格时，有些字段只是查询时显示，不需要填写（WhetherAddUpdate），所以这些字段没必要生成相应接口
|
var fkTableList = tableFieldList.Where(u => u.EffectType == "fk" && (u.WhetherAddUpdate == "Y" || u.QueryWhether == "Y")).ToList();
|
foreach (var @column in fkTableList)
|
{
|
var menuType = new SysMenu
|
{
|
Pid = pid1,
|
Title = "外键" + @column.ColumnName,
|
Type = MenuTypeEnum.Btn,
|
Permission = className[..1].ToLower() + className[1..] + ":" + column.FkEntityName + column.ColumnName + "Dropdown",
|
OrderNo = menuOrder
|
};
|
menuOrder += 10;
|
menuList.Add(menuType);
|
}
|
var treeSelectTableList = tableFieldList.Where(u => u.EffectType == "ApiTreeSelect").ToList();
|
foreach (var @column in treeSelectTableList)
|
{
|
var menuType = new SysMenu
|
{
|
Pid = pid1,
|
Title = "树型" + @column.ColumnName,
|
Type = MenuTypeEnum.Btn,
|
Permission = className[..1].ToLower() + className[1..] + ":" + column.FkEntityName + "Tree",
|
OrderNo = menuOrder
|
};
|
menuOrder += 10;
|
menuList.Add(menuType);
|
}
|
var uploadTableList = tableFieldList.Where(u => u.EffectType == "Upload").ToList();
|
foreach (var @column in uploadTableList)
|
{
|
var menuType = new SysMenu
|
{
|
Pid = pid1,
|
Title = "上传" + @column.ColumnName,
|
Type = MenuTypeEnum.Btn,
|
Permission = className[..1].ToLower() + className[1..] + ":Upload" + column.ColumnName,
|
OrderNo = menuOrder
|
};
|
menuOrder += 10;
|
menuList.Add(menuType);
|
}
|
await _db.Insertable(menuList).ExecuteCommandAsync();
|
}
|
|
|
|
/// <summary>
|
/// 获取模板文件路径集合
|
/// </summary>
|
/// <returns></returns>
|
private static List<string> GetTemplatePathList(SysCodeGen input)
|
{
|
var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "Template");
|
if (input.GenerateType.Substring(1, 1).Contains('1'))
|
{
|
return new List<string>()
|
{
|
Path.Combine(templatePath , "index.vue.vm"),
|
Path.Combine(templatePath , "editDialog.vue.vm"),
|
Path.Combine(templatePath , "manage.js.vm"),
|
};
|
}
|
else if (input.GenerateType.Substring(1, 1).Contains('2'))
|
{
|
return new List<string>()
|
{
|
Path.Combine(templatePath , "Service.cs.vm"),
|
Path.Combine(templatePath , "Input.cs.vm"),
|
Path.Combine(templatePath , "Output.cs.vm"),
|
//不再生成Dto.cs文件,这个文件没地方用到 【Editby shaocx,2024-05-22】
|
//Path.Combine(templatePath , "Dto.cs.vm"),
|
};
|
}
|
else
|
{
|
return new List<string>()
|
{
|
Path.Combine(templatePath , "Service.cs.vm"),
|
Path.Combine(templatePath , "Input.cs.vm"),
|
Path.Combine(templatePath , "Output.cs.vm"),
|
//不再生成Dto.cs文件,这个文件没地方用到 【Editby shaocx,2024-05-22】
|
//Path.Combine(templatePath , "Dto.cs.vm"),
|
Path.Combine(templatePath , "index.vue.vm"),
|
Path.Combine(templatePath , "editDialog.vue.vm"),
|
Path.Combine(templatePath , "manage.js.vm"),
|
};
|
}
|
}
|
|
/// <summary>
|
/// 获取模板文件路径集合
|
/// </summary>
|
/// <returns></returns>
|
private static List<string> GetTemplatePathList()
|
{
|
var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "Template");
|
return new List<string>()
|
{
|
Path.Combine(templatePath , "Service.cs.vm"),
|
Path.Combine(templatePath , "Input.cs.vm"),
|
Path.Combine(templatePath , "Output.cs.vm"),
|
//不再生成Dto.cs文件,这个文件没地方用到 【Editby shaocx,2024-05-22】
|
//Path.Combine(templatePath , "Dto.cs.vm"),
|
Path.Combine(templatePath , "index.vue.vm"),
|
Path.Combine(templatePath , "editDialog.vue.vm"),
|
Path.Combine(templatePath , "manage.js.vm"),
|
};
|
}
|
|
/// <summary>
|
/// 设置生成文件路径
|
/// </summary>
|
/// <param name="input"></param>
|
/// <returns></returns>
|
private List<string> GetTargetPathList(SysCodeGen input)
|
{
|
//var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, _codeGenOptions.BackendApplicationNamespace, "Service", input.TableName);
|
var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, input.NameSpace, "Service", input.ModuleName, input.TableName);
|
var servicePath = Path.Combine(backendPath, input.TableName + "Service.cs");
|
var inputPath = Path.Combine(backendPath, "Dto", input.TableName + "Input.cs");
|
var outputPath = Path.Combine(backendPath, "Dto", input.TableName + "Output.cs");
|
//不再生成Dto.cs文件,这个文件没地方用到 【Editby shaocx,2024-05-22】
|
//var viewPath = Path.Combine(backendPath, "Dto", input.TableName + "Dto.cs");
|
var frontendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.Parent.FullName, _codeGenOptions.FrontRootPath, "src", "views", "main", input.ModuleName);
|
var indexPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "index.vue");//
|
var formModalPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "component", "editDialog.vue");
|
var apiJsPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.Parent.FullName, _codeGenOptions.FrontRootPath, "src", "api", "main", input.ModuleName, input.TableName[..1].ToLower() + input.TableName[1..] + ".ts");
|
|
if (input.GenerateType.Substring(1, 1).Contains('1'))
|
{
|
// 生成到本项目(前端)
|
return new List<string>()
|
{
|
indexPath,
|
formModalPath,
|
apiJsPath
|
};
|
}
|
else if (input.GenerateType.Substring(1, 1).Contains('2'))
|
{
|
// 生成到本项目(后端)
|
return new List<string>()
|
{
|
servicePath,
|
inputPath,
|
outputPath,
|
//不再生成Dto.cs文件,这个文件没地方用到 【Editby shaocx,2024-05-22】
|
//viewPath,
|
};
|
}
|
else
|
{
|
// 前后端同时生成到本项目
|
return new List<string>()
|
{
|
servicePath,
|
inputPath,
|
outputPath,
|
//不再生成Dto.cs文件,这个文件没地方用到 【Editby shaocx,2024-05-22】
|
//viewPath,
|
indexPath,
|
formModalPath,
|
apiJsPath
|
};
|
}
|
}
|
|
/// <summary>
|
/// 设置生成文件路径
|
/// </summary>
|
/// <param name="input"></param>
|
/// <returns></returns>
|
private List<string> GetZipPathList(SysCodeGen input)
|
{
|
var zipPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName);
|
|
//var backendPath = Path.Combine(zipPath, _codeGenOptions.BackendApplicationNamespace, "Service", input.TableName);
|
var backendPath = Path.Combine(zipPath, input.NameSpace, "Service", input.TableName);
|
var servicePath = Path.Combine(backendPath, input.TableName + "Service.cs");
|
var inputPath = Path.Combine(backendPath, "Dto", input.TableName + "Input.cs");
|
var outputPath = Path.Combine(backendPath, "Dto", input.TableName + "Output.cs");
|
//不再生成Dto.cs文件,这个文件没地方用到 【Editby shaocx,2024-05-22】
|
//var viewPath = Path.Combine(backendPath, "Dto", input.TableName + "Dto.cs");
|
var frontendPath = Path.Combine(zipPath, _codeGenOptions.FrontRootPath, "src", "views", "main");
|
var indexPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "index.vue");
|
var formModalPath = Path.Combine(frontendPath, input.TableName[..1].ToLower() + input.TableName[1..], "component", "editDialog.vue");
|
var apiJsPath = Path.Combine(zipPath, _codeGenOptions.FrontRootPath, "src", "api", "main", input.TableName[..1].ToLower() + input.TableName[1..] + ".ts");
|
if (input.GenerateType.StartsWith("11"))
|
{
|
return new List<string>()
|
{
|
indexPath,
|
formModalPath,
|
apiJsPath
|
};
|
}
|
else if (input.GenerateType.StartsWith("12"))
|
{
|
return new List<string>()
|
{
|
servicePath,
|
inputPath,
|
outputPath,
|
//不再生成Dto.cs文件,这个文件没地方用到 【Editby shaocx,2024-05-22】
|
//viewPath
|
};
|
}
|
else
|
{
|
return new List<string>()
|
{
|
servicePath,
|
inputPath,
|
outputPath,
|
//不再生成Dto.cs文件,这个文件没地方用到 【Editby shaocx,2024-05-22】
|
//viewPath,
|
indexPath,
|
formModalPath,
|
apiJsPath
|
};
|
}
|
}
|
|
#region 导入相关
|
|
//update by liuwq 20240418
|
/// <summary>
|
/// 创建导入功能自定义模板代码
|
/// </summary>
|
/// <param name="input"></param>
|
/// <param name="tableFieldList"></param>
|
/// <returns></returns>
|
private static StringBuilder CreateImportExcelTemplteCode(SysCodeGen input, List<CodeGenConfig> tableFieldList)
|
{
|
StringBuilder templateContent = new StringBuilder();
|
var whetherAddUpdate = tableFieldList.Where(u => u.WhetherAddUpdate == YesNoEnum.Y.ToString()).ToList(); // 前端查询增改集合
|
templateContent.AppendLine(" ");
|
templateContent.AppendLine($" var addItem = new {input.TableName}();");
|
//templateContent.AppendLine($" var addItem = new {input.TableName}()");
|
//templateContent.AppendLine(" {");
|
//templateContent.AppendLine(" CreatedTime = SysHelper.GetNowTime(),");
|
//templateContent.AppendLine(" CreatedUserId = SysHelper.GetUserId(),");
|
//templateContent.AppendLine(" CreatedUserName = SysHelper.GetUserName(),");
|
//templateContent.AppendLine(" UpdatedTime = SysHelper.GetNowTime(),");
|
//templateContent.AppendLine(" UpdatedUserId = SysHelper.GetUserId(),");
|
//templateContent.AppendLine(" UpdatedUserName = SysHelper.GetUserName()");
|
//templateContent.AppendLine(" };");
|
|
|
templateContent.AppendLine(" #region 定义变量");
|
foreach (var item in whetherAddUpdate)
|
{
|
templateContent.AppendLine($" var _{item.ColumnName} = \"\";//{item.ColumnComment}");
|
}
|
|
templateContent.AppendLine(" #endregion");
|
templateContent.AppendLine(" ");
|
|
templateContent.AppendLine(" ");
|
templateContent.AppendLine(" #region 取值");
|
foreach (var item in whetherAddUpdate)
|
{
|
templateContent.AppendLine($" _{item.ColumnName} = row[\"{item.ColumnComment}\"]?.ToString() ;");
|
}
|
|
templateContent.AppendLine(" #endregion");
|
templateContent.AppendLine(" ");
|
|
templateContent.AppendLine(" ");
|
templateContent.AppendLine(" #region 验证");
|
|
//数值类型验证
|
List<string> numericalTypeList = new List<string>()
|
{
|
"int",
|
"long",
|
"decimal"
|
};
|
|
|
|
foreach (var item in whetherAddUpdate)
|
{
|
|
|
//必填字段验证是否为空
|
if (item.WhetherRequired.Equals("Y"))
|
{
|
templateContent.AppendLine(" ");
|
templateContent.AppendLine($" if (string.IsNullOrEmpty(_{item.ColumnName}))");
|
templateContent.AppendLine(" {");
|
templateContent.AppendLine($" throw Oops.Oh($\"第{{index}}行[{item.ColumnComment}]{{_{item.ColumnName}}}不能为空！\");");
|
templateContent.AppendLine(" }");
|
templateContent.AppendLine(" ");
|
|
|
}
|
|
|
|
|
//验证值是否有效
|
|
//验证bool值
|
if (item.NetType.Equals("bool"))
|
{
|
templateContent.AppendLine($" if(!string.IsNullOrEmpty(_{item.ColumnName}))");
|
templateContent.AppendLine(" {");
|
templateContent.AppendLine($" if(!_{item.ColumnName}.Equals(\"是\") && !_{item.ColumnName}.Equals(\"否\"))");
|
templateContent.AppendLine(" {");
|
templateContent.AppendLine($" throw Oops.Oh($\"第{{index}}行[{item.ColumnComment}]{{_{item.ColumnName}}}值不正确！\");");
|
templateContent.AppendLine(" }");
|
templateContent.AppendLine(" else");
|
templateContent.AppendLine(" {");
|
templateContent.AppendLine($" bool out{item.ColumnName} = _{item.ColumnName}.Equals(\"是\") ? true : false;");
|
templateContent.AppendLine($" addItem.{item.ColumnName} = out{item.ColumnName};");
|
templateContent.AppendLine(" }");
|
templateContent.AppendLine(" }");
|
templateContent.AppendLine(" ");
|
|
}
|
//验证枚举值是否有效
|
else if (!string.IsNullOrWhiteSpace(item.DictTypeCode))//字典code不为空 是使用枚举值
|
{
|
templateContent.AppendLine($" if(!string.IsNullOrEmpty(_{item.ColumnName}))");
|
templateContent.AppendLine(" {");
|
templateContent.AppendLine($" {item.NetType} enum{item.ColumnName} = default({item.NetType});");
|
templateContent.AppendLine(" ");
|
templateContent.AppendLine($" if(!Enum.TryParse<{item.NetType}>(_{item.ColumnName}, out enum{item.ColumnName})&&!string.IsNullOrEmpty(_{item.ColumnName}))");
|
templateContent.AppendLine(" {");
|
templateContent.AppendLine($" throw Oops.Oh($\"第{{index}}行[{item.ColumnComment}]{{_{item.ColumnName}}}值不正确！\");");
|
templateContent.AppendLine(" }");
|
templateContent.AppendLine(" else");
|
templateContent.AppendLine(" {");
|
templateContent.AppendLine($" addItem.{item.ColumnName} = enum{item.ColumnName};");
|
templateContent.AppendLine(" }");
|
templateContent.AppendLine(" ");
|
templateContent.AppendLine(" }");
|
|
}//验证数值是否有效
|
else if (numericalTypeList.Any(a => a == item.NetType))
|
{
|
|
|
templateContent.AppendLine($" if(!string.IsNullOrEmpty(_{item.ColumnName}))");
|
templateContent.AppendLine(" {");
|
templateContent.AppendLine($" if (!{item.NetType}.TryParse(_{item.ColumnName}, out {item.NetType} out{item.ColumnName})&&!string.IsNullOrEmpty(_{item.ColumnName}))");
|
templateContent.AppendLine(" {");
|
templateContent.AppendLine($" throw Oops.Oh($\"第{{index}}行[{item.ColumnComment}]{{_{item.ColumnName}}}值不正确！\");");
|
templateContent.AppendLine(" }");
|
templateContent.AppendLine($" if (out{item.ColumnName} <= 0&&!string.IsNullOrEmpty(_{item.ColumnName}))");
|
templateContent.AppendLine(" {");
|
templateContent.AppendLine($" throw Oops.Oh($\"第{{index}}行[{item.ColumnComment}]{{_{item.ColumnName}}}值不能小于等于0！\");");
|
templateContent.AppendLine(" }");
|
templateContent.AppendLine(" else");
|
templateContent.AppendLine(" {");
|
templateContent.AppendLine($" addItem.{item.ColumnName} = out{item.ColumnName};");
|
templateContent.AppendLine(" }");
|
templateContent.AppendLine(" ");
|
templateContent.AppendLine(" }");
|
|
}
|
else
|
{
|
templateContent.AppendLine($" if(!string.IsNullOrEmpty(_{item.ColumnName}))");
|
templateContent.AppendLine(" {");
|
|
//日期控件
|
if (item.EffectType.ToLower() == "datepicker")
|
{
|
templateContent.AppendLine($" addItem.{item.ColumnName} = Convert.ToDateTime(Convert.ToDateTime(_{item.ColumnName}.Trim()).ToShortDateString());");
|
}//日期时间控件
|
else if (item.EffectType.ToLower() == "datetimepicker")
|
{
|
templateContent.AppendLine($" addItem.{item.ColumnName} = Convert.ToDateTime(_{item.ColumnName}.Trim());");
|
}
|
else
|
{
|
templateContent.AppendLine($" addItem.{item.ColumnName} = ({item.NetType})(_{item.ColumnName}.Trim());");
|
}
|
templateContent.AppendLine(" }");
|
}
|
|
|
|
|
|
|
|
}
|
|
templateContent.AppendLine(" #endregion");
|
templateContent.AppendLine(" ");
|
|
|
|
return templateContent;
|
}
|
|
/// <summary>
|
/// 根据表名创建导入模版
|
/// </summary>
|
/// <param name="input"></param>
|
private async Task CreateImportTemplte(SysCodeGen input)
|
{
|
string errMsg = string.Empty;
|
//业务名
|
string _busName = input.BusName;
|
var copayPath = TemplateConst.EXCEL_TEMPLATEFILE_导入模版路径 + "\\CommonTemplate.xlsx";
|
string newPath = TemplateConst.EXCEL_TEMPLATEFILE_导入模版路径 + $"\\{_busName}{TemplateConst.EXCEL_TEMPLATEFILE_导入模版名称后缀}.xlsx";
|
var tableFieldList = await _codeGenConfigService.GetList(new CodeGenConfig() { CodeGenId = input.Id }); // 字段集合
|
CreatExcel(tableFieldList, copayPath, newPath);
|
}
|
|
|
/// <summary>
|
/// 根据实体类名 生成导入excel模版
|
/// </summary>
|
/// <param name="tableFieldList">代码生成选择配置表的字段</param>
|
/// <param name="copayPath"></param>
|
/// <param name="newPath"></param>
|
private void CreatExcel(List<CodeGenConfig> tableFieldList, string copayPath, string newPath)
|
{
|
string errMsg = string.Empty;
|
try
|
{
|
#region 验证原始导入模板文件是否存在
|
if (!File.Exists(copayPath))
|
{
|
errMsg = $"用来复制生成模版的EXCEL文件不存在";
|
throw Oops.Oh($"生成导入模版文件异常：{errMsg}");
|
}
|
#endregion
|
|
//获取模板文件
|
FileInfo copyFile = new FileInfo(copayPath);
|
|
// 检查新生成的导入模版文件是否存在，存在就删除，以避免可能的异常。
|
if (File.Exists(newPath))
|
{
|
File.Delete(newPath); // 删除文件
|
}
|
//复制原始导入模版，创建新的导入模版文件
|
copyFile.CopyTo(newPath, true);
|
FileInfo existingFile = new FileInfo(newPath);
|
using (ExcelPackage package = new ExcelPackage(existingFile))
|
{
|
//获取模板内容
|
ExcelWorksheet worksheet = package.Workbook.Worksheets[0];//获取第一个worksheet
|
//行和列都是从1开始，而不是从0开始！！！
|
int _remarkRowIndex = 1;//worksheet 行索引（说明） 注意：默认是第一行是导入模版的说明
|
int _mergeRowCount = _remarkRowIndex;//合并行 （必填字段数量） 默认是第一行合并
|
int _titleRowIndex = 2;//worksheet 行索引（标题）注意：默认是第一行是导入模版的标题
|
int _cellIndex = 1;//worksheet 列索引 注意：默认是第一列开始设置导入模版的标题
|
|
//模版第一列作为模版使用，所有新增的列样式报错一致
|
var templateCell = worksheet.Cells[_titleRowIndex, _cellIndex];
|
|
string _remark = string.Empty; //第一行添加说明，注意换行。
|
|
//获取要处理的代码生成配置的是增改的模版字段
|
var showCodeGenConfigs = tableFieldList.Where(w => w.WhetherAddUpdate.Equals("Y")).ToList();
|
showCodeGenConfigs = showCodeGenConfigs.Where(x => x.ColumnKey.ToLower() == "false").ToList();
|
int _mergeCellsCount = showCodeGenConfigs.Count();//合并列（模版赋值的标题列数）
|
//获取必填字段
|
|
//创建模版说明
|
StringBuilder _remarkContentBuilder = GetParseTemplateHint(showCodeGenConfigs);
|
string _remarkContent = _remarkContentBuilder.ToString();
|
|
#region 合并单元格
|
//合并单元格，合并行和列。默认是合并第一行的所有标题列
|
var cellRange = worksheet.Cells[1, 1, _mergeRowCount, _mergeCellsCount];
|
|
cellRange.Value = _remarkContent;//合并单元格后的内容赋值
|
var mergeCell = cellRange.Merge = true;//合并
|
int rowHeight = GetRowHeightBasedOnContent(showCodeGenConfigs.Count()); // 根据内容计算行高（这里需要你自己实现逻辑）
|
worksheet.Row(_remarkRowIndex).Height = rowHeight; // 设置行高，注意EPPlus的单位不同，需要转换
|
|
#endregion
|
|
//循环创建模版标题列
|
foreach (var item in showCodeGenConfigs)
|
{
|
var currentCell = worksheet.Cells[_titleRowIndex, _cellIndex];
|
//给新模版列标题赋值
|
currentCell.Value = item.ColumnComment;
|
//复制拷贝的excel模版列，给新模版列样式赋值
|
currentCell.StyleID = templateCell.StyleID;
|
// worksheet.Column(_cellIndex).AutoFit();//宽度自适应
|
|
_cellIndex++;
|
}
|
|
|
package.Save();//保存
|
}
|
}
|
catch (Exception ex)
|
{
|
throw Oops.Oh($"生成导入模版文件异常，请查看系统日志:" + ex.Message);
|
}
|
finally { }
|
}
|
|
|
// 这里是一个假设的方法，用于根据单元格内容计算行高。你需要根据实际情况来实现这个逻辑。
|
private static int GetRowHeightBasedOnContent(int lineCount)
|
{
|
// 这里只是一个示例逻辑，你可能需要更复杂的算法来决定合适的行高。
|
|
return (lineCount + 3) * 20; //
|
}
|
|
private StringBuilder GetParseTemplateHint(List<CodeGenConfig> requiredTableFieldList)
|
{
|
StringBuilder _remarkContentBuilder = new StringBuilder();
|
_remarkContentBuilder.AppendLine("");
|
_remarkContentBuilder.AppendLine("1.支持Excel2007及以上版本文件。");
|
_remarkContentBuilder.AppendLine("2.导入数据时不能超过5000行。");
|
_remarkContentBuilder.AppendLine("");
|
_remarkContentBuilder.AppendLine("");
|
|
Dictionary<string, string> typeNameDict = new()
|
{
|
{ "string", "输入文本。"},
|
{ "int", "输入数字。"},
|
{ "long", "输入数字。"},
|
{ "decimal", "输入小数。"},
|
{ "bool", "是,否。"},
|
|
};
|
Dictionary<string, string> effectTypeDict = new()
|
{
|
{ "datepicker", "日期。 示例: 2023/3/1。"},
|
{ "datetimepicker", "日期时间。示例: 2023/3/1 12:00:00。"},
|
|
};
|
|
requiredTableFieldList.ForEach(x =>
|
{
|
string text = "";
|
text += x.ColumnComment + "(" + (x.WhetherRequired.Equals("Y") ? "必填" : "非必填") + "): ";
|
var _dataType = x.DataType.Replace("?", "").ToLower();
|
if (effectTypeDict.ContainsKey(x.EffectType.ToLower()))
|
{
|
text += effectTypeDict[x.EffectType.ToLower()];
|
}
|
else if (x.EffectType.ToLower() == "enumselector")
|
{//注意：这个判断要放在判断_dataType前面，主要是考虑枚举和int类型的混淆
|
var queryValue = x.NetType.Split('.').Last();
|
var enumStr = _sysEnumService.GetEnumDataListStr(new EnumDataInput() { EnumName = queryValue });
|
text += enumStr + "。";
|
}
|
else if (typeNameDict.ContainsKey(_dataType))
|
{
|
text += typeNameDict[_dataType];
|
}
|
else
|
{
|
text += ("注意：类型未能识别出来，需要自己维护！！！。");
|
}
|
_remarkContentBuilder.AppendLine(text);
|
});
|
|
return _remarkContentBuilder;
|
|
}
|
|
/// <summary>
|
/// 根据属性名称判断是否是枚举类型
|
/// </summary>
|
/// <param name="type"></param>
|
/// <param name="propertyName"></param>
|
/// <returns></returns>
|
/// <exception cref="ArgumentException"></exception>
|
public static bool IsEnumProperty(Type type, string propertyName)
|
{
|
PropertyInfo propertyInfo = type.GetProperty(propertyName);
|
if (propertyInfo == null)
|
{
|
throw new ArgumentException("Property not found", nameof(propertyName));
|
}
|
return propertyInfo.PropertyType.IsEnum;
|
}
|
|
#endregion
|
}
|
