From d37a0d46defce412975f800bb16951ed876d7a52 Mon Sep 17 00:00:00 2001
From: bignutty <3515180-bignutty@users.noreply.gitlab.com>
Date: Fri, 3 Jan 2025 00:40:55 +0100
Subject: [PATCH] add initial anime search work

---
 commands/message/search/anime.js | 94 ++++++++++++++++++++++++++++++++
 labscore/api/endpoints.js        |  2 +
 labscore/api/index.js            | 13 +++++
 labscore/utils/embed.js          |  5 +-
 4 files changed, 113 insertions(+), 1 deletion(-)
 create mode 100644 commands/message/search/anime.js

diff --git a/commands/message/search/anime.js b/commands/message/search/anime.js
new file mode 100644
index 0000000..f49693d
--- /dev/null
+++ b/commands/message/search/anime.js
@@ -0,0 +1,94 @@
+const { anime } = require('#api');
+const { paginator } = require('#client');
+const { PERMISSION_GROUPS } = require('#constants');
+
+const { createEmbed, formatPaginationEmbeds, page, hexToEmbedColor } = require('#utils/embed');
+const { acknowledge } = require('#utils/interactions');
+const { smallPill, link } = require('#utils/markdown');
+const { editOrReply } = require('#utils/message');
+
+function renderAnimeResultsPage(context, res){
+  let result = createEmbed("default", context, {
+    author: {
+      name: res.title,
+      url: res.url
+    },
+    description: ``,
+    fields: []
+  })
+
+  // Render Description
+  if(res.subtitle) result.description += `-# ${res.subtitle}\n\n`;
+  if(res.genres) result.description += res.genres.map((r)=>smallPill(r)).join("   ") + "\n\n";
+  if(res.description) result.description += res.description;
+
+  // Render Images
+  if(res.cover) result.thumbnail = { url: res.cover };
+  if(res.image) result.image = { url: res.image };
+  
+  // Render Color
+  if(res.color) result.color = hexToEmbedColor(res.color);
+
+  // Render Episode Metadata
+  if(res.episodes) {
+    result.fields.push({
+      name: "Episodes",
+      value: `${res.episodes} ${res.episode_length ? `@ ${res.episode_length}` : ""}`,
+      inline: true
+    })
+  }
+
+  if(res.links){
+    result.fields.push({
+      name: "Links",
+      value: res.links.map((l)=>`${link(l.url, l.label)}`).join("\n"),
+      inline: true
+    })
+  }
+
+  return page(result);
+}
+
+module.exports = {
+  name: 'anime',
+  label: 'query',
+  aliases: ['ani'],
+  metadata: {
+    description: 'Returns search results for Anime.',
+    description_short: 'Search Anime',
+    examples: [
+      'ani dandadan',
+      'ani stranger by the shore'
+    ],
+    category: 'search',
+    usage: 'anime <query>',
+    //slashCommand: "anime"
+  },
+  permissionsClient: [...PERMISSION_GROUPS.baseline],
+  run: async (context, args) => {
+    await acknowledge(context);
+    
+    if(!args.query) return editOrReply(context, createEmbed("warning", context, `Missing Parameter (query).`))
+    try{
+      let search = await anime(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(renderAnimeResultsPage(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 anime search.`))
+    }
+  },
+};
\ No newline at end of file
diff --git a/labscore/api/endpoints.js b/labscore/api/endpoints.js
index 51794aa..950f610 100644
--- a/labscore/api/endpoints.js
+++ b/labscore/api/endpoints.js
@@ -32,6 +32,8 @@ const Api = Object.freeze({
   PHOTOFUNIA_RETRO_WAVE: '/photofunia/retro-wave',
   PHOTOFUNIA_YACHT: '/photofunia/yacht',
 
+  SEARCH_ANIME: '/search/anime',
+  SEARCH_ANIME_SUPPLEMENTAL: '/search/anime-supplemental',
   SEARCH_BING: '/search/bing',
   SEARCH_BING_IMAGES: '/search/bing-images',
   SEARCH_GOOGLE: '/search/google',
diff --git a/labscore/api/index.js b/labscore/api/index.js
index 6df9c93..b118ebe 100644
--- a/labscore/api/index.js
+++ b/labscore/api/index.js
@@ -170,6 +170,19 @@ module.exports.rule34 = async function(context, query, service){
   })
 }
 
+module.exports.anime = async function(context, query, includeAdultContent){
+  return await request(Api.SEARCH_ANIME, "GET", {}, {
+    q: query,
+    include_adult: includeAdultContent
+  })
+}
+
+module.exports.animeSupplemental = async function(context, supplementalKey){
+  return await request(Api.SEARCH_ANIME_SUPPLEMENTAL, "GET", {}, {
+    supplemental_key: supplementalKey
+  })
+}
+
 module.exports.bing = async function(context, query, nsfw){
   return await request(Api.SEARCH_BING, "GET", {}, {
     q: query,
diff --git a/labscore/utils/embed.js b/labscore/utils/embed.js
index 3d62469..e395275 100644
--- a/labscore/utils/embed.js
+++ b/labscore/utils/embed.js
@@ -181,8 +181,11 @@ module.exports.formatPaginationEmbeds = function(embeds){
 
 // Creates a page for our paginator. simple helper so we dont have to do {embeds:[]} every time
 module.exports.page = function(embed){
-
   return {
     embeds: [embed]
   }
+}
+
+module.exports.hexToEmbedColor = (color)=>{
+  return parseInt(color.split("#")[1], 16)
 }
\ No newline at end of file