Files
xiao/commands/search/anime.js
T
2019-01-31 20:02:34 +00:00

100 lines
2.7 KiB
JavaScript
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
const Command = require('../../structures/Command');
const { MessageEmbed } = require('discord.js');
const request = require('node-superfetch');
const { stripIndents } = require('common-tags');
const { shorten } = require('../../util/Util');
const searchGraphQL = stripIndents`
query ($search: String, $type: MediaType, $isAdult: Boolean) {
anime: Page (perPage: 1) {
results: media (type: $type, isAdult: $isAdult, search: $search) { id }
}
}
`;
const resultGraphQL = stripIndents`
query media($id: Int, $type: MediaType) {
Media(id: $id, type: $type) {
id
title { userPreferred }
coverImage { large }
startDate { year }
description
season
type
status
episodes
isAdult
meanScore
}
}
`;
module.exports = class AnimeCommand extends Command {
constructor(client) {
super(client, {
name: 'anime',
aliases: ['anilist-anime', 'anilist'],
group: 'search',
memberName: 'anime',
description: 'Searches AniList for your query, getting anime results.',
clientPermissions: ['EMBED_LINKS'],
args: [
{
key: 'query',
prompt: 'What anime would you like to search for?',
type: 'string'
}
]
});
}
async run(msg, { query }) {
try {
const id = await this.search(query, msg.channel.nsfw);
if (!id) return msg.say('Could not find any results.');
const anime = await this.fetchAnime(id);
const embed = new MessageEmbed()
.setColor(0x02A9FF)
.setAuthor('AniList', 'https://i.imgur.com/iUIRC7v.png', 'https://anilist.co/')
.setURL(`https://anilist.co/anime/${anime.id}`)
.setThumbnail(anime.coverImage.large || null)
.setTitle(anime.title.english || anime.title.userPreferred)
.setDescription(anime.description ? shorten(anime.description.replace(/(<br>)+/g, '\n')) : 'No description.')
.addField(' Status', anime.status, true)
.addField(' Episodes', anime.episodes, true)
.addField(' Season', `${anime.season} ${anime.startDate.year}`, true)
.addField(' Average Score', `${anime.meanScore}/100`, true);
return msg.embed(embed);
} catch (err) {
return msg.reply(`Oh no, an error occurred: \`${err.message}\`. Try again later!`);
}
}
async search(query, nsfw) {
const { body } = await request
.post('https://graphql.anilist.co/')
.send({
variables: {
search: query,
type: 'ANIME',
isAdult: Boolean(nsfw)
},
query: searchGraphQL
});
if (!body.data.anime.results.length) return null;
return body.data.anime.results[0].id;
}
async fetchAnime(id) {
const { body } = await request
.post('https://graphql.anilist.co/')
.send({
variables: {
id,
type: 'ANIME'
},
query: resultGraphQL
});
return body.data.Media;
}
};