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 } 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 } 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 {
            await msg.guild?.bans.create(user, banOptions);

            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()
            });

            // await History.create(user.id, msg.guild!, 'ban', msg.member!.user.id, typeof banOptions.reason === 'undefined' ? null : banOptions.reason, async (data: any) => undefined);
        }
        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;
        }

        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 + '')
                    }
                ])
            ]
        });
    }
}
1	/**
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 } 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 } 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	await msg.guild?.bans.create(user, banOptions);
153
154	await Punishment.create({
155	type: PunishmentType.BAN,
156	user_id: user.id,
157	guild_id: msg.guild!.id,
158	mod_id: msg.member!.user.id,
159	mod_tag: (msg.member!.user as User).tag,
160	reason: banOptions.reason ?? undefined,
161	createdAt: new Date()
162	});
163
164	// await History.create(user.id, msg.guild!, 'ban', msg.member!.user.id, typeof banOptions.reason === 'undefined' ? null : banOptions.reason, async (data: any) => undefined);
165	}
166	catch (e) {
167	await msg.reply({
168	embeds: [
169	new MessageEmbed()
170	.setColor('#f14a60')
171	.setDescription("Failed to ban this user. Maybe missing permisions or I'm not allowed to ban this user?")
172	]
173	});
174
175	return;
176	}
177
178	await msg.reply({
179	embeds: [
180	new MessageEmbed()
181	.setAuthor({
182	name: user.tag,
183	iconURL: user.displayAvatarURL(),
184	})
185	.setDescription(user.tag + " has been banned from this server.")
186	.addFields([
187	{
188	name: "Banned by",
189	value: (msg.member!.user as User).tag
190	},
191	{
192	name: "Reason",
193	value: banOptions.reason === undefined ? "No reason provided" : banOptions.reason
194	},
195	{
196	name: "Days of message deletion",
197	value: banOptions.days === undefined ? "No message will be deleted" : (banOptions.days + '')
198	}
199	])
200	]
201	});
202	}
203	}