commands/automation/BallotCreateCommand.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 { ActionRowBuilder, ButtonBuilder, ButtonStyle, ChannelType, Message, User } from "discord.js";
import Command, { BasicCommandContext, CommandMessage, CommandReturn } from "../../core/Command";
import { logError } from "../../utils/logger";
import { getComponentEmojiResolvable, isTextableChannel } from "../../utils/utils";

export default class BallotCreateCommand extends Command {
    public readonly name = "ballot__create";
    public readonly permissions = [];
    public readonly description = "Sends a poll/ballot embed.";
    public readonly supportsInteractions: boolean = false;
    public readonly supportsLegacy: boolean = false;

    async execute(message: CommandMessage, context: BasicCommandContext): Promise<CommandReturn> {
        if (context.isLegacy && context.args[0] === undefined) {
            await this.error(message, "Please provide the content to put inside the ballot/poll!");
            return;
        }

        await this.deferIfInteraction(message, { ephemeral: true });

        const content = context.isLegacy
            ? (message as Message).content
                  .substring(this.client.configManager.config[message.guildId!]?.prefix?.length ?? 1)
                  .trimStart()
                  .substring(context.argv[0] === "ballot" ? "ballot".length : 0)
                  .trimStart()
                  .substring(context.argv[0] === "ballot" ? this.name.replace("ballot__", "").length : context.argv[0].length)
                  .trim()
            : context.options.getString("content", true);

        const anonymous = (context.isLegacy ? null : context.options.getBoolean("anonymous")) ?? false;
        const channel =
            (context.isLegacy ? null : context.options.getChannel<ChannelType.GuildText>("channel")) ?? message.channel!;

        if (!isTextableChannel(channel)) {
            await this.error(message, "Cannot send messages into a non-text based channel!");
            return;
        }

        try {
            const ballotMessage = await channel.send({
                embeds: [
                    {
                        author: {
                            icon_url: anonymous
                                ? message.guild?.iconURL() ?? undefined
                                : (message.member?.user as User).displayAvatarURL(),
                            name: anonymous ? "Staff" : message.member!.user.username
                        },
                        color: 0x007bff,
                        description: content,
                        footer: {
                            text: `0 Votes • React to vote!`
                        }
                    }
                ],
                files:
                    message instanceof Message
                        ? message.attachments.map(({ name, proxyURL }) => ({
                              name,
                              attachment: proxyURL
                          }))
                        : undefined,
                components: [
                    new ActionRowBuilder<ButtonBuilder>().addComponents(
                        new ButtonBuilder()
                            .setCustomId("ballot__upvote")
                            .setEmoji(getComponentEmojiResolvable(this.client, "ArrowTop") ?? "⬆️")
                            .setStyle(ButtonStyle.Secondary),
                        new ButtonBuilder()
                            .setCustomId("ballot__downvote")
                            .setEmoji(getComponentEmojiResolvable(this.client, "ArrowDown") ?? "⬇️")
                            .setStyle(ButtonStyle.Secondary)
                    )
                ]
            });

            try {
                const ballot = await this.client.ballotManager.create({
                    content,
                    guildId: message.guildId!,
                    userId: message.member!.user.id,
                    channelId: channel.id,
                    messageId: ballotMessage.id,
                    anonymous,
                    files: message instanceof Message ? [...message.attachments.map(a => a.proxyURL).values()] : []
                });

                await this.success(
                    message,
                    `The ballot/poll has been created successfully.\nID: \`${ballot.id}\`\n${
                        message instanceof Message && message.attachments.size > 0
                            ? "Please do not delete your message, otherwise the attachments will be lost."
                            : ""
                    }`
                );
            } catch (e) {
                logError(e);
                return;
            }
        } catch (e) {
            logError(e);

            await this.error(
                message,
                "An error has occurred while sending the message. Make sure I have enough permissions to send messages here!"
            );

            return;
        }
    }
}
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 { ActionRowBuilder, ButtonBuilder, ButtonStyle, ChannelType, Message, User } from "discord.js";
21	import Command, { BasicCommandContext, CommandMessage, CommandReturn } from "../../core/Command";
22	import { logError } from "../../utils/logger";
23	import { getComponentEmojiResolvable, isTextableChannel } from "../../utils/utils";
24
25	export default class BallotCreateCommand extends Command {
26	public readonly name = "ballot__create";
27	public readonly permissions = [];
28	public readonly description = "Sends a poll/ballot embed.";
29	public readonly supportsInteractions: boolean = false;
30	public readonly supportsLegacy: boolean = false;
31
32	async execute(message: CommandMessage, context: BasicCommandContext): Promise<CommandReturn> {
33	if (context.isLegacy && context.args[0] === undefined) {
34	await this.error(message, "Please provide the content to put inside the ballot/poll!");
35	return;
36	}
37
38	await this.deferIfInteraction(message, { ephemeral: true });
39
40	const content = context.isLegacy
41	? (message as Message).content
42	.substring(this.client.configManager.config[message.guildId!]?.prefix?.length ?? 1)
43	.trimStart()
44	.substring(context.argv[0] === "ballot" ? "ballot".length : 0)
45	.trimStart()
46	.substring(context.argv[0] === "ballot" ? this.name.replace("ballot__", "").length : context.argv[0].length)
47	.trim()
48	: context.options.getString("content", true);
49
50	const anonymous = (context.isLegacy ? null : context.options.getBoolean("anonymous")) ?? false;
51	const channel =
52	(context.isLegacy ? null : context.options.getChannel<ChannelType.GuildText>("channel")) ?? message.channel!;
53
54	if (!isTextableChannel(channel)) {
55	await this.error(message, "Cannot send messages into a non-text based channel!");
56	return;
57	}
58
59	try {
60	const ballotMessage = await channel.send({
61	embeds: [
62	{
63	author: {
64	icon_url: anonymous
65	? message.guild?.iconURL() ?? undefined
66	: (message.member?.user as User).displayAvatarURL(),
67	name: anonymous ? "Staff" : message.member!.user.username
68	},
69	color: 0x007bff,
70	description: content,
71	footer: {
72	text: `0 Votes • React to vote!`
73	}
74	}
75	],
76	files:
77	message instanceof Message
78	? message.attachments.map(({ name, proxyURL }) => ({
79	name,
80	attachment: proxyURL
81	}))
82	: undefined,
83	components: [
84	new ActionRowBuilder<ButtonBuilder>().addComponents(
85	new ButtonBuilder()
86	.setCustomId("ballot__upvote")
87	.setEmoji(getComponentEmojiResolvable(this.client, "ArrowTop") ?? "⬆️")
88	.setStyle(ButtonStyle.Secondary),
89	new ButtonBuilder()
90	.setCustomId("ballot__downvote")
91	.setEmoji(getComponentEmojiResolvable(this.client, "ArrowDown") ?? "⬇️")
92	.setStyle(ButtonStyle.Secondary)
93	)
94	]
95	});
96
97	try {
98	const ballot = await this.client.ballotManager.create({
99	content,
100	guildId: message.guildId!,
101	userId: message.member!.user.id,
102	channelId: channel.id,
103	messageId: ballotMessage.id,
104	anonymous,
105	files: message instanceof Message ? [...message.attachments.map(a => a.proxyURL).values()] : []
106	});
107
108	await this.success(
109	message,
110	`The ballot/poll has been created successfully.\nID: \`${ballot.id}\`\n${
111	message instanceof Message && message.attachments.size > 0
112	? "Please do not delete your message, otherwise the attachments will be lost."
113	: ""
114	}`
115	);
116	} catch (e) {
117	logError(e);
118	return;
119	}
120	} catch (e) {
121	logError(e);
122
123	await this.error(
124	message,
125	"An error has occurred while sending the message. Make sure I have enough permissions to send messages here!"
126	);
127
128	return;
129	}
130	}
131	}