From 014ba91637ecd69cbe3c86b3df55f8d22ed70dc5 Mon Sep 17 00:00:00 2001
From: Ashley <iamashley@duck.com>
Date: Sun, 21 May 2023 20:22:44 +0000
Subject: [PATCH] Refactor code :3

---
 src/libpoketube/libpoketube-core.js | 321 +++++++++++++++-------------
 1 file changed, 170 insertions(+), 151 deletions(-)

diff --git a/src/libpoketube/libpoketube-core.js b/src/libpoketube/libpoketube-core.js
index 9c634dc4..b356d6f0 100644
--- a/src/libpoketube/libpoketube-core.js
+++ b/src/libpoketube/libpoketube-core.js
@@ -1,170 +1,189 @@
-/*
-
-    PokeTube is a Free/Libre youtube front-end !
-    
-    Copyright (C) 2021-2023 POKETUBE
- 
-    This file is Licensed under LGPL-3.0-or-later. Poketube itself is GPL, Only this file is LGPL.
-    
-    see a copy here:https://www.gnu.org/licenses/lgpl-3.0.txt
-    
-    please dont remove this comment while sharing this code 
-    
-  */
+/**
+ * PokeTube is a Free/Libre youtube front-end !
+ *
+ * This file is Licensed under LGPL-3.0-or-later. Poketube itself is GPL, Only this file is LGPL.
+ * See a copy here: https://www.gnu.org/licenses/lgpl-3.0.txt
+ * Please don't remove this comment while sharing this code.
+ */
 
 const fetch = require("node-fetch");
 const { toJson } = require("xml2json");
 const { curly } = require("node-libcurl");
-
 const fetcher = require("../libpoketube/libpoketube-fetcher.js");
 const getColors = require("get-image-colors");
-
 const wiki = require("wikipedia");
 
-// Util functions
-
-/*
- * Api functions
+/**
+ * Class representing PokeTube's core functionality.
  */
