using System.IO.Compression;
namespace Admin.NET.Core.Service;
///
/// 系统代码生成器服务
///
[ApiDescriptionSettings(Order = 270)]
public class SysCodeGenService : IDynamicApiController, ITransient
{
private readonly ISqlSugarClient _db;
private readonly SysCodeGenConfigService _codeGenConfigService;
private readonly IViewEngine _viewEngine;
private readonly CodeGenOptions _codeGenOptions;
public SysCodeGenService(ISqlSugarClient db,
SysCodeGenConfigService codeGenConfigService,
IViewEngine viewEngine,
IOptions codeGenOptions)
{
_db = db;
_codeGenConfigService = codeGenConfigService;
_viewEngine = viewEngine;
_codeGenOptions = codeGenOptions.Value;
}
///
/// 获取代码生成分页列表
///
///
///
[DisplayName("获取代码生成分页列表")]
public async Task> Page(CodeGenInput input)
{
return await _db.Queryable()
.WhereIF(!string.IsNullOrWhiteSpace(input.TableName), u => u.TableName.Contains(input.TableName.Trim()))
.WhereIF(!string.IsNullOrWhiteSpace(input.BusName), u => u.BusName.Contains(input.BusName.Trim()))
.ToPagedListAsync(input.Page, input.PageSize);
}
///
/// 增加代码生成
///
///
///
[ApiDescriptionSettings(Name = "Add"), HttpPost]
[DisplayName("增加代码生成")]
public async Task AddCodeGen(AddCodeGenInput input)
{
var isExist = await _db.Queryable().Where(u => u.TableName == input.TableName).AnyAsync();
if (isExist)
throw Oops.Oh(ErrorCodeEnum.D1400);
var codeGen = input.Adapt();
var newCodeGen = await _db.Insertable(codeGen).ExecuteReturnEntityAsync();
// 加入配置表中
_codeGenConfigService.AddList(GetColumnList(input), newCodeGen);
}
///
/// 更新代码生成
///
///
///
[ApiDescriptionSettings(Name = "Update"), HttpPost]
[DisplayName("更新代码生成")]
public async Task UpdateCodeGen(UpdateCodeGenInput input)
{
var isExist = await _db.Queryable().AnyAsync(u => u.TableName == input.TableName && u.Id != input.Id);
if (isExist)
throw Oops.Oh(ErrorCodeEnum.D1400);
await _db.Updateable(input.Adapt()).ExecuteCommandAsync();
}
///
/// 删除代码生成
///
///
///
[ApiDescriptionSettings(Name = "Delete"), HttpPost]
[DisplayName("删除代码生成")]
public async Task DeleteCodeGen(List inputs)
{
if (inputs == null || inputs.Count < 1) return;
var codeGenConfigTaskList = new List();
inputs.ForEach(u =>
{
_db.Deleteable().In(u.Id).ExecuteCommand();
// 删除配置表中
codeGenConfigTaskList.Add(_codeGenConfigService.DeleteCodeGenConfig(u.Id));
});
await Task.WhenAll(codeGenConfigTaskList);
}
///
/// 获取代码生成详情
///
///
///
[DisplayName("获取代码生成详情")]
public async Task GetDetail([FromQuery] QueryCodeGenInput input)
{
return await _db.Queryable().SingleAsync(m => m.Id == input.Id);
}
///
/// 获取数据库库集合
///
///
[DisplayName("获取数据库库集合")]
public async Task> GetDatabaseList()
{
var dbCongigs = App.GetOptions().ConnectionConfigs;
return await Task.FromResult(dbCongigs.Adapt>());
}
///
/// 获取数据库表(实体)集合
///
///
[DisplayName("获取数据库表(实体)集合")]
public async Task> GetTableList(string configId = SqlSugarConst.ConfigId)
{
var provider = _db.AsTenant().GetConnectionScope(configId);
var dbTableInfos = provider.DbMaintenance.GetTableInfoList(false); // 不能走缓存,否则切库不起作用
var config = App.GetOptions().ConnectionConfigs.FirstOrDefault(u => u.ConfigId == configId);
var dbTableNames = dbTableInfos.Select(u => u.Name.ToLower()).ToList();
IEnumerable entityInfos = await GetEntityInfos();
var tableOutputList = new List();
foreach (var item in entityInfos)
{
var table = dbTableInfos.FirstOrDefault(x => x.Name.ToLower() == (config.EnableUnderLine ? UtilMethods.ToUnderLine(item.DbTableName) : item.DbTableName).ToLower());
if (table == null) continue;
tableOutputList.Add(new TableOutput
{
ConfigId = configId,
EntityName = item.EntityName,
TableName = table.Name,
TableComment = item.TableDescription
});
}
return tableOutputList;
//return entityInfos.Where(u => dbTableNames.Contains(config.EnableUnderLine ? UtilMethods.ToUnderLine(u.DbTableName) : u.DbTableName.ToLower())).Select(u => new TableOutput()
//{
// ConfigId = configId,
// EntityName = u.EntityName,
// TableName = config.EnableUnderLine ? UtilMethods.ToUnderLine(u.DbTableName) : u.DbTableName,
// TableComment = u.TableDescription
//}).ToList();
}
///
/// 根据表名获取列集合
///
///
[DisplayName("根据表名获取列集合")]
public List GetColumnListByTableName([Required] string tableName, string configId = SqlSugarConst.ConfigId)
{
// 切库---多库代码生成用
var provider = _db.AsTenant().GetConnectionScope(configId);
var config = App.GetOptions().ConnectionConfigs.FirstOrDefault(u => u.ConfigId == 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.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
ColumnKey = u.IsPrimarykey.ToString(),
DataType = u.DataType.ToString(),
NetType = CodeGenUtil.ConvertDataType(u),
ColumnComment = u.ColumnDescription
}).ToList();
}
///
/// 获取数据表列(实体属性)集合
///
///
private List GetColumnList([FromQuery] AddCodeGenInput input)
{
var entityType = GetEntityInfos().GetAwaiter().GetResult().FirstOrDefault(m => m.EntityName == input.TableName);
if (entityType == null)
return null;
// 切库---多库代码生成用
var provider = _db.AsTenant().GetConnectionScope(!string.IsNullOrEmpty(input.ConfigId) ? input.ConfigId : SqlSugarConst.ConfigId);
var config = App.GetOptions().ConnectionConfigs.FirstOrDefault(u => u.ConfigId == input.ConfigId);
var dbTableName = config.EnableUnderLine ? UtilMethods.ToUnderLine(entityType.DbTableName) : entityType.DbTableName;
var entityBasePropertyNames = _codeGenOptions.EntityBaseColumn[nameof(EntityTenant)];
return provider.DbMaintenance.GetColumnInfosByTableName(dbTableName, false).Select(u => new ColumnOuput
{
//转下划线后的列名 需要转回来
ColumnName = config.EnableUnderLine ? CodeGenUtil.CamelColumnName(u.DbColumnName, entityBasePropertyNames) : u.DbColumnName,
ColumnKey = u.IsPrimarykey.ToString(),
NetType = CodeGenUtil.ConvertDataType(u),
DataType = CodeGenUtil.ConvertDataType(u),
ColumnComment = string.IsNullOrWhiteSpace(u.ColumnDescription) ? u.DbColumnName : u.ColumnDescription
}).ToList();
}
///
/// 获取库表信息
///
///
private async Task> GetEntityInfos()
{
var entityInfos = new List();
var type = typeof(SugarTable);
var types = new List();
if (_codeGenOptions.EntityAssemblyNames != null)
{
foreach (var assemblyName in _codeGenOptions.EntityAssemblyNames)
{
Assembly asm = Assembly.Load(assemblyName);
types.AddRange(asm.GetExportedTypes().ToList());
}
}
Func IsMyAttribute = o =>
{
foreach (Attribute a in o)
{
if (a.GetType() == type)
return true;
}
return false;
};
Type[] cosType = types.Where(o =>
{
return IsMyAttribute(Attribute.GetCustomAttributes(o, true));
}
).ToArray();
foreach (var c in cosType)
{
var sugarAttribute = c.GetCustomAttributes(type, true)?.FirstOrDefault();
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 : ((SugarTable)sugarAttribute).TableName,
TableDescription = description
});
}
return await Task.FromResult(entityInfos);
}
///
/// 代码生成到本地
///
///
[DisplayName("代码生成到本地")]
public async Task RunLocal(SysCodeGen input)
{
if (string.IsNullOrEmpty(input.GenerateType))
input.GenerateType = "200";
// 先删除该表已生成的菜单列表
var templatePathList = GetTemplatePathList(input);
List 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").ToList(); // 需要连表查询的字段
(string joinTableNames, string lowerJoinTableNames) = GetJoinTableStr(joinTableList); // 获取连表的实体名和别名
var data = new CustomViewEngine(_db)
{
ConfigId = input.ConfigId,
AuthorName = input.AuthorName,
BusName = input.BusName,
NameSpace = input.NameSpace,
ClassName = input.TableName,
ProjectLastName = input.NameSpace.Split('.').Last(),
QueryWhetherList = queryWhetherList,
TableField = tableFieldList,
IsJoinTable = joinTableList.Count > 0,
IsUpload = joinTableList.Where(u => u.EffectType == "Upload").Any(),
};
for (var i = 0; i < templatePathList.Count; i++)
{
var tContent = File.ReadAllText(templatePathList[i]);
var tResult = _viewEngine.RunCompile(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);
}
await AddMenu(input.TableName, input.BusName, input.MenuPid);
// 非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 = $"{CommonUtil.GetLocalhost()}/CodeGen/{input.TableName}.zip" };
}
}
///
/// 获取连表的实体名和别名
///
///
///
private (string, string) GetJoinTableStr(List configs)
{
var uploads = configs.Where(u => u.EffectType == "Upload").ToList();
var fks = configs.Where(u => u.EffectType == "fk").ToList();
string str = ""; //
string lowerStr = ""; //(o, i, c)
foreach (var item in uploads)
{
lowerStr += "sysFile_FK_" + item.LowerColumnName + ",";
str += "SysFile,";
}
foreach (var item in fks)
{
lowerStr += item.LowerFkEntityName + "_FK_" + item.LowerFkColumnName + ",";
str += item.FkEntityName + ",";
}
return (str.TrimEnd(','), lowerStr.TrimEnd(','));
}
///
/// 增加菜单
///
///
///
///
///
private async Task AddMenu(string className, string busName, long pid)
{
string pPath = null;//父路径
// 如果 pid 为 0 说明为顶级菜单, 需要创建顶级目录
if (pid == 0)
{
// 目录
var menuType0 = new SysMenu
{
Pid = 0,
Title = busName + "管理",
Type = MenuTypeEnum.Dir,
Icon = "robot",
Path = "/" + className.ToLower(),
Component = "LAYOUT",
};
{ //如果之前存在那么就删除本级和下级
var list = await _db.Queryable().Where(e => e.Title == menuType0.Title && e.Type == menuType0.Type).ToListAsync();
if (list.Count > 0)
{
var listIds = list.Select(f => f.Id).ToList();
var _ChildlistIds = new List();
foreach (var item in listIds)
{
var _Childlist = await _db.Queryable().ToChildListAsync(u => u.Pid, item);
_ChildlistIds.AddRange(_Childlist.Select(f => f.Id).ToList());
}
listIds.AddRange(_ChildlistIds);
await _db.Deleteable().Where(e => listIds.Contains(e.Id)).ExecuteCommandAsync();
await _db.Deleteable().Where(e => listIds.Contains(e.MenuId)).ExecuteCommandAsync();
}
}
pid = (await _db.Insertable(menuType0).ExecuteReturnEntityAsync()).Id;
}
// 由于后续菜单会有修改, 需要判断下 pid 是否存在, 不存在报错
else
{
var pMenu = await _db.Queryable().FirstAsync(e => e.Id == pid);
if (pMenu == null)
throw Oops.Oh(ErrorCodeEnum.D1505);
else
{
pPath = pMenu.Path;
}
}
// 菜单
var menuType1 = new SysMenu
{
Pid = pid,
Title = busName + "管理",
Name = className + "Management",
Type = MenuTypeEnum.Menu,
Path = pPath + "/" + className.ToLower(),
Component = "/main/" + className[..1].ToLower() + className[1..] + "/index",
};
{ // 如果之前存在那么就删除本级和下级
var list = await _db.Queryable().Where(e => e.Title == menuType1.Title && e.Type == menuType1.Type).ToListAsync();
if (list.Count > 0)
{
var listIds = list.Select(f => f.Id).ToList();
var _ChildlistIds = new List();
foreach (var item in listIds)
{
var _Childlist = await _db.Queryable().ToChildListAsync(u => u.Pid, item);
_ChildlistIds.AddRange(_Childlist.Select(f => f.Id).ToList());
}
listIds.AddRange(_ChildlistIds);
await _db.Deleteable().Where(e => listIds.Contains(e.Id)).ExecuteCommandAsync();
await _db.Deleteable().Where(e => listIds.Contains(e.MenuId)).ExecuteCommandAsync();
}
}
var pid1 = (await _db.Insertable(menuType1).ExecuteReturnEntityAsync()).Id;
// 按钮-page
var menuType2 = new SysMenu
{
Pid = pid1,
Title = busName + "查询",
Type = MenuTypeEnum.Btn,
Permission = className[..1].ToLower() + className[1..] + ":page",
};
// 按钮-detail
var menuType2_1 = new SysMenu
{
Pid = pid1,
Title = busName + "详情",
Type = MenuTypeEnum.Btn,
Permission = className[..1].ToLower() + className[1..] + ":detail",
};
// 按钮-add
var menuType2_2 = new SysMenu
{
Pid = pid1,
Title = busName + "增加",
Type = MenuTypeEnum.Btn,
Permission = className[..1].ToLower() + className[1..] + ":add",
};
// 按钮-delete
var menuType2_3 = new SysMenu
{
Pid = pid1,
Title = busName + "删除",
Type = MenuTypeEnum.Btn,
Permission = className[..1].ToLower() + className[1..] + ":delete",
};
// 按钮-edit
var menuType2_4 = new SysMenu
{
Pid = pid1,
Title = busName + "编辑",
Type = MenuTypeEnum.Btn,
Permission = className[..1].ToLower() + className[1..] + ":edit",
};
var menuList = new List() { menuType2, menuType2_1, menuType2_2, menuType2_3, menuType2_4 };
await _db.Insertable(menuList).ExecuteCommandAsync();
}
///
/// 获取模板文件路径集合
///
///
private static List GetTemplatePathList(SysCodeGen input)
{
var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "Template");
if (input.GenerateType.Substring(1, 1).Contains('1'))
return new List()
{
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()
{
Path.Combine(templatePath , "Service.cs.vm"),
Path.Combine(templatePath , "Input.cs.vm"),
Path.Combine(templatePath , "Output.cs.vm"),
Path.Combine(templatePath , "Dto.cs.vm"),
};
else
return new List()
{
Path.Combine(templatePath , "Service.cs.vm"),
Path.Combine(templatePath , "Input.cs.vm"),
Path.Combine(templatePath , "Output.cs.vm"),
Path.Combine(templatePath , "Dto.cs.vm"),
Path.Combine(templatePath , "index.vue.vm"),
Path.Combine(templatePath , "editDialog.vue.vm"),
Path.Combine(templatePath , "manage.js.vm"),
};
}
///
/// 获取模板文件路径集合
///
///
private static List GetTemplatePathList()
{
var templatePath = Path.Combine(App.WebHostEnvironment.WebRootPath, "Template");
return new List()
{
Path.Combine(templatePath , "Service.cs.vm"),
Path.Combine(templatePath , "Input.cs.vm"),
Path.Combine(templatePath , "Output.cs.vm"),
Path.Combine(templatePath , "Dto.cs.vm"),
Path.Combine(templatePath , "index.vue.vm"),
Path.Combine(templatePath , "editDialog.vue.vm"),
Path.Combine(templatePath , "manage.js.vm"),
};
}
///
/// 设置生成文件路径
///
///
///
private List GetTargetPathList(SysCodeGen input)
{
var backendPath = Path.Combine(new DirectoryInfo(App.WebHostEnvironment.ContentRootPath).Parent.FullName, _codeGenOptions.BackendApplicationNamespace, "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");
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");
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.TableName[..1].ToLower() + input.TableName[1..] + ".ts");
return new List()
{
servicePath,
inputPath,
outputPath,
viewPath,
indexPath,
formModalPath,
apiJsPath
};
}
///
/// 设置生成文件路径
///
///
///
private List GetZipPathList(SysCodeGen input)
{
var zipPath = Path.Combine(App.WebHostEnvironment.WebRootPath, "CodeGen", input.TableName);
var backendPath = Path.Combine(zipPath, _codeGenOptions.BackendApplicationNamespace, "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");
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()
{
indexPath,
formModalPath,
apiJsPath
};
else if (input.GenerateType.StartsWith("12"))
return new List()
{
servicePath,
inputPath,
outputPath,
viewPath
};
else
return new List()
{
servicePath,
inputPath,
outputPath,
viewPath,
indexPath,
formModalPath,
apiJsPath
};
}
}