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

export default class NoteCreateCommand extends Command {
    public readonly name = "note__create";
    public readonly validationRules: ValidationRule[] = [];
    public readonly permissions = [PermissionsBitField.Flags.ModerateMembers, PermissionsBitField.Flags.ViewAuditLog];
    public readonly permissionMode = "or";
    public readonly description = "Create a note";
    public readonly argumentSyntaxes = ["<user> <content>"];

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

        if (context.isLegacy && context.args[1] === undefined) {
            await this.error(message, "Please specify the note contents!");
            return;
        }

        let user: User | null | undefined = context.isLegacy ? undefined : context.options.getUser("user", true);

        if (context.isLegacy) {
            user = await safeUserFetch(
                this.client,
                isSnowflake(context.args[0])
                    ? context.args[0]
                    : context.args[0].substring(context.args[0].includes("!") ? 3 : 2, context.args[0].length - 1)
            );
        }

        if (!user) {
            await this.error(message, "Invalid user specified!");
            return;
        }

        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 { id } = await this.client.prisma.infraction.create({
            data: {
                guildId: message.guildId!,
                moderatorId: message.member!.user.id,
                userId: user.id,
                type: InfractionType.NOTE,
                reason: content
            }
        });

        await this.deferredReply(
            message,
            `${this.emoji("check")} Successfully created note for user **${user.tag}**. The note ID is \`${id}\`.`
        );
    }
}
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 { Message, PermissionsBitField, User } from "discord.js";
22			import Command, { BasicCommandContext, CommandMessage, CommandReturn, ValidationRule } from "../../core/Command";
23			import { safeUserFetch } from "../../utils/fetch";
24			import { isSnowflake } from "../../utils/utils";
25
26			export default class NoteCreateCommand extends Command {
27			public readonly name = "note__create";
28			public readonly validationRules: ValidationRule[] = [];
29			public readonly permissions = [PermissionsBitField.Flags.ModerateMembers, PermissionsBitField.Flags.ViewAuditLog];
30			public readonly permissionMode = "or";
31			public readonly description = "Create a note";
32			public readonly argumentSyntaxes = ["<user> <content>"];
33
34			async execute(message: CommandMessage, context: BasicCommandContext): Promise<CommandReturn> {
35			if (context.isLegacy && context.args[0] === undefined) {
36			await this.error(message, "Please specify a user to take note for!");
37			return;
38			}
39
40			if (context.isLegacy && context.args[1] === undefined) {
41			await this.error(message, "Please specify the note contents!");
42			return;
43			}
44
45			let user: User \| null \| undefined = context.isLegacy ? undefined : context.options.getUser("user", true);
46
47			if (context.isLegacy) {
48			user = await safeUserFetch(
49			this.client,
50			isSnowflake(context.args[0])
51			? context.args[0]
52			: context.args[0].substring(context.args[0].includes("!") ? 3 : 2, context.args[0].length - 1)
53			);
54			}
55
56			if (!user) {
57			await this.error(message, "Invalid user specified!");
58			return;
59			}
60
61			const content = context.isLegacy
62			? (message as Message).content
63			.substring(this.client.configManager.config[message.guildId!]?.prefix?.length ?? 1)
64			.trimStart()
65			.substring("note".length)
66			.trimStart()
67			.substring(this.name.replace("note__", "").length)
68			.trimStart()
69			.substring(context.args[0].length)
70			.trimEnd()
71			: context.options.getString("content", true);
72
73			const { id } = await this.client.prisma.infraction.create({
74			data: {
75			guildId: message.guildId!,
76			moderatorId: message.member!.user.id,
77			userId: user.id,
78			type: InfractionType.NOTE,
79			reason: content
80			}
81			});
82
83			await this.deferredReply(
84			message,
85			`${this.emoji("check")} Successfully created note for user ${user.tag}. The note ID is \`${id}\`.`
86			);
87			}
88			}