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	/**
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	}