diff --git a/src/commands/fun/happy.ts b/src/commands/fun/happy.ts index 02f1073..2294475 100644 --- a/src/commands/fun/happy.ts +++ b/src/commands/fun/happy.ts @@ -27,7 +27,7 @@ export class HappyCommand extends Command { ...options, name: 'happy', description: 'Express your happiness', - preconditions: ['PatreonOnly'], + preconditions: [['PatreonOnly', 'CoordinatorOnly']], }); } diff --git a/src/commands/fun/hug.ts b/src/commands/fun/hug.ts index 35bea0b..cec1b58 100644 --- a/src/commands/fun/hug.ts +++ b/src/commands/fun/hug.ts @@ -27,7 +27,7 @@ export class HugCommand extends Command { ...options, name: 'hug', description: 'Hug a user', - preconditions: ['PatreonOnly'], + preconditions: [['PatreonOnly', 'CoordinatorOnly']], }); } diff --git a/src/commands/fun/kill.ts b/src/commands/fun/kill.ts index 66b39c3..b584180 100644 --- a/src/commands/fun/kill.ts +++ b/src/commands/fun/kill.ts @@ -27,7 +27,7 @@ export class KillCommand extends Command { ...options, name: 'kill', description: 'Kill a user', - preconditions: ['PatreonOnly'], + preconditions: [['PatreonOnly', 'CoordinatorOnly']], }); } diff --git a/src/commands/fun/poke.ts b/src/commands/fun/poke.ts index 0b011fe..c0c46fe 100644 --- a/src/commands/fun/poke.ts +++ b/src/commands/fun/poke.ts @@ -27,7 +27,7 @@ export class PokeCommand extends Command { ...options, name: 'poke', description: 'Poke a user', - preconditions: ['PatreonOnly'], + preconditions: [['PatreonOnly', 'CoordinatorOnly']], }); } diff --git a/src/commands/fun/sad.ts b/src/commands/fun/sad.ts index 8bafcdb..64655b5 100644 --- a/src/commands/fun/sad.ts +++ b/src/commands/fun/sad.ts @@ -27,7 +27,7 @@ export class SadCommand extends Command { ...options, name: 'sad', description: 'Express your sadness', - preconditions: ['PatreonOnly'], + preconditions: [['PatreonOnly', 'CoordinatorOnly']], }); } diff --git a/src/preconditions/CoordinatorOnly.ts b/src/preconditions/CoordinatorOnly.ts new file mode 100644 index 0000000..90cbf55 --- /dev/null +++ b/src/preconditions/CoordinatorOnly.ts @@ -0,0 +1,52 @@ +// SPDX-License-Identifier: GPL-3.0-or-later +/* + Animal Rights Advocates Discord Bot + Copyright (C) 2022 Anthony Berg + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see . +*/ + +import { AllFlowsPrecondition } from '@sapphire/framework'; +import type { CommandInteraction, ContextMenuInteraction, Message } from 'discord.js'; +import { IDs } from '../utils/ids'; +import type { GuildMember } from 'discord.js'; + +export class CoordinatorOnlyPrecondition extends AllFlowsPrecondition { + public override async messageRun(message: Message) { + // for message command + return this.checkCoordinator(message.member!); + } + + public override async chatInputRun(interaction: CommandInteraction) { + // for slash command + return this.checkCoordinator(interaction.member! as GuildMember); + } + + public override async contextMenuRun(interaction: ContextMenuInteraction) { + // for context menu command + return this.checkCoordinator(interaction.member! as GuildMember); + } + + private async checkCoordinator(user: GuildMember) { + return user.roles.cache.has(IDs.roles.staff.coordinator) + ? this.ok() + : this.error({ message: 'Only coordinators can run this command!' }); + } +} + +declare module '@sapphire/framework' { + interface Preconditions { + CoordinatorOnly: never; + } +}