commands/moderation/BeanCommand.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 { CommandInteraction, GuildMember, Message, Permissions, User } from 'discord.js';
import BaseCommand from '../../utils/structures/BaseCommand';
import DiscordClient from '../../client/Client';
import CommandOptions from '../../types/CommandOptions';
import InteractionOptions from '../../types/InteractionOptions';
import MessageEmbed from '../../client/MessageEmbed';
import getMember from '../../utils/getMember';
import Punishment from '../../models/Punishment';
import PunishmentType from '../../types/PunishmentType';

export default class BeanCommand extends BaseCommand {
    supportsInteractions: boolean = true;
    supportsContextMenu: boolean = true;
    permissions = [Permissions.FLAGS.MANAGE_MESSAGES];

    constructor() {
        super('bean', 'moderation', []);
    }

    async run(client: DiscordClient, msg: Message | CommandInteraction, options: CommandOptions | InteractionOptions) {
        if (!options.isInteraction && typeof options.args[0] === 'undefined') {
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`This command requires at least one argument.`)
                ]
            });

            return;
        }

        let user: GuildMember;
        let dm = true;
        let reason: string | undefined;
        
        if (msg instanceof CommandInteraction) {
            await msg.deferReply();   
        }
        
        if (options.isInteraction) {
            user = await <GuildMember> options.options.getMember('member');

            if (!user) {
                await this.deferReply(msg, {
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription("Invalid member given.")
                    ]
                });
    
                return;
            }

            if (options.options.getString('reason')) {
                reason = await <string> options.options.getString('reason');
            }
        }
        else {
            try {
                const user2 = await getMember((msg as Message), options);

                if (!user2) {
                    throw new Error('Invalid user');
                }

                user = user2;
            }
            catch (e) {
                await this.deferReply(msg, {
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription(`Invalid user given.`)
                    ]
                });
    
                return;
            }

            console.log(user);

            if (options.args[1]) {
                const args = [...options.args];
                args.shift();
                reason = await args.join(' ');
            }
        }

        if (user.id === client.user?.id) {
            const random = Math.random() >= 0.5;
            
            await this.deferReply(msg, {
                content: `Oh no no no... wait wait, you can't just do that with me${random ? "... I'm such an innocent bot :innocent:, PWEASE don't do that :pleading_face:" : "!?!? Can you?"}`,
                files: [random ? "https://tenor.com/view/folded-hands-emoji-funny-animals-gray-cat-cute-pwease-gif-14039745.gif" : 'https://tenor.com/view/are-you-even-capable-vera-bennett-wentworth-can-you-handle-this-are-you-qualified-gif-22892513.gif']
            });

            return;
        }

        try {            
            const { numericId: id } = await Punishment.create({
                type: PunishmentType.BEAN,
                user_id: user.id,
                guild_id: msg.guild!.id,
                mod_id: msg.member!.user.id,
                mod_tag: (msg.member!.user as User).tag,
                reason,
                createdAt: new Date()
            });

            try {
                await user.send({
                    embeds: [
                        new MessageEmbed()
                            .setAuthor({
                                iconURL: <string> msg.guild!.iconURL(),
                                name: `\tYou've been beaned in ${msg.guild!.name}`
                            })
                            .addFields([
                                {
                                    name: "Reason",
                                    value: typeof reason === 'undefined' ? '*No reason provided*' : reason
                                },
                                {
                                    name: 'Bean ID',
                                    value: id + ''
                                }
                            ])
                    ]
                });
            }
            catch (e) {
                console.log(e);
                dm = false;
            }

            await this.deferReply(msg, {
                embeds: [
                    new MessageEmbed()
                    .setAuthor({
                        name: user.user.tag,
                        iconURL: user.user.displayAvatarURL(),
                    })
                    .setDescription(user.user.tag + " has been beaned." + (!dm ? "\nThey have DMs disabled. They will not know that they were beaned." : ''))
                    .addFields([
                        {
                            name: "Beaned by",
                            value: (msg.member!.user as User).tag
                        },
                        {
                            name: "Reason",
                            value: reason === undefined ? "*No reason provided*" : reason
                        },
                        {
                            name: 'Bean ID',
                            value: id + ''
                        }
                    ])
                ]
            });
        }
        catch (e) {
            console.log(e);            
        }
    }
}
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 { CommandInteraction, GuildMember, Message, Permissions, User } from 'discord.js';
21			import BaseCommand from '../../utils/structures/BaseCommand';
22			import DiscordClient from '../../client/Client';
23			import CommandOptions from '../../types/CommandOptions';
24			import InteractionOptions from '../../types/InteractionOptions';
25			import MessageEmbed from '../../client/MessageEmbed';
26			import getMember from '../../utils/getMember';
27			import Punishment from '../../models/Punishment';
28			import PunishmentType from '../../types/PunishmentType';
29
30			export default class BeanCommand extends BaseCommand {
31			supportsInteractions: boolean = true;
32			supportsContextMenu: boolean = true;
33			permissions = [Permissions.FLAGS.MANAGE_MESSAGES];
34
35			constructor() {
36			super('bean', 'moderation', []);
37			}
38
39			async run(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
40			if (!options.isInteraction && typeof options.args[0] === 'undefined') {
41			await msg.reply({
42			embeds: [
43			new MessageEmbed()
44			.setColor('#f14a60')
45			.setDescription(`This command requires at least one argument.`)
46			]
47			});
48
49			return;
50			}
51
52			let user: GuildMember;
53			let dm = true;
54			let reason: string \| undefined;
55
56			if (msg instanceof CommandInteraction) {
57			await msg.deferReply();
58			}
59
60			if (options.isInteraction) {
61			user = await <GuildMember> options.options.getMember('member');
62
63			if (!user) {
64			await this.deferReply(msg, {
65			embeds: [
66			new MessageEmbed()
67			.setColor('#f14a60')
68			.setDescription("Invalid member given.")
69			]
70			});
71
72			return;
73			}
74
75			if (options.options.getString('reason')) {
76			reason = await <string> options.options.getString('reason');
77			}
78			}
79			else {
80			try {
81			const user2 = await getMember((msg as Message), options);
82
83			if (!user2) {
84			throw new Error('Invalid user');
85			}
86
87			user = user2;
88			}
89			catch (e) {
90			await this.deferReply(msg, {
91			embeds: [
92			new MessageEmbed()
93			.setColor('#f14a60')
94			.setDescription(`Invalid user given.`)
95			]
96			});
97
98			return;
99			}
100
101			console.log(user);
102
103			if (options.args[1]) {
104			const args = [...options.args];
105			args.shift();
106			reason = await args.join(' ');
107			}
108			}
109
110			if (user.id === client.user?.id) {
111			const random = Math.random() >= 0.5;
112
113			await this.deferReply(msg, {
114			content: `Oh no no no... wait wait, you can't just do that with me${random ? "... I'm such an innocent bot :innocent:, PWEASE don't do that :pleading_face:" : "!?!? Can you?"}`,
115			files: [random ? "https://tenor.com/view/folded-hands-emoji-funny-animals-gray-cat-cute-pwease-gif-14039745.gif" : 'https://tenor.com/view/are-you-even-capable-vera-bennett-wentworth-can-you-handle-this-are-you-qualified-gif-22892513.gif']
116			});
117
118			return;
119			}
120
121			try {
122			const { numericId: id } = await Punishment.create({
123			type: PunishmentType.BEAN,
124			user_id: user.id,
125			guild_id: msg.guild!.id,
126			mod_id: msg.member!.user.id,
127			mod_tag: (msg.member!.user as User).tag,
128			reason,
129			createdAt: new Date()
130			});
131
132			try {
133			await user.send({
134			embeds: [
135			new MessageEmbed()
136			.setAuthor({
137			iconURL: <string> msg.guild!.iconURL(),
138			name: `\tYou've been beaned in ${msg.guild!.name}`
139			})
140			.addFields([
141			{
142			name: "Reason",
143			value: typeof reason === 'undefined' ? 'No reason provided' : reason
144			},
145			{
146			name: 'Bean ID',
147			value: id + ''
148			}
149			])
150			]
151			});
152			}
153			catch (e) {
154			console.log(e);
155			dm = false;
156			}
157
158			await this.deferReply(msg, {
159			embeds: [
160			new MessageEmbed()
161			.setAuthor({
162			name: user.user.tag,
163			iconURL: user.user.displayAvatarURL(),
164			})
165			.setDescription(user.user.tag + " has been beaned." + (!dm ? "\nThey have DMs disabled. They will not know that they were beaned." : ''))
166			.addFields([
167			{
168			name: "Beaned by",
169			value: (msg.member!.user as User).tag
170			},
171			{
172			name: "Reason",
173			value: reason === undefined ? "No reason provided" : reason
174			},
175			{
176			name: 'Bean ID',
177			value: id + ''
178			}
179			])
180			]
181			});
182			}
183			catch (e) {
184			console.log(e);
185			}
186			}
187			}