commands/moderation/UnbanCommand.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, Message, User, Permissions, 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 UnbanQueue from '../../queues/UnbanQueue';

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

    constructor() {
        super('unban', '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: User;

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

            if (!user) {
                await msg.reply({
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription("Invalid user given.")
                    ]
                });
    
                return;
            }
        }
        else {
            try {
                const user2 = await getUser(client, (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);
        }

        try {
            for await (const queue of client.queueManager.queues.values()) {
                if (queue instanceof UnbanQueue && queue.data!.userID === user.id && queue.data!.guildID === msg.guild!.id) {
                    await queue.cancel();
                }
            }

            await msg.guild?.bans.remove(user);

            client.logger.onGuildBanRemove({
                guild: msg.guild!,
                user
            } as GuildBan, msg.member!.user as User).catch(console.error);

            await Punishment.create({
                type: PunishmentType.UNBAN,
                user_id: user.id,
                guild_id: msg.guild!.id,
                mod_id: msg.member!.user.id,
                mod_tag: (msg.member!.user as User).tag,
                createdAt: new Date()
            });
        }
        catch (e) {
            console.log(e);            
        }

        await msg.reply({
            embeds: [
                new MessageEmbed()
                .setAuthor({
                    name: user.tag,
                    iconURL: user.displayAvatarURL(),
                })
                .setDescription(user.tag + " has been unbanned.")
                .addFields([
                    {
                        name: "Unbanned by",
                        value: (msg.member!.user as User).tag
                    },
                ])
            ]
        });
    }
}
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, Message, User, Permissions, 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 UnbanQueue from '../../queues/UnbanQueue';
30
31	export default class UnbanCommand extends BaseCommand {
32	supportsInteractions: boolean = true;
33	permissions = [Permissions.FLAGS.BAN_MEMBERS];
34
35	constructor() {
36	super('unban', '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: User;
53
54	if (options.isInteraction) {
55	user = await <User> options.options.getUser('user');
56
57	if (!user) {
58	await msg.reply({
59	embeds: [
60	new MessageEmbed()
61	.setColor('#f14a60')
62	.setDescription("Invalid user given.")
63	]
64	});
65
66	return;
67	}
68	}
69	else {
70	try {
71	const user2 = await getUser(client, (msg as Message), options);
72
73	if (!user2) {
74	throw new Error('Invalid user');
75	}
76
77	user = user2;
78	}
79	catch (e) {
80	await msg.reply({
81	embeds: [
82	new MessageEmbed()
83	.setColor('#f14a60')
84	.setDescription(`Invalid user given.`)
85	]
86	});
87
88	return;
89	}
90
91	console.log(user);
92	}
93
94	try {
95	for await (const queue of client.queueManager.queues.values()) {
96	if (queue instanceof UnbanQueue && queue.data!.userID === user.id && queue.data!.guildID === msg.guild!.id) {
97	await queue.cancel();
98	}
99	}
100
101	await msg.guild?.bans.remove(user);
102
103	client.logger.onGuildBanRemove({
104	guild: msg.guild!,
105	user
106	} as GuildBan, msg.member!.user as User).catch(console.error);
107
108	await Punishment.create({
109	type: PunishmentType.UNBAN,
110	user_id: user.id,
111	guild_id: msg.guild!.id,
112	mod_id: msg.member!.user.id,
113	mod_tag: (msg.member!.user as User).tag,
114	createdAt: new Date()
115	});
116	}
117	catch (e) {
118	console.log(e);
119	}
120
121	await msg.reply({
122	embeds: [
123	new MessageEmbed()
124	.setAuthor({
125	name: user.tag,
126	iconURL: user.displayAvatarURL(),
127	})
128	.setDescription(user.tag + " has been unbanned.")
129	.addFields([
130	{
131	name: "Unbanned by",
132	value: (msg.member!.user as User).tag
133	},
134	])
135	]
136	});
137	}
138	}