using System; using System.Collections.Generic; using System.Text; using System.Threading.Tasks; using Discord; namespace Discord.Commands { /// /// Require that the command is invoked by the owner of the bot. /// /// This precondition will only work if the bot is a bot account. [AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, AllowMultiple = false, Inherited = true)] public class RequireOwnerAttribute : PreconditionAttribute { public override async Task CheckPermissions(CommandContext context, CommandInfo command, IDependencyMap map) { var application = await context.Client.GetApplicationInfoAsync(); if (context.User.Id == application.Owner.Id) return PreconditionResult.FromSuccess(); return PreconditionResult.FromError("Command can only be run by the owner of the bot"); } } }