Files
xiao/commands/search/anime.js
T
Daniel Odendahl Jr d43bdb18c0 Exact anime matches
2019-02-21 22:42:32 +00:00

129 lines
3.3 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: 10) {
results: media (type: $type, isAdult: $isAdult, search: $search) {
id
title {
english
userPreferred
}
}
}
}
`;
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;
const found = body.data.anime.results.find(
anime => anime.title.english.toLowerCase() === query || anime.title.userPreferred.toLowerCase() === query
);
if (found) return found.id;
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;
}
};