Files
xiao/commands/search/google.js
T
2020-03-14 11:57:15 -04:00

65 lines
1.5 KiB
JavaScript

const Command = require('../../structures/Command');
const request = require('node-superfetch');
const { GOOGLE_KEY, CUSTOM_SEARCH_ID } = process.env;
module.exports = class GoogleCommand extends Command {
constructor(client) {
super(client, {
name: 'google',
aliases: ['search'],
group: 'search',
memberName: 'google',
description: 'Searches Google for your query.',
credit: [
{
name: 'Google',
url: 'https://www.google.com/',
reason: 'Custom Search API',
reasonURL: 'https://cse.google.com/cse/all'
},
{
name: 'LMGTFY',
url: 'https://lmgtfy.com/',
reason: 'API'
}
],
args: [
{
key: 'query',
prompt: 'What would you like to search for?',
type: 'string',
validate: query => {
if (encodeURIComponent(query).length < 1950) return true;
return 'Invalid query, your query is too long.';
}
}
]
});
}
async run(msg, { query }) {
let href;
const nsfw = msg.channel.nsfw || false;
try {
href = await this.customSearch(query, nsfw);
} catch {
href = `http://lmgtfy.com/?iie=1&q=${encodeURIComponent(query)}`;
}
if (!href) return msg.say('Could not find any results.');
return msg.say(href);
}
async customSearch(query, nsfw) {
const { body } = await request
.get('https://www.googleapis.com/customsearch/v1')
.query({
key: GOOGLE_KEY,
cx: CUSTOM_SEARCH_ID,
safe: nsfw ? 'off' : 'active',
q: query
});
if (!body.items) return null;
return body.items[0].formattedUrl;
}
};