59 lines
3.7 KiB
C#
59 lines
3.7 KiB
C#
// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
|
||
//
|
||
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
|
||
//
|
||
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！
|
||
|
||
namespace Admin.NET.Core.CodeGen;
|
||
|
||
/// <summary>
|
||
/// 表实体代码生成策略类
|
||
/// </summary>
|
||
[CodeGenStrategy(CodeGenSceneEnum.TableEntity)]
|
||
public class TableEntityCodeGenStrategy : CodeGenEntityStrategyBase<CreateEntityInput>, ISingleton
|
||
{
|
||
private readonly TemplateContextOutput _template = new() { Name = "Entity.cs.vm", OutPath = "Entity/{ModuleName}.cs" };
|
||
|
||
public override async Task<List<TemplateContextOutput>> GenerateCode(CreateEntityInput input)
|
||
{
|
||
var config = _dbConnectionOption.ConnectionConfigs.FirstOrDefault(u => u.ConfigId.ToString() == input.ConfigId) ?? throw Oops.Oh(ErrorCodeEnum.db1004);
|
||
input.EntityName = string.IsNullOrWhiteSpace(input.EntityName) ? (config.DbSettings.EnableUnderLine ? CodeGenHelper.CamelColumnName(input.TableName, null) : input.TableName) : input.EntityName;
|
||
input.Position = string.IsNullOrWhiteSpace(input.Position) ? "X.Application" : input.Position;
|
||
|
||
// Entity.cs.vm中是允许创建没有基类的实体的，所以这里也要做出相同的判断
|
||
string[] dbColumnNames = [];
|
||
if (!string.IsNullOrWhiteSpace(input.BaseClassName))
|
||
{
|
||
_codeGenOption.EntityBaseColumn.TryGetValue(input.BaseClassName, out dbColumnNames);
|
||
if (dbColumnNames is null or { Length: 0 }) throw Oops.Oh("基类配置文件不存在此类型");
|
||
}
|
||
|
||
var db = _db.AsTenant().GetConnectionScope(input.ConfigId);
|
||
var dbColumnInfos = db.DbMaintenance.GetColumnInfosByTableName(input.TableName, false);
|
||
dbColumnInfos.ForEach(u =>
|
||
{
|
||
u.PropertyName = config.DbSettings.EnableUnderLine ? CodeGenHelper.CamelColumnName(u.DbColumnName, dbColumnNames) : u.DbColumnName; // 转下划线后的列名需要再转回来
|
||
u.DataType = CodeGenHelper.ConvertDataType(u, config.DbType);
|
||
});
|
||
if (_codeGenOption.BaseEntityNames.Contains(input.BaseClassName, StringComparer.OrdinalIgnoreCase))
|
||
dbColumnInfos = dbColumnInfos.Where(u => !dbColumnNames.Contains(u.PropertyName, StringComparer.OrdinalIgnoreCase)).ToList();
|
||
|
||
var dbTableInfo = db.DbMaintenance.GetTableInfoList(false).FirstOrDefault(u => u.Name == input.TableName || u.Name == input.TableName.ToLower()) ?? throw Oops.Oh(ErrorCodeEnum.db1001);
|
||
|
||
var engine = new TableEntityEngine
|
||
{
|
||
NameSpace = $"{input.Position}.Entity",
|
||
TableName = input.TableName,
|
||
EntityName = input.EntityName,
|
||
BaseClassName = string.IsNullOrWhiteSpace(input.BaseClassName) ? "" : $": {input.BaseClassName}",
|
||
ConfigId = input.ConfigId,
|
||
Description = string.IsNullOrWhiteSpace(dbTableInfo.Description)
|
||
? input.EntityName + "业务表"
|
||
: dbTableInfo.Description,
|
||
TableFields = dbColumnInfos
|
||
};
|
||
var result = await RenderAsync(_template, (context, builder) => _viewEngine.RunCompile(context, engine, builder));
|
||
result.OutPath = Path.Combine(Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent!.FullName, input.Position), result.OutPath.Replace("{ModuleName}", input.EntityName));
|
||
return [result];
|
||
}
|
||
} |