-function getJson(str) {
-  try {
-    return JSON.parse(str);
-  } catch {
-    return null;
+class PokeTubeCore {
+  /**
+   * Create an instance of PokeTubeCore.
+   * @param {object} config - Configuration object for PokeTubeCore.
+   * @param {string} config.tubeApi - Tube API URL.
+   * @param {string} config.invapi - Invid API URL.
+   * @param {string} config.dislikes - Dislikes API URL.
+   * @param {string} config.t_url - Matomo URL.
+   */
+  constructor(config) {
+    this.config = config;
+    this.cache = {};
+    this.sqp = "-oaymwEbCKgBEF5IVfKriqkDDggBFQAAiEIYAXABwAEG&rs=AOn4CLBy_x4UUHLNDZtJtH0PXeQGoRFTgw";
   }
-}
 
-function checkUnexistingObject(obj) {
-  if (obj) {
-    if ("authorId" in obj) {
-      return true;
+  /**
+   * Fetch JSON from API response.
+   * @param {string} str - String response from the API.
+   * @returns {object|null} Parsed JSON object or null if parsing failed.
+   */
+  getJson(str) {
+    try {
+      return JSON.parse(str);
+    } catch {
+      return null;
     }
   }
+
+  /**
+   * Check if the provided object has the required properties.
+   * @param {object} obj - Object to check.
+   * @returns {boolean} True if the object has the required properties, false otherwise.
+   */
+  checkUnexistingObject(obj) {
+    if (obj) {
+      if ("authorId" in obj) {
+        return true;
+      }
+    }
+  }
+
+  /**
+   * Fetch video information.
+   * @param {string} v - Video ID.
+   * @returns {Promise<object>} Promise resolving to the video information.
+   */
+  async video(v) {
+    if (v == null) return "Gib ID";
+
+    // Check if result is already cached
+    if (this.cache[v] && Date.now() - this.cache[v].timestamp < 3600000) {
+      console.log("Returning cached result");
+      return this.cache[v].result;
+    }
+
+    let desc = "";
+
+    try {
+      const inv_comments = await fetch(`${this.config.invapi}/comments/${v}`).then((res) =>
+        res.text()
+      );
+      var comments = await this.getJson(inv_comments);
+    } catch (error) {
+      this.initError("Error getting comments", error);
+      var comments = "";
+    }
+
+    let vid;
+
+    try {
+      const videoInfo = await fetch(`${this.config.invapi}/videos/${v}`).then((res) =>
+        res.text()
+      );
+      vid = await this.getJson(videoInfo);
+    } catch (error) {
+      this.initError("Error getting video info", error);
+    }
+
+    if (!vid) {
+      console.log(`Sorry nya, we couldn't find any information about that video qwq`);
+    }
+
+    if (this.checkUnexistingObject(vid)) {
+      let a;
+
+      try {
+        a = await fetch(`${this.config.tubeApi}channel?id=${vid.authorId}&tab=about`)
+          .then((res) => res.text())
+          .then((xml) => this.getJson(toJson(xml)));
+      } catch (error) {
+        this.initError("Error getting channel info", error);
+        a = "";
+      }
+
+      desc = a.Channel?.Contents?.ItemSection?.About?.Description;
+      const fe = await fetcher(v);
+
+      try {
+        const summary = await wiki
+          .summary(vid.author + " ")
+          .then((summary_) =>
+            summary_.title !== "Not found." ? summary_ : "none"
+          );
+
+        const headers = {};
+
+        const { data } = await curly.get(`${this.config.tubeApi}video?v=${v}`, {
+          httpHeader: Object.entries(headers).map(([k, v]) => `${k}: ${v}`),
+        });
+        const json = toJson(data);
+        const video = this.getJson(json);
+
+        // Store result in cache
+        this.cache[v] = {
+          result: {
+            json: fe?.video?.Player,
+            video,
+            vid,
+            comments,
+            engagement: fe.engagement,
+            wiki: summary,
+            desc: desc,
+            color: await getColors(
+              `https://i.ytimg.com/vi/${v}/hqdefault.jpg?sqp=${this.sqp}`
+            ).then((colors) => colors[0].hex()),
+            color2: await getColors(
+              `https://i.ytimg.com/vi/${v}/hqdefault.jpg?sqp=${this.sqp}`
+            ).then((colors) => colors[1].hex()),
+          },
+          timestamp: Date.now(),
+        };
+
+        return this.cache[v].result;
+      } catch (error) {
+        this.initError("Error getting video", error);
+      }
+    }
+  }
+
+  /**
+   * Check if a video ID is valid.
+   * @param {string} v - Video ID.
+   * @returns {boolean} True if the video ID is valid, false otherwise.
+   */
+  isvalidvideo(v) {
+    if (v != "assets" && v != "cdn-cgi" && v != "404") {
+      return true;
+    } else {
+      return false;
+    }
+  }
+
+  /**
+   * Initialize an error.
+   * @param {string} args - Error message.
+   * @param {Error} error - Error object.
+   */
+  initError(args, error) {
+    console.error("[LIBPT CORE ERROR]" + args, error);
+  }
 }
 
-const cache = {};
-const sqp = "-oaymwEbCKgBEF5IVfKriqkDDggBFQAAiEIYAXABwAEG&rs=AOn4CLBy_x4UUHLNDZtJtH0PXeQGoRFTgw";
-
-const config = {
+// Create an instance of PokeTubeCore with the provided config
+const pokeTubeCore = new PokeTubeCore({
   tubeApi: "https://inner-api.poketube.fun/api/",
   invapi: "https://invid-api.poketube.fun/api/v1",
   dislikes: "https://returnyoutubedislikeapi.com/votes?videoId=",
-  t_url: "https://t.poketube.fun/", //  def matomo url
-};
+  t_url: "https://t.poketube.fun/",
+});
 
-function initerr(args){
-  console.error("[LIBPT CORE ERROR]" + args) 
-}
-
-async function video(v) {
-  if (v == null) return "Gib ID";
-
-  // Check if result is already cached
-  if (cache[v] && Date.now() - cache[v].timestamp < 3600000) {
-    console.log("Returning cached result");
-    return cache[v].result;
-  }
-
-   var desc = "";
-
-  try {
-    var inv_comments = await fetch(`${config.invapi}/comments/${v}`).then(
-      (res) => res.text()
-    );
-
-    var comments = await getJson(inv_comments);
-  } catch (error) {
-    initerr("Error getting comments", error);
-    var comments = "";
-  }
-
-  let vid;
-
-  try {
-    const videoInfo = await fetch(`${config.invapi}/videos/${v}`).then((res) =>
-      res.text()
-    );
-    vid = await getJson(videoInfo);
-  } catch (error) {
-    initerr("Error getting video info", error);
-  }
-
-  if (!vid) {
-    console.log(
-      `Sorry nya, we couldn't find any information about that video qwq`
-    );
-  }
-
-  if (checkUnexistingObject(vid)) {
-    var a;
-
-    try {
-      var a = await fetch(
-        `${config.tubeApi}channel?id=${vid.authorId}&tab=about`
-      )
-        .then((res) => res.text())
-        .then((xml) => getJson(toJson(xml)));
-    } catch (error) {
-      initerr("Error getting channel info", error);
-      var a = "";
-    }
-    
-    desc = a.Channel?.Contents?.ItemSection?.About?.Description;
-    const fe = await fetcher(v);
-
-    try {
-      
-    const summary = await wiki
-      .summary(vid.author + " ")
-      .then((summary_) =>
-        summary_.title !== "Not found." ? summary_ : "none"
-      );
-      
-      const headers = {};
-
-      var { data } = await curly.get(`${config.tubeApi}video?v=${v}`, {
-        httpHeader: Object.entries(headers).map(([k, v]) => `${k}: ${v}`),
-      });
-      var json = toJson(data);
-      const video = getJson(json);
-
-      // Store result in cache
-      cache[v] = {
-        result: {
-          json: fe?.video?.Player,
-          video,
-          vid,
-          comments,
-          engagement: fe.engagement,
-          wiki: summary,
-          desc: desc,
-          color: await getColors(
-            `https://i.ytimg.com/vi/${v}/hqdefault.jpg?sqp=${sqp}`
-          ).then((colors) => colors[0].hex()),
-          color2: await getColors(
-            `https://i.ytimg.com/vi/${v}/hqdefault.jpg?sqp=${sqp}`
-          ).then((colors) => colors[1].hex()),
-        },
-        timestamp: Date.now(),
-      };
-
-      return cache[v].result;
-    } catch (error) {
-      initerr("Error getting video", error);
-    }
-  }
-}
-
-
-async function isvalidvideo(v) {
-    if (v != "assets" && v != "cdn-cgi" && v != "404") {
-    return true;
-    } else {
-    return false;
-    }
- }
-
-module.exports = {
-  video,
-  isvalidvideo
-};
+module.exports = pokeTubeCore;