commands/tools/SnippetEditCommand.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 { Message, PermissionsBitField } from "discord.js";
import Command, { ArgumentType, CommandReturn, ValidationRule } from "../../core/Command";
import { LegacyCommandContext } from "../../services/CommandManager";

export default class SnippetEditCommand extends Command {
    public readonly name = "snippet__edit";
    public readonly validationRules: ValidationRule[] = [
        {
            types: [ArgumentType.String],
            requiredErrorMessage: "Please specify the name of the snippet/tag to edit!",
            typeErrorMessage: "Please specify a valid snippet/tag name!",
            name: "name"
        },
        {
            types: [ArgumentType.String],
            requiredErrorMessage: "Please specify what to edit in the snippet! One of these can be specified: `level`, `permission` (`perm`), `content`",
            typeErrorMessage: "Please specify a valid attribute of the snippet to edit! One of these can be specified: `level`, `permission` (`perm`), `content`",
            name: "attr"
        }
    ];
    public readonly permissions = [
        PermissionsBitField.Flags.BanMembers,
        PermissionsBitField.Flags.KickMembers,
        PermissionsBitField.Flags.ManageGuild,
        PermissionsBitField.Flags.ModerateMembers,
        PermissionsBitField.Flags.ManageMessages
    ];
    public readonly permissionMode = "or";
    public readonly aliases: string[] = ["editsnippet", "snippetedit", "tagedit", "edittag", "updatetag"];
    public readonly beta: boolean = true;
    public readonly supportsInteractions = false;

