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 subCommandCheck = true;
    public readonly validationRules: ValidationRule[] = [
        {
            types: [ArgumentType.String],
            errors: {
                required: `Please provide a valid subcommand! The valid commands are: \`${this.subcommands.join(
                    "`, `"
                )}\``,
                "type:invalid": "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("votelist")
                .setDescription("Shows a list of each vote in a poll/ballot")
                .addIntegerOption(option => option.setName("id").setDescription("The ballot ID").setRequired(true))
                .addStringOption(option =>
                    option.setName("mode").setDescription("Determines what kind of data is shown").setChoices(
                        {
                            name: "All Votes",
                            value: "all"
                        },
                        {
                            name: "Upvotes",
                            value: "upvotes"
                        },
                        {
                            name: "Downvotes",
                            value: "downvotes"
                        }
                    )
                )
        )
        .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) {
            await this.error(message, this.validationRules[0].errors!.required!);
            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 subCommandCheck = true;
27	public readonly validationRules: ValidationRule[] = [
28	{
29	types: [ArgumentType.String],
30	errors: {
31	required: `Please provide a valid subcommand! The valid commands are: \`${this.subcommands.join(
32	"`, `"
33	)}\``,
34	"type:invalid": "Please provide a valid subcommand!",
35	},
36	name: "subcommand"
37	}
38	];
39	public readonly permissions = [];
40	public readonly aliases = ["poll"];
41	public readonly description = "Create and manage ballots/polls.";
42	public readonly slashCommandBuilder = new SlashCommandBuilder()
43	.addSubcommand(subcommand =>
44	subcommand
45	.setName("create")
46	.setDescription("Sends a poll/ballot embed")
47	.addStringOption(option => option.setName("content").setDescription("The ballot/poll content").setRequired(true))
48	.addBooleanOption(option =>
49	option
50	.setName("anonymous")
51	.setDescription("Anonymous mode won't show your name in the ballot. Default is true")
52	)
53	.addChannelOption(option =>
54	option
55	.setName("channel")
56	.setDescription("The channel where the message will be sent, defaults to the current channel")
57	)
58	)
59	.addSubcommand(subcommand =>
60	subcommand
61	.setName("view")
62	.setDescription("Shows a poll/ballot")
63	.addIntegerOption(option => option.setName("id").setDescription("The ballot ID").setRequired(true))
64	)
65	.addSubcommand(subcommand =>
66	subcommand
67	.setName("votelist")
68	.setDescription("Shows a list of each vote in a poll/ballot")
69	.addIntegerOption(option => option.setName("id").setDescription("The ballot ID").setRequired(true))
70	.addStringOption(option =>
71	option.setName("mode").setDescription("Determines what kind of data is shown").setChoices(
72	{
73	name: "All Votes",
74	value: "all"
75	},
76	{
77	name: "Upvotes",
78	value: "upvotes"
79	},
80	{
81	name: "Downvotes",
82	value: "downvotes"
83	}
84	)
85	)
86	)
87	.addSubcommand(subcommand =>
88	subcommand
89	.setName("delete")
90	.setDescription("Deletes a poll/ballot")
91	.addIntegerOption(option => option.setName("id").setDescription("The ballot ID").setRequired(true))
92	);
93
94	async execute(message: CommandMessage, context: BasicCommandContext): Promise<CommandReturn> {
95	const subcommand = context.isLegacy ? context.parsedNamedArgs.subcommand : context.options.getSubcommand(true);
96	const command = this.client.commands.get(`ballot__${subcommand}`);
97
98	if (!command) {
99	await this.error(message, this.validationRules[0].errors!.required!);
100	return;
101	}
102
103	if (context.isLegacy) context.args.shift();
104
105	return await command.execute(message, context);
106	}
107	}