commands/moderation/InfractionCommand.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 { MessageMentions, PermissionsBitField, SlashCommandBuilder } from "discord.js";
import Command, { ArgumentType, BasicCommandContext, CommandMessage, CommandReturn, ValidationRule } from "../../core/Command";
import { safeUserFetch } from "../../utils/fetch";
import { isSnowflake } from "../../utils/utils";

const infractionTypes = [
    {
        name: "Ban",
        value: InfractionType.BAN
    },
    {
        name: "Kick",
        value: InfractionType.KICK
    },
    {
        name: "Mute",
        value: InfractionType.MUTE
    },
    {
        name: "Warning",
        value: InfractionType.WARNING
    },
    {
        name: "Unmute",
        value: InfractionType.UNMUTE
    },
    {
        name: "Unban",
        value: InfractionType.UNBAN
    },
    {
        name: "Bulk message delete",
        value: InfractionType.BULK_DELETE_MESSAGE
    },
    {
        name: "Temporary Ban",
        value: InfractionType.TEMPBAN
    },
    {
        name: "Softban",
        value: InfractionType.SOFTBAN
    },
    {
        name: "Timeout",
        value: InfractionType.TIMEOUT
    },
    {
        name: "Timeout remove",
        value: InfractionType.TIMEOUT_REMOVE
    },
    {
        name: "Note",
        value: InfractionType.TIMEOUT_REMOVE
    }
].map(o => ({ ...o, value: o.value.toLowerCase() }));

export default class InfractionCommand extends Command {
    public readonly name = "infraction";
    public readonly subcommands = ["view", "create", "edit", "delete", "list", "clear", "s", "l"];
    public readonly subcommandShortList = ["view", "create", "edit", "delete", "list", "clear"];
    public readonly subCommandCheck = true;
    public readonly validationRules: ValidationRule[] = [
        {
            types: [ArgumentType.String],
            name: "subcommand",
            errors: {
                required: `Please provide a valid subcommand! The available subcommands are: \`${this.subcommandShortList.join(
                    "`, `"
                )}\`.`
            }
        }
    ];
    public readonly permissions = [PermissionsBitField.Flags.ModerateMembers, PermissionsBitField.Flags.ViewAuditLog];
    public readonly permissionMode = "or";
    public readonly description = "Manage infractions.";
    public readonly detailedDescription = "Use this command to manage infractions.";
    public readonly argumentSyntaxes = ["<subcommand> [...args]"];
    public readonly aliases = ["i", "inf", "infs"];

    public readonly slashCommandBuilder = new SlashCommandBuilder()
        .addSubcommand(subcommand =>
            subcommand
                .setName("view")
                .setDescription("View information about an infraction")
                .addIntegerOption(option => option.setName("id").setDescription("The infraction ID").setRequired(true))
        )
        .addSubcommand(subcommand =>
            subcommand
                .setName("edit")
                .setDescription("Update reason/duration of an infraction")
                .addIntegerOption(option => option.setName("id").setDescription("The infraction ID").setRequired(true))
                .addStringOption(option => option.setName("new_reason").setDescription("New reason to set"))
                .addStringOption(option => option.setName("new_duration").setDescription("New duration to set"))
                .addBooleanOption(option =>
                    option
                        .setName("silent")
                        .setDescription("Specify if the bot should not let the user know about this, defaults to true")
                )
        )
        .addSubcommand(subcommand =>
            subcommand
                .setName("delete")
                .setDescription("Delete an infraction")
                .addIntegerOption(option => option.setName("id").setDescription("The infraction ID").setRequired(true))
        )
        .addSubcommand(subcommand =>
            subcommand
                .setName("clear")
                .setDescription("Clear infractions for a user")
                .addUserOption(option => option.setName("user").setDescription("The target user").setRequired(true))
                .addStringOption(option =>
                    option
                        .setName("type")
                        .setDescription("Specify infraction type")
                        .setChoices(...infractionTypes)
                )
        )
        .addSubcommand(subcommand =>
            subcommand
                .setName("list")
                .setDescription("List infractions for a user")
                .addUserOption(option => option.setName("user").setDescription("The target user").setRequired(true))
        )
        .addSubcommand(subcommand =>
            subcommand
                .setName("create")
                .setDescription("Add infractions to a user")
                .addUserOption(option => option.setName("user").setDescription("The target user").setRequired(true))
                .addStringOption(option =>
                    option
                        .setName("type")
                        .setDescription("Specify infraction type")
                        .setChoices(...infractionTypes)
                        .setRequired(true)
                )
                .addStringOption(option => option.setName("reason").setDescription("The reason for giving this infraction"))
                .addStringOption(option =>
                    option.setName("duration").setDescription("The duration of this infraction (e.g. 45, 1h30m)")
                )
        );

