commands/automation/BallotViewCommand.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 } from "discord.js";
import Command, { BasicCommandContext, CommandMessage, CommandReturn } from "../../core/Command";
import { displayDate } from "../../utils/datetime";
import { userInfo } from "../../utils/embed";
import { safeUserFetch } from "../../utils/fetch";

export default class BallotViewCommand extends Command {
    public readonly name = "ballot__view";
    public readonly permissions = [];
    public readonly description = "Shows a poll/ballot.";
    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 ballot ID!");
            return;
        }

        await this.deferIfInteraction(message);

        const id = context.isLegacy
            ? parseInt(context.args[0])
            : context.options.getInteger("id", true);

        if (isNaN(id)) {
            await this.error(
                message,
                "Invalid ballot ID given! Ballot IDs must be numeric values."
            );
            return;
        }

        const ballot = await this.client.ballotManager.get({
            id,
            guildId: message.guildId!
        });

        if (!ballot) {
            await this.error(message, "No such ballot exists with that ID!");
            return;
        }

        const user = await safeUserFetch(this.client, ballot.userId);
        const url = `https://discord.com/channels/${encodeURIComponent(
            ballot.guildId
        )}/${encodeURIComponent(ballot.channelId)}/${encodeURIComponent(ballot.messageId)}`;

        await this.deferredReply(message, {
            files: ballot.files.map(url => ({ attachment: url })),
            embeds: [
                {
                    author: {
                        name: ballot.anonymous ? "Staff" : user?.username ?? "Unknown",
                        icon_url: ballot.anonymous
                            ? message.guild!.iconURL() ?? undefined
                            : user?.displayAvatarURL(),
                        url
                    },
                    description: ballot.content,
                    color: 0x007bff,
                    fields: [
                        {
                            name: "Total Votes",
                            value: `⚪ **${ballot.upvotes.length - ballot.downvotes.length}**`,
                            inline: true
                        },
                        {
                            name: "Upvotes",
                            value: `${this.emoji("ArrowTop")} ${ballot.upvotes.length}`,
                            inline: true
                        },
                        {
                            name: "Downvotes",
                            value: `${this.emoji("ArrowDown")} ${ballot.downvotes.length}`,
                            inline: true
                        },
                        {
                            name: "Created At",
                            value: `${displayDate(ballot.createdAt)}`,
                            inline: true
                        },
                        {
                            name: "Updated At",
                            value: `${displayDate(ballot.updatedAt)}`,
                            inline: true
                        },
                        {
                            name: "Created By",
                            value: user ? userInfo(user) : "*Not available*"
                        }
                    ]
                }
            ],
            components: [
                new ActionRowBuilder<ButtonBuilder>().addComponents(
                    new ButtonBuilder()
                        .setURL(url)
                        .setStyle(ButtonStyle.Link)
                        .setLabel("Go to ballot message")
                )
            ]
        });
    }
}
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 } from "discord.js";
21	import Command, { BasicCommandContext, CommandMessage, CommandReturn } from "../../core/Command";
22	import { displayDate } from "../../utils/datetime";
23	import { userInfo } from "../../utils/embed";
24	import { safeUserFetch } from "../../utils/fetch";
25
26	export default class BallotViewCommand extends Command {
27	public readonly name = "ballot__view";
28	public readonly permissions = [];
29	public readonly description = "Shows a poll/ballot.";
30	public readonly supportsInteractions: boolean = true;
31	public readonly supportsLegacy: boolean = true;
32
33	async execute(message: CommandMessage, context: BasicCommandContext): Promise<CommandReturn> {
34	if (context.isLegacy && context.args[0] === undefined) {
35	await this.error(message, "Please provide the ballot ID!");
36	return;
37	}
38
39	await this.deferIfInteraction(message);
40
41	const id = context.isLegacy
42	? parseInt(context.args[0])
43	: context.options.getInteger("id", true);
44
45	if (isNaN(id)) {
46	await this.error(
47	message,
48	"Invalid ballot ID given! Ballot IDs must be numeric values."
49	);
50	return;
51	}
52
53	const ballot = await this.client.ballotManager.get({
54	id,
55	guildId: message.guildId!
56	});
57
58	if (!ballot) {
59	await this.error(message, "No such ballot exists with that ID!");
60	return;
61	}
62
63	const user = await safeUserFetch(this.client, ballot.userId);
64	const url = `https://discord.com/channels/${encodeURIComponent(
65	ballot.guildId
66	)}/${encodeURIComponent(ballot.channelId)}/${encodeURIComponent(ballot.messageId)}`;
67
68	await this.deferredReply(message, {
69	files: ballot.files.map(url => ({ attachment: url })),
70	embeds: [
71	{
72	author: {
73	name: ballot.anonymous ? "Staff" : user?.username ?? "Unknown",
74	icon_url: ballot.anonymous
75	? message.guild!.iconURL() ?? undefined
76	: user?.displayAvatarURL(),
77	url
78	},
79	description: ballot.content,
80	color: 0x007bff,
81	fields: [
82	{
83	name: "Total Votes",
84	value: `⚪ ${ballot.upvotes.length - ballot.downvotes.length}`,
85	inline: true
86	},
87	{
88	name: "Upvotes",
89	value: `${this.emoji("ArrowTop")} ${ballot.upvotes.length}`,
90	inline: true
91	},
92	{
93	name: "Downvotes",
94	value: `${this.emoji("ArrowDown")} ${ballot.downvotes.length}`,
95	inline: true
96	},
97	{
98	name: "Created At",
99	value: `${displayDate(ballot.createdAt)}`,
100	inline: true
101	},
102	{
103	name: "Updated At",
104	value: `${displayDate(ballot.updatedAt)}`,
105	inline: true
106	},
107	{
108	name: "Created By",
109	value: user ? userInfo(user) : "Not available"
110	}
111	]
112	}
113	],
114	components: [
115	new ActionRowBuilder<ButtonBuilder>().addComponents(
116	new ButtonBuilder()
117	.setURL(url)
118	.setStyle(ButtonStyle.Link)
119	.setLabel("Go to ballot message")
120	)
121	]
122	});
123	}
124	}