xiao/commands/random-img/xiao.js

const Command = require('../../framework/Command');
const { PermissionFlagsBits } = require('discord.js');
const { embedURL } = require('../../util/Util');
const { SAUCENAO_KEY } = process.env;
const fs = require('fs');
const path = require('path');
const images = fs.readdirSync(path.join(__dirname, '..', '..', 'assets', 'images', 'xiao'));
const sourceRegex = /([A-Z ]+)-?([0-9A-Z]+)?(-[0-9]+)?(\.[A-Z])?/i;

module.exports = class XiaoCommand extends Command {
	constructor(client) {
		super(client, {
			name: 'xiao',
			aliases: ['xiao-pai', 'iao'],
			group: 'random-img',
			description: 'Responds with a random image of Xiao Pai.',
			clientPermissions: [PermissionFlagsBits.AttachFiles],
			credit: [
				{
					name: 'Marvelous',
					url: 'http://www.marv.jp/',
					reasonURL: 'http://www.runefactory4.com/index1.html',
					reason: 'Images, Original "Rune Factory 4" Game'
				},
				{
					name: 'SauceNAO',
					url: 'https://saucenao.com/',
					reason: 'API'
				}
			]
		});

		this.cache = new Map();
		this.sagiri = null;
		this.sagiriClient = null;
	}

	async run(msg) {
		const image = images[Math.floor(Math.random() * images.length)];
		return msg.say(await this.getSource(image), {
			files: [path.join(__dirname, '..', '..', 'assets', 'images', 'xiao', image)]
		});
	}

	async getSource(img) {
		if (this.cache.has(img)) return this.cache.get(img);
		const source = img.match(sourceRegex);
		const site = source[1];
		if (site === 'official') return 'Official Art';
		let result;
		if (site === 'unknown') result = 'Artist Unknown';
		else if (site === 'deviant') result = `Art by <https://www.deviantart.com/${source[2]}>`;
		else if (site === 'pixiv') result = `Art Source: <https://www.pixiv.net/en/artworks/${source[2]}>`;
		else result = `Art by ${site}`;
		if (!SAUCENAO_KEY) return result;
		try {
			const data = await this.sauceNao(img);
			if (data && data[0].similarity > 90) {
				result = '';
				const sauce = data[0];
				if (sauce.authorName && sauce.authorUrl) {
					result += `Art by ${embedURL(sauce.authorName, `<${sauce.authorUrl}>`)} | `;
				}
				if (sauce.authorName && !sauce.authorUrl) result += `Art by ${sauce.authorName} | `;
				result += `Art Source: <${sauce.url}>`;
			}
		} catch {
			this.cache.set(img, result);
			return result;
		}
		this.cache.set(img, result);
		return result;
	}

	async sauceNao(img) {
		if (!this.sagiri) this.sagiri = await import('sagiri');
		if (!this.sagiriClient) this.sagiriClient = this.sagiri(SAUCENAO_KEY);
		return this.sagiriClient(path.join(__dirname, '..', '..', 'assets', 'images', 'xiao', img));
	}
};