315 lines
10 KiB
C#
315 lines
10 KiB
C#
using Dlhis.Entity.Sys.Entities;
|
|
using Dlhis.Entity.Sys.Enums;
|
|
using Dlhis.Sys.Identity.Services;
|
|
using Dlhis.Sys.Menu.Request;
|
|
using Dlhis.Sys.Menu.Response;
|
|
using Kurisu.AspNetCore.Scope;
|
|
using Kurisu.AspNetCore.Utils.Extensions;
|
|
using Kurisu.Core.Result;
|
|
using Kurisu.Core.User.Abstractions;
|
|
using Kurisu.SqlSugar.Attributes;
|
|
using Kurisu.SqlSugar.Services;
|
|
using Mapster;
|
|
|
|
namespace Dlhis.Sys.Menu.Services.Implements;
|
|
|
|
public class MenuService : IMenuService
|
|
{
|
|
private readonly IDbContext _dbContext;
|
|
private readonly ICurrentUser _currentUser;
|
|
|
|
public MenuService(IDbContext dbContext, ICurrentUser currentUser)
|
|
{
|
|
_dbContext = dbContext;
|
|
_currentUser = currentUser;
|
|
}
|
|
|
|
public async Task<List<MenuResponse>> GetListAsync(PlatformEnumType platform)
|
|
{
|
|
var result = await _dbContext.Queryable<MenuEntity>()
|
|
.Where(x => x.Platform == platform)
|
|
.OrderBy(x => x.Sequence)
|
|
.Select<MenuResponse>()
|
|
.ToListAsync();
|
|
|
|
var top = result.Where(x => x.PCode.IsEmpty()).ToList();
|
|
foreach (var menu in top)
|
|
{
|
|
RecursionGetNext(menu, result);
|
|
}
|
|
|
|
return top;
|
|
}
|
|
|
|
|
|
public async Task<string> SaveMenuAsync(SaveMenuRequest request)
|
|
{
|
|
await CheckAsync(request);
|
|
|
|
if (!request.Id.HasValue)
|
|
{
|
|
var entity = request.Adapt<MenuEntity>();
|
|
await _dbContext.InsertAsync(entity);
|
|
return entity.Code;
|
|
}
|
|
else
|
|
{
|
|
var entity = await _dbContext.Queryable<MenuEntity>().Where(x => x.Id == request.Id).FirstAsync();
|
|
await ChangeIfCodeEditAsync(entity.Code, request.Code);
|
|
|
|
request.Adapt(entity);
|
|
|
|
await _dbContext.UpdateAsync(entity);
|
|
return entity.Code;
|
|
}
|
|
}
|
|
|
|
public async Task AssignmentAsync(MenuPermissionAssignmentRequest request)
|
|
{
|
|
(request.UniqueType == AssignTypeEnumType.Role).ThrowIfTrue("暂不支持角色分配");
|
|
|
|
//清空覆盖
|
|
var permissionsToDelete = await _dbContext.Queryable<MenuPermissionEntity>()
|
|
.Where(x => x.UniqueType == request.UniqueType)
|
|
.Where(x => x.UniqueValue == request.UniqueValue)
|
|
.ToListAsync();
|
|
|
|
//触发日志记录
|
|
await _dbContext.DeleteAsync(permissionsToDelete);
|
|
|
|
var permissions = request.Codes.Select(x => new MenuPermissionEntity
|
|
{
|
|
Code = x,
|
|
UniqueType = request.UniqueType,
|
|
UniqueValue = request.UniqueValue
|
|
|
|
}).ToList();
|
|
|
|
|
|
await _dbContext.InsertAsync(permissions);
|
|
}
|
|
|
|
public async Task<List<string>> GetAssignmentCodesAsync(AssignTypeEnumType uniqueType, string uniqueData)
|
|
{
|
|
var permissionCodes = await _dbContext.Queryable<MenuPermissionEntity>()
|
|
.Where(x => x.UniqueType == uniqueType)
|
|
.Where(x => x.UniqueValue == uniqueData)
|
|
.Select(x => x.Code)
|
|
.ToListAsync();
|
|
|
|
var menus = await GetListAsync(PlatformEnumType.PCAdmin);
|
|
|
|
foreach (var menu in menus)
|
|
{
|
|
if (menu.Items.Any())
|
|
{
|
|
var codes = RecursionHandler(menu.Items, permissionCodes);
|
|
permissionCodes.RemoveAll(x => codes.Contains(x));
|
|
}
|
|
}
|
|
|
|
return permissionCodes;
|
|
|
|
static List<string> RecursionHandler(List<MenuResponse> menus, List<string> innerPermissions)
|
|
{
|
|
var codes = new HashSet<string>();
|
|
foreach (var menu in menus)
|
|
{
|
|
if (menu.Items.Any())
|
|
{
|
|
var innerPermissionCodes = RecursionHandler(menu.Items, innerPermissions);
|
|
foreach (var item in innerPermissionCodes)
|
|
{
|
|
codes.Add(item);
|
|
}
|
|
}
|
|
|
|
if (!innerPermissions.Contains(menu.Code))
|
|
{
|
|
foreach (var item in menu.Items.Select(x => x.Code))
|
|
{
|
|
codes.Add(item);
|
|
}
|
|
}
|
|
}
|
|
|
|
return codes.ToList();
|
|
}
|
|
}
|
|
|
|
public async Task<List<string>> GetAllAssignmentPermissionsAsync()
|
|
{
|
|
var all = await GetAllAssignmentCodesAsync();
|
|
return await _dbContext.Queryable<MenuEntity>().Where(x => all.Contains(x.Code)).Select(x => x.Permission).ToListAsync();
|
|
}
|
|
|
|
public async Task<List<MenuResponse>> GetCurrentMenusAsync()
|
|
{
|
|
//我能看到的权限code,非permissions
|
|
var allCodes = await GetAllAssignmentCodesAsync();
|
|
if (!allCodes.Any())
|
|
{
|
|
return await GetListAsync(PlatformEnumType.PCAdmin);
|
|
}
|
|
|
|
var allMenus = await _dbContext.Queryable<MenuEntity>()
|
|
.Where(x => x.Platform == PlatformEnumType.PCAdmin)
|
|
.Where(x => x.Visible)
|
|
.Where(x => allCodes.Contains(x.Code))
|
|
.Select<MenuResponse>()
|
|
.ToListAsync();
|
|
|
|
//获取顶级数据
|
|
var menus = await _dbContext.Queryable<MenuEntity>()
|
|
.Where(x => x.Platform == PlatformEnumType.PCAdmin)
|
|
.Where(x => x.Visible)
|
|
.Where(x => string.IsNullOrEmpty(x.PCode))
|
|
.Select<MenuResponse>()
|
|
.ToListAsync();
|
|
|
|
foreach (var menu in menus)
|
|
{
|
|
RecursionGetNext(menu, allMenus);
|
|
}
|
|
|
|
menus.RemoveAll(x => !x.Items.Any());
|
|
return menus;
|
|
}
|
|
|
|
public async Task<bool> IsPass(IEnumerable<string> p)
|
|
{
|
|
var permissions = await GetAllAssignmentPermissionsAsync();
|
|
return permissions.Count == 0 || permissions.All(x => p.Contains(x));
|
|
}
|
|
|
|
public async Task ResetAsync(List<SaveMenuRequest> request)
|
|
{
|
|
var menus = await _dbContext.Queryable<MenuEntity>().Where(x => x.Platform == PlatformEnumType.PCAdmin).ToListAsync();
|
|
await _dbContext.Client.Deleteable(menus).ExecuteCommandAsync();
|
|
|
|
foreach (var item in request)
|
|
{
|
|
var code = await SaveMenuAsync(item);
|
|
if (item.Items?.Any() == true)
|
|
{
|
|
item.Items.ForEach(x => x.PCode = code);
|
|
await RecursionSaveNextAsync(item.Items);
|
|
}
|
|
}
|
|
|
|
//递归处理下一级
|
|
async Task RecursionSaveNextAsync(List<SaveMenuRequest> menus)
|
|
{
|
|
foreach (var item in menus)
|
|
{
|
|
var code = await SaveMenuAsync(item);
|
|
|
|
if (item.Items?.Any() == true)
|
|
{
|
|
item.Items.ForEach(x => x.PCode = code);
|
|
await RecursionSaveNextAsync(item.Items);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
private void RecursionGetNext(MenuResponse current, List<MenuResponse> menus)
|
|
{
|
|
var nextMenus = menus.Where(x => x.PCode == current.Code).ToList();
|
|
current.Items = nextMenus;
|
|
foreach (var menu in nextMenus)
|
|
{
|
|
RecursionGetNext(menu, menus);
|
|
}
|
|
}
|
|
|
|
private async Task CheckAsync(SaveMenuRequest request)
|
|
{
|
|
|
|
//检查重复code
|
|
var exists = await _dbContext.Queryable<MenuEntity>()
|
|
.Where(x => x.Code == request.Code)
|
|
.WhereIF(request.Id.HasValue, x => x.Id != request.Id).AnyAsync();
|
|
exists.ThrowIfTrue($"{request.Title}存在重复code");
|
|
|
|
//检查name
|
|
exists = await _dbContext.Queryable<MenuEntity>()
|
|
.Where(x => x.Name == request.Name)
|
|
.WhereIF(request.Id.HasValue, x => x.Id != request.Id).AnyAsync();
|
|
exists.ThrowIfTrue($"{request.Title}存在重复name");
|
|
|
|
//判断权限重复
|
|
if (request.Permission.IsPresent())
|
|
{
|
|
exists = await _dbContext.Queryable<MenuEntity>()
|
|
.Where(x => x.Permission == request.Permission && (request.MenuType == MenuEnumType.Menu || request.MenuType == MenuEnumType.Button))
|
|
.WhereIF(request.Id.HasValue, x => x.Id != request.Id).AnyAsync();
|
|
exists.ThrowIfTrue($"{request.Title}存在重复权限标识");
|
|
}
|
|
|
|
//request.PCode
|
|
var menu = await _dbContext.Queryable<MenuEntity>().Where(x => x.Code == request.PCode).FirstAsync();
|
|
(menu == null && request.MenuType != MenuEnumType.Folder).ThrowIfTrue($"{request.Title}上级不存在");
|
|
|
|
if (request.MenuType == MenuEnumType.Folder)
|
|
{
|
|
//request.PCode = null;
|
|
if (string.IsNullOrEmpty(request.PCode))
|
|
{
|
|
request.PCode = null;
|
|
}
|
|
}
|
|
else if (request.MenuType == MenuEnumType.Menu)
|
|
{
|
|
if (menu.MenuType != MenuEnumType.Folder)
|
|
{
|
|
throw new Exception($"{request.Title}菜单上级需要为目录");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
//button
|
|
if (menu.MenuType != MenuEnumType.Menu)
|
|
{
|
|
throw new Exception($"{request.Title}按钮上级需要为菜单");
|
|
}
|
|
}
|
|
}
|
|
|
|
private async Task<List<string>> GetAllAssignmentCodesAsync()
|
|
{
|
|
var user = await Scoped.Request.Value.CreateAsync(sp =>
|
|
{
|
|
return sp.GetService<IUserService>().GetUserByIdAsync(_currentUser.GetIntSubjectId());
|
|
});
|
|
|
|
// var groupCodes = await GetAssignmentCodesAsync(MenuPermissionTypeEnumType.Group, manager.usergrp);
|
|
|
|
//user
|
|
var userCodes = await GetAssignmentCodesAsync(AssignTypeEnumType.User, user.Id.ToString());
|
|
var all = userCodes;// gP.Concat(uP).Distinct().ToList();
|
|
|
|
return all;
|
|
}
|
|
|
|
|
|
private async Task ChangeIfCodeEditAsync(string beforeCode, string pcode)
|
|
{
|
|
if (beforeCode != pcode)
|
|
{
|
|
//修改下级的PCode为当前的code
|
|
var menus = await _dbContext.Queryable<MenuEntity>().Where(x => x.PCode == beforeCode).ToListAsync();
|
|
menus.ForEach(x => { x.PCode = pcode; });
|
|
|
|
//修改权限code为当前的code,避免权限丢失
|
|
var permissions = await _dbContext.Queryable<MenuPermissionEntity>().Where(x => x.Code == beforeCode).ToListAsync();
|
|
permissions.ForEach(x => { x.Code = pcode; });
|
|
|
|
await _dbContext.UpdateAsync(menus);
|
|
await _dbContext.UpdateAsync(permissions);
|
|
}
|
|
}
|
|
|
|
|
|
} |