commands/automation/BallotCommand.ts

import { CommandInteraction, EmojiIdentifierResolvable, GuildMember, Message, TextChannel, User } from 'discord.js';
import BaseCommand from '../../utils/structures/BaseCommand';
import DiscordClient from '../../client/Client';
import MessageEmbed from '../../client/MessageEmbed';
import CommandOptions from '../../types/CommandOptions';
import InteractionOptions from '../../types/InteractionOptions';
import { fetchEmoji } from '../../utils/Emoji';

export default class BallotCommand extends BaseCommand {
    supportsInteractions = true;

    constructor() {
        super('ballot', 'automation', []);
    }

    async ballotCreate(client: DiscordClient, msg: Message | CommandInteraction, options: CommandOptions | InteractionOptions) {
        if (!options.isInteraction && typeof options.args[0] === 'undefined') {
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`This command requires at least one argument.`)
                ]
            });

            return;
        }

        if (msg instanceof CommandInteraction)
            await msg.deferReply({
                ephemeral: true
            });

        let content: string;
        let channel: TextChannel = msg.channel as TextChannel;
        let anonymous = false;

        if (options.isInteraction) {
            content = <string> await options.options.getString('content');

            if (options.options.getChannel('channel'))
                channel = <TextChannel> await options.options.getChannel('channel');

            if (options.options.getBoolean('anonymous'))
                anonymous = <boolean> await options.options.getBoolean('anonymous');
        }
        else {
            if ((msg as Message).mentions.channels.last() && /\<#(\d+)\>/g.test([...options.args].pop()!)) {
                channel = <TextChannel> (msg as Message).mentions.channels.last();
                options.args.pop();
            }

            content = await options.args.join(' ').trim();
        }

        if (!channel.send) {
            await this.deferReply(msg, {
                content: 'Invalid text channel.'
            });

            return;
        }
        
        const message = await channel.send({
            embeds: [
                new MessageEmbed()
                .setAuthor({
                    name: anonymous ? 'Staff' : (msg.member?.user as User).tag!,
                    iconURL: anonymous ? msg.guild!.iconURL()! : (msg.member as GuildMember)?.displayAvatarURL()
                })
                .setDescription(content)
                .setTimestamp()
            ]
        });

        await client.db.runAsync('INSERT INTO ballots(content, author, msg_id, guild_id, date, channel_id) VALUES(?, ?, ?, ?, ?, ?)', [content, anonymous ? null : msg.member?.user.id, message.id, msg.guild!.id, new Date().toISOString(), msg.channel!.id]);
        const ballot = await client.db.getAsync("SELECT * FROM ballots WHERE msg_id = ? AND guild_id = ? ORDER BY id DESC LIMIT 0, 1", [message.id, msg.guild!.id]);

        await message.react(<EmojiIdentifierResolvable> await fetchEmoji('check'));
        await message.react(<EmojiIdentifierResolvable> await fetchEmoji('error'));

        await this.deferReply(msg, {
            content: `${(await fetchEmoji('check'))!.toString()} Your message has been delivered. The ballot ID is ${ballot.id}.`,
        });
    }

    async ballotView(client: DiscordClient, msg: Message | CommandInteraction, options: CommandOptions | InteractionOptions) {
        if (!options.isInteraction && typeof options.args[0] === 'undefined') {
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`This command requires at least one argument.`)
                ]
            });

            return;
        }
        
        try {
            const id = options.isInteraction ? options.options.getInteger('id') : options.args[0];
            const ballot = await client.db.getAsync("SELECT * FROM ballots WHERE id = ? ORDER BY id DESC LIMIT 0, 1", [id]);

            if (!ballot)
                throw new Error();

            const ballotAuthor = ballot.author ? (await msg.guild!.members.fetch(ballot.author)) : null;

            const channel = <TextChannel> await msg.guild?.channels.fetch(ballot.channel_id);
            
            if (!channel)
                throw new Error();

            const message = await channel.messages.fetch(ballot.msg_id);

            if (!message)
                throw new Error();

            const upEmote = await fetchEmoji('check');
            const downEmote = await fetchEmoji('error');
            const upVotes = message.reactions.cache.find(r => r.emoji.id === upEmote!.id)!.count - 1;
            const downVotes = message.reactions.cache.find(r => r.emoji.id === downEmote!.id)!.count - 1;

            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setAuthor({
                        name: ballot.author ? ballotAuthor!.user!.tag! : 'Staff',
                        iconURL: ballot.author ? ballotAuthor?.displayAvatarURL()! : msg.guild!.iconURL()!
                    })
                    .setDescription(ballot.content)
                    .addFields([
                        {
                            name: 'Upvotes',
                            value: `${upVotes}`
                        },
                        {
                            name: 'Downvotes',
                            value: `${downVotes}`
                        }
                    ])
                    .setTimestamp()
                ]
            });
        }
        catch (e) {
            console.log(e);
            
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`Invalid ID or failed to fetch data.`)
                ]
            });
        }
    }

    async run(client: DiscordClient, msg: Message | CommandInteraction, options: CommandOptions | InteractionOptions) {
        if (!options.isInteraction && typeof options.args[0] === 'undefined') {
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`This command requires at least one subcommand.`)
                ]
            });

            return;
        }

        const subcmd = options.isInteraction ? options.options.getSubcommand(true) : options.args[0];

        if (!options.isInteraction)
            await options.args.shift();

        if (subcmd === 'create') 
            await this.ballotCreate(client, msg, options);
        else if (subcmd === 'view')
            await this.ballotView(client, msg, options);
        else {
            await msg.reply({
                embeds: [
                    new MessageEmbed()
                    .setColor('#f14a60')
                    .setDescription(`Invalid subcommand provided.`)
                ]
            });
        }
    }
}
1	rakin	55	import { CommandInteraction, EmojiIdentifierResolvable, GuildMember, Message, TextChannel, User } from 'discord.js';
2			import BaseCommand from '../../utils/structures/BaseCommand';
3			import DiscordClient from '../../client/Client';
4			import MessageEmbed from '../../client/MessageEmbed';
5			import CommandOptions from '../../types/CommandOptions';
6			import InteractionOptions from '../../types/InteractionOptions';
7			import { fetchEmoji } from '../../utils/Emoji';
8
9			export default class BallotCommand extends BaseCommand {
10			supportsInteractions = true;
11
12			constructor() {
13			super('ballot', 'automation', []);
14			}
15
16			async ballotCreate(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
17			if (!options.isInteraction && typeof options.args[0] === 'undefined') {
18			await msg.reply({
19			embeds: [
20			new MessageEmbed()
21			.setColor('#f14a60')
22			.setDescription(`This command requires at least one argument.`)
23			]
24			});
25
26			return;
27			}
28
29			if (msg instanceof CommandInteraction)
30			await msg.deferReply({
31			ephemeral: true
32			});
33
34			let content: string;
35			let channel: TextChannel = msg.channel as TextChannel;
36			let anonymous = false;
37
38			if (options.isInteraction) {
39			content = <string> await options.options.getString('content');
40
41			if (options.options.getChannel('channel'))
42			channel = <TextChannel> await options.options.getChannel('channel');
43
44			if (options.options.getBoolean('anonymous'))
45			anonymous = <boolean> await options.options.getBoolean('anonymous');
46			}
47			else {
48			if ((msg as Message).mentions.channels.last() && /\<#(\d+)\>/g.test([...options.args].pop()!)) {
49			channel = <TextChannel> (msg as Message).mentions.channels.last();
50			options.args.pop();
51			}
52
53			content = await options.args.join(' ').trim();
54			}
55
56			if (!channel.send) {
57			await this.deferReply(msg, {
58			content: 'Invalid text channel.'
59			});
60
61			return;
62			}
63
64			const message = await channel.send({
65			embeds: [
66			new MessageEmbed()
67			.setAuthor({
68			name: anonymous ? 'Staff' : (msg.member?.user as User).tag!,
69			iconURL: anonymous ? msg.guild!.iconURL()! : (msg.member as GuildMember)?.displayAvatarURL()
70			})
71			.setDescription(content)
72			.setTimestamp()
73			]
74			});
75
76			await client.db.runAsync('INSERT INTO ballots(content, author, msg_id, guild_id, date, channel_id) VALUES(?, ?, ?, ?, ?, ?)', [content, anonymous ? null : msg.member?.user.id, message.id, msg.guild!.id, new Date().toISOString(), msg.channel!.id]);
77			const ballot = await client.db.getAsync("SELECT * FROM ballots WHERE msg_id = ? AND guild_id = ? ORDER BY id DESC LIMIT 0, 1", [message.id, msg.guild!.id]);
78
79			await message.react(<EmojiIdentifierResolvable> await fetchEmoji('check'));
80			await message.react(<EmojiIdentifierResolvable> await fetchEmoji('error'));
81
82			await this.deferReply(msg, {
83			content: `${(await fetchEmoji('check'))!.toString()} Your message has been delivered. The ballot ID is ${ballot.id}.`,
84			});
85			}
86
87			async ballotView(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
88			if (!options.isInteraction && typeof options.args[0] === 'undefined') {
89			await msg.reply({
90			embeds: [
91			new MessageEmbed()
92			.setColor('#f14a60')
93			.setDescription(`This command requires at least one argument.`)
94			]
95			});
96
97			return;
98			}
99
100			try {
101			const id = options.isInteraction ? options.options.getInteger('id') : options.args[0];
102			const ballot = await client.db.getAsync("SELECT * FROM ballots WHERE id = ? ORDER BY id DESC LIMIT 0, 1", [id]);
103
104			if (!ballot)
105			throw new Error();
106
107			const ballotAuthor = ballot.author ? (await msg.guild!.members.fetch(ballot.author)) : null;
108
109			const channel = <TextChannel> await msg.guild?.channels.fetch(ballot.channel_id);
110
111			if (!channel)
112			throw new Error();
113
114			const message = await channel.messages.fetch(ballot.msg_id);
115
116			if (!message)
117			throw new Error();
118
119			const upEmote = await fetchEmoji('check');
120			const downEmote = await fetchEmoji('error');
121			const upVotes = message.reactions.cache.find(r => r.emoji.id === upEmote!.id)!.count - 1;
122			const downVotes = message.reactions.cache.find(r => r.emoji.id === downEmote!.id)!.count - 1;
123
124			await msg.reply({
125			embeds: [
126			new MessageEmbed()
127			.setAuthor({
128			name: ballot.author ? ballotAuthor!.user!.tag! : 'Staff',
129			iconURL: ballot.author ? ballotAuthor?.displayAvatarURL()! : msg.guild!.iconURL()!
130			})
131			.setDescription(ballot.content)
132			.addFields([
133			{
134			name: 'Upvotes',
135			value: `${upVotes}`
136			},
137			{
138			name: 'Downvotes',
139			value: `${downVotes}`
140			}
141			])
142			.setTimestamp()
143			]
144			});
145			}
146			catch (e) {
147			console.log(e);
148
149			await msg.reply({
150			embeds: [
151			new MessageEmbed()
152			.setColor('#f14a60')
153			.setDescription(`Invalid ID or failed to fetch data.`)
154			]
155			});
156			}
157			}
158
159			async run(client: DiscordClient, msg: Message \| CommandInteraction, options: CommandOptions \| InteractionOptions) {
160			if (!options.isInteraction && typeof options.args[0] === 'undefined') {
161			await msg.reply({
162			embeds: [
163			new MessageEmbed()
164			.setColor('#f14a60')
165			.setDescription(`This command requires at least one subcommand.`)
166			]
167			});
168
169			return;
170			}
171
172			const subcmd = options.isInteraction ? options.options.getSubcommand(true) : options.args[0];
173
174			if (!options.isInteraction)
175			await options.args.shift();
176
177			if (subcmd === 'create')
178			await this.ballotCreate(client, msg, options);
179			else if (subcmd === 'view')
180			await this.ballotView(client, msg, options);
181			else {
182			await msg.reply({
183			embeds: [
184			new MessageEmbed()
185			.setColor('#f14a60')
186			.setDescription(`Invalid subcommand provided.`)
187			]
188			});
189			}
190			}
191			}