commands/moderation/InfractionCreateCommand.ts

/**
 * This file is part of SudoBot.
 *
 * Copyright (C) 2021-2023 OSN Developers.
 *
 * 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 { InfractionType } from "@prisma/client";
import { ChatInputCommandInteraction, GuildMember, PermissionsBitField } from "discord.js";
import Command, { CommandReturn, ValidationRule } from "../../core/Command";
import { ChatInputCommandContext } from "../../services/CommandManager";
import { stringToTimeInterval } from "../../utils/datetime";
import { logError } from "../../utils/logger";

export default class InfractionCreateCommand extends Command {
    public readonly name = "infraction__create";
    public readonly validationRules: ValidationRule[] = [];
    public readonly permissions = [PermissionsBitField.Flags.ModerateMembers, PermissionsBitField.Flags.ViewAuditLog];
    public readonly supportsLegacy: boolean = false;
    public readonly permissionMode = "or";

    public readonly description = "Create infractions.";
    public readonly detailedDescription = "Create and assign an infraction to someone.";
    public readonly argumentSyntaxes = ["<user> <type> [reason] [duration]"];

    async execute(interaction: ChatInputCommandInteraction, context: ChatInputCommandContext): Promise<CommandReturn> {
        const user = interaction.options.getUser("user", true);
        const type = interaction.options.getString("type", true);
        const reason = interaction.options.getString("reason");
        const duration = interaction.options.getString("duration");
        const parsedDuration = duration ? stringToTimeInterval(duration) : null;

        if (parsedDuration && parsedDuration.error) {
            await interaction.editReply(`${this.emoji("error")} ${parsedDuration.error} provided in the \`duration\` field`);
            return;
        }

        if (!(type in InfractionType)) {
            await interaction.editReply(`${this.emoji("error")} Invalid infraction type provided in the \`type\` field`);
            return;
        }

        try {
            const member = interaction.guild!.members.cache.get(user.id) ?? (await interaction.guild!.members.fetch(user.id));

            if (!(await this.client.permissionManager.shouldModerate(member, interaction.member! as GuildMember))) {
                await this.error(interaction, "You don't have permission to create infractions for this user!");
                return;
            }
        } catch (e) {
            logError(e);
        }

        const infraction = await this.client.prisma.infraction.create({
            data: {
                userId: user.id,
                guildId: interaction.guildId!,
                moderatorId: interaction.user.id,
                type: type as InfractionType,
                reason,
                metadata: parsedDuration?.result
                    ? {
                          duration: parsedDuration.result * 1000
                      }
                    : undefined,
                expiresAt: parsedDuration?.result ? new Date(parsedDuration?.result * 1000 + Date.now()) : undefined
            }
        });

        await interaction.editReply({
            embeds: [
                this.client.infractionManager.generateInfractionDetailsEmbed(user, infraction).setTitle("Infraction Created")
            ]
        });
    }
}
1	/**
2	* This file is part of SudoBot.
3	*
4	* Copyright (C) 2021-2023 OSN Developers.
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 { InfractionType } from "@prisma/client";
21	import { ChatInputCommandInteraction, GuildMember, PermissionsBitField } from "discord.js";
22	import Command, { CommandReturn, ValidationRule } from "../../core/Command";
23	import { ChatInputCommandContext } from "../../services/CommandManager";
24	import { stringToTimeInterval } from "../../utils/datetime";
25	import { logError } from "../../utils/logger";
26
27	export default class InfractionCreateCommand extends Command {
28	public readonly name = "infraction__create";
29	public readonly validationRules: ValidationRule[] = [];
30	public readonly permissions = [PermissionsBitField.Flags.ModerateMembers, PermissionsBitField.Flags.ViewAuditLog];
31	public readonly supportsLegacy: boolean = false;
32	public readonly permissionMode = "or";
33
34	public readonly description = "Create infractions.";
35	public readonly detailedDescription = "Create and assign an infraction to someone.";
36	public readonly argumentSyntaxes = ["<user> <type> [reason] [duration]"];
37
38	async execute(interaction: ChatInputCommandInteraction, context: ChatInputCommandContext): Promise<CommandReturn> {
39	const user = interaction.options.getUser("user", true);
40	const type = interaction.options.getString("type", true);
41	const reason = interaction.options.getString("reason");
42	const duration = interaction.options.getString("duration");
43	const parsedDuration = duration ? stringToTimeInterval(duration) : null;
44
45	if (parsedDuration && parsedDuration.error) {
46	await interaction.editReply(`${this.emoji("error")} ${parsedDuration.error} provided in the \`duration\` field`);
47	return;
48	}
49
50	if (!(type in InfractionType)) {
51	await interaction.editReply(`${this.emoji("error")} Invalid infraction type provided in the \`type\` field`);
52	return;
53	}
54
55	try {
56	const member = interaction.guild!.members.cache.get(user.id) ?? (await interaction.guild!.members.fetch(user.id));
57
58	if (!(await this.client.permissionManager.shouldModerate(member, interaction.member! as GuildMember))) {
59	await this.error(interaction, "You don't have permission to create infractions for this user!");
60	return;
61	}
62	} catch (e) {
63	logError(e);
64	}
65
66	const infraction = await this.client.prisma.infraction.create({
67	data: {
68	userId: user.id,
69	guildId: interaction.guildId!,
70	moderatorId: interaction.user.id,
71	type: type as InfractionType,
72	reason,
73	metadata: parsedDuration?.result
74	? {
75	duration: parsedDuration.result * 1000
76	}
77	: undefined,
78	expiresAt: parsedDuration?.result ? new Date(parsedDuration?.result * 1000 + Date.now()) : undefined
79	}
80	});
81
82	await interaction.editReply({
83	embeds: [
84	this.client.infractionManager.generateInfractionDetailsEmbed(user, infraction).setTitle("Infraction Created")
85	]
86	});
87	}
88	}