arthur/xiao
1
0
Fork 0
mirror of https://github.com/arthur-pbty/xiao.git synced 2026-06-03 23:36:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

PokemonStore for a shared pokemon cache

Browse Source
This commit is contained in:
Daniel Odendahl Jr
2018-08-29 20:49:23 +00:00
parent dbcb80a7b5
commit 632cea6ec5
6 changed files with 80 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Xiao.js
+2 -2
View File
@@ -1,8 +1,8 @@
require('dotenv').config();
const { XIAO_TOKEN, OWNERS, XIAO_PREFIX, INVITE } = process.env;
const path = require('path');
const { CommandoClient } = require('discord.js-commando');
const client = new CommandoClient({
const Client = require('./structures/Client');
const client = new Client({
commandPrefix: XIAO_PREFIX,
owner: OWNERS.split(','),
invite: INVITE,
commands/games/whos-that-pokemon.js
+10 -23
View File
@@ -25,49 +25,36 @@ module.exports = class WhosThatPokemonCommand extends Command {
}
]
});
this.cache = new Map();
}
async run(msg, { hide }) {
const pokemon = Math.floor(Math.random() * 802) + 1;
try {
const data = await this.fetchPokemon(pokemon);
const data = await this.client.pokemon.fetch(pokemon.toString());
const names = data.names.map(name => name.name.toLowerCase());
const displayName = data.names.filter(name => name.language.name === 'en')[0].name;
const id = data.id.toString().padStart(3, '0');
const attachment = await this.fetchImage(id, hide);
await msg.reply('**You have 15 seconds, who\'s that Pokémon?**', { files: [{ attachment, name: `${id}.png` }] });
const attachment = await this.fetchImage(data, hide);
await msg.reply('**You have 15 seconds, who\'s that Pokémon?**', { files: [attachment] });
const msgs = await msg.channel.awaitMessages(res => res.author.id === msg.author.id, {
max: 1,
time: 15000
});
if (!msgs.size) return msg.reply(`Sorry, time is up! It was ${displayName}.`);
if (!names.includes(msgs.first().content.toLowerCase())) return msg.reply(`Nope, sorry, it's ${displayName}.`);
if (!msgs.size) return msg.reply(`Sorry, time is up! It was ${data.name}.`);
if (!names.includes(msgs.first().content.toLowerCase())) return msg.reply(`Nope, sorry, it's ${data.name}.`);
return msg.reply('Nice job! 10/10! You deserve some cake!');
} catch (err) {
return msg.reply(`Oh no, an error occurred: \`${err.message}\`. Try again later!`);
}
}
async fetchPokemon(pokemon) {
if (this.cache.has(pokemon)) return this.cache.get(pokemon);
const { body } = await request.get(`https://pokeapi.co/api/v2/pokemon-species/${pokemon}/`);
this.cache.set(body.id, {
id: body.id,
names: body.names
});
return body;
}
async fetchImage(id, hide = false) {
const image = await request.get(`https://www.serebii.net/sunmoon/pokemon/${id}.png`);
if (!hide) return image.body;
async fetchImage(pokemon, hide = false) {
const name = `${pokemon.id}.png`;
const image = await request.get(pokemon.spriteImageURL);
if (!hide) return { attachment: image.body, name };
const base = await loadImage(image.body);
const canvas = createCanvas(base.width, base.height);
const ctx = canvas.getContext('2d');
ctx.drawImage(base, 0, 0);
silhouette(ctx, 0, 0, base.width, base.height);
return canvas.toBuffer();
return { attachment: canvas.toBuffer(), name };
}
};
commands/search/pokedex.js
+6 -44
View File
@@ -1,6 +1,5 @@
const Command = require('../../structures/Command');
const { Collection, MessageEmbed } = require('discord.js');
const request = require('node-superfetch');
const { MessageEmbed } = require('discord.js');
const { stripIndents } = require('common-tags');
module.exports = class PokedexCommand extends Command {
@@ -16,64 +15,27 @@ module.exports = class PokedexCommand extends Command {
{
key: 'pokemon',
prompt: 'What Pokémon would you like to get information on?',
type: 'string',
parse: pokemon => this.makeSlug(pokemon)
type: 'string'
}
]
});
this.cache = new Collection();
}
async run(msg, { pokemon }) {
try {
const data = await this.fetchPokemon(pokemon);
const data = await this.client.pokemon.fetch(pokemon);
const embed = new MessageEmbed()
.setColor(0xED1C24)
.setAuthor(
`#${data.displayID} - ${data.name}`,
`https://www.serebii.net/pokedex-sm/icon/${data.displayID}.png`,
`https://www.serebii.net/pokedex-sm/${data.displayID}.shtml`
)
.setAuthor(`#${data.displayID} - ${data.name}`, data.boxImageURL, data.serebiiURL)
.setDescription(stripIndents`
**The ${data.genus}**
**${data.genus}**
${data.entries[Math.floor(Math.random() * data.entries.length)]}
`)
.setThumbnail(`https://www.serebii.net/sunmoon/pokemon/${data.displayID}.png`);
.setThumbnail(data.spriteImageURL);
return msg.embed(embed);
} catch (err) {
if (err.status === 404) return msg.say('Could not find any results.');
return msg.reply(`Oh no, an error occurred: \`${err.message}\`. Try again later!`);
}
}
async fetchPokemon(query) {
const num = Number.parseInt(query, 10);
if (this.cache.has(num)) return this.cache.get(num);
const found = this.cache.find(pokemon => pokemon.slug === query);
if (found) return found;
const { body } = await request.get(`https://pokeapi.co/api/v2/pokemon-species/${query}/`);
const entries = body.flavor_text_entries
.filter(entry => entry.language.name === 'en')
.map(entry => entry.flavor_text.replace(/\n|\f|\r/g, ' '));
const { name } = this.filterPokemonData(body.names);
this.cache.set(body.id, {
id: body.id,
displayID: body.id.toString().padStart(3, '0'),
name,
slug: this.makeSlug(name),
genus: this.filterPokemonData(body.genera).genus,
entries
});
return this.cache.get(body.id);
}
filterPokemonData(arr) {
const filtered = arr.filter(entry => entry.language.name === 'en');
return filtered[0];
}
makeSlug(name) {
return encodeURIComponent(name.toLowerCase().replace(/ /g, '-').replace(/[^a-zA-Z0-9-]/g, ''));
}
};
structures/Client.js
+10
View File
@@ -0,0 +1,10 @@
const { CommandoClient } = require('discord.js-commando');
const PokemonStore = require('./PokemonStore');
module.exports = class XiaoClient extends CommandoClient {
constructor(options) {
super(options);
this.pokemon = new PokemonStore();
}
};
structures/Pokemon.js
+31
View File
@@ -0,0 +1,31 @@
module.exports = class Pokemon {
constructor(data) {
this.id = data.id;
this.name = data.names.filter(entry => entry.language.name === 'en')[0].name;
this.entries = data.flavor_text_entries
.filter(entry => entry.language.name === 'en')
.map(entry => entry.flavor_text.replace(/\n|\f|\r/g, ' '));
this.names = data.names;
this.genus = `The ${data.genera.filter(entry => entry.language.name === 'en')[0].genus}`;
}
get displayID() {
return this.id.toString().padStart(3, '0');
}
get slug() {
return encodeURIComponent(this.name.toLowerCase().replace(/ /g, '-').replace(/[^a-zA-Z0-9-]/g, ''));
}
get spriteImageURL() {
return `https://www.serebii.net/sunmoon/pokemon/${this.displayID}.png`;
}
get boxImageURL() {
return `https://www.serebii.net/pokedex-sm/icon/${this.displayID}.png`;
}
get serebiiURL() {
return `https://www.serebii.net/pokedex-sm/${this.displayID}.shtml`;
}
};
structures/PokemonStore.js
+21
View File
@@ -0,0 +1,21 @@
const { Collection } = require('discord.js');
const request = require('node-superfetch');
const Pokemon = require('./Pokemon');
module.exports = class PokemonStore extends Collection {
constructor(iterable) {
super(iterable);
}
async fetch(query) {
query = encodeURIComponent(query.toLowerCase().replace(/ /g, '-').replace(/[^a-zA-Z0-9-]/g, ''));
const num = Number.parseInt(query, 10);
if (this.has(num)) return this.get(num);
const found = this.find(pokemon => pokemon.slug === query);
if (found) return found;
const { body } = await request.get(`https://pokeapi.co/api/v2/pokemon-species/${query}/`);
const pokemon = new Pokemon(body);
this.set(pokemon.id, pokemon);
return pokemon;
}
};