commands/moderation/MuteCommand.ts

import { BanOptions, CommandInteraction, GuildMember, Interaction, 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 ms from 'ms';
import PunishmentType from '../../types/PunishmentType';
import { hasPermission, shouldNotModerate } from '../../utils/util';

export async function mute(client: DiscordClient, dateTime: number | undefined, user: GuildMember, msg: Message | CommandInteraction, timeInterval: number | undefined, reason: string | undefined, hard: boolean = false) {
    try {
        const { default: Punishment } = await import('../../models/Punishment');
        
        const { getTimeouts, clearTimeoutv2, setTimeoutv2 } = await import('../../utils/setTimeout');

        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('unmute-job')) {
                            await clearTimeoutv2(timeout);
                        }
                    }
                }
                catch (e) {
                    console.log(e);                    
                }
            }
        }

        if (dateTime && timeInterval) {
            await setTimeoutv2('unmute-job', timeInterval, msg.guild!.id, `unmute ${user.id}`, msg.guild!.id, user.id);
        }
        
        if (hard) {
            const { default: Hardmute } = await import("../../models/Hardmute");
            const roles = await user.roles.cache.filter(r => r.id !== msg.guild!.id);
            await user.roles.remove(roles, reason);

            await Hardmute.create({
                user_id: user.id,
                roles: roles.map(role => role.id),
                guild_id: msg.guild!.id,
                createdAt: new Date()
            });
        }

        const role = await msg.guild!.roles.fetch(client.config.get('mute_role'));
        await user.roles.add(role!, reason);

        await Punishment.create({
            type: hard ? PunishmentType.HARDMUTE : PunishmentType.MUTE,
            user_id: user.id,
            guild_id: msg.guild!.id,
            mod_id: msg.member!.user.id,
            mod_tag: (msg.member!.user as User).tag,
            reason,
            meta: {
                time: timeInterval ? ms(timeInterval) : undefined
            },
            createdAt: new Date()
        });
        
        await client.logger.logMute(user, reason === undefined || reason.trim() === '' ? "*No reason provided*" : reason, timeInterval, msg.member!.user as User, hard);

        try {
            await user.send({
                embeds: [
                    new MessageEmbed()
                    .setAuthor({
                        iconURL: <string> msg.guild!.iconURL(),
                        name: `\tYou have been muted in ${msg.guild!.name}`
                    })
                    .addField("Reason", reason === undefined || reason.trim() === '' ? "*No reason provided*" : reason)
                ]
            });
        }
        catch (e) {
            console.log(e);
        }
    }
    catch (e) {
        console.log(e);
        
        if (msg instanceof Message)
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription("Failed to assign the muted role to this user. Maybe missing permisions/roles or I'm not allowed to assign roles this user?")
                ]
            });
        else
            await msg.editReply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription("Failed to assign the muted role to this user. Maybe missing permisions/roles or I'm not allowed to assign roles this user?")
                ]
            });

        return;
    }
}

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

    constructor() {
        super('mute', '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;
        }

        if (msg instanceof CommandInteraction)
            await msg.deferReply();

        let user: GuildMember;
        let reason: string | undefined;
        let time: string | undefined;
        let timeInterval: number | undefined;
        let dateTime: number | undefined;
        let hard: boolean = false;

        if (options.isInteraction) {
            user = await <GuildMember> options.options.getMember('member');

            if (options.options.getString('reason')) {
                reason = await <string> options.options.getString('reason');
            }

            if (options.options.getBoolean('hardmute')) {
                hard = await <boolean> options.options.getBoolean('hardmute');
            }

            if (options.options.getString('time')) {
                time = await options.options.getString('time') as string;
                timeInterval = await ms(time);

                if (!timeInterval) {
                    await this.deferReply(msg, {
                        embeds: [
                            new MessageEmbed()
                            .setColor('#f14a60')
                            .setDescription(`Option \`-t\` (time) requires an argument which must be a valid time interval.`)
                        ]
                    });
        
                    return;
                }
            }
        }
        else {
            const user2 = await getMember((msg as Message), options);

            if (!user2) {
                await this.deferReply(msg, {
                    embeds: [
                        new MessageEmbed()
                        .setColor('#f14a60')
                        .setDescription(`Invalid user given.`)
                    ]
                });
    
                return;
            }

            user = user2;

            const index = await options.args.indexOf('-t');

            if (options.args[1]) {
                const args = [...options.args];
                args.shift();

                if (index !== -1) {
                    args.splice(index - 1, 2);
                }

                reason = await args.join(' ');
            }

            if (index !== -1) {
                time = await options.args[index + 1];

                if (time === undefined) {
                    await this.deferReply(msg, {
                        embeds: [
                            new MessageEmbed()
                            .setColor('#f14a60')
                            .setDescription(`Option \`-t\` (time) requires an argument.`)
                        ]
                    });
        
                    return;
                }

                if (!ms(time)) {
                    await this.deferReply(msg, {
                        embeds: [
                            new MessageEmbed()
                            .setColor('#f14a60')
                            .setDescription(`Option \`-t\` (time) requires an argument which must be a valid time interval.`)
                        ]
                    });
        
                    return;
                }

                timeInterval = await ms(time);
            }
        }

        if (timeInterval) {
            dateTime = Date.now() + timeInterval;
        }

        if (!(await hasPermission(client, user, msg, null, "You don't have permission to mute this user."))) {
            return;
        }
        
        if (shouldNotModerate(client, user)) {
            await msg.reply({
                embeds: [
                    {
                        description: "This user cannot be muted."
                    }
                ]
            });

            return;
        }
        
        await mute(client, dateTime, user, msg, timeInterval, reason, hard);

        const fields = [
            {
                name: "Muted by",
                value: (msg.member!.user as User).tag
            },
            {
                name: "Reason",
                value: reason === undefined || reason.trim() === '' ? "*No reason provided*" : reason
            },
            {
                name: "Duration",
                value: time === undefined ? "*No duration set*" : (time + '')
            },
            {
                name: "Role Takeout",
                value: hard ? 'Yes' : 'No'
            }
        ];

        console.log(fields);        

        await this.deferReply(msg, {
            embeds: [
                new MessageEmbed()
                .setAuthor({
                    name: user.user.tag,
                    iconURL: user.displayAvatarURL(),
                })
                .setDescription(user.user.tag + " has been muted.")
                .addFields(fields)
            ]
        });
    }
}
1	import { BanOptions, CommandInteraction, GuildMember, Interaction, Message, Permissions, 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 getMember from '../../utils/getMember';
8	import ms from 'ms';
9	import PunishmentType from '../../types/PunishmentType';
10	import { hasPermission, shouldNotModerate } from '../../utils/util';
11
12	export async function mute(client: DiscordClient, dateTime: number \| undefined, user: GuildMember, msg: Message \| CommandInteraction, timeInterval: number \| undefined, reason: string \| undefined, hard: boolean = false) {
13	try {
14	const { default: Punishment } = await import('../../models/Punishment');
15
16	const { getTimeouts, clearTimeoutv2, setTimeoutv2 } = await import('../../utils/setTimeout');
17
18	const timeouts = getTimeouts();
19
20	for (const timeout of timeouts.values()) {
21	if (timeout.row.params) {
22	try {
23	const json = JSON.parse(timeout.row.params);
24
25	if (json) {
26	if (json[1] === user.id && timeout.row.filePath.endsWith('unmute-job')) {
27	await clearTimeoutv2(timeout);
28	}
29	}
30	}
31	catch (e) {
32	console.log(e);
33	}
34	}
35	}
36
37	if (dateTime && timeInterval) {
38	await setTimeoutv2('unmute-job', timeInterval, msg.guild!.id, `unmute ${user.id}`, msg.guild!.id, user.id);
39	}
40
41	if (hard) {
42	const { default: Hardmute } = await import("../../models/Hardmute");
43	const roles = await user.roles.cache.filter(r => r.id !== msg.guild!.id);
44	await user.roles.remove(roles, reason);
45
46	await Hardmute.create({
47	user_id: user.id,
48	roles: roles.map(role => role.id),
49	guild_id: msg.guild!.id,
50	createdAt: new Date()
51	});
52	}
53
54	const role = await msg.guild!.roles.fetch(client.config.get('mute_role'));
55	await user.roles.add(role!, reason);
56
57	await Punishment.create({
58	type: hard ? PunishmentType.HARDMUTE : PunishmentType.MUTE,
59	user_id: user.id,
60	guild_id: msg.guild!.id,
61	mod_id: msg.member!.user.id,
62	mod_tag: (msg.member!.user as User).tag,
63	reason,
64	meta: {
65	time: timeInterval ? ms(timeInterval) : undefined
66	},
67	createdAt: new Date()
68	});
69
70	await client.logger.logMute(user, reason === undefined \|\| reason.trim() === '' ? "No reason provided" : reason, timeInterval, msg.member!.user as User, hard);
71
72	try {
73	await user.send({
74	embeds: [
75	new MessageEmbed()
76	.setAuthor({
77	iconURL: <string> msg.guild!.iconURL(),
78	name: `\tYou have been muted in ${msg.guild!.name}`
79	})
80	.addField("Reason", reason === undefined \|\| reason.trim() === '' ? "No reason provided" : reason)
81	]
82	});
83	}
84	catch (e) {
85	console.log(e);
86	}
87	}
88	catch (e) {
89	console.log(e);
90
91	if (msg instanceof Message)
92	await msg.reply({
93	embeds: [
94	new MessageEmbed()
95	.setColor('#f14a60')
96	.setDescription("Failed to assign the muted role to this user. Maybe missing permisions/roles or I'm not allowed to assign roles this user?")
97	]
98	});
99	else
100	await msg.editReply({
101	embeds: [
102	new MessageEmbed()
103	.setColor('#f14a60')
104	.setDescription("Failed to assign the muted role to this user. Maybe missing permisions/roles or I'm not allowed to assign roles this user?")
105	]
106	});
107
108	return;
109	}
110	}
111
112	export default class MuteCommand extends BaseCommand {
113	supportsInteractions: boolean = true;
114	permissions = [Permissions.FLAGS.MODERATE_MEMBERS];
115
116	constructor() {
117	super('mute', 'moderation', []);
118	}
119
120	async run(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
121	if (!options.isInteraction && typeof options.args[0] === 'undefined') {
122	await msg.reply({
123	embeds: [
124	new MessageEmbed()
125	.setColor('#f14a60')
126	.setDescription(`This command requires at least one argument.`)
127	]
128	});
129
130	return;
131	}
132
133	if (msg instanceof CommandInteraction)
134	await msg.deferReply();
135
136	let user: GuildMember;
137	let reason: string \| undefined;
138	let time: string \| undefined;
139	let timeInterval: number \| undefined;
140	let dateTime: number \| undefined;
141	let hard: boolean = false;
142
143	if (options.isInteraction) {
144	user = await <GuildMember> options.options.getMember('member');
145
146	if (options.options.getString('reason')) {
147	reason = await <string> options.options.getString('reason');
148	}
149
150	if (options.options.getBoolean('hardmute')) {
151	hard = await <boolean> options.options.getBoolean('hardmute');
152	}
153
154	if (options.options.getString('time')) {
155	time = await options.options.getString('time') as string;
156	timeInterval = await ms(time);
157
158	if (!timeInterval) {
159	await this.deferReply(msg, {
160	embeds: [
161	new MessageEmbed()
162	.setColor('#f14a60')
163	.setDescription(`Option \`-t\` (time) requires an argument which must be a valid time interval.`)
164	]
165	});
166
167	return;
168	}
169	}
170	}
171	else {
172	const user2 = await getMember((msg as Message), options);
173
174	if (!user2) {
175	await this.deferReply(msg, {
176	embeds: [
177	new MessageEmbed()
178	.setColor('#f14a60')
179	.setDescription(`Invalid user given.`)
180	]
181	});
182
183	return;
184	}
185
186	user = user2;
187
188	const index = await options.args.indexOf('-t');
189
190	if (options.args[1]) {
191	const args = [...options.args];
192	args.shift();
193
194	if (index !== -1) {
195	args.splice(index - 1, 2);
196	}
197
198	reason = await args.join(' ');
199	}
200
201	if (index !== -1) {
202	time = await options.args[index + 1];
203
204	if (time === undefined) {
205	await this.deferReply(msg, {
206	embeds: [
207	new MessageEmbed()
208	.setColor('#f14a60')
209	.setDescription(`Option \`-t\` (time) requires an argument.`)
210	]
211	});
212
213	return;
214	}
215
216	if (!ms(time)) {
217	await this.deferReply(msg, {
218	embeds: [
219	new MessageEmbed()
220	.setColor('#f14a60')
221	.setDescription(`Option \`-t\` (time) requires an argument which must be a valid time interval.`)
222	]
223	});
224
225	return;
226	}
227
228	timeInterval = await ms(time);
229	}
230	}
231
232	if (timeInterval) {
233	dateTime = Date.now() + timeInterval;
234	}
235
236	if (!(await hasPermission(client, user, msg, null, "You don't have permission to mute this user."))) {
237	return;
238	}
239
240	if (shouldNotModerate(client, user)) {
241	await msg.reply({
242	embeds: [
243	{
244	description: "This user cannot be muted."
245	}
246	]
247	});
248
249	return;
250	}
251
252	await mute(client, dateTime, user, msg, timeInterval, reason, hard);
253
254	const fields = [
255	{
256	name: "Muted by",
257	value: (msg.member!.user as User).tag
258	},
259	{
260	name: "Reason",
261	value: reason === undefined \|\| reason.trim() === '' ? "No reason provided" : reason
262	},
263	{
264	name: "Duration",
265	value: time === undefined ? "No duration set" : (time + '')
266	},
267	{
268	name: "Role Takeout",
269	value: hard ? 'Yes' : 'No'
270	}
271	];
272
273	console.log(fields);
274
275	await this.deferReply(msg, {
276	embeds: [
277	new MessageEmbed()
278	.setAuthor({
279	name: user.user.tag,
280	iconURL: user.displayAvatarURL(),
281	})
282	.setDescription(user.user.tag + " has been muted.")
283	.addFields(fields)
284	]
285	});
286	}
287	}