mirror of
https://github.com/NoiTheCat/BirthdayBot.git
synced 2024-10-17 00:09:58 +00:00
Noi
6c498045b3
The current way of loading database information makes things confusing and difficult to follow, particularly with retrieving guild data and its second parameter. A much better alternative should be considered, and accessed by means of extension methods.
31 lines
1.3 KiB
C#
31 lines
1.3 KiB
C#
using BirthdayBot.Data;
|
|
using Discord.Interactions;
|
|
|
|
namespace BirthdayBot.ApplicationCommands;
|
|
|
|
// Contains preconditions used by our interaction modules.
|
|
|
|
/// <summary>
|
|
/// Precondition requiring the executing user be considered a bot moderator.
|
|
/// That is, they must either have the Manage Server permission or be a member of the designated bot moderator role.
|
|
/// </summary>
|
|
class RequireBotModeratorAttribute : PreconditionAttribute {
|
|
public const string FailMsg = "User did not pass the mod check.";
|
|
public const string Reply = ":x: You must be a moderator to use this command.";
|
|
|
|
public override string ErrorMessage => FailMsg;
|
|
|
|
public override async Task<PreconditionResult> CheckRequirementsAsync(
|
|
IInteractionContext context, ICommandInfo commandInfo, IServiceProvider services) {
|
|
if (context.User is not SocketGuildUser user) {
|
|
return PreconditionResult.FromError("Failed due to non-guild context.");
|
|
}
|
|
|
|
if (user.GuildPermissions.ManageGuild) return PreconditionResult.FromSuccess();
|
|
var gconf = await ((SocketGuild)context.Guild).GetConfigAsync().ConfigureAwait(false);
|
|
if (gconf.ModeratorRole.HasValue && user.Roles.Any(r => r.Id == gconf.ModeratorRole.Value))
|
|
return PreconditionResult.FromSuccess();
|
|
|
|
return PreconditionResult.FromError(FailMsg);
|
|
}
|
|
}
|