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);
|
|||
|
|
}
|
|||
|
|
}
|