xiao/commands/search/flickr.js

const Command = require('../../structures/Command');
const request = require('node-superfetch');
const { isImageNSFW } = require('../../util/Util');
const { FLICKR_KEY } = process.env;

module.exports = class FlickrCommand extends Command {
	constructor(client) {
		super(client, {
			name: 'flickr',
			group: 'search',
			memberName: 'flickr',
			description: 'Searches Flickr for your query.',
			throttling: {
				usages: 2,
				duration: 30
			},
			credit: [
				{
					name: 'Flickr',
					url: 'https://www.flickr.com/',
					reason: 'API',
					reasonURL: 'https://www.flickr.com/services/api/'
				}
			],
			args: [
				{
					key: 'query',
					prompt: 'What photo would you like to search for?',
					type: 'string'
				}
			]
		});
	}

	async run(msg, { query }) {
		try {
			const { body } = await request
				.get('https://api.flickr.com/services/rest/')
				.query({
					api_key: FLICKR_KEY,
					format: 'json',
					method: 'flickr.photos.search',
					text: query,
					nojsoncallback: true
				});
			if (!body.photos.photo.length) return msg.say('Could not find any results.');
			const data = body.photos.photo[Math.floor(Math.random() * body.photos.photo.length)];
			const url = `https://farm${data.farm}.staticflickr.com/${data.server}/${data.id}_${data.secret}.jpg`;
			if (!msg.channel.nsfw) {
				const { body: imageBody } = await request.get(url);
				const aiDetect = await isImageNSFW(this.client.nsfwModel, imageBody);
				if (aiDetect) return msg.reply('Found an NSFW image. Sorry, please try again.');
			}
			return msg.say(url);
		} catch (err) {
			return msg.say(`Oh no, an error occurred: \`${err.message}\`. Try again later!`);
		}
	}
};