mirror of
https://github.com/arthur-pbty/xiao.git
synced 2026-06-04 07:46:43 +02:00
75 lines
2.1 KiB
JavaScript
75 lines
2.1 KiB
JavaScript
const Command = require('../../framework/Command');
|
|
const request = require('node-superfetch');
|
|
const { list } = require('../../util/Util');
|
|
const { DEVIANTART_ID, DEVIANTART_SECRET } = process.env;
|
|
const sections = ['dailydeviations', 'hot', 'newest', 'popular', 'undiscovered'];
|
|
|
|
module.exports = class DeviantartCommand extends Command {
|
|
constructor(client) {
|
|
super(client, {
|
|
name: 'deviantart',
|
|
group: 'search',
|
|
memberName: 'deviantart',
|
|
description: 'Responds with an image from a DeviantArt section, with optional query.',
|
|
details: `**Sections:** ${sections.join(', ')}`,
|
|
credit: [
|
|
{
|
|
name: 'DeviantArt',
|
|
url: 'https://www.deviantart.com/',
|
|
reason: 'API',
|
|
reasonURL: 'https://www.deviantart.com/developers/'
|
|
}
|
|
],
|
|
args: [
|
|
{
|
|
key: 'section',
|
|
prompt: `What section would you like to search? Either ${list(sections, 'or')}.`,
|
|
type: 'string',
|
|
oneOf: sections,
|
|
parse: section => section.toLowerCase()
|
|
},
|
|
{
|
|
key: 'query',
|
|
prompt: 'What image would you like to search for?',
|
|
type: 'string',
|
|
default: ''
|
|
}
|
|
]
|
|
});
|
|
|
|
this.token = null;
|
|
}
|
|
|
|
async run(msg, { section, query }) {
|
|
try {
|
|
if (!this.token) await this.fetchToken();
|
|
const { body } = await request
|
|
.get(`https://www.deviantart.com/api/v1/oauth2/browse/${section}`)
|
|
.query({
|
|
q: query,
|
|
limit: 120,
|
|
access_token: this.token,
|
|
mature_content: msg.channel.nsfw || false
|
|
});
|
|
const images = body.results.filter(image => image.content && image.content.src);
|
|
if (!images.length) return msg.say('Could not find any results.');
|
|
return msg.say(images[Math.floor(Math.random() * images.length)].content.src);
|
|
} catch (err) {
|
|
return msg.reply(`Oh no, an error occurred: \`${err.message}\`. Try again later!`);
|
|
}
|
|
}
|
|
|
|
async fetchToken() {
|
|
const { body } = await request
|
|
.get('https://www.deviantart.com/oauth2/token')
|
|
.query({
|
|
grant_type: 'client_credentials',
|
|
client_id: DEVIANTART_ID,
|
|
client_secret: DEVIANTART_SECRET
|
|
});
|
|
this.token = body.access_token;
|
|
setTimeout(() => { this.token = null; }, body.expires_in * 1000);
|
|
return body;
|
|
}
|
|
};
|