diff --git a/commands/message/search/reverse-image.js b/commands/message/search/reverse-image.js new file mode 100644 index 0000000..627fee5 --- /dev/null +++ b/commands/message/search/reverse-image.js @@ -0,0 +1,67 @@ +const { createEmbed, formatPaginationEmbeds } = require('../../../labscore/utils/embed') +const { link } = require('../../../labscore/utils/markdown') +const { editOrReply } = require('../../../labscore/utils/message') +const { STATICS } = require('../../../labscore/utils/statics') +const { getRecentImage } = require("../../../labscore/utils/attachment"); + +const { paginator } = require('../../../labscore/client'); +const { tineye } = require('../../../labscore/api'); + +function createTineyeResultPage(context, result){ + let res = { + "embeds": [ + createEmbed("default", context, { + description: `**${link(result.url, result.name)}**\nLast indexed: ${result.date}`, + image: { + url: result.image + }, + thumbnail: { + url: result.cached + }, + footer: { + iconUrl: STATICS.tineye, + text: `TinEye • ${context.application.name}` + } + }) + ] + } + if(result.thumbnail) res.embeds[0].thumbnail = { url: result.thumbnail }; + return res; +} + +module.exports = { + name: 'reverse-image', + aliases: ['reverse', 'tineye','reverseimage'], + metadata: { + description: 'reverse image search', + examples: ['reverseimage'], + category: 'search', + usage: 'reverse ' + }, + run: async (context, args) => { + context.triggerTyping(); + try{ + let image = await getRecentImage(context, 50) + if(!image) return editOrReply(context, { embeds: [createEmbed("warning", context, "No images found.")] }) + + let search = await tineye(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(createTineyeResultPage(context, res)) + } + + 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.`)]}) + } + }, +}; \ No newline at end of file diff --git a/labscore/api/endpoints.js b/labscore/api/endpoints.js index 7fe8dad..5159ebc 100644 --- a/labscore/api/endpoints.js +++ b/labscore/api/endpoints.js @@ -32,6 +32,7 @@ const Api = Object.freeze({ SEARCH_GOOGLE_IMAGES: '/search/google-images', SEARCH_LYRICS: '/search/lyrics', SEARCH_RULE34: '/search/booru', + SEARCH_TINEYE: '/search/tineye', SEARCH_URBANDICTIONARY: '/search/urbandictionary', SEARCH_WIKIHOW: '/search/wikihow', SEARCH_WOLFRAM_ALPHA: '/search/wolfram-alpha', diff --git a/labscore/api/index.js b/labscore/api/index.js index 8d0f49d..8206a45 100644 --- a/labscore/api/index.js +++ b/labscore/api/index.js @@ -135,6 +135,12 @@ module.exports.bingImages = async function(context, query){ }) } +module.exports.tineye = async function(context, url){ + return await request(Api.SEARCH_TINEYE, "GET", {}, { + url: url + }) +} + module.exports.urbandictionary = async function(context, query){ return await request(Api.SEARCH_URBANDICTIONARY, "GET", {}, { q: query diff --git a/labscore/utils/statics.js b/labscore/utils/statics.js index 3d2a49d..e0c3e8a 100644 --- a/labscore/utils/statics.js +++ b/labscore/utils/statics.js @@ -39,11 +39,15 @@ const Statics = Object.freeze({ wikihow: { file: "brands/wikihow.png", revision: 1 + }, + tineye: { + file: "brands/tineye.png", + revision: 1 } } }) -function staticAsset(static){ +function staticAsset(static) { return Hosts.statics + `assets/` + static.file + "?r=" + static.revision } @@ -56,5 +60,6 @@ module.exports.STATICS = Object.freeze({ inferkit: staticAsset(Statics.brands.inferkit), youtube: staticAsset(Statics.brands.youtube), urbandictionary: staticAsset(Statics.brands.urbandictionary), - wikihow: staticAsset(Statics.brands.wikihow) + wikihow: staticAsset(Statics.brands.wikihow), + tineye: staticAsset(Statics.brands.tineye) }) \ No newline at end of file