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	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 } 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			}