Discord/chaosbot
3
0
Fork 0
Code Issues 12 Pull Requests Packages Projects Releases Wiki Activity

Updating Role -> Name

Browse Source
This commit is contained in:
Sean "Solao Bajiuik" Stoves 2020-06-15 21:35:53 -04:00
parent d7c997f0cf
commit 4e2f03bf4c
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
ChaosBot/Discord/Services/RequireRole.cs
View File

@ -8,33 +8,35 @@ using Discord.WebSocket;
public class RequireRole : PreconditionAttribute public class RequireRole : PreconditionAttribute
{ {
private string _name { get; set; } private string _role { get; set; }
public RequireRole(string name) => _name = name; public RequireRole(string role) => _role = role;
public override Task<PreconditionResult> CheckPermissionsAsync(ICommandContext context, CommandInfo command, IServiceProvider services) public override Task<PreconditionResult> CheckPermissionsAsync(ICommandContext context, CommandInfo command, IServiceProvider services)
{ {
string requiredGroup;
if (context.User is SocketGuildUser gUser) if (context.User is SocketGuildUser gUser)
{ {
switch (_name) switch (_role)
{ {
case "Admin": case "Admin":
_name = ConfigurationRepository.GetValue<string>("Role:Admin", context.Guild.Id); requiredGroup = ConfigurationRepository.GetValue<string>("Role:Admin", context.Guild.Id);
break; break;
case "Officer": case "Officer":
_name = ConfigurationRepository.GetValue<string>("Role:Officer", context.Guild.Id); requiredGroup = ConfigurationRepository.GetValue<string>("Role:Officer", context.Guild.Id);
break; break;
case "Member": case "Member":
_name = ConfigurationRepository.GetValue<string>("Role:Member", context.Guild.Id); requiredGroup = ConfigurationRepository.GetValue<string>("Role:Member", context.Guild.Id);
break; break;
default: default:
return Task.FromResult(PreconditionResult.FromError($"{_name} is not a valid Permission Node.")); return Task.FromResult(PreconditionResult.FromError($"{_role} is not a valid Permission Node."));
} }
if (gUser.Roles.Any(r => r.Name == _name)) if (gUser.Roles.Any(r => r.Name == requiredGroup))
return Task.FromResult(PreconditionResult.FromSuccess()); return Task.FromResult(PreconditionResult.FromSuccess());
else else
return Task.FromResult(PreconditionResult.FromError($"You must have a role named {_name} to run this command.")); return Task.FromResult(PreconditionResult.FromError($"You must have a role named {requiredGroup} to run this command."));
} }
else else
return Task.FromResult(PreconditionResult.FromError("You must be in a guild to run this command.")); return Task.FromResult(PreconditionResult.FromError("You must be in a guild to run this command."));