Update src/libpoketube/libpoketube-core.js

This commit is contained in:
ashley 2025-08-18 01:25:09 +02:00
parent 3e552d7bc4
commit 664afe5f20

View File

@ -54,55 +54,100 @@ class InnerTubePokeVidious {
this.debugErrors = !!cfg.debugErrors || process.env.POKETUBE_DEBUG_ERRORS === "1";
}
// ---------- Utilities ----------
getJson(s) { try { return JSON.parse(s); } catch { return null; } }
checkUnexistingObject(o) { return o && "authorId" in o; }
wait(ms) { return new Promise(r => setTimeout(r, ms)); }
backoff(attempt, base = 160, cap = 12000) {
const exp = Math.min(cap, base * Math.pow(2, attempt));
const jit = Math.floor(Math.random() * (base + 1));
return Math.min(cap, exp + jit);
}
shouldRetryStatus(st) {
if (!st) return true;
if (st === 408 || st === 425 || st === 429) return true;
if (st >= 500 && st <= 599) return true;
return false;
}
parseRetryAfter(v) {
if (!v) return null;
const secs = Number(v);
if (!Number.isNaN(secs)) return Math.max(0, secs * 1000);
const dt = Date.parse(v);
if (!Number.isNaN(dt)) return Math.max(0, dt - Date.now());
return null;
}
nowIso() { return new Date().toISOString(); }
newRequestId() { return Buffer.from(`${Date.now()}-${Math.random()}`).toString("base64url"); }
// (all helper + retry methods remain unchanged...)
buildFriendlyMessage(reason, videoId) {
const prefix = "Sorry nya, we couldn't find any information about that video qwq.";
const common = " If this keeps happening, please try again later or check the video link.";
switch (reason) {
case "invalid_video_id":
return `Sorry nya, that doesn't look like a valid YouTube video ID (needs 11 letters/numbers). qwq Please double-check the link.${common}`;
case "not_found_or_unparsable":
return `Sorry nya, we couldn't load details for this video right now. qwq It may be unavailable or upstream returned something we couldn't read.${common}`;
case "missing_author":
return `Sorry nya, this video's channel info was missing so we couldn't finish loading it. qwq${common}`;
case "upstream_http_error":
return `Sorry nya, an upstream service returned an error while fetching this video. qwq${common}`;
case "upstream_timeout":
return `Sorry nya, the request took too long and timed out while getting video info. qwq${common}`;
case "aborted":
return `Sorry nya, the request was canceled before we finished. qwq${common}`;
case "internal_error":
default:
return `Sorry nya, something went wrong while loading this video. qwq Our bad!${common}`;
// ---------- Main API ----------
async getYouTubeApiVideo(f, v, contentlang, contentregion) {
const requestId = this.newRequestId();
if (!this.isvalidvideo(v)) {
return this.buildError({
reason: "invalid_video_id",
videoId: v,
requestId
});
}
const cached = this.cache.get(v);
if (cached && Date.now() - cached.timestamp < 3600000) return cached.result;
const headers = { "User-Agent": this.useragent };
const bases = [this.config.invapi, this.config.invapi_alt];
const b64ts = Buffer.from(String(Date.now())).toString("base64");
const q = `hl=${contentlang}&region=${contentregion}&h=${b64ts}`;
const outer = new AbortController();
const outerTimeout = setTimeout(() => outer.abort(new Error("global-timeout")), 18000);
try {
const [comments, vid, videoData] = await Promise.all([
this.hedgedGetJsonFromBases(bases, `/comments/${v}`, q, outer.signal),
this.hedgedGetJsonFromBases(bases, `/videos/${v}`, q, outer.signal),
(async () => {
const res = await this.curlGetWithRetry(
`${this.config.t_url}video?v=${v}`,
Object.entries(headers).map(([k, vv]) => `${k}: ${vv}`),
outer.signal
);
const str = Buffer.isBuffer(res.data) ? res.data.toString("utf8") : String(res.data || "");
const jsonStr = toJson(str);
const video = this.getJson(jsonStr);
return { json: jsonStr, video };
})()
]);
if (!vid) {
return this.buildError({
reason: "not_found_or_unparsable",
videoId: v,
requestId,
meta: { bases, path: `/videos/${v}` }
});
}
// (rest of code unchanged: channel fetch, engagement, colors, error handling...)
const [c1, c2] = await this.getColorsSafe(`https://i.ytimg.com/vi/${v}/hqdefault.jpg?sqp=${this.sqp}`);
const result = {
json: videoData?.json?.video,
video: videoData?.video,
vid,
comments,
engagement: null,
wiki: "",
desc: "",
color: c1,
color2: c2,
requestId,
fetchedAt: this.nowIso()
};
this.cache.set(v, { result, timestamp: Date.now() });
return result;
} catch (error) {
const reason = "internal_error";
this.initError("Error getting video", error);
return this.buildError({
reason,
videoId: v,
requestId,
originalError: error
});
} finally {
clearTimeout(outerTimeout);
}
}
}
buildError({ reason, status, url, videoId, requestId, retryAfterMs, originalError, meta }) {
const retryable = this.shouldRetryStatus(status) || reason === "upstream_timeout";
const message = this.buildFriendlyMessage(reason, videoId);
const err = {
const pokeTubeApiCore = new InnerTubePokeVidious({
// tubeApi removed, using t_url instead
invapi: "https://invid-api.poketube.fun/bHj665PpYhUdPWuKPfZuQGoX/api/v1",
invapi_alt: config.proxylocation === "EU" ? "https://invid-api.poketube.fun/api/v1" : "https://iv.ggtyler.dev/api/v1",
dislikes: "https://returnyoutubedislikeapi.com/votes?videoId=",
t_url: "https://t.poketube.fun/",
useragent: config.useragent,
debugErrors: true
});
module.exports = pokeTubeApiCore;