using FlexJobApi.Core;
|
using Furion;
|
using Furion.DatabaseAccessor;
|
using Mapster;
|
using MediatR;
|
using Microsoft.AspNetCore.Http;
|
using Microsoft.EntityFrameworkCore;
|
using System;
|
using System.Collections.Generic;
|
using System.Linq;
|
using System.Text;
|
using System.Threading.Tasks;
|
|
namespace FlexJobApi.UserServer.Application
|
{
|
/// <summary>
|
/// 获取当前登录用户菜单
|
/// </summary>
|
public class GetCurrentLogierMenusQueryHandler(
|
IRepository<Menu> repMenu,
|
IRepository<RoleMenu> repRoleMenu,
|
IRepository<UserRole> repUserInfoRole
|
)
|
: IRequestHandler<GetCurrentLogierMenusQuery, List<GetMenusQueryResultItem>>
|
{
|
private readonly IRepository<Menu> repMenu = repMenu;
|
private readonly IRepository<RoleMenu> repRoleMenu = repRoleMenu;
|
private readonly IRepository<UserRole> repUserInfoRole = repUserInfoRole;
|
|
/// <inheritdoc/>
|
public async Task<List<GetMenusQueryResultItem>> Handle(GetCurrentLogierMenusQuery request, CancellationToken cancellationToken)
|
{
|
var logier = JwtUtils.GetCurrentLogier();
|
var q = repMenu.AsQueryable().AsNoTracking()
|
.OrderBy(it => it.Sort).ThenBy(it => it.CreatedTime)
|
.Where(it => it.UserType == logier.Type && it.ClientType == logier.ClientType)
|
.Where(it => it.Type == EnumMenuType.Menu || it.Type == EnumMenuType.Page || it.Type == EnumMenuType.Modal);
|
if (logier.Level != 999)
|
{
|
var menuIds = await repRoleMenu.AsQueryable().AsNoTracking()
|
.Where(it => it.Role.UserRoles.Any(ur => ur.UserId == logier.Id))
|
.Select(it => it.MenuId)
|
.ToListAsync(cancellationToken);
|
q = q.Where(it => menuIds.Contains(it.Id));
|
}
|
var all = await q.ProjectToType<GetMenusQueryResultItem>().ToListAsync();
|
var models = all.Where(it => it.ParentId == null).ToList();
|
LoopChildrens(models, all, []);
|
return models;
|
}
|
|
/// <summary>
|
/// 递归赋值下级菜单
|
/// </summary>
|
/// <param name="models"></param>
|
/// <param name="all"></param>
|
/// <param name="roleMenuIds"></param>
|
private void LoopChildrens(List<GetMenusQueryResultItem> models, List<GetMenusQueryResultItem> all, List<Guid> roleMenuIds)
|
{
|
foreach (var item in models)
|
{
|
item.IsChecked = roleMenuIds.Contains(item.Id);
|
item.Children = all.Where(it => it.ParentId == item.Id).ToList();
|
if (item.Children.IsNotNull())
|
{
|
LoopChildrens(item.Children, all, roleMenuIds);
|
}
|
}
|
}
|
}
|
}
|
