utils/structures/BaseCommand.ts

/**
* This file is part of SudoBot.
* 
* Copyright (C) 2021-2022 OSN Inc.
*
* SudoBot is free software; you can redistribute it and/or modify it
* under the terms of the GNU Affero General Public License as published by 
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
* 
* SudoBot 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License 
* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
*/

import { ModalSubmitInteraction } from 'discord-modals';
import { PermissionResolvable, AutocompleteInteraction, CommandInteraction, ContextMenuInteraction, Interaction, Message, MessageEditOptions, MessageOptions, MessagePayload, WebhookEditMessageOptions, GuildMember } from 'discord.js';
import DiscordClient from '../../client/Client';
import AutoCompleteOptions from '../../types/AutoCompleteOptions';
import CommandOptions from '../../types/CommandOptions';
import InteractionOptions from '../../types/InteractionOptions';

export default abstract class BaseCommand {
    supportsInteractions: boolean = false;
    supportsLegacy: boolean = true;
    supportsContextMenu: boolean = false;
    coolDown?: number;
    ownerOnly: boolean = false;
    permissions: PermissionResolvable[] = [];

    
    constructor(private name: string, private category: string, private aliases: Array<string>) {
        
    }
    
    getName(): string { 
        return this.name; 
    }
    
    getCategory(): string { 
        return this.category; 
    }
    
    getAliases(): Array<string> { 
        return this.aliases; 
    }
    
    async permissionValidation(client: DiscordClient, member: GuildMember): Promise <boolean> {
        return true;
    }

    async autoComplete(client: DiscordClient, interaction: AutocompleteInteraction, options: AutoCompleteOptions): Promise <void> {

    }

    async default(client: DiscordClient, interaction: Interaction): Promise <void> {
        
    }

    async modalSubmit(client: DiscordClient, interaction: ModalSubmitInteraction): Promise <void> {
        
    }

    async deferReply(msg: Message | CommandInteraction | ContextMenuInteraction, options: MessageOptions | string | MessagePayload | WebhookEditMessageOptions, edit: boolean = false): Promise<Message | CommandInteraction> {
        if (msg instanceof Message) {
            return await msg[edit ? 'edit' : 'reply'](options as (MessageOptions & MessageEditOptions));
        }
        
        return (await msg.editReply(options as string | MessagePayload | WebhookEditMessageOptions)) as Message;
    }

    async perms(client: DiscordClient, message: Message | Interaction) {
        let member: GuildMember | null = null;

        if (message.member && !(message.member instanceof GuildMember)) {
            try {
                member = (await message.guild?.members.fetch(message.member!.user.id)) ?? null;
            }
            catch (e) {
                console.log(e);
                return false;
            }
        }
        else {
            member = message.member;
        }

        for await (let permission of this.permissions) {
            if (!member?.permissions.has(permission, true)) {                
                if (message instanceof Interaction && !message.isRepliable())
                    return; 
    
                await message.reply({
                    embeds: [
                        {
                            description: ":x: You don't have enough permissions to run this command.",
                            color: 0xf14a60
                        }
                    ]
                });
    
                return false;
            }
        }

        if (!(await this.permissionValidation(client, member!))) {
            if (message instanceof Interaction && !message.isRepliable())
                return; 
            
            await message.reply({
                embeds: [
                    {
                        description: ":x: You don't have enough permissions to run this command.",
                        color: 0xf14a60
                    }
                ]
            });

            return false;
        }

        return true;
    }

    async execute(client: DiscordClient, message: Message | Interaction, options: CommandOptions | InteractionOptions) {
        if (!(await this.perms(client, message))) {
            return;
        } 

        await this.run(client, message, options);
    }

