pissbot-9000/commands/message/search/reverse-image.js

74 lines
No EOL
2.4 KiB
JavaScript

const { createEmbed, formatPaginationEmbeds } = require('../../../labscore/utils/embed')
const { editOrReply } = require('../../../labscore/utils/message')
const { STATICS } = require('../../../labscore/utils/statics')
const { getRecentImage } = require("../../../labscore/utils/attachment");
const { paginator } = require('../../../labscore/client');
const { reverseImageSearch } = require('../../../labscore/api');
const { Permissions } = require("detritus-client/lib/constants");
function createReverseImageSearchResultPage(context, result, source){
let res = {
"embeds": [
createEmbed("default", context, {
author: {
iconUrl: `https://www.google.com/s2/favicons?domain=${encodeURIComponent(result.url)}&sz=256`,
name: result.name,
url: result.url
},
image: {
url: result.image
},
thumbnail: {
url: source
},
footer: {
iconUrl: STATICS.google,
text: `Google Cloud Vision • ${context.application.name}`
}
})
]
}
if(result.thumbnail) res.embeds[0].thumbnail = { url: result.thumbnail };
return res;
}
module.exports = {
name: 'reverse-image',
aliases: ['reverse','reverseimage'],
metadata: {
description: 'Performs a reverse-image-search.',
description_short: 'Reverse image search',
examples: ['reverseimage'],
category: 'search',
usage: 'reverse <image>'
},
permissionsClient: [Permissions.EMBED_LINKS, Permissions.SEND_MESSAGES, Permissions.USE_EXTERNAL_EMOJIS],
run: async (context) => {
context.triggerTyping();
try{
let image = await getRecentImage(context, 50)
if(!image) return editOrReply(context, { embeds: [createEmbed("warning", context, "No images found.")] })
let search = await reverseImageSearch(context, image)
search = search.response
if(search.body.status == 2) return editOrReply(context, {embeds:[createEmbed("error", context, search.body.message)]})
let pages = []
for(const res of search.body.results){
pages.push(createReverseImageSearchResultPage(context, res, image))
}
pages = formatPaginationEmbeds(pages)
const paging = await paginator.createPaginator({
context,
pages
});
}catch(e){
console.log(e)
return editOrReply(context, {embeds:[createEmbed("error", context, `Unable to perform reverse image search.`)]})
}
},
};