BookingHeChuan/Myshipping.Core/Service/Menu/SysMenuService.cs

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;

/// <summary>
/// 系统菜单服务
/// </summary>
[ApiDescriptionSettings(Name = "Menu", Order = 146)]
public class SysMenuService : ISysMenuService, IDynamicApiController, ITransient
{
    private readonly SqlSugarRepository<SysMenu> _sysMenuRep;  // 菜单表仓储  

    private readonly ISysCacheService _sysCacheService;
    private readonly ISysUserRoleService _sysUserRoleService;
    private readonly ISysRoleMenuService _sysRoleMenuService;

    public SysMenuService(SqlSugarRepository<SysMenu> sysMenuRep,
                          ISysCacheService sysCacheService,
                          ISysUserRoleService sysUserRoleService,
                          ISysRoleMenuService sysRoleMenuService)
    {
        _sysMenuRep = sysMenuRep;
        _sysCacheService = sysCacheService;
        _sysUserRoleService = sysUserRoleService;
        _sysRoleMenuService = sysRoleMenuService;
    }

    /// <summary>
    /// 获取用户权限(按钮权限标识集合)
    /// </summary>
    /// <param name="userId"></param>
    /// <returns></returns>
    [NonAction]
    public async Task<List<string>> 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;
    }

    /// <summary>
    /// 获取所有权限集合
    /// </summary>
    /// <returns></returns>
    [NonAction]
    public async Task<List<string>> 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;
    }


    /// <summary>
    /// 获取用户AntDesign菜单集合
    /// </summary>
    /// <param name="userId"></param>
    /// <param name="appCode"></param>
    /// <returns></returns>
    [NonAction]
    public async Task<List<AntDesignTreeNode>> GetLoginMenusAntDesign(long userId, string appCode)
    {
        var antDesignTreeNodes = await _sysCacheService.GetMenu(userId, appCode); // 先从缓存里面读取
        if (antDesignTreeNodes == null || antDesignTreeNodes.Count < 1)
        {
            var sysMenuList = new List<SysMenu>();
            // 管理员则展示所有系统菜单
            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;
    }

    /// <summary>
    /// 获取用户菜单所属的应用编码集合
    /// </summary>
    /// <param name="userId"></param>
    /// <returns></returns>
    [NonAction]
    public async Task<List<string>> 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();
    }

    /// <summary>
    /// 系统菜单列表（树表）
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [HttpGet("/sysMenu/list")]
    public async Task<dynamic> 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<MenuOutput>()
             .ToListAsync();
        return new TreeBuildUtil<MenuOutput>().DoTreeBuild(menus);
    }

    /// <summary>
    /// 创建Pids格式 
    /// 如果pid是0顶级节点，pids就是 [0];
    /// 如果pid不是顶级节点，pids就是 pid菜单的 pids + [pid] + ,
    /// </summary>
    /// <param name="pid"></param>
    /// <returns></returns>
    private async Task<string> CreateNewPids(long pid)
    {
        if (pid == 0L)
        {
            return "[0],";
        }
        else
        {
            var pmenu = await _sysMenuRep.FirstOrDefaultAsync(u => u.Id == pid);
            return pmenu.Pids + "[" + pid + "],";
        }
    }

    /// <summary>
    /// 增加和编辑时检查参数
    /// </summary>
    /// <param name="input"></param>
    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);                
        }
    }

    /// <summary>
    /// 增加系统菜单
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [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<SysMenu>();
        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);
    }

    /// <summary>
    /// 删除系统菜单
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [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;
        }

    }

    /// <summary>
    /// 更新系统菜单
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [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<SysMenu>();

        // 如果是编辑，父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<SysMenu>();
        oldMenu.Pids = newPids;

        menuList.Add(oldMenu);

        await _sysMenuRep.UpdateAsync(menuList);

        // 清除缓存
        await _sysCacheService.DelByPatternAsync(CommonConst.CACHE_KEY_MENU);
        await _sysCacheService.DelByPatternAsync(CommonConst.CACHE_KEY_ALLPERMISSION);
    }

    /// <summary>
    /// 获取系统菜单
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [HttpPost("/sysMenu/detail")]
    public async Task<dynamic> GetMenu(QueryMenuInput input)
    {
        return await _sysMenuRep.FirstOrDefaultAsync(u => u.Id == input.Id);
    }

    /// <summary>
    /// 获取系统菜单树，用于新增、编辑时选择上级节点
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [HttpGet("/sysMenu/tree")]
    public async Task<dynamic> 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<MenuTreeOutput>().DoTreeBuild(menus);
    }

    /// <summary>
    /// 获取系统菜单树，用于给角色授权时选择
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [HttpGet("/sysMenu/treeForGrant")]
    public async Task<dynamic> TreeForGrant([FromQuery] MenuInput input)
    {
        var menuIdList = new List<long>();
        if (!UserManager.IsSuperAdmin)
        {
            var roleIdList = await _sysUserRoleService.GetUserRoleIdList(UserManager.UserId);
            menuIdList = await _sysRoleMenuService.GetRoleMenuIdList(roleIdList);
        }

        List<MenuTreeOutput> menus = await _sysMenuRep.AsQueryable()
                                     .LeftJoin<SysApp>((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<MenuTreeOutput>().DoTreeBuild(menus) };
    }

    /// <summary>
    /// 为菜单列表中，缺失的父节点的菜单补充父节点菜单或目录
    /// </summary>
    public async Task<List<MenuTreeOutput>> AddParentMenus(List<MenuTreeOutput> menus)
    {
        List<long> 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<MenuTreeOutput> addMenus = await _sysMenuRep.AsQueryable()
                                                         .LeftJoin<SysApp>((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;
    }
    /// <summary>
    /// 根据应用编码判断该机构下是否有状态为正常的菜单
    /// </summary>
    /// <param name="appCode"></param>
    /// <returns></returns>
    [NonAction]
    public async Task<bool> HasMenu(string appCode)
    {
        return await _sysMenuRep.AnyAsync(u => u.Application == appCode && u.Status != CommonStatus.DELETED);
    }

    /// <summary>
    /// 根据系统应用切换菜单
    /// </summary>
    /// <param name="input"></param>
    /// <returns></returns>
    [AllowAnonymous]
    [HttpPost("/sysMenu/change")]
    public async Task<List<AntDesignTreeNode>> ChangeAppMenu(ChangeAppMenuInput input)
    {
        return await GetLoginMenusAntDesign(UserManager.UserId, input.Application);
    }
}