    abstract run(client: DiscordClient, message: Message | Interaction, options: CommandOptions | InteractionOptions): Promise<void>;
}
1	rakinar2	577	/**
2			* This file is part of SudoBot.
3			*
4			* Copyright (C) 2021-2022 OSN Inc.
5			*
6			* SudoBot is free software; you can redistribute it and/or modify it
7			* under the terms of the GNU Affero General Public License as published by
8			* the Free Software Foundation, either version 3 of the License, or
9			* (at your option) any later version.
10			*
11			* SudoBot is distributed in the hope that it will be useful, but
12			* WITHOUT ANY WARRANTY; without even the implied warranty of
13			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14			* GNU Affero General Public License for more details.
15			*
16			* You should have received a copy of the GNU Affero General Public License
17			* along with SudoBot. If not, see <https://www.gnu.org/licenses/>.
18			*/
19
20			import { ModalSubmitInteraction } from 'discord-modals';
21			import { PermissionResolvable, AutocompleteInteraction, CommandInteraction, ContextMenuInteraction, Interaction, Message, MessageEditOptions, MessageOptions, MessagePayload, WebhookEditMessageOptions, GuildMember } from 'discord.js';
22			import DiscordClient from '../../client/Client';
23			import AutoCompleteOptions from '../../types/AutoCompleteOptions';
24			import CommandOptions from '../../types/CommandOptions';
25			import InteractionOptions from '../../types/InteractionOptions';
26
27			export default abstract class BaseCommand {
28			supportsInteractions: boolean = false;
29			supportsLegacy: boolean = true;
30			supportsContextMenu: boolean = false;
31			coolDown?: number;
32			ownerOnly: boolean = false;
33			permissions: PermissionResolvable[] = [];
34
35
36			constructor(private name: string, private category: string, private aliases: Array<string>) {
37
38			}
39
40			getName(): string {
41			return this.name;
42			}
43
44			getCategory(): string {
45			return this.category;
46			}
47
48			getAliases(): Array<string> {
49			return this.aliases;
50			}
51
52			async permissionValidation(client: DiscordClient, member: GuildMember): Promise <boolean> {
53			return true;
54			}
55
56			async autoComplete(client: DiscordClient, interaction: AutocompleteInteraction, options: AutoCompleteOptions): Promise <void> {
57
58			}
59
60			async default(client: DiscordClient, interaction: Interaction): Promise <void> {
61
62			}
63
64			async modalSubmit(client: DiscordClient, interaction: ModalSubmitInteraction): Promise <void> {
65
66			}
67
68			async deferReply(msg: Message \| CommandInteraction \| ContextMenuInteraction, options: MessageOptions \| string \| MessagePayload \| WebhookEditMessageOptions, edit: boolean = false): Promise<Message \| CommandInteraction> {
69			if (msg instanceof Message) {
70			return await msg[edit ? 'edit' : 'reply'](options as (MessageOptions & MessageEditOptions));
71			}
72
73			return (await msg.editReply(options as string \| MessagePayload \| WebhookEditMessageOptions)) as Message;
74			}
75
76			async perms(client: DiscordClient, message: Message \| Interaction) {
77			let member: GuildMember \| null = null;
78
79			if (message.member && !(message.member instanceof GuildMember)) {
80			try {
81			member = (await message.guild?.members.fetch(message.member!.user.id)) ?? null;
82			}
83			catch (e) {
84			console.log(e);
85			return false;
86			}
87			}
88			else {
89			member = message.member;
90			}
91
92			for await (let permission of this.permissions) {
93			if (!member?.permissions.has(permission, true)) {
94			if (message instanceof Interaction && !message.isRepliable())
95			return;
96
97			await message.reply({
98			embeds: [
99			{
100			description: ":x: You don't have enough permissions to run this command.",
101			color: 0xf14a60
102			}
103			]
104			});
105
106			return false;
107			}
108			}
109
110			if (!(await this.permissionValidation(client, member!))) {
111			if (message instanceof Interaction && !message.isRepliable())
112			return;
113
114			await message.reply({
115			embeds: [
116			{
117			description: ":x: You don't have enough permissions to run this command.",
118			color: 0xf14a60
119			}
120			]
121			});
122
123			return false;
124			}
125
126			return true;
127			}
128
129			async execute(client: DiscordClient, message: Message \| Interaction, options: CommandOptions \| InteractionOptions) {
130			if (!(await this.perms(client, message))) {
131			return;
132			}
133
134			await this.run(client, message, options);
135			}
136
137			abstract run(client: DiscordClient, message: Message \| Interaction, options: CommandOptions \| InteractionOptions): Promise<void>;
138			}