commands/moderation/TempBanCommand.ts

import { BanOptions, CommandInteraction, GuildMember, Interaction, 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 History from '../../automod/History';
import Punishment from '../../models/Punishment';
import PunishmentType from '../../types/PunishmentType';
import { fetchEmojiStr } from '../../utils/Emoji';
import ms from 'ms';
import { clearTimeoutv2, getTimeouts, setTimeoutv2 } from '../../utils/setTimeout';
import { shouldNotModerate } from '../../utils/util';

export default class tempBanCommand extends BaseCommand {
    supportsInteractions: boolean = true;

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

    async run(client: DiscordClient, msg: Message | CommandInteraction, options: CommandOptions | InteractionOptions) {
        if (!options.isInteraction && typeof options.args[1] === 'undefined') {
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`This command requires at least two arguments.`)
                ]
            });

            return;
        }

        let user: User;
        let banOptions: BanOptions = {
            days: 7
        };
        let time;

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

            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;

            options.args.shift();

            time = options.args[0];

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

        console.log(time);        

        if (!time || !ms(time)) {
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`The time must be a valid time identifier.`)
                ]
            });

            return;
        }

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

                if (member && shouldNotModerate(client, member)) {
                    await msg.reply({
                        embeds: [
                            new MessageEmbed()
                            .setColor('#f14a60')
                            .setDescription(`This user cannot be tempbanned.`)
                        ]
                    });

                    return;     
                }
            }
            catch (e) {
                console.log(e);
            }
        
            await msg.guild?.bans.create(user, banOptions);

            const punishment = await Punishment.create({
                type: PunishmentType.TEMPBAN,
                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,
                meta: {
                    days: banOptions.days,
                    time
                }
            });

            const timeouts = getTimeouts();
            
            for (const timeout of timeouts.values()) {
                if (timeout.row.params) {
                    try {
                        const json = JSON.parse(timeout.row.params);

                        if (json) {
                            if (json[1] === user.id && timeout.row.filePath.endsWith('tempban-remove')) {
                                await clearTimeoutv2(timeout);
                            }
                        }
                    }
                    catch (e) {
                        console.log(e);                    
                    }
                }
            }

            await setTimeoutv2('tempban-remove', time, msg.guild!.id, 'unban ' + user.id, user.id, msg.guild!.id);

            await client.logger.logTempBan(banOptions, msg.guild!, user, punishment);

            await msg.reply({
                embeds: [
                    new MessageEmbed({
                        author: {
                            name: user.tag,
                            iconURL: user.displayAvatarURL()
                        },
                        description: `${await fetchEmojiStr('check')} Temporarily banned user ${user.tag}`,
                        fields: [
                            {
                                name: 'Banned by',
                                value: (<User> msg.member?.user).tag
                            },
                            {
                                name: 'Reason',
                                value: banOptions.reason ?? '*No reason provided*'
                            }
                        ]
                    })
                ]
            });
        }
        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	import { BanOptions, CommandInteraction, GuildMember, Interaction, Message, User } from 'discord.js';
