commands/automation/BallotCommand.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 { SlashCommandBuilder } from "discord.js";
import Command, { ArgumentType, BasicCommandContext, CommandMessage, CommandReturn, ValidationRule } from "../../core/Command";

export default class BallotCommand extends Command {
    public readonly name = "ballot";
    public readonly subcommands = ["create", "delete", "edit"];
    public readonly validationRules: ValidationRule[] = [
        {
            types: [ArgumentType.String],
            requiredErrorMessage: `Please provide a subcommand! The valid commands are: \`${this.subcommands.join("`, `")}\``,
            typeErrorMessage: "Please provide a valid subcommand!",
            name: "subcommand"
        }
    ];
    public readonly permissions = [];
    public readonly aliases = ["poll"];
    public readonly description = "Create and manage ballots/polls.";
    public readonly slashCommandBuilder = new SlashCommandBuilder()
        .addSubcommand(subcommand =>
            subcommand
                .setName("create")
                .setDescription("Sends a poll/ballot embed")
                .addStringOption(option => option.setName("content").setDescription("The ballot/poll content").setRequired(true))
                .addBooleanOption(option =>
                    option
                        .setName("anonymous")
                        .setDescription("Anonymous mode won't show your name in the ballot. Default is true")
                )
                .addChannelOption(option =>
                    option
                        .setName("channel")
                        .setDescription("The channel where the message will be sent, defaults to the current channel")
                )
        )
        .addSubcommand(subcommand =>
            subcommand
                .setName("view")
                .setDescription("Shows a poll/ballot")
                .addIntegerOption(option => option.setName("id").setDescription("The ballot ID").setRequired(true))
        )
        .addSubcommand(subcommand =>
            subcommand
                .setName("delete")
                .setDescription("Deletes a poll/ballot")
                .addIntegerOption(option => option.setName("id").setDescription("The ballot ID").setRequired(true))
        );

    async execute(message: CommandMessage, context: BasicCommandContext): Promise<CommandReturn> {
        const subcommand = context.isLegacy ? context.parsedNamedArgs.subcommand : context.options.getSubcommand(true);
        const command = this.client.commands.get(`ballot__${subcommand}`);

        if (!command) {
            return;
        }

        if (context.isLegacy) context.args.shift();

        return await command.execute(message, context);
    }
}
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 { SlashCommandBuilder } from "discord.js";
21	import Command, { ArgumentType, BasicCommandContext, CommandMessage, CommandReturn, ValidationRule } from "../../core/Command";
22
23	export default class BallotCommand extends Command {
24	public readonly name = "ballot";
25	public readonly subcommands = ["create", "delete", "edit"];
26	public readonly validationRules: ValidationRule[] = [
27	{
28	types: [ArgumentType.String],
29	requiredErrorMessage: `Please provide a subcommand! The valid commands are: \`${this.subcommands.join("`, `")}\``,
30	typeErrorMessage: "Please provide a valid subcommand!",
31	name: "subcommand"
32	}
33	];
34	public readonly permissions = [];
35	public readonly aliases = ["poll"];
36	public readonly description = "Create and manage ballots/polls.";
37	public readonly slashCommandBuilder = new SlashCommandBuilder()
38	.addSubcommand(subcommand =>
39	subcommand
40	.setName("create")
41	.setDescription("Sends a poll/ballot embed")
42	.addStringOption(option => option.setName("content").setDescription("The ballot/poll content").setRequired(true))
43	.addBooleanOption(option =>
44	option
45	.setName("anonymous")
46	.setDescription("Anonymous mode won't show your name in the ballot. Default is true")
47	)
48	.addChannelOption(option =>
49	option
50	.setName("channel")
51	.setDescription("The channel where the message will be sent, defaults to the current channel")
52	)
53	)
54	.addSubcommand(subcommand =>
55	subcommand
56	.setName("view")
57	.setDescription("Shows a poll/ballot")
58	.addIntegerOption(option => option.setName("id").setDescription("The ballot ID").setRequired(true))
59	)
60	.addSubcommand(subcommand =>
61	subcommand
62	.setName("delete")
63	.setDescription("Deletes a poll/ballot")
64	.addIntegerOption(option => option.setName("id").setDescription("The ballot ID").setRequired(true))
65	);
66
67	async execute(message: CommandMessage, context: BasicCommandContext): Promise<CommandReturn> {
68	const subcommand = context.isLegacy ? context.parsedNamedArgs.subcommand : context.options.getSubcommand(true);
69	const command = this.client.commands.get(`ballot__${subcommand}`);
70
71	if (!command) {
72	return;
73	}
74
75	if (context.isLegacy) context.args.shift();
76
77	return await command.execute(message, context);
78	}
79	}