From 2be710d907042701646320e806708c83ab6c5aca Mon Sep 17 00:00:00 2001
From: derpystuff <3515180-derpystuff@users.noreply.gitlab.com>
Date: Fri, 19 Jan 2024 21:50:29 +0100
Subject: [PATCH] xkcd command

---
 commands/message/search/xkcd.js | 97 +++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 commands/message/search/xkcd.js

diff --git a/commands/message/search/xkcd.js b/commands/message/search/xkcd.js
new file mode 100644
index 0000000..a967e4f
--- /dev/null
+++ b/commands/message/search/xkcd.js
@@ -0,0 +1,97 @@
+const { createEmbed, formatPaginationEmbeds, page } = require('../../../labscore/utils/embed')
+const { editOrReply } = require('../../../labscore/utils/message')
+
+const { paginator } = require('../../../labscore/client');
+const superagent = require('superagent')
+
+const { Permissions } = require("detritus-client/lib/constants");
+
+module.exports = {
+  name: 'xkcd',
+  label: 'query',
+  metadata: {
+    description: 'Search XKCD comics.',
+    description_short: 'Search XKCD comics.',
+    examples: ['xkcd lead pipe'],
+    category: 'search',
+    usage: 'xkcd <query>'
+  },
+  permissionsClient: [Permissions.EMBED_LINKS, Permissions.SEND_MESSAGES, Permissions.USE_EXTERNAL_EMOJIS, Permissions.READ_MESSAGE_HISTORY],
+  run: async (context, args) => {
+    context.triggerTyping();
+    if (!args.query) return editOrReply(context, createEmbed("warning", context, `Missing Parameter (query).`))
+
+    if(!isNaN(parseInt(args.query))){
+      try{
+        let search = await superagent.get(`https://xkcd.com/${parseInt(args.query)}/info.0.json`)
+
+        return editOrReply(context, createEmbed("default", context, {
+          author: {
+            name: search.body.safe_title,
+            url: `https://xkcd.com/${search.body.num}/`
+          },
+          description: search.body.alt,
+          image: {
+            url: search.body.img
+          },
+          footer: {
+            iconUrl: `https://www.google.com/s2/favicons?domain=xkcd.com&sz=256`,
+            text: `xkcd • ${context.application.name}`
+          }
+        }))
+      }catch(e){
+        return editOrReply(context, createEmbed("error", context, `Comic not found.`))
+      }
+    } else {
+      try {
+        let search = await superagent.post(`https://qtg5aekc2iosjh93p.a1.typesense.net/multi_search?use_cache=true&x-typesense-api-key=${process.env.XKCD_KEY}`)
+          .send({
+            "searches": [
+              {
+                "query_by": "title,altTitle,transcript,topics,embedding",
+                "query_by_weights": "127,80,80,1,1",
+                "num_typos": 1,
+                "exclude_fields": "embedding",
+                "vector_query": "embedding:([], k: 30, distance_threshold: 0.1, alpha: 0.9)",
+                "highlight_full_fields": "title,altTitle,transcript,topics,embedding",
+                "collection": "xkcd",
+                "q": args.query,
+                "facet_by": "topics,publishDateYear",
+                "max_facet_values": 100,
+                "page": 1,
+                "per_page": 100
+              }
+            ]
+          })
+  
+        let pages = []
+        for (const res of search.body.results[0].hits) {
+          pages.push(page(createEmbed("default", context, {
+            author: {
+              name: res.document.title,
+              url: `https://xkcd.com/${res.document.id}/`
+            },
+            description: res.document.altTitle,
+            image: {
+              url: res.document.imageUrl
+            },
+            footer: {
+              iconUrl: `https://www.google.com/s2/favicons?domain=xkcd.com&sz=256`,
+              text: `xkcd • ${context.application.name}`
+            }
+          })))
+        }
+  
+        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 xkcd search.`))
+      }
+    }
+  },
+};
\ No newline at end of file