265 lines
11 KiB
C#
265 lines
11 KiB
C#
using Mapster;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using SqlSugar;
|
|
using Volo.Abp.Application.Dtos;
|
|
using Volo.Abp.Application.Services;
|
|
using Volo.Abp.Domain.Entities;
|
|
using Volo.Abp.Uow;
|
|
using Yi.Framework.Ddd.Application;
|
|
using Yi.Framework.Rbac.Application.Contracts.Dtos.Role;
|
|
using Yi.Framework.Rbac.Application.Contracts.Dtos.User;
|
|
using Yi.Framework.Rbac.Application.Contracts.IServices;
|
|
using Yi.Framework.Rbac.Domain.Entities;
|
|
using Yi.Framework.Rbac.Domain.Managers;
|
|
using Yi.Framework.Rbac.Domain.Shared.Consts;
|
|
using Yi.Framework.Rbac.Domain.Shared.Enums;
|
|
using Yi.Framework.SqlSugarCore.Abstractions;
|
|
|
|
namespace Yi.Framework.Rbac.Application.Services.System
|
|
{
|
|
/// <summary>
|
|
/// Role服务实现
|
|
/// </summary>
|
|
public class RoleService : YiCrudAppService<RoleAggregateRoot, RoleGetOutputDto, RoleGetListOutputDto, Guid,
|
|
RoleGetListInputVo, RoleCreateInputVo, RoleUpdateInputVo>,
|
|
IRoleService
|
|
{
|
|
public RoleService(RoleManager roleManager, ISqlSugarRepository<RoleDeptEntity> roleDeptRepository,
|
|
ISqlSugarRepository<UserRoleEntity> userRoleRepository,
|
|
ISqlSugarRepository<RoleAggregateRoot, Guid> repository,
|
|
ISqlSugarRepository<MenuAggregateRoot, Guid> menuRepository,
|
|
ISqlSugarRepository<DeptAggregateRoot, Guid> deptRepository) : base(repository)
|
|
{
|
|
(_roleManager, _roleDeptRepository, _userRoleRepository, _repository, _menuRepository, _deptRepository) =
|
|
(roleManager, roleDeptRepository, userRoleRepository, repository, menuRepository, deptRepository);
|
|
}
|
|
|
|
private ISqlSugarRepository<RoleAggregateRoot, Guid> _repository;
|
|
private RoleManager _roleManager { get; set; }
|
|
|
|
private ISqlSugarRepository<RoleDeptEntity> _roleDeptRepository;
|
|
|
|
private ISqlSugarRepository<UserRoleEntity> _userRoleRepository;
|
|
|
|
private ISqlSugarRepository<MenuAggregateRoot, Guid> _menuRepository;
|
|
|
|
private ISqlSugarRepository<DeptAggregateRoot, Guid> _deptRepository;
|
|
|
|
public async Task UpdateDataScopeAsync(UpdateDataScopeInput input)
|
|
{
|
|
//只有自定义的需要特殊处理
|
|
if (input.DataScope == DataScopeEnum.CUSTOM)
|
|
{
|
|
await _roleDeptRepository.DeleteAsync(x => x.RoleId == input.RoleId);
|
|
var insertEntities = input.DeptIds.Select(x => new RoleDeptEntity { DeptId = x, RoleId = input.RoleId })
|
|
.ToList();
|
|
await _roleDeptRepository.InsertRangeAsync(insertEntities);
|
|
}
|
|
|
|
var entity = new RoleAggregateRoot() { DataScope = input.DataScope };
|
|
EntityHelper.TrySetId(entity, () => input.RoleId);
|
|
await _repository._Db.Updateable(entity).UpdateColumns(x => x.DataScope).ExecuteCommandAsync();
|
|
}
|
|
|
|
public override async Task<PagedResultDto<RoleGetListOutputDto>> GetListAsync(RoleGetListInputVo input)
|
|
{
|
|
RefAsync<int> total = 0;
|
|
|
|
var entities = await _repository._DbQueryable.WhereIF(!string.IsNullOrEmpty(input.RoleCode),
|
|
x => x.RoleCode.Contains(input.RoleCode!))
|
|
.WhereIF(!string.IsNullOrEmpty(input.RoleName), x => x.RoleName.Contains(input.RoleName!))
|
|
.WhereIF(input.State is not null, x => x.State == input.State)
|
|
.ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
|
|
return new PagedResultDto<RoleGetListOutputDto>(total, await MapToGetListOutputDtosAsync(entities));
|
|
}
|
|
|
|
/// <summary>
|
|
/// 添加角色
|
|
/// </summary>
|
|
/// <param name="input"></param>
|
|
/// <returns></returns>
|
|
public override async Task<RoleGetOutputDto> CreateAsync(RoleCreateInputVo input)
|
|
{
|
|
var isExist =
|
|
await _repository.IsAnyAsync(x => x.RoleCode == input.RoleCode || x.RoleName == input.RoleName);
|
|
if (isExist)
|
|
{
|
|
throw new UserFriendlyException(RoleConst.Exist);
|
|
}
|
|
|
|
var entity = await MapToEntityAsync(input);
|
|
await _repository.InsertAsync(entity);
|
|
var outputDto = await MapToGetOutputDtoAsync(entity);
|
|
|
|
return outputDto;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 修改角色
|
|
/// </summary>
|
|
/// <param name="id"></param>
|
|
/// <param name="input"></param>
|
|
/// <returns></returns>
|
|
public override async Task<RoleGetOutputDto> UpdateAsync(Guid id, RoleUpdateInputVo input)
|
|
{
|
|
var entity = await _repository.GetByIdAsync(id);
|
|
|
|
var isExist = await _repository._DbQueryable.Where(x => x.Id != entity.Id).AnyAsync(x => x.RoleCode == input.RoleCode || x.RoleName == input.RoleName);
|
|
if (isExist)
|
|
{
|
|
throw new UserFriendlyException(RoleConst.Exist);
|
|
}
|
|
|
|
await MapToEntityAsync(input, entity);
|
|
await _repository.UpdateAsync(entity);
|
|
|
|
await _roleManager.GiveRoleSetMenuAsync(new List<Guid> { id }, input.MenuIds);
|
|
|
|
var dto = await MapToGetOutputDtoAsync(entity);
|
|
return dto;
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// 更新状态
|
|
/// </summary>
|
|
/// <param name="id"></param>
|
|
/// <param name="state"></param>
|
|
/// <returns></returns>
|
|
[Route("role/{id}/{state}")]
|
|
public async Task<RoleGetOutputDto> UpdateStateAsync([FromRoute] Guid id, [FromRoute] bool state)
|
|
{
|
|
var entity = await _repository.GetByIdAsync(id);
|
|
if (entity is null)
|
|
{
|
|
throw new ApplicationException("角色未存在");
|
|
}
|
|
|
|
entity.State = state;
|
|
await _repository.UpdateAsync(entity);
|
|
return await MapToGetOutputDtoAsync(entity);
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// 获取角色下的用户
|
|
/// </summary>
|
|
/// <param name="roleId"></param>
|
|
/// <param name="input"></param>
|
|
/// <param name="isAllocated">是否在该角色下</param>
|
|
/// <returns></returns>
|
|
[Route("role/auth-user/{roleId}/{isAllocated}")]
|
|
public async Task<PagedResultDto<UserGetListOutputDto>> GetAuthUserByRoleIdAsync([FromRoute] Guid roleId,
|
|
[FromRoute] bool isAllocated, [FromQuery] RoleAuthUserGetListInput input)
|
|
{
|
|
PagedResultDto<UserGetListOutputDto> output;
|
|
//角色下已授权用户
|
|
if (isAllocated == true)
|
|
{
|
|
output = await GetAllocatedAuthUserByRoleIdAsync(roleId, input);
|
|
}
|
|
//角色下未授权用户
|
|
else
|
|
{
|
|
output = await GetNotAllocatedAuthUserByRoleIdAsync(roleId, input);
|
|
}
|
|
|
|
return output;
|
|
}
|
|
|
|
private async Task<PagedResultDto<UserGetListOutputDto>> GetAllocatedAuthUserByRoleIdAsync(Guid roleId,
|
|
RoleAuthUserGetListInput input)
|
|
{
|
|
RefAsync<int> total = 0;
|
|
var output = await _userRoleRepository._DbQueryable
|
|
.LeftJoin<UserAggregateRoot>((ur, u) => ur.UserId == u.Id && ur.RoleId == roleId)
|
|
.Where((ur, u) => ur.RoleId == roleId)
|
|
.WhereIF(!string.IsNullOrEmpty(input.UserName), (ur, u) => u.UserName.Contains(input.UserName))
|
|
.WhereIF(input.Phone is not null, (ur, u) => u.Phone.ToString().Contains(input.Phone.ToString()))
|
|
.Select((ur, u) => new UserGetListOutputDto { Id = u.Id }, true)
|
|
.ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
|
|
return new PagedResultDto<UserGetListOutputDto>(total, output);
|
|
}
|
|
|
|
private async Task<PagedResultDto<UserGetListOutputDto>> GetNotAllocatedAuthUserByRoleIdAsync(Guid roleId,
|
|
RoleAuthUserGetListInput input)
|
|
{
|
|
RefAsync<int> total = 0;
|
|
var entities = await _userRoleRepository._Db.Queryable<UserAggregateRoot>()
|
|
.Where(u => SqlFunc.Subqueryable<UserRoleEntity>().Where(x => x.RoleId == roleId)
|
|
.Where(x => x.UserId == u.Id).NotAny())
|
|
.WhereIF(!string.IsNullOrEmpty(input.UserName), u => u.UserName.Contains(input.UserName))
|
|
.WhereIF(input.Phone is not null, u => u.Phone.ToString().Contains(input.Phone.ToString()))
|
|
.ToPageListAsync(input.SkipCount, input.MaxResultCount, total);
|
|
var output = entities.Adapt<List<UserGetListOutputDto>>();
|
|
return new PagedResultDto<UserGetListOutputDto>(total, output);
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// 批量给用户授权
|
|
/// </summary>
|
|
/// <param name="input"></param>
|
|
/// <returns></returns>
|
|
public async Task CreateAuthUserAsync([FromBody] RoleAuthUserCreateOrDeleteInput input)
|
|
{
|
|
var userRoleEntities = input.UserIds.Select(u => new UserRoleEntity { RoleId = input.RoleId, UserId = u })
|
|
.ToList();
|
|
await _userRoleRepository.InsertRangeAsync(userRoleEntities);
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// 批量取消授权
|
|
/// </summary>
|
|
/// <param name="input"></param>
|
|
/// <returns></returns>
|
|
public async Task DeleteAuthUserAsync([FromBody] RoleAuthUserCreateOrDeleteInput input)
|
|
{
|
|
await _userRoleRepository._Db.Deleteable<UserRoleEntity>().Where(x => x.RoleId == input.RoleId)
|
|
.Where(x => input.UserIds.Contains(x.UserId))
|
|
.ExecuteCommandAsync();
|
|
;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 获取角色菜单树
|
|
/// </summary>
|
|
/// <param name="roleId"></param>
|
|
/// <returns></returns>
|
|
public async Task<ActionResult> GetMenuTreeAsync(Guid roleId, MenuSourceEnum menuSource = MenuSourceEnum.Vben5)
|
|
{
|
|
var checkedKeys = await _menuRepository._DbQueryable
|
|
.Where(m => SqlFunc.Subqueryable<RoleMenuEntity>().Where(rm => rm.RoleId == roleId && rm.MenuId == m.Id).Any())
|
|
.Select(x => x.Id).ToListAsync();
|
|
var menus = await _menuRepository._DbQueryable.Where(x => x.State == true && x.MenuSource == menuSource).ToListAsync();
|
|
var menuTrees = menus.TreeDtoBuild();
|
|
return new JsonResult(new
|
|
{
|
|
checkedKeys,
|
|
menus = menuTrees
|
|
});
|
|
}
|
|
|
|
/// <summary>
|
|
/// 获取角色部门树
|
|
/// </summary>
|
|
/// <param name="roleId"></param>
|
|
/// <returns></returns>
|
|
public async Task<ActionResult> GetDeptTreeAsync(Guid roleId)
|
|
{
|
|
var checkedKeys = await _deptRepository._DbQueryable
|
|
.Where(d => SqlFunc.Subqueryable<RoleDeptEntity>().Where(rd => rd.RoleId == roleId && rd.DeptId == d.Id).Any())
|
|
.Select(x => x.Id).ToListAsync();
|
|
var deptList = await _deptRepository._DbQueryable
|
|
.Where(x => x.State == true)
|
|
.OrderBy(x => x.OrderNum, OrderByType.Asc)
|
|
.ToListAsync();
|
|
return new JsonResult(new
|
|
{
|
|
checkedKeys,
|
|
depts = deptList.DeptTreeBuild()
|
|
});
|
|
}
|
|
}
|
|
} |