Major multiple choice question improvements

src/commands/train.ts

@@ -1,5 +1,5 @@
 import { SlashCommandBuilder } from '@discordjs/builders';
-import { MessageEmbed } from 'discord.js';
+import { MessageEmbed, MessageActionRow, MessageButton } from 'discord.js';
 
 import { decode } from 'html-entities';
 import axios from 'axios';
@@ -97,13 +97,17 @@ export async function execute(interaction) {
 			const questionData = res.data.question;
 			const tossupQuestion = questionData.tossup_question;
 			const tossupAnswer = questionData.tossup_answer;
+			const tossupFormat = questionData.tossup_format;
 			let answers = tossupAnswer.split(' (ACCEPT: ');
 			if (answers.length > 1) {
 				answers[1] = answers[1].slice(0, answers[1].length - 1); // If there are multiple elements, it means there was an 'accept' and therefore a trailing ')' which should be removed
 				answers = [answers[0], ...answers[1].split(new RegExp(' OR ', 'i'))]; // Use the first element plus the last element split by 'OR' case insensitive
 			}
-			interaction.followUp({ content: decode(tossupQuestion) + `\n\n||Source: ${questionData.uri}||` })
+			interaction.followUp({ content: decode(tossupQuestion) + `\n\n||Source: ${questionData.uri}||`, fetchMessage: true })
-				.then(() => {
+				.then(questionMessage => {
+					switch (tossupFormat) {
+					case 'Short Answer': {
+						// eslint-disable-next-line no-case-declarations
 						const messageFilter = m => m.author.id === interaction.user.id || m.author.id === interaction.client.user.id;
 						interaction.channel.awaitMessages({
 							filter: messageFilter,
@@ -115,15 +119,7 @@ export async function execute(interaction) {
 								if (answerMsg.author.id === interaction.client.user.id) return;
 
 								let predicted = '';
-							if (questionData.tossup_format === 'Multiple Choice') {
+								if (answerMsg.content.toLowerCase() === tossupAnswer.toLowerCase() || answers.includes(answerMsg.content.toUpperCase())) {
-								if (answerMsg.content.charAt(0).toLowerCase() === tossupAnswer.charAt(0).toLowerCase()) {
-									predicted = 'correct';
-								}
-								else {
-									predicted = 'incorrect';
-								}
-							}
-							else if (answerMsg.content.toLowerCase() === tossupAnswer.toLowerCase() || answers.includes(answerMsg.content.toUpperCase())) {
 									predicted = 'correct';
 								}
 								else {
@@ -166,6 +162,47 @@ export async function execute(interaction) {
 										}).catch(err => log({ logger: 'train', content: `Failed to send override message: ${err}`, level: 'error' }));
 								}
 							}).catch(err => log({ logger: 'train', content: `${err}`, level: 'error' }));
+						break;
+					}
+					case 'Multiple Choice': {
+						const choices = new MessageActionRow()
+							.addComponents(
+								new MessageButton()
+									.setCustomId('w')
+									.setLabel('W')
+									.setStyle('SECONDARY'),
+								new MessageButton()
+									.setCustomId('x')
+									.setLabel('X')
+									.setStyle('SECONDARY'),
+								new MessageButton()
+									.setCustomId('y')
+									.setLabel('Y')
+									.setStyle('SECONDARY'),
+								new MessageButton()
+									.setCustomId('z')
+									.setLabel('Z')
+									.setStyle('SECONDARY'),
+							);
+						interaction.editReply({ components: [choices] });
+						const mcFilter = i => ['w', 'x', 'y', 'z'].includes(i.customId) && i.user.id === interaction.user.id;
+						questionMessage.awaitMessageComponent({ filter: mcFilter })
+							.then(mcChoice => {
+								if (tossupAnswer.charAt(0).toLowerCase() === mcChoice.customId) {
+									updateScore(true, score, authorId).then((msgToReply) =>
+										mcChoice.reply(msgToReply),
+									);
+								}
+								else {
+									mcChoice.reply({ embeds: [
+										new MessageEmbed()
+											.setDescription(`Unfortunately, ${mcChoice.customId.toUpperCase()} was not the correct answer. The correct answer was actually \`${tossupAnswer}\`.`),
+									] });
+								}
+							});
+						break;
+					}
+					}
 				}).catch(err => log({ logger: 'train', content: `${err}`, level: 'error' }));
 		}).catch(err => log({ logger: 'train', content: `${err}`, level: 'error' }));
 }