mirror of
https://github.com/arthur-pbty/xiao.git
synced 2026-06-03 23:36:43 +02:00
176 lines
5.6 KiB
JavaScript
176 lines
5.6 KiB
JavaScript
const Command = require('../../framework/Command');
|
|
const { MessageEmbed, MessageActionRow, MessageButton } = require('discord.js');
|
|
const { Aki, regions } = require('aki-api');
|
|
const { list } = require('../../util/Util');
|
|
|
|
module.exports = class AkinatorCommand extends Command {
|
|
constructor(client) {
|
|
super(client, {
|
|
name: 'akinator',
|
|
aliases: ['aki'],
|
|
group: 'games-sp',
|
|
memberName: 'akinator',
|
|
description: 'Think about a real or fictional character, I will try to guess who it is.',
|
|
details: `**Regions:** ${regions.join(', ')}`,
|
|
clientPermissions: ['EMBED_LINKS'],
|
|
credit: [
|
|
{
|
|
name: 'Akinator',
|
|
url: 'https://en.akinator.com/',
|
|
reason: 'API'
|
|
}
|
|
],
|
|
args: [
|
|
{
|
|
key: 'region',
|
|
prompt: `What region do you want to use? Either ${list(regions, 'or')}.`,
|
|
type: 'string',
|
|
default: 'en',
|
|
oneOf: regions,
|
|
parse: region => region.toLowerCase()
|
|
}
|
|
]
|
|
});
|
|
}
|
|
|
|
async run(msg, { region }) {
|
|
const aki = new Aki({ region, childMode: !msg.channel.nsfw });
|
|
let ans = null;
|
|
let win = false;
|
|
let timesGuessed = 0;
|
|
let guessResetNum = 0;
|
|
let wentBack = false;
|
|
let forceGuess = false;
|
|
const initialRow = new MessageActionRow().addComponents(
|
|
new MessageButton().setCustomId('true').setLabel('Ready!').setStyle('PRIMARY'),
|
|
new MessageButton().setCustomId('false').setLabel('Nevermind').setStyle('SECONDARY')
|
|
);
|
|
const gameMsg = await msg.reply('Welcome to Akinator! Think of a character, and I will try to guess it.', {
|
|
components: [initialRow]
|
|
});
|
|
try {
|
|
const initialVerify = await gameMsg.awaitMessageComponent({
|
|
filter: res => res.user.id === msg.author.id,
|
|
max: 1,
|
|
time: 30000
|
|
});
|
|
if (initialVerify.customId === 'false') return initialVerify.update('Too bad...', { components: [] });
|
|
} catch {
|
|
return gameMsg.edit('Guess you didn\'t want to play after all...', { components: [] });
|
|
}
|
|
await this.sendLoadingMessage(buttonPress, [initialRow]);
|
|
const guessBlacklist = [];
|
|
while (timesGuessed < 3) {
|
|
if (guessResetNum > 0) guessResetNum--;
|
|
if (ans === null) {
|
|
await aki.start();
|
|
} else if (wentBack) {
|
|
wentBack = false;
|
|
} else {
|
|
try {
|
|
await aki.step(ans);
|
|
} catch {
|
|
await aki.step(ans);
|
|
}
|
|
}
|
|
if (!aki.answers || aki.currentStep >= 79) forceGuess = true;
|
|
const row = new MessageActionRow();
|
|
for (const answer of aki.answers) {
|
|
row.addComponents(new MessageButton().setCustomId(answer).setStyle('PRIMARY').setLabel(answer));
|
|
}
|
|
const sRow = new MessageActionRow();
|
|
if (aki.currentStep > 0) {
|
|
sRow.addComponents(new MessageButton().setCustomId('back').setStyle('SECONDARY').setLabel('Back'));
|
|
}
|
|
sRow.addComponents(new MessageButton().setCustomId('end').setStyle('DANGER').setLabel('End'));
|
|
await buttonPress.editReply(
|
|
`**${aki.currentStep + 1}.** ${aki.question} (${Math.round(Number.parseInt(aki.progress, 10))}%)`,
|
|
{ components: [row, sRow] }
|
|
);
|
|
let buttonPress;
|
|
try {
|
|
buttonPress = await gameMsg.awaitMessageComponent({
|
|
filter: res => res.user.id === msg.author.id,
|
|
max: 1,
|
|
time: 30000
|
|
});
|
|
} catch {
|
|
win = 'time';
|
|
break;
|
|
}
|
|
await this.sendLoadingMessage(buttonPress, [row, sRow]);
|
|
const choice = interactions.first().customId;
|
|
if (choice === 'end') {
|
|
forceGuess = true;
|
|
} else if (choice === 'back') {
|
|
if (guessResetNum > 0) guessResetNum++;
|
|
wentBack = true;
|
|
await aki.back();
|
|
continue;
|
|
} else {
|
|
ans = aki.answers.indexOf(choice);
|
|
}
|
|
if ((aki.progress >= 90 && !guessResetNum) || forceGuess) {
|
|
timesGuessed++;
|
|
guessResetNum += 10;
|
|
await aki.win();
|
|
const guess = aki.answers.filter(g => !guessBlacklist.includes(g.id))[0];
|
|
if (!guess) {
|
|
win = true;
|
|
break;
|
|
}
|
|
guessBlacklist.push(guess.id);
|
|
const embed = new MessageEmbed()
|
|
.setColor(0xF78B26)
|
|
.setTitle(`I'm ${Math.round(guess.proba * 100)}% sure it's...`)
|
|
.setDescription(`${guess.name}${guess.description ? `\n_${guess.description}_` : ''}`)
|
|
.setThumbnail(guess.absolute_picture_path || null)
|
|
.setFooter(forceGuess ? 'Final Guess' : `Guess ${timesGuessed}`);
|
|
const guessRow = new MessageActionRow().addComponents(
|
|
new MessageButton().setCustomId('true').setLabel('Yes').setStyle('SUCCESS'),
|
|
new MessageButton().setCustomId('false').setLabel('No').setStyle('DANGER')
|
|
);
|
|
await buttonPress.editReply('Is this your character?', { embeds: [embed], components: [guessRow] });
|
|
try {
|
|
buttonPress = await gameMsg.awaitMessageComponent({
|
|
filter: res => res.user.id === msg.author.id,
|
|
max: 1,
|
|
time: 30000
|
|
});
|
|
} catch {
|
|
win = 'time';
|
|
break;
|
|
}
|
|
await this.sendLoadingMessage(buttonPress, [guessRow]);
|
|
if (buttonPress.customId === 'true') {
|
|
win = false;
|
|
break;
|
|
} else if (timesGuessed >= 3 || forceGuess) {
|
|
win = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
const row = new MessageActionRow();
|
|
row.addComponents(
|
|
new MessageButton().setLabel('Akinator Website').setStyle('LINK').setURL('https://akinator.com/')
|
|
);
|
|
if (win === 'time') {
|
|
return buttonPress.editReply('I guess your silence means I have won.', { components: [row] });
|
|
}
|
|
if (win) {
|
|
return buttonPress.editReply('Bravo, you have defeated me.', { components: [row] });
|
|
}
|
|
return buttonPress.editReply('Guessed right one more time! I love playing with you!', { components: [row] });
|
|
}
|
|
|
|
sendLoadingMessage(buttonPress, rows) {
|
|
for (const row of rows) {
|
|
for (const button of row.components) {
|
|
button.setDisabled(true);
|
|
}
|
|
}
|
|
return buttonPress.update('Loading...', { components: rows });
|
|
}
|
|
};
|