75 lines
3.3 KiB
C#
75 lines
3.3 KiB
C#
using System;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace Discord.Commands
|
|
{
|
|
/// <summary>
|
|
/// This attribute requires that the user invoking the command has a specified permission.
|
|
/// </summary>
|
|
[AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, AllowMultiple = true, Inherited = true)]
|
|
public class RequireUserPermissionAttribute : PreconditionAttribute
|
|
{
|
|
public GuildPermission? GuildPermission { get; }
|
|
public ChannelPermission? ChannelPermission { get; }
|
|
|
|
/// <summary>
|
|
/// Require that the user invoking the command has a specified GuildPermission
|
|
/// </summary>
|
|
/// <remarks>This precondition will always fail if the command is being invoked in a private channel.</remarks>
|
|
/// <param name="permission">The GuildPermission that the user must have. Multiple permissions can be specified by ORing or ANDing the permissions together.</param>
|
|
public RequireUserPermissionAttribute(GuildPermission permission)
|
|
{
|
|
GuildPermission = permission;
|
|
ChannelPermission = null;
|
|
}
|
|
/// <summary>
|
|
/// Require that the user invoking the command has a specified ChannelPermission.
|
|
/// </summary>
|
|
/// <param name="permission">The ChannelPermission that the user must have. Multiple permissions can be specified by ORing or ANDing the permissions together.</param>
|
|
/// <example>
|
|
/// <code language="c#">
|
|
/// [Command("permission")]
|
|
/// [RequireUserPermission(ChannelPermission.ReadMessageHistory & ChannelPermission.ReadMessages)]
|
|
/// public async Task HasPermission()
|
|
/// {
|
|
/// await ReplyAsync("You can read messages and the message history!");
|
|
/// }
|
|
/// </code>
|
|
/// </example>
|
|
public RequireUserPermissionAttribute(ChannelPermission permission)
|
|
{
|
|
ChannelPermission = permission;
|
|
GuildPermission = null;
|
|
}
|
|
|
|
public override Task<PreconditionResult> CheckPermissions(CommandContext context, CommandInfo command, IDependencyMap map)
|
|
{
|
|
var guildUser = context.User as IGuildUser;
|
|
|
|
if (GuildPermission.HasValue)
|
|
{
|
|
if (guildUser == null)
|
|
return Task.FromResult(PreconditionResult.FromError("Command must be used in a guild channel"));
|
|
if (!guildUser.GuildPermissions.Has(GuildPermission.Value))
|
|
return Task.FromResult(PreconditionResult.FromError($"Command requires guild permission {GuildPermission.Value}"));
|
|
}
|
|
|
|
if (ChannelPermission.HasValue)
|
|
{
|
|
var guildChannel = context.Channel as IGuildChannel;
|
|
|
|
ChannelPermissions perms;
|
|
if (guildChannel != null)
|
|
perms = guildUser.GetPermissions(guildChannel);
|
|
else
|
|
perms = ChannelPermissions.All(guildChannel);
|
|
|
|
if (!perms.Has(ChannelPermission.Value))
|
|
return Task.FromResult(PreconditionResult.FromError($"Command requires channel permission {ChannelPermission.Value}"));
|
|
}
|
|
|
|
return Task.FromResult(PreconditionResult.FromSuccess());
|
|
}
|
|
}
|
|
}
|