    async execute(message: Message, context: LegacyCommandContext): Promise<CommandReturn> {
        if (!this.client.configManager.config[message.guildId!])
            return;

        const name: string = context.parsedNamedArgs.name;

        if (!this.client.snippetManager.checkPermissionInSnippetCommands(name, message, this)) {
            return;
        }

        const index = name !== context.args[0] ? 1 : 0;

        if (!context.args[index + 1]) {
            await this.error(message, this.validationRules[1].requiredErrorMessage!);
            return;
        }

        const attrs = ["level", "permission", "perm", "content"] as const;

        if (!attrs.includes(context.args[index + 1]?.toLowerCase() as typeof attrs[number])) {
            await this.error(message, this.validationRules[1].typeErrorMessage!);
            return;
        }

        const attr = context.args[index + 1]?.toLowerCase() as typeof attrs[number];

        switch (attr) {
            case "level": {
                if (this.client.configManager.config[message.guildId!]?.permissions.mode !== "levels") {
                    await this.error(message, "This server does not use the level-based permission system. Please switch to the level-based system to set a permission level for this snippet. Alternatively, if you're using the named permission system, please edit the `permission` (or `perm`) attribute instead.");
                    return;
                }

                if (!context.args[index + 2]) {
                    await this.error(message, "Please specify a permission level to set!");
                    return;
                }

                const level = parseInt(context.args[index + 2]);

                if (isNaN(level) || level < 0 || level > 100) {
                    await this.error(message, "Please specify a valid permission level (0-100) to set!");
                    return;
                }

                const { error } = await this.client.snippetManager.updateSnippet({
                    name,
                    guildId: message.guildId!,
                    level
                });

                if (error) {
                    await this.error(message, error);
                    return;
                }

                await this.success(message, "Sucessfully updated snippet permission level.");
                break;
            }

            case "perm":
            case "permission": {
                if (this.client.configManager.config[message.guildId!]?.permissions.mode !== "advanced") {
                    await this.error(message, "This server does not use the named permission system. Please switch to the named permission system to set a named permission for this snippet. Alternatively, if you're using the level-based permission system, please edit the `level` attribute instead.");
                    return;
                }

                if (!context.args[index + 2]) {
                    await this.error(message, "Please specify a permission role name to set!");
                    return;
                }

                const { error } = await this.client.snippetManager.updateSnippet({
                    name,
                    guildId: message.guildId!,
                    permissionRoleName: context.args[index + 2]
                });

                if (error) {
                    await this.error(message, error);
                    return;
                }

                await this.success(message, "Sucessfully updated snippet permissions.");
                break;
            }

            case "content": {
                if (!context.args[index + 2]) {
                    await this.error(message, "Please specify new content for the snippet!");
                    return;
                }

                const content = message.content
                    .slice(this.client.configManager.config[message.guildId!]!.prefix.length)
                    .trimStart()
                    .slice(context.argv[0].length)
                    .trimStart()
                    .slice(context.args[0].length)
                    .trimStart()
                    .slice(context.args[1].length)
                    .trimStart()
                    .slice(index === 1 ? context.args[2].length : 0)
                    .trimStart();

                const { error } = await this.client.snippetManager.updateSnippet({
                    name,
                    guildId: message.guildId!,
                    content
                });

                if (error) {
                    await this.error(message, error);
                    return;
                }

                await this.success(message, "Sucessfully updated snippet content.");
                break;
            }
        }
    }
}
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 { Message, PermissionsBitField } from "discord.js";
21			import Command, { ArgumentType, CommandReturn, ValidationRule } from "../../core/Command";
22			import { LegacyCommandContext } from "../../services/CommandManager";
23
24			export default class SnippetEditCommand extends Command {
25			public readonly name = "snippet__edit";
26			public readonly validationRules: ValidationRule[] = [
27			{
28			types: [ArgumentType.String],
29			requiredErrorMessage: "Please specify the name of the snippet/tag to edit!",
30			typeErrorMessage: "Please specify a valid snippet/tag name!",
31			name: "name"
32			},
33			{
34			types: [ArgumentType.String],
35			requiredErrorMessage: "Please specify what to edit in the snippet! One of these can be specified: `level`, `permission` (`perm`), `content`",
36			typeErrorMessage: "Please specify a valid attribute of the snippet to edit! One of these can be specified: `level`, `permission` (`perm`), `content`",
37			name: "attr"
38			}
39			];
40			public readonly permissions = [
41			PermissionsBitField.Flags.BanMembers,
42			PermissionsBitField.Flags.KickMembers,
43			PermissionsBitField.Flags.ManageGuild,
44			PermissionsBitField.Flags.ModerateMembers,
45			PermissionsBitField.Flags.ManageMessages
46			];
47			public readonly permissionMode = "or";
48			public readonly aliases: string[] = ["editsnippet", "snippetedit", "tagedit", "edittag", "updatetag"];
49			public readonly beta: boolean = true;
50			public readonly supportsInteractions = false;
51
52			async execute(message: Message, context: LegacyCommandContext): Promise<CommandReturn> {
53			if (!this.client.configManager.config[message.guildId!])
54			return;
55
56			const name: string = context.parsedNamedArgs.name;
57
58			if (!this.client.snippetManager.checkPermissionInSnippetCommands(name, message, this)) {
59			return;
60			}
61
62			const index = name !== context.args[0] ? 1 : 0;
63
64			if (!context.args[index + 1]) {
65			await this.error(message, this.validationRules[1].requiredErrorMessage!);
66			return;
67			}
68
69			const attrs = ["level", "permission", "perm", "content"] as const;
70
71			if (!attrs.includes(context.args[index + 1]?.toLowerCase() as typeof attrs[number])) {
72			await this.error(message, this.validationRules[1].typeErrorMessage!);
73			return;
74			}
75
76			const attr = context.args[index + 1]?.toLowerCase() as typeof attrs[number];
77
78			switch (attr) {
79			case "level": {
80			if (this.client.configManager.config[message.guildId!]?.permissions.mode !== "levels") {
81			await this.error(message, "This server does not use the level-based permission system. Please switch to the level-based system to set a permission level for this snippet. Alternatively, if you're using the named permission system, please edit the `permission` (or `perm`) attribute instead.");
82			return;
83			}
84
85			if (!context.args[index + 2]) {
86			await this.error(message, "Please specify a permission level to set!");
87			return;
88			}
89
90			const level = parseInt(context.args[index + 2]);
91
92			if (isNaN(level) \|\| level < 0 \|\| level > 100) {
93			await this.error(message, "Please specify a valid permission level (0-100) to set!");
94			return;
95			}
96
97			const { error } = await this.client.snippetManager.updateSnippet({
98			name,
99			guildId: message.guildId!,
100			level
101			});
102
103			if (error) {
104			await this.error(message, error);
105			return;
106			}
107
108			await this.success(message, "Sucessfully updated snippet permission level.");
109			break;
110			}
111
112			case "perm":
113			case "permission": {
114			if (this.client.configManager.config[message.guildId!]?.permissions.mode !== "advanced") {
115			await this.error(message, "This server does not use the named permission system. Please switch to the named permission system to set a named permission for this snippet. Alternatively, if you're using the level-based permission system, please edit the `level` attribute instead.");
116			return;
117			}
118
119			if (!context.args[index + 2]) {
120			await this.error(message, "Please specify a permission role name to set!");
121			return;
122			}
123
124			const { error } = await this.client.snippetManager.updateSnippet({
125			name,
126			guildId: message.guildId!,
127			permissionRoleName: context.args[index + 2]
128			});
129
130			if (error) {
131			await this.error(message, error);
132			return;
133			}
134
135			await this.success(message, "Sucessfully updated snippet permissions.");
136			break;
137			}
138
139			case "content": {
140			if (!context.args[index + 2]) {
141			await this.error(message, "Please specify new content for the snippet!");
142			return;
143			}
144
145			const content = message.content
146			.slice(this.client.configManager.config[message.guildId!]!.prefix.length)
147			.trimStart()
148			.slice(context.argv[0].length)
149			.trimStart()
150			.slice(context.args[0].length)
151			.trimStart()
152			.slice(context.args[1].length)
153			.trimStart()
154			.slice(index === 1 ? context.args[2].length : 0)
155			.trimStart();
156
157			const { error } = await this.client.snippetManager.updateSnippet({
158			name,
159			guildId: message.guildId!,
160			content
161			});
162
163			if (error) {
164			await this.error(message, error);
165			return;
166			}
167
168			await this.success(message, "Sucessfully updated snippet content.");
169			break;
170			}
171			}
172			}
173			}