290 lines
10 KiB
C#
290 lines
10 KiB
C#
// 大名科技（天津）有限公司版权所有 电话：18020030720 QQ：515096995
|
||
//
|
||
// 此源代码遵循位于源代码树根目录中的 LICENSE 文件的许可证
|
||
|
||
namespace Admin.NET.Core.Service;
|
||
|
||
/// <summary>
|
||
/// 系统菜单服务
|
||
/// </summary>
|
||
[ApiDescriptionSettings(Order = 450)]
|
||
public class SysMenuService : IDynamicApiController, ITransient
|
||
{
|
||
private readonly UserManager _userManager;
|
||
private readonly SqlSugarRepository<SysMenu> _sysMenuRep;
|
||
private readonly SysRoleMenuService _sysRoleMenuService;
|
||
private readonly SysUserRoleService _sysUserRoleService;
|
||
private readonly SysCacheService _sysCacheService;
|
||
|
||
public SysMenuService(UserManager userManager,
|
||
SqlSugarRepository<SysMenu> sysMenuRep,
|
||
SysRoleMenuService sysRoleMenuService,
|
||
SysUserRoleService sysUserRoleService,
|
||
SysCacheService sysCacheService)
|
||
{
|
||
_userManager = userManager;
|
||
_sysMenuRep = sysMenuRep;
|
||
_sysRoleMenuService = sysRoleMenuService;
|
||
_sysUserRoleService = sysUserRoleService;
|
||
_sysCacheService = sysCacheService;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取登录菜单树
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
[DisplayName("获取登录菜单树")]
|
||
public async Task<List<MenuOutput>> GetLoginMenuTree()
|
||
{
|
||
if (_userManager.SuperAdmin)
|
||
{
|
||
var menuList = await _sysMenuRep.AsQueryable()
|
||
.Where(u => u.Type != MenuTypeEnum.Btn && u.Status == StatusEnum.Enable)
|
||
.OrderBy(u => new { u.OrderNo, u.Id }).ToTreeAsync(u => u.Children, u => u.Pid, 0);
|
||
return menuList.Adapt<List<MenuOutput>>();
|
||
}
|
||
else
|
||
{
|
||
var menuIdList = await GetMenuIdList();
|
||
var menuTree = await _sysMenuRep.AsQueryable()
|
||
.Where(u => u.Status == StatusEnum.Enable)
|
||
.OrderBy(u => new { u.OrderNo, u.Id }).ToTreeAsync(u => u.Children, u => u.Pid, 0, menuIdList.Select(d => (object)d).ToArray());
|
||
DeleteBtnFromMenuTree(menuTree);
|
||
return menuTree.Adapt<List<MenuOutput>>();
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 删除登录菜单树里面的按钮
|
||
/// </summary>
|
||
private void DeleteBtnFromMenuTree(List<SysMenu> menuList)
|
||
{
|
||
if (menuList == null) return;
|
||
for (var i = menuList.Count - 1; i >= 0; i--)
|
||
{
|
||
var menu = menuList[i];
|
||
if (menu.Type == MenuTypeEnum.Btn)
|
||
menuList.Remove(menu);
|
||
else if (menu.Children.Count > 0)
|
||
DeleteBtnFromMenuTree(menu.Children);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取菜单列表
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
[AllowAnonymous]
|
||
[DisplayName("获取菜单列表")]
|
||
public async Task<List<SysMenu>> GetList([FromQuery] MenuInput input)
|
||
{
|
||
var menuIdList = _userManager.SuperAdmin ? new List<long>() : await GetMenuIdList();
|
||
|
||
// 有筛选条件时返回list列表（防止构造不出树）
|
||
if (!string.IsNullOrWhiteSpace(input.Title) || input.Type is > 0)
|
||
{
|
||
return await _sysMenuRep.AsQueryable()
|
||
.WhereIF(!string.IsNullOrWhiteSpace(input.Title), u => u.Title.Contains(input.Title))
|
||
.WhereIF(input.Type is > 0, u => u.Type == input.Type)
|
||
.WhereIF(menuIdList.Count > 1, u => menuIdList.Contains(u.Id))
|
||
.OrderBy(u => u.OrderNo).ToListAsync();
|
||
}
|
||
|
||
return _userManager.SuperAdmin ?
|
||
await _sysMenuRep.AsQueryable().OrderBy(u => u.OrderNo).ToTreeAsync(u => u.Children, u => u.Pid, 0) :
|
||
await _sysMenuRep.AsQueryable()
|
||
.OrderBy(u => u.OrderNo).ToTreeAsync(u => u.Children, u => u.Pid, 0, menuIdList.Select(d => (object)d).ToArray()); // 角色菜单授权时
|
||
}
|
||
|
||
/// <summary>
|
||
/// 增加菜单
|
||
/// </summary>
|
||
/// <param name="input"></param>
|
||
/// <returns></returns>
|
||
[ApiDescriptionSettings(Name = "Add"), HttpPost]
|
||
[DisplayName("增加菜单")]
|
||
public async Task AddMenu(AddMenuInput input)
|
||
{
|
||
var isExist = input.Type != MenuTypeEnum.Btn
|
||
? await _sysMenuRep.IsAnyAsync(u => u.Title == input.Title)
|
||
: await _sysMenuRep.IsAnyAsync(u => u.Permission == input.Permission);
|
||
if (isExist)
|
||
throw Oops.Oh(ErrorCodeEnum.D4000);
|
||
|
||
if (!string.IsNullOrWhiteSpace(input.Name))
|
||
{
|
||
if (await _sysMenuRep.IsAnyAsync(u => u.Name == input.Name))
|
||
throw Oops.Oh(ErrorCodeEnum.D4009);
|
||
}
|
||
|
||
if (input.Pid != 0)
|
||
{
|
||
if (await _sysMenuRep.IsAnyAsync(u => u.Id == input.Pid && u.Type == MenuTypeEnum.Btn))
|
||
throw Oops.Oh(ErrorCodeEnum.D4010);
|
||
}
|
||
|
||
// 校验菜单参数
|
||
var sysMenu = input.Adapt<SysMenu>();
|
||
CheckMenuParam(sysMenu);
|
||
|
||
await _sysMenuRep.InsertAsync(sysMenu);
|
||
|
||
// 清除缓存
|
||
DeleteMenuCache();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 更新菜单
|
||
/// </summary>
|
||
/// <param name="input"></param>
|
||
/// <returns></returns>
|
||
[ApiDescriptionSettings(Name = "Update"), HttpPost]
|
||
[DisplayName("更新菜单")]
|
||
public async Task UpdateMenu(UpdateMenuInput input)
|
||
{
|
||
if (input.Id == input.Pid)
|
||
throw Oops.Oh(ErrorCodeEnum.D4008);
|
||
|
||
var isExist = input.Type != MenuTypeEnum.Btn
|
||
? await _sysMenuRep.IsAnyAsync(u => u.Title == input.Title && u.Type == input.Type && u.Id != input.Id)
|
||
: await _sysMenuRep.IsAnyAsync(u => u.Permission == input.Permission && u.Type == input.Type && u.Id != input.Id);
|
||
if (isExist)
|
||
throw Oops.Oh(ErrorCodeEnum.D4000);
|
||
|
||
if (!string.IsNullOrWhiteSpace(input.Name))
|
||
{
|
||
if (await _sysMenuRep.IsAnyAsync(u => u.Id != input.Id && u.Name == input.Name))
|
||
throw Oops.Oh(ErrorCodeEnum.D4009);
|
||
}
|
||
|
||
if (input.Pid != 0)
|
||
{
|
||
if (await _sysMenuRep.IsAnyAsync(u => u.Id == input.Pid && u.Type == MenuTypeEnum.Btn))
|
||
throw Oops.Oh(ErrorCodeEnum.D4010);
|
||
}
|
||
|
||
// 校验菜单参数
|
||
var sysMenu = input.Adapt<SysMenu>();
|
||
CheckMenuParam(sysMenu);
|
||
|
||
await _sysMenuRep.AsUpdateable(sysMenu).ExecuteCommandAsync();
|
||
|
||
// 清除缓存
|
||
DeleteMenuCache();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 删除菜单
|
||
/// </summary>
|
||
/// <param name="input"></param>
|
||
/// <returns></returns>
|
||
[UnitOfWork]
|
||
[ApiDescriptionSettings(Name = "Delete"), HttpPost]
|
||
[DisplayName("删除菜单")]
|
||
public async Task DeleteMenu(DeleteMenuInput input)
|
||
{
|
||
var menuTreeList = await _sysMenuRep.AsQueryable().ToChildListAsync(u => u.Pid, input.Id, true);
|
||
var menuIdList = menuTreeList.Select(u => u.Id).ToList();
|
||
|
||
await _sysMenuRep.DeleteAsync(u => menuIdList.Contains(u.Id));
|
||
|
||
// 级联删除角色菜单数据
|
||
await _sysRoleMenuService.DeleteRoleMenuByMenuIdList(menuIdList);
|
||
|
||
// 清除缓存
|
||
DeleteMenuCache();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 增加和编辑时检查菜单数据
|
||
/// </summary>
|
||
/// <param name="menu"></param>
|
||
private static void CheckMenuParam(SysMenu menu)
|
||
{
|
||
var permission = menu.Permission;
|
||
if (menu.Type == MenuTypeEnum.Btn)
|
||
{
|
||
menu.Name = null;
|
||
menu.Path = null;
|
||
menu.Component = null;
|
||
menu.Icon = null;
|
||
menu.Redirect = null;
|
||
menu.OutLink = null;
|
||
menu.IsHide = false;
|
||
menu.IsKeepAlive = true;
|
||
menu.IsAffix = false;
|
||
menu.IsIframe = false;
|
||
|
||
if (string.IsNullOrEmpty(permission))
|
||
throw Oops.Oh(ErrorCodeEnum.D4003);
|
||
if (!permission.Contains(':'))
|
||
throw Oops.Oh(ErrorCodeEnum.D4004);
|
||
}
|
||
else
|
||
{
|
||
menu.Permission = null;
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取用户拥有按钮权限集合（缓存）
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
[DisplayName("获取按钮权限集合")]
|
||
public async Task<List<string>> GetOwnBtnPermList()
|
||
{
|
||
var userId = _userManager.UserId;
|
||
var permissions = _sysCacheService.Get<List<string>>(CacheConst.KeyUserButton + userId);
|
||
if (permissions == null)
|
||
{
|
||
var menuIdList = _userManager.SuperAdmin ? new List<long>() : await GetMenuIdList();
|
||
permissions = menuIdList.Count > 0 || _userManager.SuperAdmin
|
||
? await _sysMenuRep.AsQueryable()
|
||
.Where(u => u.Type == MenuTypeEnum.Btn)
|
||
.WhereIF(menuIdList.Count > 0, u => menuIdList.Contains(u.Id))
|
||
.Select(u => u.Permission).ToListAsync()
|
||
: new List<string>();
|
||
_sysCacheService.Set(CacheConst.KeyUserButton + userId, permissions, TimeSpan.FromDays(7));
|
||
}
|
||
|
||
return permissions;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取系统所有按钮权限集合（缓存）
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
[NonAction]
|
||
public async Task<List<string>> GetAllBtnPermList()
|
||
{
|
||
var permissions = _sysCacheService.Get<List<string>>(CacheConst.KeyUserButton + 0);
|
||
if (permissions == null || permissions.Count == 0)
|
||
{
|
||
permissions = await _sysMenuRep.AsQueryable()
|
||
.Where(u => u.Type == MenuTypeEnum.Btn)
|
||
.Select(u => u.Permission).ToListAsync();
|
||
_sysCacheService.Set(CacheConst.KeyUserButton + 0, permissions);
|
||
}
|
||
|
||
return permissions;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 清除菜单和按钮缓存
|
||
/// </summary>
|
||
private void DeleteMenuCache()
|
||
{
|
||
// _sysCacheService.RemoveByPrefixKey(CacheConst.KeyUserMenu);
|
||
_sysCacheService.RemoveByPrefixKey(CacheConst.KeyUserButton);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取当前用户菜单Id集合
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
private async Task<List<long>> GetMenuIdList()
|
||
{
|
||
var roleIdList = await _sysUserRoleService.GetUserRoleIdList(_userManager.UserId);
|
||
return await _sysRoleMenuService.GetRoleMenuIdList(roleIdList);
|
||
}
|
||
} |