xiao/commands/search/anime.js

const Command = require('../../framework/Command');
const { EmbedBuilder, PermissionFlagsBits } = require('discord.js');
const request = require('node-superfetch');
const cheerio = require('cheerio');
const { stripIndents } = require('common-tags');
const { cleanAnilistHTML, embedURL } = require('../../util/Util');
const logos = require('../../assets/json/logos');
const ANILIST_USERNAME = process.env.ANILIST_USERNAME || 'AniList';
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
					romaji
				}
			}
		}
	}
`;
const resultGraphQL = stripIndents`
	query media($id: Int, $type: MediaType) {
		Media(id: $id, type: $type) {
			id
			idMal
			title {
				english
				romaji
			}
			coverImage {
				large
				medium
			}
			startDate { year }
			description(asHtml: false)
			season
			type
			format
			siteUrl
			status
			episodes
			isAdult
			meanScore
			averageScore
			externalLinks {
				url
				site
			}
		}
	}
`;
const personalGraphQL = stripIndents`
	query ($name: String, $type: MediaType) {
		MediaListCollection(userName: $name, type: $type) {
			lists {
				entries {
					mediaId
					score(format: POINT_10)
					status
				}
				name
			}
		}
	}
`;
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', 'ani', 'myanimelist', 'mal', 'mal-score'],
			group: 'search',
			description: 'Searches AniList for your query, getting anime results.',
			clientPermissions: [PermissionFlagsBits.EmbedLinks],
			credit: [
				{
					name: 'AniList',
					url: 'https://anilist.co/',
					reason: 'API',
					reasonURL: 'https://anilist.gitbook.io/anilist-apiv2-docs/'
				},
				{
					name: 'MyAnimeList',
					url: 'https://myanimelist.net/',
					reason: 'Score Data'
				}
			],
			args: [
				{
					key: 'query',
					type: 'string'
				}
			]
		});

		this.personalList = null;
	}

	async run(msg, { query }) {
		const id = await this.search(query);
		if (!id) return msg.say('Could not find any results.');
		const anime = await this.fetchAnime(id);
		if (!this.personalList) await this.fetchPersonalList();
		const entry = this.personalList.find(ani => ani.mediaId === id);
		const malScore = await this.fetchMALScore(anime.idMal);
		const malURL = `https://myanimelist.net/anime/${anime.idMal}`;
		const embed = new EmbedBuilder()
			.setColor(0x02A9FF)
			.setAuthor({ name: 'AniList', iconURL: logos.anilist, url: 'https://anilist.co/' })
			.setURL(anime.siteUrl)
			.setThumbnail(anime.coverImage.large || anime.coverImage.medium || null)
			.setTitle(anime.title.english || anime.title.romaji)
			.setDescription(anime.description ? cleanAnilistHTML(anime.description) : 'No description.')
			.addField('❯ Status', statuses[anime.status], true)
			.addField('❯ Episodes', anime.episodes?.toString() || '???', true)
			.addField('❯ Season', anime.season ? `${seasons[anime.season]} ${anime.startDate.year}` : '???', true)
			.addField('❯ Average Score', anime.averageScore ? `${anime.averageScore}%` : '???', true)
			.addField(`❯ MAL Score`, malScore ? embedURL(malScore, malURL) : '???', true)
			.addField(`❯ ${ANILIST_USERNAME}'s Score`, entry && entry.score ? `${entry.score}/10` : '???', true)
			.addField('❯ External Links', anime.externalLinks.length
				? anime.externalLinks.map(link => embedURL(link.site, link.url)).join(', ')
				: 'None');
		return msg.embed(embed);
	}

	async search(query) {
		const { body } = await request
			.post('https://graphql.anilist.co/')
			.set({ referer: 'nick is dum dum' })
			.send({
				variables: {
					search: query,
					type: 'ANIME'
				},
				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/')
			.set({ referer: 'nick is dum dum' })
			.send({
				variables: {
					id,
					type: 'ANIME'
				},
				query: resultGraphQL
			});
		return body.data.Media;
	}

	async fetchMALScore(id) {
		try {
			const { text } = await request.get(`https://myanimelist.net/anime/${id}`);
			const $ = cheerio.load(text);
			return $('span[itemprop="ratingValue"]').first().text();
		} catch {
			return null;
		}
	}

	async fetchPersonalList() {
		if (this.personalList) return this.personalList;
		const { body } = await request
			.post('https://graphql.anilist.co/')
			.set({ referer: 'nick is dum dum' })
			.send({
				variables: {
					name: ANILIST_USERNAME,
					type: 'ANIME'
				},
				query: personalGraphQL
			});
		const { lists } = body.data.MediaListCollection;
		this.personalList = [];
		for (const list of Object.values(lists)) this.personalList.push(...list.entries);
		setTimeout(() => { this.personalList = null; }, 3.6e+6);
		return this.personalList;
	}
};