const { Command } = require('discord.js-commando');
const snekfetch = require('snekfetch');

module.exports = class DanbooruCommand extends Command {
	constructor(client) {
		super(client, {
			name: 'danbooru',
			aliases: ['danbooru-image'],
			group: 'search',
			memberName: 'danbooru',
			description: 'Searches Danbooru for your query.',
			nsfw: true,
			args: [
				{
					key: 'query',
					prompt: 'What image would you like to search for?',
					type: 'string',
					validate: query => {
						if (!query.includes(' ')) return true;
						return 'Invalid query, please only search for one tag at a time.';
					}
				}
			]
		});
	}

	async run(msg, { query }) {
		try {
			const { body } = await snekfetch
				.get('https://danbooru.donmai.us/posts.json')
				.query({
					tags: `${query} order:random`,
					limit: 1
				});
			if (!body.length || !body[0].file_url) return msg.say('Could not find any results.');
			return msg.say(`https://danbooru.donmai.us${body[0].file_url}`);
		} catch (err) {
			return msg.reply(`Oh no, an error occurred: \`${err.message}\`. Try again later!`);
		}
	}
};