    async execute(message: CommandMessage, context: BasicCommandContext): Promise<CommandReturn> {
        const subcommand = context.isLegacy ? context.parsedNamedArgs.subcommand : context.options.getSubcommand(true);

        if ((subcommand === "edit" || subcommand === "create") && context.isLegacy) {
            await message.reply(
                `${this.emoji("error")} Please use the slash command \`/infraction ${subcommand}\` to perform this action.`
            );
            return;
        }

        if (["clear", "list", "s", "l"].includes(subcommand)) {
            if (context.isLegacy && !context.args[1]) {
                await message.reply(`${this.emoji("error")} Please provide a user to perform this action!`);
                return;
            }

            if (context.isLegacy) {
                let userId = "";

                if (isSnowflake(context.args[1])) {
                    userId = context.args[1];
                } else if (MessageMentions.UsersPattern.test(context.args[1])) {
                    userId = context.args[1].substring(context.args[1].includes("!") ? 3 : 2, context.args[1].length - 1);
                } else {
                    await message.reply(
                        `${this.emoji("error")} Please provide a valid user mention or ID to perform this action!`
                    );
                    return;
                }

                const user = await safeUserFetch(this.client, userId);

                if (!user) {
                    await message.reply(`${this.emoji("error")} That user does not exist!`);
                    return;
                }

                context.parsedNamedArgs.user = user;
            }
        } else {
            if (context.isLegacy && !context.args[1]) {
                await message.reply(`${this.emoji("error")} Please provide an infraction ID to perform this action!`);
                return;
            }

            if (context.isLegacy && isNaN(parseInt(context.args[1]))) {
                await message.reply(`${this.emoji("error")} Please provide a __valid__ infraction ID to perform this action!`);
                return;
            }

            if (context.isLegacy) context.parsedNamedArgs.id = parseInt(context.args[1]);
        }

        await this.deferIfInteraction(message);

        const command = this.client.commands.get(`infraction__${subcommand}`);

        if (command) {
            return await command.execute(message, {
                ...context
            });
        }
    }
}
1	rakinar2	577	/**
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 { MessageMentions, PermissionsBitField, SlashCommandBuilder } from "discord.js";
22			import Command, { ArgumentType, BasicCommandContext, CommandMessage, CommandReturn, ValidationRule } from "../../core/Command";
23			import { safeUserFetch } from "../../utils/fetch";
24			import { isSnowflake } from "../../utils/utils";
25
26			const infractionTypes = [
27			{
28			name: "Ban",
29			value: InfractionType.BAN
30			},
31			{
32			name: "Kick",
33			value: InfractionType.KICK
34			},
35			{
36			name: "Mute",
37			value: InfractionType.MUTE
38			},
39			{
40			name: "Warning",
41			value: InfractionType.WARNING
42			},
43			{
44			name: "Unmute",
45			value: InfractionType.UNMUTE
46			},
47			{
48			name: "Unban",
49			value: InfractionType.UNBAN
50			},
51			{
52			name: "Bulk message delete",
53			value: InfractionType.BULK_DELETE_MESSAGE
54			},
55			{
56			name: "Temporary Ban",
57			value: InfractionType.TEMPBAN
58			},
59			{
60			name: "Softban",
61			value: InfractionType.SOFTBAN
62			},
63			{
64			name: "Timeout",
65			value: InfractionType.TIMEOUT
66			},
67			{
68			name: "Timeout remove",
69			value: InfractionType.TIMEOUT_REMOVE
70			},
71			{
72			name: "Note",
73			value: InfractionType.TIMEOUT_REMOVE
74			}
75			].map(o => ({ ...o, value: o.value.toLowerCase() }));
76
77			export default class InfractionCommand extends Command {
78			public readonly name = "infraction";
79			public readonly subcommands = ["view", "create", "edit", "delete", "list", "clear", "s", "l"];
80			public readonly subcommandShortList = ["view", "create", "edit", "delete", "list", "clear"];
81			public readonly subCommandCheck = true;
82			public readonly validationRules: ValidationRule[] = [
83			{
84			types: [ArgumentType.String],
85			name: "subcommand",
86			errors: {
87			required: `Please provide a valid subcommand! The available subcommands are: \`${this.subcommandShortList.join(
88			"`, `"
89			)}\`.`
90			}
91			}
92			];
93			public readonly permissions = [PermissionsBitField.Flags.ModerateMembers, PermissionsBitField.Flags.ViewAuditLog];
94			public readonly permissionMode = "or";
95			public readonly description = "Manage infractions.";
96			public readonly detailedDescription = "Use this command to manage infractions.";
97			public readonly argumentSyntaxes = ["<subcommand> [...args]"];
98			public readonly aliases = ["i", "inf", "infs"];
99
100			public readonly slashCommandBuilder = new SlashCommandBuilder()
101			.addSubcommand(subcommand =>
102			subcommand
103			.setName("view")
104			.setDescription("View information about an infraction")
105			.addIntegerOption(option => option.setName("id").setDescription("The infraction ID").setRequired(true))
106			)
107			.addSubcommand(subcommand =>
108			subcommand
109			.setName("edit")
110			.setDescription("Update reason/duration of an infraction")
111			.addIntegerOption(option => option.setName("id").setDescription("The infraction ID").setRequired(true))
112			.addStringOption(option => option.setName("new_reason").setDescription("New reason to set"))
113			.addStringOption(option => option.setName("new_duration").setDescription("New duration to set"))
114			.addBooleanOption(option =>
115			option
116			.setName("silent")
117			.setDescription("Specify if the bot should not let the user know about this, defaults to true")
118			)
119			)
120			.addSubcommand(subcommand =>
121			subcommand
122			.setName("delete")
123			.setDescription("Delete an infraction")
124			.addIntegerOption(option => option.setName("id").setDescription("The infraction ID").setRequired(true))
125			)
126			.addSubcommand(subcommand =>
127			subcommand
128			.setName("clear")
129			.setDescription("Clear infractions for a user")
130			.addUserOption(option => option.setName("user").setDescription("The target user").setRequired(true))
131			.addStringOption(option =>
132			option
133			.setName("type")
134			.setDescription("Specify infraction type")
135			.setChoices(...infractionTypes)
136			)
137			)
138			.addSubcommand(subcommand =>
139			subcommand
140			.setName("list")
141			.setDescription("List infractions for a user")
142			.addUserOption(option => option.setName("user").setDescription("The target user").setRequired(true))
143			)
144			.addSubcommand(subcommand =>
145			subcommand
146			.setName("create")
147			.setDescription("Add infractions to a user")
148			.addUserOption(option => option.setName("user").setDescription("The target user").setRequired(true))
149			.addStringOption(option =>
150			option
151			.setName("type")
152			.setDescription("Specify infraction type")
153			.setChoices(...infractionTypes)
154			.setRequired(true)
155			)
156			.addStringOption(option => option.setName("reason").setDescription("The reason for giving this infraction"))
157			.addStringOption(option =>
158			option.setName("duration").setDescription("The duration of this infraction (e.g. 45, 1h30m)")
159			)
160			);
161
162			async execute(message: CommandMessage, context: BasicCommandContext): Promise<CommandReturn> {
163			const subcommand = context.isLegacy ? context.parsedNamedArgs.subcommand : context.options.getSubcommand(true);
164
165			if ((subcommand === "edit" \|\| subcommand === "create") && context.isLegacy) {
166			await message.reply(
167			`${this.emoji("error")} Please use the slash command \`/infraction ${subcommand}\` to perform this action.`
168			);
169			return;
170			}
171
172			if (["clear", "list", "s", "l"].includes(subcommand)) {
173			if (context.isLegacy && !context.args[1]) {
174			await message.reply(`${this.emoji("error")} Please provide a user to perform this action!`);
175			return;
176			}
177
178			if (context.isLegacy) {
179			let userId = "";
180
181			if (isSnowflake(context.args[1])) {
182			userId = context.args[1];
183			} else if (MessageMentions.UsersPattern.test(context.args[1])) {
184			userId = context.args[1].substring(context.args[1].includes("!") ? 3 : 2, context.args[1].length - 1);
185			} else {
186			await message.reply(
187			`${this.emoji("error")} Please provide a valid user mention or ID to perform this action!`
188			);
189			return;
190			}
191
192			const user = await safeUserFetch(this.client, userId);
193
194			if (!user) {
195			await message.reply(`${this.emoji("error")} That user does not exist!`);
196			return;
197			}
198
199			context.parsedNamedArgs.user = user;
200			}
201			} else {
202			if (context.isLegacy && !context.args[1]) {
203			await message.reply(`${this.emoji("error")} Please provide an infraction ID to perform this action!`);
204			return;
205			}
206
207			if (context.isLegacy && isNaN(parseInt(context.args[1]))) {
208			await message.reply(`${this.emoji("error")} Please provide a __valid__ infraction ID to perform this action!`);
209			return;
210			}
211
212			if (context.isLegacy) context.parsedNamedArgs.id = parseInt(context.args[1]);
213			}
214
215			await this.deferIfInteraction(message);
216
217			const command = this.client.commands.get(`infraction__${subcommand}`);
218
219			if (command) {
220			return await command.execute(message, {
221			...context
222			});
223			}
224			}
225			}