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 = true;
    public readonly supportsLegacy: boolean = true;

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