Added user rights management
This commit is contained in:
parent
f00418d23c
commit
d472889ae1
9 changed files with 419 additions and 17 deletions
|
@ -13,23 +13,48 @@ namespace Lieb.Data
|
|||
_contextFactory = contextFactory;
|
||||
}
|
||||
|
||||
public async Task<LiebUser> GetLiebUser(ulong discordId)
|
||||
public List<LiebUser> GetLiebUsers()
|
||||
{
|
||||
using var context = _contextFactory.CreateDbContext();
|
||||
return context.LiebUsers
|
||||
.Include(u => u.GuildWars2Accounts)
|
||||
.ThenInclude(a => a.EquippedBuilds)
|
||||
.ThenInclude(b => b.GuildWars2Build)
|
||||
.Include(u => u.RoleAssignments)
|
||||
.ThenInclude(r => r.LiebRole)
|
||||
.ToList();
|
||||
}
|
||||
|
||||
public LiebUser GetLiebUser(ulong discordId)
|
||||
{
|
||||
if (discordId > 0)
|
||||
{
|
||||
using var context = _contextFactory.CreateDbContext();
|
||||
return await context.LiebUsers
|
||||
return context.LiebUsers
|
||||
.Include(u => u.GuildWars2Accounts)
|
||||
.ThenInclude(a => a.EquippedBuilds)
|
||||
.ThenInclude(b => b.GuildWars2Build)
|
||||
.Include(u => u.RoleAssignments)
|
||||
.ThenInclude(r => r.LiebRole)
|
||||
.FirstOrDefaultAsync(u => u.DiscordUserId == discordId);
|
||||
.FirstOrDefault(u => u.DiscordUserId == discordId);
|
||||
}
|
||||
else
|
||||
return new LiebUser();
|
||||
}
|
||||
|
||||
public LiebUser GetLiebUser(int userId)
|
||||
{
|
||||
using var context = _contextFactory.CreateDbContext();
|
||||
return context.LiebUsers
|
||||
.Include(u => u.GuildWars2Accounts)
|
||||
.ThenInclude(a => a.EquippedBuilds)
|
||||
.ThenInclude(b => b.GuildWars2Build)
|
||||
.Include(u => u.RoleAssignments)
|
||||
.ThenInclude(r => r.LiebRole)
|
||||
.AsNoTracking()
|
||||
.FirstOrDefault(u => u.LiebUserId == userId);
|
||||
}
|
||||
|
||||
public LiebUser GetLiebUserSmall(ulong discordId)
|
||||
{
|
||||
if (discordId > 0)
|
||||
|
@ -43,6 +68,14 @@ namespace Lieb.Data
|
|||
return new LiebUser();
|
||||
}
|
||||
|
||||
public LiebUser GetLiebUserSmall(int userId)
|
||||
{
|
||||
using var context = _contextFactory.CreateDbContext();
|
||||
return context.LiebUsers
|
||||
.Include(u => u.GuildWars2Accounts)
|
||||
.FirstOrDefault(u => u.LiebUserId == userId);
|
||||
}
|
||||
|
||||
public async Task<int> GetLiebUserId(ulong discordId)
|
||||
{
|
||||
if (discordId > 0)
|
||||
|
@ -69,5 +102,73 @@ namespace Lieb.Data
|
|||
}
|
||||
await context.SaveChangesAsync();
|
||||
}
|
||||
|
||||
public async Task EditUserRoles(LiebUser user)
|
||||
{
|
||||
if (user != null)
|
||||
{
|
||||
using var context = _contextFactory.CreateDbContext();
|
||||
LiebUser? userToChange = await context.LiebUsers
|
||||
.Include(u => u.RoleAssignments)
|
||||
.FirstOrDefaultAsync(u => u.LiebUserId == user.LiebUserId);
|
||||
|
||||
if (userToChange == null)
|
||||
return;
|
||||
|
||||
userToChange.BannedUntil = user.BannedUntil;
|
||||
|
||||
List<RoleAssignment> toDelete = new List<RoleAssignment>();
|
||||
foreach (RoleAssignment assignment in userToChange.RoleAssignments)
|
||||
{
|
||||
RoleAssignment? newAssignment = user.RoleAssignments.FirstOrDefault(r => r.RoleAssignmentId == assignment.RoleAssignmentId);
|
||||
if (newAssignment == null)
|
||||
{
|
||||
toDelete.Add(assignment);
|
||||
}
|
||||
}
|
||||
foreach (RoleAssignment assignment in toDelete)
|
||||
{
|
||||
userToChange.RoleAssignments.Remove(assignment);
|
||||
context.RoleAssignments.Remove(assignment);
|
||||
}
|
||||
foreach (RoleAssignment assignment in user.RoleAssignments.Where(r => r.RoleAssignmentId == 0))
|
||||
{
|
||||
userToChange.RoleAssignments.Add(assignment);
|
||||
}
|
||||
|
||||
await context.SaveChangesAsync();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
public List<LiebRole> GetLiebRoles()
|
||||
{
|
||||
using var context = _contextFactory.CreateDbContext();
|
||||
return context.LiebRoles
|
||||
.Include(u => u.RoleAssignments)
|
||||
.ThenInclude(r => r.LiebUser)
|
||||
.ToList();
|
||||
}
|
||||
|
||||
public async Task AddRole(LiebRole role)
|
||||
{
|
||||
using var context = _contextFactory.CreateDbContext();
|
||||
if (context.LiebRoles.FirstOrDefault(r => r.RoleName == role.RoleName) == null)
|
||||
{
|
||||
context.LiebRoles.Add(role);
|
||||
}
|
||||
await context.SaveChangesAsync();
|
||||
}
|
||||
|
||||
public async Task DeleteRole(int roleId)
|
||||
{
|
||||
using var context = _contextFactory.CreateDbContext();
|
||||
LiebRole role = context.LiebRoles.FirstOrDefault(r => r.LiebRoleId == roleId);
|
||||
if (role != null)
|
||||
{
|
||||
context.LiebRoles.Remove(role);
|
||||
await context.SaveChangesAsync();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue