AiDOPWarehouse/server/Admin.NET.Core/Service/Role/SysRoleMenuService.cs

// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
//
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
//
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！

namespace Admin.NET.Core.Service;

/// <summary>
/// 系统角色菜单服务
/// </summary>
public class SysRoleMenuService : ITransient
{
    private readonly SqlSugarRepository<SysRoleMenu> _sysRoleMenuRep;
    private readonly SysCacheService _sysCacheService;

    public SysRoleMenuService(SqlSugarRepository<SysRoleMenu> sysRoleMenuRep,
        SysCacheService sysCacheService)
    {
        _sysRoleMenuRep = sysRoleMenuRep;
        _sysCacheService = sysCacheService;
    }

    /// <summary>
    /// 根据角色Id集合获取菜单Id集合
    /// </summary>
    /// <param name="roleIdList"></param>
    /// <returns></returns>
    public async Task<List<long>> GetRoleMenuIdList(List<long> roleIdList)
    {
        return await _sysRoleMenuRep.AsQueryable()
            .Where(u => roleIdList.Contains(u.RoleId))
            .Select(u => u.MenuId).ToListAsync();
    }

    /// <summary>
    /// 授权角色菜单 🔖
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [UnitOfWork]
    [DisplayName("授权角色菜单")]
    public async Task GrantRoleMenu(RoleMenuInput input)
    {
        var targetRoleIds = new[] { input.Id }
            .Concat(input.RoleIdList ?? new List<long>())
            .Distinct()
            .ToList();
        if (targetRoleIds.Count == 0) return;

        var allIdList = await _sysRoleMenuRep.Context.Queryable<SysMenu>().ToListAsync();
        var requestedMenuIds = ExpandMenuIdsWithAncestors(input.MenuIdList ?? new List<long>(), allIdList);

        var roleTenants = await _sysRoleMenuRep.Context.Queryable<SysRole>()
            .Where(u => targetRoleIds.Contains(u.Id))
            .Select(u => new { u.Id, u.TenantId })
            .ToListAsync();
        if (roleTenants.Count == 0) return;

        var tenantIds = roleTenants
            .Where(u => u.TenantId.HasValue)
            .Select(u => u.TenantId!.Value)
            .Distinct()
            .ToList();
        var tenantMenus = await _sysRoleMenuRep.Context.Queryable<SysTenantMenu>()
            .Where(u => tenantIds.Contains(u.TenantId))
            .Select(u => new { u.TenantId, u.MenuId })
            .ToListAsync();
        var tenantMenuMap = tenantMenus
            .GroupBy(u => u.TenantId)
            .ToDictionary(u => u.Key, u => u.Select(x => x.MenuId).ToHashSet());

        await _sysRoleMenuRep.DeleteAsync(u => targetRoleIds.Contains(u.RoleId));

        // 保存授权数据
        var menus = new List<SysRoleMenu>();
        foreach (var roleTenant in roleTenants)
        {
            if (!roleTenant.TenantId.HasValue) continue;
            if (!tenantMenuMap.TryGetValue(roleTenant.TenantId.Value, out var allowedMenuIds)) continue;

            menus.AddRange(requestedMenuIds
                .Where(allowedMenuIds.Contains)
                .Select(menuId => new SysRoleMenu
            {
                    RoleId = roleTenant.Id,
                    MenuId = menuId
                }));
        }

        if (menus.Count > 0) await _sysRoleMenuRep.InsertRangeAsync(menus);

        // 清除缓存
        // _sysCacheService.RemoveByPrefixKey(CacheConst.KeyUserMenu);
        _sysCacheService.RemoveByPrefixKey(CacheConst.KeyUserButton);
    }

    private static List<long> ExpandMenuIdsWithAncestors(List<long> menuIds, IEnumerable<SysMenu> allMenus)
    {
        var menuMap = allMenus.ToDictionary(u => u.Id, u => u.Pid);
        var result = new HashSet<long>(menuIds.Where(u => u != 0));

        foreach (var menuId in menuIds)
        {
            var current = menuId;
            while (menuMap.TryGetValue(current, out var pid) && pid != 0)
            {
                if (!result.Add(pid)) break;
                current = pid;
            }
        }

        return result.ToList();
    }

    /// <summary>
    /// 根据菜单Id集合删除角色菜单
    /// </summary>
    /// <param name="menuIdList"></param>
    /// <returns></returns>
    public async Task DeleteRoleMenuByMenuIdList(List<long> menuIdList)
    {
        await _sysRoleMenuRep.DeleteAsync(u => menuIdList.Contains(u.MenuId));
    }

    /// <summary>
    /// 根据角色Id删除角色菜单
    /// </summary>
    /// <param name="roleId"></param>
    /// <returns></returns>
    public async Task DeleteRoleMenuByRoleId(long roleId)
    {
        await _sysRoleMenuRep.DeleteAsync(u => u.RoleId == roleId);
    }
}