2	import BaseCommand from '../../utils/structures/BaseCommand';
3	import DiscordClient from '../../client/Client';
4	import CommandOptions from '../../types/CommandOptions';
5	import InteractionOptions from '../../types/InteractionOptions';
6	import MessageEmbed from '../../client/MessageEmbed';
7	import getUser from '../../utils/getUser';
8	import History from '../../automod/History';
9	import Punishment from '../../models/Punishment';
10	import PunishmentType from '../../types/PunishmentType';
11	import { fetchEmojiStr } from '../../utils/Emoji';
12	import ms from 'ms';
13	import { clearTimeoutv2, getTimeouts, setTimeoutv2 } from '../../utils/setTimeout';
14	import { shouldNotModerate } from '../../utils/util';
15
16	export default class tempBanCommand extends BaseCommand {
17	supportsInteractions: boolean = true;
18
19	constructor() {
20	super('tempban', 'moderation', []);
21	}
22
23	async run(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
24	if (!options.isInteraction && typeof options.args[1] === 'undefined') {
25	await msg.reply({
26	embeds: [
27	new MessageEmbed()
28	.setColor('#f14a60')
29	.setDescription(`This command requires at least two arguments.`)
30	]
31	});
32
33	return;
34	}
35
36	let user: User;
37	let banOptions: BanOptions = {
38	days: 7
39	};
40	let time;
41
42	if (options.isInteraction) {
43	user = await <User> options.options.getUser('user');
44	time = await <string> options.options.getString('time');
45
46	if (options.options.getString('reason')) {
47	banOptions.reason = await <string> options.options.getString('reason');
48	}
49
50	if (options.options.getInteger('days')) {
51	banOptions.days = await <number> options.options.getInteger('days');
52	}
53	}
54	else {
55	const user2 = await getUser(client, (msg as Message), options);
56
57	if (!user2) {
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	user = user2;
70
71	options.args.shift();
72
73	time = options.args[0];
74
75	const index = await options.args.indexOf('-d');
76
77	if (options.args[1]) {
78	const args = [...options.args];
79	args.shift();
80
81	if (index !== -1) {
82	args.splice(index - 1, 2)
83	}
84
85	banOptions.reason = await args.join(' ');
86	}
87
88	if (index !== -1) {
89	const days = await options.args[index + 1];
90
91	if (days === undefined) {
92	await msg.reply({
93	embeds: [
94	new MessageEmbed()
95	.setColor('#f14a60')
96	.setDescription(`Option \`-d\` (days) requires an argument.`)
97	]
98	});
99
100	return;
101	}
102
103	if (!parseInt(days) \|\| parseInt(days) < 0 \|\| parseInt(days) > 7) {
104	await msg.reply({
105	embeds: [
106	new MessageEmbed()
107	.setColor('#f14a60')
108	.setDescription(`Option \`-d\` (days) requires an argument which must be a valid number and in range of 0-7.`)
109	]
110	});
111
112	return;
113	}
114
115	banOptions.days = await parseInt(days);
116	}
117	}
118
119	console.log(time);
120
121	if (!time \|\| !ms(time)) {
122	await msg.reply({
123	embeds: [
124	new MessageEmbed()
125	.setColor('#f14a60')
126	.setDescription(`The time must be a valid time identifier.`)
127	]
128	});
129
130	return;
131	}
132
133	time = ms(time);
134
135	try {
136	try {
137	const member = await msg.guild?.members.fetch(user.id);
138
139	if (member && shouldNotModerate(client, member)) {
140	await msg.reply({
141	embeds: [
142	new MessageEmbed()
143	.setColor('#f14a60')
144	.setDescription(`This user cannot be tempbanned.`)
145	]
146	});
147
148	return;
149	}
150	}
151	catch (e) {
152	console.log(e);
153	}
154
155	await msg.guild?.bans.create(user, banOptions);
156
157	const punishment = await Punishment.create({
158	type: PunishmentType.TEMPBAN,
159	user_id: user.id,
160	guild_id: msg.guild!.id,
161	mod_id: msg.member!.user.id,
162	mod_tag: (msg.member!.user as User).tag,
163	reason: banOptions.reason ?? undefined,
164	meta: {
165	days: banOptions.days,
166	time
167	}
168	});
169
170	const timeouts = getTimeouts();
171
172	for (const timeout of timeouts.values()) {
173	if (timeout.row.params) {
174	try {
175	const json = JSON.parse(timeout.row.params);
176
177	if (json) {
178	if (json[1] === user.id && timeout.row.filePath.endsWith('tempban-remove')) {
179	await clearTimeoutv2(timeout);
180	}
181	}
182	}
183	catch (e) {
184	console.log(e);
185	}
186	}
187	}
188
189	await setTimeoutv2('tempban-remove', time, msg.guild!.id, 'unban ' + user.id, user.id, msg.guild!.id);
190
191	await client.logger.logTempBan(banOptions, msg.guild!, user, punishment);
192
193	await msg.reply({
194	embeds: [
195	new MessageEmbed({
196	author: {
197	name: user.tag,
198	iconURL: user.displayAvatarURL()
199	},
200	description: `${await fetchEmojiStr('check')} Temporarily banned user ${user.tag}`,
201	fields: [
202	{
203	name: 'Banned by',
204	value: (<User> msg.member?.user).tag
205	},
206	{
207	name: 'Reason',
208	value: banOptions.reason ?? 'No reason provided'
209	}
210	]
211	})
212	]
213	});
214	}
215	catch (e) {
216	await msg.reply({
217	embeds: [
218	new MessageEmbed()
219	.setColor('#f14a60')
220	.setDescription("Failed to ban this user. Maybe missing permisions or I'm not allowed to ban this user?")
221	]
222	});
223
224	return;
225	}
226	}
227	}