using Furion.DependencyInjection;
using Furion.DynamicApiController;
using Furion.FriendlyException;
using Myshipping.Core.Entity;
using Mapster;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using SqlSugar;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
namespace Myshipping.Core.Service;
///
/// 系统菜单服务
///
[ApiDescriptionSettings(Name = "Menu", Order = 146)]
public class SysMenuService : ISysMenuService, IDynamicApiController, ITransient
{
private readonly SqlSugarRepository _sysMenuRep; // 菜单表仓储
private readonly ISysCacheService _sysCacheService;
private readonly ISysUserRoleService _sysUserRoleService;
private readonly ISysRoleMenuService _sysRoleMenuService;
public SysMenuService(SqlSugarRepository sysMenuRep,
ISysCacheService sysCacheService,
ISysUserRoleService sysUserRoleService,
ISysRoleMenuService sysRoleMenuService)
{
_sysMenuRep = sysMenuRep;
_sysCacheService = sysCacheService;
_sysUserRoleService = sysUserRoleService;
_sysRoleMenuService = sysRoleMenuService;
}
///
/// 获取用户权限(按钮权限标识集合)
///
///
///
[NonAction]
public async Task> GetLoginPermissionList(long userId)
{
var permissions = await _sysCacheService.GetPermission(userId); // 先从缓存里面读取
if (permissions == null || permissions.Count < 1)
{
if (!UserManager.IsSuperAdmin)
{
var roleIdList = await _sysUserRoleService.GetUserRoleIdList(userId);
var menuIdList = await _sysRoleMenuService.GetRoleMenuIdList(roleIdList);
permissions = await _sysMenuRep.Where(u => menuIdList.Contains(u.Id))
.Where(u => u.Type == (int)MenuType.BTN)
.Where(u => u.Status == (int)CommonStatus.ENABLE)
.Select(u => u.Permission).ToListAsync();
}
else
{
permissions = await _sysMenuRep.Where(u => u.Type == (int)MenuType.BTN)
.Where(u => u.Status == (int)CommonStatus.ENABLE)
.Select(u => u.Permission).ToListAsync();
}
await _sysCacheService.SetPermission(userId, permissions); // 缓存结果
}
return permissions;
}
///
/// 获取所有权限集合
///
///
[NonAction]
public async Task> GetAllPermission()
{
var permissions = await _sysCacheService.GetAllPermission(); // 先从缓存里面读取
if (permissions == null || permissions.Count < 1)
{
permissions = await _sysMenuRep.Where(u => u.Type == (int)MenuType.BTN)
.Where(u => u.Status == (int)CommonStatus.ENABLE)
.Select(u => u.Permission).ToListAsync();
await _sysCacheService.SetAllPermission(permissions); // 缓存结果
}
return permissions;
}
///
/// 获取用户AntDesign菜单集合
///
///
///
///
[NonAction]
public async Task> GetLoginMenusAntDesign(long userId, string appCode)
{
var antDesignTreeNodes = await _sysCacheService.GetMenu(userId, appCode); // 先从缓存里面读取
if (antDesignTreeNodes == null || antDesignTreeNodes.Count < 1)
{
var sysMenuList = new List();
// 管理员则展示所有系统菜单
if (UserManager.IsSuperAdmin)
{
sysMenuList = await _sysMenuRep
.Where(u => u.Status == (int)CommonStatus.ENABLE)
.WhereIF(!string.IsNullOrWhiteSpace(appCode), u => u.Application == appCode)
.Where(u => u.Type != (int)MenuType.BTN)
//.Where(u => u.Weight != (int)MenuWeight.DEFAULT_WEIGHT)
.OrderBy(u => u.Sort).ToListAsync();
}
else
{
// 非管理员则获取自己角色所拥有的菜单集合
var roleIdList = await _sysUserRoleService.GetUserRoleIdList(userId);
var menuIdList = await _sysRoleMenuService.GetRoleMenuIdList(roleIdList);
sysMenuList = await _sysMenuRep
.Where(u => menuIdList.Contains(u.Id))
.Where(u => u.Status == (int)CommonStatus.ENABLE)
.WhereIF(!string.IsNullOrWhiteSpace(appCode), u => u.Application == appCode)
.Where(u => u.Type != (int)MenuType.BTN)
.OrderBy(u => u.Sort).ToListAsync();
}
// 转换成登录菜单
antDesignTreeNodes = sysMenuList.Select(u => new AntDesignTreeNode
{
Application = u.Application,
Id = u.Id,
Pid = u.Pid,
Name = u.Code,
Component = u.Component,
Redirect = u.OpenType == (int)MenuOpenType.OUTER ? u.Link : u.Redirect,
Path = u.OpenType == (int)MenuOpenType.OUTER ? u.Link : u.Router,
Meta = new Meta
{
Title = u.Name,
Icon = u.Icon,
Show = u.Visible == YesOrNot.Y.ToString(),
Link = u.Link,
Target = u.OpenType == (int)MenuOpenType.OUTER ? "_blank" : ""
}
}).ToList();
await _sysCacheService.SetMenu(userId, appCode, antDesignTreeNodes); // 缓存结果
}
return antDesignTreeNodes;
}
///
/// 获取用户菜单所属的应用编码集合
///
///
///
[NonAction]
public async Task> GetUserMenuAppCodeList(long userId)
{
var roleIdList = await _sysUserRoleService.GetUserRoleIdList(userId);
var menuIdList = await _sysRoleMenuService.GetRoleMenuIdList(roleIdList);
return await _sysMenuRep.Where(u => menuIdList.Contains(u.Id))
.Where(u => u.Status == (int)CommonStatus.ENABLE)
.Select(u => u.Application).ToListAsync();
}
///
/// 系统菜单列表(树表)
///
///
///
[HttpGet("/sysMenu/list")]
public async Task GetMenuList([FromQuery] MenuInput input)
{
var menus = await _sysMenuRep.AsQueryable()
.WhereIF(!string.IsNullOrWhiteSpace(input.Application), u => u.Application == input.Application.Trim())
.WhereIF(!string.IsNullOrWhiteSpace(input.Name), u => u.Name.Contains(input.Name.Trim()))
.Where(u => u.Status == (int)CommonStatus.ENABLE).OrderBy(u => u.Sort)
.Select()
.ToListAsync();
return new TreeBuildUtil().DoTreeBuild(menus);
}
///
/// 创建Pids格式
/// 如果pid是0顶级节点,pids就是 [0];
/// 如果pid不是顶级节点,pids就是 pid菜单的 pids + [pid] + ,
///
///
///
private async Task CreateNewPids(long pid)
{
if (pid == 0L)
{
return "[0],";
}
else
{
var pmenu = await _sysMenuRep.FirstOrDefaultAsync(u => u.Id == pid);
return pmenu.Pids + "[" + pid + "],";
}
}
///
/// 增加和编辑时检查参数
///
///
private static void CheckMenuParam(MenuInput input)
{
var type = input.Type;
var router = input.Router;
var permission = input.Permission;
var openType = input.OpenType;
if (type.Equals((int)MenuType.DIR))
{
if (string.IsNullOrEmpty(router))
throw Oops.Oh(ErrorCode.D4001);
}
else if (type.Equals((int)MenuType.MENU))
{
if (string.IsNullOrEmpty(router))
throw Oops.Oh(ErrorCode.D4001);
if (string.IsNullOrEmpty(openType))
throw Oops.Oh(ErrorCode.D4002);
}
else if (type.Equals((int)MenuType.BTN))
{
if (string.IsNullOrEmpty(permission))
throw Oops.Oh(ErrorCode.D4003);
if (!permission.Contains(":"))
throw Oops.Oh(ErrorCode.D4004);
// 判断该资源是否存在
//permission = ":" + permission;
//var urlSet = resourceCache.getAllResources();
//if (!urlSet.Contains(permission.Replace(":","/")))
// throw Oops.Oh(ErrorCode.meu1005);
}
}
///
/// 增加系统菜单
///
///
///
[HttpPost("/sysMenu/add")]
public async Task AddMenu(AddMenuInput input)
{
var isExist = await _sysMenuRep.AnyAsync(u => u.Code == input.Code); // u.Name == input.Name
if (isExist)
throw Oops.Oh(ErrorCode.D4000);
// 校验参数
CheckMenuParam(input);
var menu = input.Adapt();
menu.Pids = await CreateNewPids(input.Pid);
menu.Status = (int)CommonStatus.ENABLE;
await _sysMenuRep.InsertAsync(menu);
// 清除缓存
await _sysCacheService.DelByPatternAsync(CommonConst.CACHE_KEY_MENU);
await _sysCacheService.DelByPatternAsync(CommonConst.CACHE_KEY_ALLPERMISSION);
}
///
/// 删除系统菜单
///
///
///
[HttpPost("/sysMenu/delete")]
public async Task DeleteMenu(DeleteMenuInput input)
{
var childIdList = await _sysMenuRep.Where(u => u.Pids.Contains(input.Id.ToString()))
.Select(u => u.Id).ToListAsync();
childIdList.Add(input.Id);
try
{
_sysMenuRep.CurrentBeginTran();
await _sysMenuRep.DeleteAsync(u => childIdList.Contains(u.Id));
// 级联删除该菜单及子菜单对应的角色-菜单表信息
await _sysRoleMenuService.DeleteRoleMenuListByMenuIdList(childIdList);
// 清除缓存
await _sysCacheService.DelByPatternAsync(CommonConst.CACHE_KEY_MENU);
await _sysCacheService.DelByPatternAsync(CommonConst.CACHE_KEY_ALLPERMISSION);
_sysMenuRep.CurrentCommitTran();
}
catch (System.Exception)
{
_sysMenuRep.CurrentRollbackTran();
throw;
}
}
///
/// 更新系统菜单
///
///
///
[HttpPost("/sysMenu/edit"),]
public async Task UpdateMenu(UpdateMenuInput input)
{
// Pid和Id不能一致,一致会导致无限递归
if (input.Id == input.Pid)
throw Oops.Oh(ErrorCode.D4006);
var isExist = await _sysMenuRep.AnyAsync(u => u.Code == input.Code && u.Id != input.Id); // u.Name == input.Name
if (isExist)
throw Oops.Oh(ErrorCode.D4000);
// 校验参数
CheckMenuParam(input);
var menuList = new List();
// 如果是编辑,父id不能为自己的子节点
var childIdList = await _sysMenuRep.Where(u => u.Pids.Contains(input.Id.ToString()))
.Select(u => u.Id).ToListAsync();
if (childIdList.Contains(input.Pid))
throw Oops.Oh(ErrorCode.D4006);
var oldMenu = await _sysMenuRep.FirstOrDefaultAsync(u => u.Id == input.Id);
// 生成新的pids
var newPids = await CreateNewPids(input.Pid);
// 是否更新子应用的标识
var updateSubAppsFlag = false;
// 是否更新子节点的pids的标识
var updateSubPidsFlag = false;
// 如果应用有变化
if (input.Application != oldMenu.Application)
{
// 父节点不是根节点不能移动应用
if (oldMenu.Pid != 0L)
throw Oops.Oh(ErrorCode.D4007);
updateSubAppsFlag = true;
}
// 父节点有变化
if (input.Pid != oldMenu.Pid)
updateSubPidsFlag = true;
// 开始更新所有子节点的配置
if (updateSubAppsFlag || updateSubPidsFlag)
{
// 查找所有叶子节点,包含子节点的子节点
menuList = await _sysMenuRep.Where(u => u.Pids.Contains(oldMenu.Id.ToString())).ToListAsync();
// 更新所有子节点的应用为当前菜单的应用
if (menuList.Count > 0)
{
// 更新所有子节点的application
if (updateSubAppsFlag)
{
menuList.ForEach(u =>
{
u.Application = input.Application;
});
}
// 更新所有子节点的pids
if (updateSubPidsFlag)
{
menuList.ForEach(u =>
{
// 子节点pids组成 = 当前菜单新pids + 当前菜单id + 子节点自己的pids后缀
var oldParentCodesPrefix = oldMenu.Pids + "[" + oldMenu.Id + "],";
var oldParentCodesSuffix = u.Pids.Substring(oldParentCodesPrefix.Length);
var menuParentCodes = newPids + "[" + oldMenu.Id + "]," + oldParentCodesSuffix;
u.Pids = menuParentCodes;
});
}
}
}
// 更新当前菜单
oldMenu = input.Adapt();
oldMenu.Pids = newPids;
menuList.Add(oldMenu);
await _sysMenuRep.UpdateAsync(menuList);
// 清除缓存
await _sysCacheService.DelByPatternAsync(CommonConst.CACHE_KEY_MENU);
await _sysCacheService.DelByPatternAsync(CommonConst.CACHE_KEY_ALLPERMISSION);
}
///
/// 获取系统菜单
///
///
///
[HttpPost("/sysMenu/detail")]
public async Task GetMenu(QueryMenuInput input)
{
return await _sysMenuRep.FirstOrDefaultAsync(u => u.Id == input.Id);
}
///
/// 获取系统菜单树,用于新增、编辑时选择上级节点
///
///
///
[HttpGet("/sysMenu/tree")]
public async Task GetMenuTree([FromQuery] MenuInput input)
{
var menus = await _sysMenuRep.AsQueryable()
.WhereIF(!string.IsNullOrWhiteSpace(input.Application), u => u.Application == input.Application.Trim())
.Where(u => u.Status == (int)CommonStatus.ENABLE)
.Where(u => u.Type == (int)MenuType.DIR || u.Type == (int)MenuType.MENU)
.OrderBy(u => u.Sort)
.Select(u => new MenuTreeOutput
{
Id = u.Id,
ParentId = u.Pid,
Value = u.Id.ToString(),
Title = u.Name,
Weight = u.Sort
}).ToListAsync();
return new TreeBuildUtil().DoTreeBuild(menus);
}
///
/// 获取系统菜单树,用于给角色授权时选择
///
///
///
[HttpGet("/sysMenu/treeForGrant")]
public async Task TreeForGrant([FromQuery] MenuInput input)
{
var menuIdList = new List();
if (!UserManager.IsSuperAdmin)
{
var roleIdList = await _sysUserRoleService.GetUserRoleIdList(UserManager.UserId);
menuIdList = await _sysRoleMenuService.GetRoleMenuIdList(roleIdList);
}
List menus = await _sysMenuRep.AsQueryable()
.LeftJoin((t1, t2) => t1.Application == t2.Code)
.Where(t1 => t1.Status == (int)CommonStatus.ENABLE)
.WhereIF(input.IsEnableDataScope != null, t1 => t1.IsEnableDataScope == input.IsEnableDataScope)
.WhereIF(menuIdList.Any(), t1 => menuIdList.Contains(t1.Id))
.OrderBy(t1 => t1.Sort)
.Select((t1, t2) => new MenuTreeOutput
{
Id = t1.Id,
ParentId = t1.Pid,
Value = t1.Id.ToString(),
Title = t1.Name,
Weight = t1.Sort,
AppName = t2.Name,
AppCode = t2.Code,
AppSort = t2.Sort,
Type = t1.Type
}).ToListAsync();
// menus内不含父项菜单或目录,所以要手动补充进去
await AddParentMenus(menus);
// 获取所有应用
var appList = menus.OrderBy(ob => ob.AppSort)
.Select(sl => new MenuTreeAppOutput { AppName = sl.AppName, AppCode = sl.AppCode }).DistinctBy(db => db.AppCode)
.ToList();
return new { AppList = appList, MenuTree = new TreeBuildUtil().DoTreeBuild(menus) };
}
///
/// 为菜单列表中,缺失的父节点的菜单补充父节点菜单或目录
///
public async Task> AddParentMenus(List menus)
{
List parentIdList = new();
foreach (var item in menus)
{
if (item.ParentId != 0 && !menus.Exists(x => x.Id == item.ParentId))
{
parentIdList.Add(item.ParentId);
}
}
if (parentIdList.Count == 0)
{
return menus;
}
parentIdList = parentIdList.Distinct().ToList();
List addMenus = await _sysMenuRep.AsQueryable()
.LeftJoin((t1, t2) => t1.Application == t2.Code)
.Where(t1 => t1.Status == (int)CommonStatus.ENABLE && parentIdList.Contains(t1.Id))
.OrderBy(t1 => t1.Sort)
.Select((t1, t2) => new MenuTreeOutput
{
Id = t1.Id,
ParentId = t1.Pid,
Value = t1.Id.ToString(),
Title = t1.Name,
Weight = t1.Sort,
AppName = t2.Name,
AppCode = t2.Code,
AppSort = t2.Sort
}).ToListAsync();
menus.AddRange(addMenus);
// 这里是为了防止某个菜单的父项菜单一直查询不到,导致死循环
if (addMenus.Count != parentIdList.Count)
{
return menus;
}
// 递归调用,防止新增的父项菜单或目录也缺失父项菜单或目录
await AddParentMenus(menus);
return menus;
}
///
/// 根据应用编码判断该机构下是否有状态为正常的菜单
///
///
///
[NonAction]
public async Task HasMenu(string appCode)
{
return await _sysMenuRep.AnyAsync(u => u.Application == appCode && u.Status != CommonStatus.DELETED);
}
///
/// 根据系统应用切换菜单
///
///
///
[AllowAnonymous]
[HttpPost("/sysMenu/change")]
public async Task> ChangeAppMenu(ChangeAppMenuInput input)
{
return await GetLoginMenusAntDesign(UserManager.UserId, input.Application);
}
}