awesomescibo/commands/train.js

const { SlashCommandBuilder } = require('@discordjs/builders');
const { MessageEmbed } = require('discord.js');

const decode = require('html-entities').decode;
const axios = require('axios');

const userScore = require('../models/userScore');

const { log } = require('../helpers/log.js');
const { updateScore } = require('../helpers/db.js');

module.exports = {
	data: new SlashCommandBuilder()
		.setName('train')
		.setDescription('Sends a training question to be answered')
		.addStringOption(option => {
			option
				.setName('subject')
				.setDescription('Optional subject to be used as a filter')
				.setRequired(false)
				.addChoice('astro', 'astro')
				.addChoice('bio', 'bio')
				.addChoice('ess', 'ess')
				.addChoice('chem', 'chem')
				.addChoice('phys', 'phys')
				.addChoice('math', 'math')
				.addChoice('energy', 'energy')
				.setRequired(false);
			return option;
		}),
	async execute(interaction) {
		await interaction.deferReply();

		const subject = interaction.options.get('subject') ? interaction.options.get('subject').value : null;
		const authorId = interaction.user.id;
		let score;
		userScore
			.findOne({ authorID: authorId })
			.lean()
			.then((obj, err) => {
				if (!obj) {
					score = 0;
				}
				else if (obj) {
					score = obj.score;
				}
				else {
					log({ logger: 'train', content: `Getting user score failed: ${err}`, level: 'error' });
				}
			});

		let categoryArray = [];

		switch (subject) {
		case null:
			categoryArray = ['BIOLOGY', 'PHYSICS', 'CHEMISTRY', 'EARTH AND SPACE', 'ASTRONOMY', 'MATH'];
			break;
		case 'astro':
		case 'astronomy':
			categoryArray = ['ASTRONOMY'];
			break;
		case 'bio':
		case 'biology':
			categoryArray = ['BIOLOGY'];
			break;
		case 'ess':
		case 'earth science':
		case 'es':
			categoryArray = ['EARTH SCIENCE'];
			break;
		case 'chem':
		case 'chemistry':
			categoryArray = ['CHEMISTRY'];
			break;
		case 'phys':
		case 'physics':
			categoryArray = ['PHYSICS'];
			break;
		case 'math':
			categoryArray = ['MATH'];
			break;
		case 'energy':
			categoryArray = ['ENERGY'];
			break;
		default:
			interaction.followUp(
				new MessageEmbed()
					.setDescription('<:red_x:816791117671825409> Not a valid subject!')
					.setColor('#ffffff'),
			);
			return;
		}

		axios
			.post('https://scibowldb.com/api/questions/random', { categories: categoryArray })
			.then((res) => {
				const data = res.data.question;
				const tossupQuestion = data.tossup_question;
				const tossupAnswer = data.tossup_answer;
				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: ${data.uri}||` })
					.then(() => {
						const messageFilter = m => m.author.id === interaction.user.id || m.author.id === interaction.client.user.id;
						interaction.channel.awaitMessages({
							filter: messageFilter,
							max: 1,
						})
							.then(collected => {
								const answerMsg = collected.first();

								if (answerMsg.author.id === interaction.client.user.id) return;

								let predicted = null;
								if (data.tossup_format === 'Multiple Choice') {
									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 {
									predicted = 'incorrect';
								}

								if (predicted === 'correct') {
									updateScore(true, score, authorId).then((msgToReply) =>
										answerMsg.reply(msgToReply),
									);
								}
								else {
									const overrideEmbed = new MessageEmbed()
										.setAuthor({ name: answerMsg.author.tag, iconURL: answerMsg.author.displayAvatarURL() })
										.addField('Correct answer', `\`${tossupAnswer}\``)
										.setDescription('It seems your answer was incorrect. Please react with <:override:955265585086857236> to override your answer if you think you got it right.')
										.setColor('#ffffff')
										.setTimestamp();
									answerMsg.channel.send({
										embeds: [overrideEmbed],
									})
										.then(overrideMsg => {
											overrideMsg.react('<:override:955265585086857236>');
											const filter = (reaction, user) => {
												return (
													['override'].includes(reaction.emoji.name) &&
									user.id === answerMsg.author.id
												);
											};
											overrideMsg
												.awaitReactions({
													filter: filter,
													max: 1,
												})
												.then(() => {
													updateScore(true, score, authorId).then((msgToReply) =>
														answerMsg.reply(msgToReply),
													);
												}).catch(err => log({ logger: 'train', content: `Failed to override score: ${err}`, level: 'error' }));
										}).catch(err => log({ logger: 'train', content: `Failed to send override message: ${err}`, level: 'error' }));
								}
							}).catch(err => log({ logger: 'train', content: `${err}`, level: 'error' }));
					}).catch(err => log({ logger: 'train', content: `${err}`, level: 'error' }));
			}).catch(err => log({ logger: 'train', content: `${err}`, level: 'error' }));
	},
};