mirror of
https://github.com/arthur-pbty/xiao.git
synced 2026-06-03 23:36:43 +02:00
65 lines
1.5 KiB
JavaScript
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;
|
|
}
|
|
};
|