namespace Admin.NET.Core.Service;
///
/// 系统菜单服务
///
[ApiDescriptionSettings(Name = "系统菜单", Order = 195)]
public class SysMenuService : IDynamicApiController, ITransient
{
private readonly SqlSugarRepository _sysMenuRep;
private readonly IUserManager _userManager;
private readonly SysRoleMenuService _sysRoleMenuService;
private readonly SysUserRoleService _sysUserRoleService;
private readonly ISysCacheService _sysCacheService;
public SysMenuService(SqlSugarRepository sysMenuRep,
IUserManager userManager,
SysRoleMenuService sysRoleMenuService,
SysUserRoleService sysUserRoleService,
ISysCacheService sysCacheService)
{
_sysMenuRep = sysMenuRep;
_userManager = userManager;
_sysRoleMenuService = sysRoleMenuService;
_sysUserRoleService = sysUserRoleService;
_sysCacheService = sysCacheService;
}
///
/// 获取登录菜单树
///
///
[HttpGet("/getMenuList")]
public async Task> GetLoginMenuTree()
{
if (_userManager.SuperAdmin)
{
var menuList = await _sysMenuRep.AsQueryable()
.Where(u => u.Type != MenuTypeEnum.Btn)
.OrderBy(u => u.OrderNo).ToTreeAsync(u => u.Children, u => u.Pid, 0);
return menuList.Adapt>();
}
else
{
var menuIdList = await GetMenuIdList();
var menuList = await _sysMenuRep.AsQueryable()
.Where(u => u.Type != MenuTypeEnum.Btn)
.Where(u => menuIdList.Contains(u.Id))
.OrderBy(u => u.OrderNo).ToTreeAsync(u => u.Children, u => u.Pid, 0);
return menuList.Adapt>();
}
}
///
/// 获取菜单列表
///
///
[HttpGet("/sysMenu/list")]
public async Task> GetMenuList([FromQuery] MenuInput input)
{
var menuIdList = new List();
if (!_userManager.SuperAdmin)
menuIdList = await GetMenuIdList();
if (!string.IsNullOrWhiteSpace(input.Title) || input.Type > 0)
{
return await _sysMenuRep.AsQueryable()
.WhereIF(input.Type > 0, u => u.Type == (MenuTypeEnum)input.Type)
.WhereIF(menuIdList.Count > 1, u => menuIdList.Contains(u.Id))
.WhereIF(!string.IsNullOrWhiteSpace(input.Title), u => u.Title.Contains(input.Title))
.OrderBy(u => u.OrderNo).ToListAsync();
}
return await _sysMenuRep.AsQueryable()
.WhereIF(menuIdList.Count > 1, u => menuIdList.Contains(u.Id))
.OrderBy(u => u.OrderNo)
.ToTreeAsync(u => u.Children, u => u.Pid, 0);
}
///
/// 增加菜单
///
///
///
[HttpPost("/sysMenu/add")]
public async Task AddMenu(AddMenuInput input)
{
var isExist = input.Type != 3
? await _sysMenuRep.IsAnyAsync(u => u.Name == input.Name)
: await _sysMenuRep.IsAnyAsync(u => u.Permission == input.Permission);
if (isExist)
throw Oops.Oh(ErrorCodeEnum.D4000);
// 校验菜单参数
CheckMenuParam(input.Adapt());
await _sysMenuRep.InsertAsync(input.Adapt());
// 清除缓存
DeleteMenuCache();
}
///
/// 更新菜单
///
///
///
[HttpPost("/sysMenu/update"),]
public async Task UpdateMenu(UpdateMenuInput input)
{
var isExist = await _sysMenuRep.IsAnyAsync(u => u.Name == input.Name && u.Id != input.Id);
if (isExist)
throw Oops.Oh(ErrorCodeEnum.D4000);
// 校验菜单参数
CheckMenuParam(input.Adapt());
await _sysMenuRep.AsUpdateable(input.Adapt()).IgnoreColumns(true).ExecuteCommandAsync();
// 清除缓存
DeleteMenuCache();
}
///
/// 删除菜单
///
///
///
[HttpPost("/sysMenu/delete")]
[UnitOfWork]
public async Task DeleteMenu(DeleteMenuInput input)
{
var menuTreeList = await _sysMenuRep.AsQueryable().ToChildListAsync(u => u.Pid, input.Id);
var menuIdList = menuTreeList.Select(u => u.Id).ToList();
await _sysMenuRep.DeleteAsync(u => menuIdList.Contains(u.Id));
// 级联删除角色菜单数据
await _sysRoleMenuService.DeleteRoleMenuByMenuIdList(menuIdList);
// 清除缓存
DeleteMenuCache();
}
///
/// 增加和编辑时检查菜单数据
///
///
private static void CheckMenuParam(MenuInput input)
{
var type = input.Type;
//var component = input.Component;
var permission = input.Permission;
//if (type == (int)MenuTypeEnum.Dir || type == (int)MenuTypeEnum.Menu)
//{
// if (string.IsNullOrEmpty(component))
// throw Oops.Oh(ErrorCodeEnum.D4001);
//}
//else
if (type == (int)MenuTypeEnum.Btn)
{
if (string.IsNullOrEmpty(permission))
throw Oops.Oh(ErrorCodeEnum.D4003);
if (!permission.Contains(':'))
throw Oops.Oh(ErrorCodeEnum.D4004);
}
}
///
/// 获取按钮权限列表(登录)
///
///
[HttpGet("getPermCode")]
public async Task> GetPermCodeList()
{
var userId = _userManager.UserId;
var permissions = await _sysCacheService.GetPermission(userId); // 先从缓存里面读取
if (permissions == null || permissions.Count == 0)
{
if (_userManager.SuperAdmin)
{
permissions = await _sysMenuRep.AsQueryable()
.Where(u => u.Type == MenuTypeEnum.Btn)
.Select(u => u.Permission).ToListAsync();
}
else
{
var menuIdList = await GetMenuIdList();
permissions = await _sysMenuRep.AsQueryable()
.Where(u => u.Type == MenuTypeEnum.Btn)
.Where(u => menuIdList.Contains(u.Id))
.Select(u => u.Permission).ToListAsync();
}
await _sysCacheService.SetPermission(userId, permissions); // 缓存结果
}
return permissions;
}
///
/// 获取所有按钮权限集合
///
///
[NonAction]
public async Task> GetAllPermCodeList()
{
var permissions = await _sysCacheService.GetPermission(0); // 先从缓存里面读取
if (permissions == null || permissions.Count == 0)
{
permissions = await _sysMenuRep.AsQueryable()
.Where(u => u.Type == MenuTypeEnum.Btn)
.Select(u => u.Permission).ToListAsync();
await _sysCacheService.SetPermission(0, permissions); // 缓存结果
}
return permissions;
}
///
/// 清除菜单和按钮缓存
///
private async void DeleteMenuCache()
{
await _sysCacheService.DelByPatternAsync(CacheConst.KeyMenu);
await _sysCacheService.DelByPatternAsync(CacheConst.KeyPermission);
}
///
/// 获取当前用户菜单Id集合
///
///
private async Task> GetMenuIdList()
{
var roleIdList = await _sysUserRoleService.GetUserRoleIdList(_userManager.UserId);
return await _sysRoleMenuService.GetRoleMenuIdList(roleIdList);
}
}