pissbot-9000/commands/interaction/slash/search/google.js

96 lines
No EOL
3.3 KiB
JavaScript

const { Constants } = require('detritus-client');
const { InteractionCallbackTypes, ApplicationCommandOptionTypes } = Constants;
const { createEmbed, page, formatPaginationEmbeds } = require('../../../../labscore/utils/embed');
const { paginator } = require('../../../../labscore/client');
const { STATICS } = require('../../../../labscore/utils/statics');
const { google } = require('../../../../labscore/api');
const { link, citation } = require('../../../../labscore/utils/markdown');
const { editOrReply } = require('../../../../labscore/utils/message');
function createSearchResultPage(context, result){
let res;
switch(result.type){
case 1: // Search Result Entry
res = page(createEmbed("default", context, {
author: {
iconUrl: `https://www.google.com/s2/favicons?domain=${encodeURIComponent(result.url)}&sz=256`,
name: result.title,
url: result.url
},
description: result.content,
footer: {
iconUrl: STATICS.google,
text: `Google • ${context.application.name}`
}
}))
if(result.thumbnail) res.embeds[0].thumbnail = { url: result.thumbnail };
break;
case 2: // Knowledge Graph Entry
let header = result.card.title;
if(result.card.url) header = link(result.card.url, result.card.title)
res = page(createEmbed("default", context, {
description: `**${header}**\n`,
footer: {
iconUrl: STATICS.google,
text: `Google Knowledge Graph • ${context.application.name}`
}
}))
if(result.card.image) res.embeds[0].thumbnail = { url: result.card.image };
if(result.card.description) res.embeds[0].description += `*${result.card.description}*\n`
if(result.card.content){
let cnt = result.card.content.replace(/\n/g, '')
if(cnt.endsWith(" ")) cnt = cnt.substr(0,cnt.length - 1)
res.embeds[0].description += "\n" + cnt + citation(1, result.card.url, "Source")
}
break;
default:
res = page(createEmbed("error", context, "Unknown GoogleResult Type: " + result.type))
break;
}
return res;
}
module.exports = {
description: 'Search on Google.',
name: 'google',
options: [
{
name: 'query',
description: 'Search query',
type: ApplicationCommandOptionTypes.STRING,
required: true
}
],
run: async (context, args) => {
await context.respond({data: {}, type: InteractionCallbackTypes.DEFERRED_CHANNEL_MESSAGE_WITH_SOURCE})
if(!args.query) return editOrReply(context, createEmbed("warning", context, `Missing Parameter (query).`))
try{
let search = await google(context, args.query, context.channel.nsfw)
search = search.response
if(search.body.status == 2) return editOrReply(context, createEmbed("error", context, search.body.message))
let pages = []
for(const res of search.body.results){
pages.push(createSearchResultPage(context, res))
}
if(!pages.length) return editOrReply(context, createEmbed("warning", context, `No results found.`))
await paginator.createPaginator({
context,
pages: formatPaginationEmbeds(pages)
});
}catch(e){
console.log(e)
return editOrReply(context, createEmbed("error", context, `Unable to perform google search.`))
}
},
};