[cmd](search) reverse-image

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 <image>'
+  },
+  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.`)]})
+    }
+  },
+};

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',

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

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)
 })