commands/moderation/BanCommand.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 { Permissions, BanOptions, CommandInteraction, Message, User, GuildBan } 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 getUser from '../../utils/getUser';
import Punishment from '../../models/Punishment';
import PunishmentType from '../../types/PunishmentType';
import { shouldNotModerate, hasPermission, generateInfractionDescription } from '../../utils/util';

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

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

    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: User;
        let banOptions: BanOptions = {};

        if (options.isInteraction) {
            user = await <User> options.options.getUser('user');

            if (options.options.getString('reason')) {
                banOptions.reason = await <string> options.options.getString('reason');
            }

            if (options.options.getInteger('days')) {
                banOptions.days = await <number> options.options.getInteger('days');
            }
        }
        else {
            const user2 = await getUser(client, (msg as Message), options);

            if (!user2) {
                await msg.reply({
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription(`Invalid user given.`)
                    ]
                });
    
                return;
            }

            user = user2;

            const index = await options.args.indexOf('-d');

            if (options.args[1]) {
                const args = [...options.args];
                args.shift();

                if (index !== -1) {
                    args.splice(index - 1, 2)
                }

                banOptions.reason = await args.join(' ');
            }

            if (index !== -1) {
                const days = await options.args[index + 1];

                if (days === undefined) {
                    await msg.reply({
                        embeds: [
                            new MessageEmbed()
                            .setColor('#f14a60')
                            .setDescription(`Option \`-d\` (days) requires an argument.`)
                        ]
                    });
        
                    return;
                }

                if (!parseInt(days) || parseInt(days) < 0 || parseInt(days) > 7) {
                    await msg.reply({
                        embeds: [
                            new MessageEmbed()
                            .setColor('#f14a60')
                            .setDescription(`Option \`-d\` (days) requires an argument which must be a valid number and in range of 0-7.`)
                        ]
                    });
        
                    return;
                }

                banOptions.days = await parseInt(days);
            }
        }

        try {
            const member = await msg.guild?.members.fetch(user.id);

            if (member && !(await hasPermission(client, member, msg, null, "You don't have permission to ban this user."))) {
                return;
            }

            if (member && shouldNotModerate(client, member)) {
                await msg.reply({
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription('Cannot ban this user: Operation not permitted')    
                    ]
                });
                
                return;
            }
        }
        catch (e) {
            console.log(e);
        }
        
        try {
            const { numericId: id } = await Punishment.create({
                type: PunishmentType.BAN,
                user_id: user.id,
                guild_id: msg.guild!.id,
                mod_id: msg.member!.user.id,
                mod_tag: (msg.member!.user as User).tag,
                reason: banOptions.reason ?? undefined,
                createdAt: new Date()
            });

            try {
                await user.send({
                    embeds: [
                        new MessageEmbed({
                            author: {
                                name: `You have been banned in ${msg.guild!.name}`,
                                iconURL: msg.guild!.iconURL() ?? undefined
                            },
                            color: 0xf14a60,
                            description: generateInfractionDescription(client, msg.guildId!, 'ban_message'),
                            fields: [
                                {
                                    name: 'Reason',
                                    value: banOptions.reason === undefined ? "*No reason provided*" : banOptions.reason
                                },
                                {
                                    name: 'Infraction ID',
                                    value: id + ''
                                }
                            ]
                        })
                    ]
                });
            }
            catch (e) {
                console.log(e);
            }

            await msg.guild?.bans.create(user, { ...banOptions, reason: `[BAN] ${banOptions.reason ?? '**No reason provided**'}` });

            client.logger.onGuildBanAdd({
                guild: msg.guild!,
                user,
                reason: banOptions.reason === undefined ? "*No reason provided*" : banOptions.reason
            } as GuildBan, msg.member!.user as User, id).catch(console.error);

            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setAuthor({
                        name: user.tag,
                        iconURL: user.displayAvatarURL(),
                    })
                    .setDescription(user.tag + " has been banned from this server.")
                    .addFields([
                        {
                            name: "Banned by",
                            value: (msg.member!.user as User).tag
                        },
                        {
                            name: "Reason",
                            value: banOptions.reason === undefined ? "*No reason provided*" : banOptions.reason
                        },
                        {
                            name: "Days of message deletion",
                            value: banOptions.days === undefined ? "*No message will be deleted*" : (banOptions.days + '')
                        },
                        {
                            name: 'Infraction ID',
                            value: id + ''
                        }
                    ])
                ]
            });
        }
        catch (e) {
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription("Failed to ban this user. Maybe missing permisions or I'm not allowed to ban this user?")
                ]
            });

            return;
        }
    }
}
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 { Permissions, BanOptions, CommandInteraction, Message, User, GuildBan } 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 getUser from '../../utils/getUser';
27			import Punishment from '../../models/Punishment';
28			import PunishmentType from '../../types/PunishmentType';
29			import { shouldNotModerate, hasPermission, generateInfractionDescription } from '../../utils/util';
30
31			export default class BanCommand extends BaseCommand {
32			supportsInteractions: boolean = true;
33			supportsContextMenu: boolean = true;
34			permissions = [Permissions.FLAGS.BAN_MEMBERS];
35
36			constructor() {
37			super('ban', 'moderation', ['Ban']);
38			}
39
40			async run(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
41			if (!options.isInteraction && typeof options.args[0] === 'undefined') {
42			await msg.reply({
43			embeds: [
44			new MessageEmbed()
45			.setColor('#f14a60')
46			.setDescription(`This command requires at least one argument.`)
47			]
48			});
49
50			return;
51			}
52
53			let user: User;
54			let banOptions: BanOptions = {};
55
56			if (options.isInteraction) {
57			user = await <User> options.options.getUser('user');
58
59			if (options.options.getString('reason')) {
60			banOptions.reason = await <string> options.options.getString('reason');
61			}
62
63			if (options.options.getInteger('days')) {
64			banOptions.days = await <number> options.options.getInteger('days');
65			}
66			}
67			else {
68			const user2 = await getUser(client, (msg as Message), options);
69
70			if (!user2) {
71			await msg.reply({
72			embeds: [
73			new MessageEmbed()
74			.setColor('#f14a60')
75			.setDescription(`Invalid user given.`)
76			]
77			});
78
79			return;
80			}
81
82			user = user2;
83
84			const index = await options.args.indexOf('-d');
85
86			if (options.args[1]) {
87			const args = [...options.args];
88			args.shift();
89
90			if (index !== -1) {
91			args.splice(index - 1, 2)
92			}
93
94			banOptions.reason = await args.join(' ');
95			}
96
97			if (index !== -1) {
98			const days = await options.args[index + 1];
99
100			if (days === undefined) {
101			await msg.reply({
102			embeds: [
103			new MessageEmbed()
104			.setColor('#f14a60')
105			.setDescription(`Option \`-d\` (days) requires an argument.`)
106			]
107			});
108
109			return;
110			}
111
112			if (!parseInt(days) \|\| parseInt(days) < 0 \|\| parseInt(days) > 7) {
113			await msg.reply({
114			embeds: [
115			new MessageEmbed()
116			.setColor('#f14a60')
117			.setDescription(`Option \`-d\` (days) requires an argument which must be a valid number and in range of 0-7.`)
118			]
119			});
120
121			return;
122			}
123
124			banOptions.days = await parseInt(days);
125			}
126			}
127
128			try {
129			const member = await msg.guild?.members.fetch(user.id);
130
131			if (member && !(await hasPermission(client, member, msg, null, "You don't have permission to ban this user."))) {
132			return;
133			}
134
135			if (member && shouldNotModerate(client, member)) {
136			await msg.reply({
137			embeds: [
138			new MessageEmbed()
139			.setColor('#f14a60')
140			.setDescription('Cannot ban this user: Operation not permitted')
141			]
142			});
143
144			return;
145			}
146			}
147			catch (e) {
148			console.log(e);
149			}
150
151			try {
152			const { numericId: id } = await Punishment.create({
153			type: PunishmentType.BAN,
154			user_id: user.id,
155			guild_id: msg.guild!.id,
156			mod_id: msg.member!.user.id,
157			mod_tag: (msg.member!.user as User).tag,
158			reason: banOptions.reason ?? undefined,
159			createdAt: new Date()
160			});
161
162			try {
163			await user.send({
164			embeds: [
165			new MessageEmbed({
166			author: {
167			name: `You have been banned in ${msg.guild!.name}`,
168			iconURL: msg.guild!.iconURL() ?? undefined
169			},
170			color: 0xf14a60,
171			description: generateInfractionDescription(client, msg.guildId!, 'ban_message'),
172			fields: [
173			{
174			name: 'Reason',
175			value: banOptions.reason === undefined ? "No reason provided" : banOptions.reason
176			},
177			{
178			name: 'Infraction ID',
179			value: id + ''
180			}
181			]
182			})
183			]
184			});
185			}
186			catch (e) {
187			console.log(e);
188			}
189
190			await msg.guild?.bans.create(user, { ...banOptions, reason: `[BAN] ${banOptions.reason ?? 'No reason provided'}` });
191
192			client.logger.onGuildBanAdd({
193			guild: msg.guild!,
194			user,
195			reason: banOptions.reason === undefined ? "No reason provided" : banOptions.reason
196			} as GuildBan, msg.member!.user as User, id).catch(console.error);
197
198			await msg.reply({
199			embeds: [
200			new MessageEmbed()
201			.setAuthor({
202			name: user.tag,
203			iconURL: user.displayAvatarURL(),
204			})
205			.setDescription(user.tag + " has been banned from this server.")
206			.addFields([
207			{
208			name: "Banned by",
209			value: (msg.member!.user as User).tag
210			},
211			{
212			name: "Reason",
213			value: banOptions.reason === undefined ? "No reason provided" : banOptions.reason
214			},
215			{
216			name: "Days of message deletion",
217			value: banOptions.days === undefined ? "No message will be deleted" : (banOptions.days + '')
218			},
219			{
220			name: 'Infraction ID',
221			value: id + ''
222			}
223			])
224			]
225			});
226			}
227			catch (e) {
228			await msg.reply({
229			embeds: [
230			new MessageEmbed()
231			.setColor('#f14a60')
232			.setDescription("Failed to ban this user. Maybe missing permisions or I'm not allowed to ban this user?")
233			]
234			});
235
236			return;
237			}
238			}
239			}