commands/moderation/NoteEditCommand.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 { Message, PermissionsBitField } from "discord.js";
import Command, { BasicCommandContext, CommandMessage, CommandReturn, ValidationRule } from "../../core/Command";

export default class NoteEditCommand extends Command {
    public readonly name = "note__edit";
    public readonly validationRules: ValidationRule[] = [];
    public readonly permissions = [PermissionsBitField.Flags.ModerateMembers, PermissionsBitField.Flags.ViewAuditLog];
    public readonly permissionMode = "or";
    public readonly description = "Edit a note";
    public readonly argumentSyntaxes = ["<id> <new_content>"];

    async execute(message: CommandMessage, context: BasicCommandContext): Promise<CommandReturn> {
        if (context.isLegacy && context.args[0] === undefined) {
            await this.error(message, "Please specify the ID of the note to delete!");
            return;
        }

        let id = context.isLegacy ? undefined : context.options.getInteger("id", true);

        if (context.isLegacy) {
            id = parseInt(context.args[0]);

            if (isNaN(id)) {
                await this.error(message, "Invalid note ID given - note IDs must be valid numbers!");
                return;
            }
        }

        await this.deferIfInteraction(message);

        const content = context.isLegacy
            ? (message as Message).content
                  .substring(this.client.configManager.config[message.guildId!]?.prefix?.length ?? 1)
                  .trimStart()
                  .substring("note".length)
                  .trimStart()
                  .substring(this.name.replace("note__", "").length)
                  .trimStart()
                  .substring(context.args[0].length)
                  .trimEnd()
            : context.options.getString("content", true);

        const { count } = await this.client.prisma.infraction.updateMany({
            where: {
                id,
                guildId: message.guildId!,
                type: InfractionType.NOTE
            },
            data: {
                reason: content
            }
        });

        if (count === 0) {
            await this.error(message, "No such note found with that ID!");
            return;
        }

        await this.deferredReply(message, `${this.emoji("check")} Note updated successfully.`);
    }
}
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 { Message, PermissionsBitField } from "discord.js";
22	import Command, { BasicCommandContext, CommandMessage, CommandReturn, ValidationRule } from "../../core/Command";
23
24	export default class NoteEditCommand extends Command {
25	public readonly name = "note__edit";
26	public readonly validationRules: ValidationRule[] = [];
27	public readonly permissions = [PermissionsBitField.Flags.ModerateMembers, PermissionsBitField.Flags.ViewAuditLog];
28	public readonly permissionMode = "or";
29	public readonly description = "Edit a note";
30	public readonly argumentSyntaxes = ["<id> <new_content>"];
31
32	async execute(message: CommandMessage, context: BasicCommandContext): Promise<CommandReturn> {
33	if (context.isLegacy && context.args[0] === undefined) {
34	await this.error(message, "Please specify the ID of the note to delete!");
35	return;
36	}
37
38	let id = context.isLegacy ? undefined : context.options.getInteger("id", true);
39
40	if (context.isLegacy) {
41	id = parseInt(context.args[0]);
42
43	if (isNaN(id)) {
44	await this.error(message, "Invalid note ID given - note IDs must be valid numbers!");
45	return;
46	}
47	}
48
49	await this.deferIfInteraction(message);
50
51	const content = context.isLegacy
52	? (message as Message).content
53	.substring(this.client.configManager.config[message.guildId!]?.prefix?.length ?? 1)
54	.trimStart()
55	.substring("note".length)
56	.trimStart()
57	.substring(this.name.replace("note__", "").length)
58	.trimStart()
59	.substring(context.args[0].length)
60	.trimEnd()
61	: context.options.getString("content", true);
62
63	const { count } = await this.client.prisma.infraction.updateMany({
64	where: {
65	id,
66	guildId: message.guildId!,
67	type: InfractionType.NOTE
68	},
69	data: {
70	reason: content
71	}
72	});
73
74	if (count === 0) {
75	await this.error(message, "No such note found with that ID!");
76	return;
77	}
78
79	await this.deferredReply(message, `${this.emoji("check")} Note updated successfully.`);
80	}
81	}