commands/moderation/KickCommand.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, ContextMenuInteraction, 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';
import { generateInfractionDescription, hasPermission, shouldNotModerate } from '../../utils/util';

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

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

    async run(client: DiscordClient, msg: Message | CommandInteraction | ContextMenuInteraction, 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 reason: string | undefined;

        if (options.isInteraction) {
            user = await <GuildMember> (msg instanceof ContextMenuInteraction ? options.options.getMember('user') : options.options.getMember('member'));

            if (!user) {
                await msg.reply({
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription("Invalid user 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 msg.reply({
                    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(' ');
            }
        }

        try {
            if (!(await hasPermission(client, user, msg, null, "You don't have permission to kick this user."))) {
                return;
            }
            
            if (!user.kickable || shouldNotModerate(client, user)) 
                throw new Error('User not kickable');

            const { numericId: id } = await Punishment.create({
                type: PunishmentType.KICK,
                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({
                            description: generateInfractionDescription(client, msg.guildId!, 'kick_message')
                        })
                        .setAuthor({
                            iconURL: <string> msg.guild!.iconURL(),
                            name: `\tYou have been kicked from ${msg.guild!.name}`
                        })
                        .setColor('#f14a60')
                        .addField("Reason", reason === undefined || reason.trim() === '' ? "*No reason provided*" : reason)
                        .addFields({
                            name: 'Infraction ID',
                            value: id + ''
                        })
                    ]
                });
            }
            catch (e) {
                console.log(e);
            }
            
            await user.kick(reason);

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

            return;
        }
    }
}
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 { CommandInteraction, ContextMenuInteraction, 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	import { generateInfractionDescription, hasPermission, shouldNotModerate } from '../../utils/util';
30
31	export default class KickCommand extends BaseCommand {
32	supportsInteractions: boolean = true;
33	supportsContextMenu: boolean = true;
34
35	permissions = [Permissions.FLAGS.KICK_MEMBERS];
36
37	constructor() {
38	super('kick', 'moderation', ['Kick']);
39	}
40
41	async run(client: DiscordClient, msg: Message \| CommandInteraction \| ContextMenuInteraction, options: CommandOptions \| InteractionOptions) {
42	if (!options.isInteraction && typeof options.args[0] === 'undefined') {
43	await msg.reply({
44	embeds: [
45	new MessageEmbed()
46	.setColor('#f14a60')
47	.setDescription(`This command requires at least one argument.`)
48	]
49	});
50
51	return;
52	}
53
54	let user: GuildMember;
55	let reason: string \| undefined;
56
57	if (options.isInteraction) {
58	user = await <GuildMember> (msg instanceof ContextMenuInteraction ? options.options.getMember('user') : options.options.getMember('member'));
59
60	if (!user) {
61	await msg.reply({
62	embeds: [
63	new MessageEmbed()
64	.setColor('#f14a60')
65	.setDescription("Invalid user given.")
66	]
67	});
68
69	return;
70	}
71
72	if (options.options.getString('reason')) {
73	reason = await <string> options.options.getString('reason');
74	}
75	}
76	else {
77	try {
78	const user2 = await getMember((msg as Message), options);
79
80	if (!user2) {
81	throw new Error('Invalid user');
82	}
83
84	user = user2;
85	}
86	catch (e) {
87	await msg.reply({
88	embeds: [
89	new MessageEmbed()
90	.setColor('#f14a60')
91	.setDescription(`Invalid user given.`)
92	]
93	});
94
95	return;
96	}
97
98	console.log(user);
99
100	if (options.args[1]) {
101	const args = [...options.args];
102	args.shift();
103	reason = await args.join(' ');
104	}
105	}
106
107	try {
108	if (!(await hasPermission(client, user, msg, null, "You don't have permission to kick this user."))) {
109	return;
110	}
111
112	if (!user.kickable \|\| shouldNotModerate(client, user))
113	throw new Error('User not kickable');
114
115	const { numericId: id } = await Punishment.create({
116	type: PunishmentType.KICK,
117	user_id: user.id,
118	guild_id: msg.guild!.id,
119	mod_id: msg.member!.user.id,
120	mod_tag: (msg.member!.user as User).tag,
121	reason,
122	createdAt: new Date()
123	});
124
125	try {
126	await user.send({
127	embeds: [
128	new MessageEmbed({
129	description: generateInfractionDescription(client, msg.guildId!, 'kick_message')
130	})
131	.setAuthor({
132	iconURL: <string> msg.guild!.iconURL(),
133	name: `\tYou have been kicked from ${msg.guild!.name}`
134	})
135	.setColor('#f14a60')
136	.addField("Reason", reason === undefined \|\| reason.trim() === '' ? "No reason provided" : reason)
137	.addFields({
138	name: 'Infraction ID',
139	value: id + ''
140	})
141	]
142	});
143	}
144	catch (e) {
145	console.log(e);
146	}
147
148	await user.kick(reason);
149
150	await msg.reply({
151	embeds: [
152	new MessageEmbed()
153	.setAuthor({
154	name: user.user.tag,
155	iconURL: user.user.displayAvatarURL(),
156	})
157	.setDescription(user.user.tag + " has been kicked from this server.")
158	.addFields([
159	{
160	name: "Kicked by",
161	value: (msg.member!.user as User).tag
162	},
163	{
164	name: "Reason",
165	value: reason === undefined ? "No reason provided" : reason
166	},
167	{
168	name: 'Infraction ID',
169	value: id + ''
170	}
171	])
172	]
173	});
174	}
175	catch (e) {
176	await msg.reply({
177	embeds: [
178	new MessageEmbed()
179	.setColor('#f14a60')
180	.setDescription("Failed to kick this user. Maybe missing permisions or I'm not allowed to kick this user?")
181	]
182	});
183
184	return;
185	}
186	}
187	}