poke/src/libpoketube/libpoketube-core.js
2023-08-24 18:45:54 +00:00

176 lines
4.8 KiB
JavaScript

/**
* 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 { 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");
const language = "hl=en-US"
const region = "region=US"
/**
* Class representing PokeTube's core functionality.
*/
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";
}
/**
* 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) {
const { fetch } = await import("undici");
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;
}
const headers = {};
let desc = "";
try {
const [invComments, videoInfo, videoData] = await Promise.all([
fetch(`${this.config.invapi}/comments/${v}?${language}`).then((res) => res.text()),
fetch(`${this.config.invapi}/videos/${v}`).then((res) => res.text()),
curly
.get(`${this.config.tubeApi}video?v=${v}`, {
httpHeader: Object.entries(headers).map(([k, v]) => `${k}: ${v}`),
})
.then((res) => {
const json = toJson(res.data);
const video = this.getJson(json);
return { json, video };
}),
]);
const comments = await this.getJson(invComments);
const vid = await this.getJson(videoInfo);
const { json, video } = videoData;
if (!vid) {
console.log(
`Sorry nya, we couldn't find any information about that video qwq`
);
}
if (this.checkUnexistingObject(vid)) {
const fe = await fetcher(v);
try {
const headers = {};
// Store result in cache
this.cache[v] = {
result: {
json: json?.video,
video,
vid,
comments,
engagement: fe.engagement,
wiki: "",
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);
}
}
} catch {
}
}
/**
* 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);
}
}
// Create an instance of PokeTubeCore with the provided config
const pokeTubeApiCore = 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/",
});
module.exports = pokeTubeApiCore;