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