Files
xiao/commands/search/anime.js
T
Daniel Odendahl Jr d535db7a08 Fix
2019-02-16 02:50:37 +00:00

119 lines
3.0 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 { cleanAnilistHTML } = 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 {
english
userPreferred
}
coverImage {
large
medium
}
startDate { year }
description(asHtml: false)
season
type
siteUrl
status
episodes
isAdult
meanScore
}
}
`;
const seasons = {
WINTER: 'Winter',
SPRING: 'Spring',
SUMMER: 'Summer',
FALL: 'Fall'
};
const statuses = {
FINISHED: 'Finished',
RELEASING: 'Releasing',
NOT_YET_RELEASED: 'Unreleased',
CANCELLED: 'Cancelled'
};
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(anime.siteUrl)
.setThumbnail(anime.coverImage.large || anime.coverImage.medium || null)
.setTitle(anime.title.english || anime.title.userPreferred)
.setDescription(anime.description ? cleanAnilistHTML(anime.description) : 'No description.')
.addField(' Status', statuses[anime.status], true)
.addField(' Episodes', anime.episodes || '???', true)
.addField(' Season', anime.season ? `${seasons[anime.season]} ${anime.startDate.year}` : '???', true)
.addField(' Average Score', anime.meanScore ? `${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;
}
};