Update src/libpoketube/libpoketube-core.js
This commit is contained in:
ashley
parent
18dae9b530
commit
3e552d7bc4
@ -51,8 +51,10 @@ class InnerTubePokeVidious {
|
|||||||
this.region = "region=US";
|
this.region = "region=US";
|
||||||
this.sqp = "-oaymwEbCKgBEF5IVfKriqkDDggBFQAAiEIYAXABwAEG&rs=AOn4CLBy_x4UUHLNDZtJtH0PXeQGoRFTgw";
|
this.sqp = "-oaymwEbCKgBEF5IVfKriqkDDggBFQAAiEIYAXABwAEG&rs=AOn4CLBy_x4UUHLNDZtJtH0PXeQGoRFTgw";
|
||||||
this.hedge = true;
|
this.hedge = true;
|
||||||
|
this.debugErrors = !!cfg.debugErrors || process.env.POKETUBE_DEBUG_ERRORS === "1";
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ---------- Utilities ----------
|
||||||
getJson(s) { try { return JSON.parse(s); } catch { return null; } }
|
getJson(s) { try { return JSON.parse(s); } catch { return null; } }
|
||||||
checkUnexistingObject(o) { return o && "authorId" in o; }
|
checkUnexistingObject(o) { return o && "authorId" in o; }
|
||||||
wait(ms) { return new Promise(r => setTimeout(r, ms)); }
|
wait(ms) { return new Promise(r => setTimeout(r, ms)); }
|
||||||
@ -75,187 +77,32 @@ class InnerTubePokeVidious {
|
|||||||
if (!Number.isNaN(dt)) return Math.max(0, dt - Date.now());
|
if (!Number.isNaN(dt)) return Math.max(0, dt - Date.now());
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
nowIso() { return new Date().toISOString(); }
|
||||||
|
newRequestId() { return Buffer.from(`${Date.now()}-${Math.random()}`).toString("base64url"); }
|
||||||
|
|
||||||
async fetchWithRetry(url, options = {}, cfg = {}, outerSignal) {
|
buildFriendlyMessage(reason, videoId) {
|
||||||
const { fetch } = await import("undici");
|
const prefix = "Sorry nya, we couldn't find any information about that video qwq.";
|
||||||
const maxRetries = Number.isInteger(cfg.retries) ? Math.max(0, cfg.retries) : 8;
|
const common = " If this keeps happening, please try again later or check the video link.";
|
||||||
const baseDelay = cfg.baseDelay ?? 160;
|
switch (reason) {
|
||||||
const maxDelay = cfg.maxDelay ?? 12000;
|
case "invalid_video_id":
|
||||||
const perAttemptTimeout = cfg.timeout ?? 12000;
|
return `Sorry nya, that doesn't look like a valid YouTube video ID (needs 11 letters/numbers). qwq Please double-check the link.${common}`;
|
||||||
const extraRetryOn = cfg.retryOnStatuses || [];
|
case "not_found_or_unparsable":
|
||||||
const uah = { "User-Agent": this.useragent };
|
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}`;
|
||||||
let lastErr = null;
|
case "missing_author":
|
||||||
|
return `Sorry nya, this video's channel info was missing so we couldn't finish loading it. qwq${common}`;
|
||||||
for (let attempt = 0; attempt <= maxRetries; attempt++) {
|
case "upstream_http_error":
|
||||||
const ac = new AbortController();
|
return `Sorry nya, an upstream service returned an error while fetching this video. qwq${common}`;
|
||||||
const onOuterAbort = () => ac.abort(outerSignal.reason || new Error("aborted"));
|
case "upstream_timeout":
|
||||||
if (outerSignal) {
|
return `Sorry nya, the request took too long and timed out while getting video info. qwq${common}`;
|
||||||
if (outerSignal.aborted) throw outerSignal.reason || new Error("aborted");
|
case "aborted":
|
||||||
outerSignal.addEventListener("abort", onOuterAbort, { once: true });
|
return `Sorry nya, the request was canceled before we finished. qwq${common}`;
|
||||||
}
|
case "internal_error":
|
||||||
const t = setTimeout(() => ac.abort(new Error("timeout")), perAttemptTimeout);
|
default:
|
||||||
try {
|
return `Sorry nya, something went wrong while loading this video. qwq Our bad!${common}`;
|
||||||
const res = await fetch(url, {
|
|
||||||
...options,
|
|
||||||
signal: ac.signal,
|
|
||||||
headers: { ...(options.headers || {}), ...uah },
|
|
||||||
});
|
|
||||||
clearTimeout(t);
|
|
||||||
if (outerSignal) outerSignal.removeEventListener("abort", onOuterAbort);
|
|
||||||
if (res.ok) return res;
|
|
||||||
const should = this.shouldRetryStatus(res.status) || extraRetryOn.includes(res.status);
|
|
||||||
if (!should || attempt === maxRetries) return res;
|
|
||||||
let delay = this.backoff(attempt, baseDelay, maxDelay);
|
|
||||||
if (res.status === 429 || res.status === 503) {
|
|
||||||
const ra = this.parseRetryAfter(res.headers.get("retry-after"));
|
|
||||||
if (ra != null) delay = Math.min(Math.max(ra, baseDelay), maxDelay);
|
|
||||||
}
|
|
||||||
await this.wait(delay);
|
|
||||||
} catch (e) {
|
|
||||||
clearTimeout(t);
|
|
||||||
if (outerSignal) outerSignal.removeEventListener("abort", onOuterAbort);
|
|
||||||
lastErr = e;
|
|
||||||
if (attempt === maxRetries) throw e;
|
|
||||||
await this.wait(this.backoff(attempt, baseDelay, maxDelay));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (lastErr) throw lastErr;
|
|
||||||
throw new Error("fetchWithRetry failed");
|
|
||||||
}
|
|
||||||
|
|
||||||
async hedgedGetJsonFromBases(bases, path, query, outerSignal) {
|
|
||||||
const qs = query ? (query.startsWith("?") ? query : "?" + query) : "";
|
|
||||||
const mk = (b) => `${b}${path}${qs}`;
|
|
||||||
const attemptOnce = async (u) => {
|
|
||||||
const r = await this.fetchWithRetry(u, {}, { retries: 4, baseDelay: 120, maxDelay: 6000, timeout: 10000 }, outerSignal);
|
|
||||||
const tx = await r.text();
|
|
||||||
return this.getJson(tx);
|
|
||||||
};
|
|
||||||
if (!this.hedge || !bases[1]) return attemptOnce(mk(bases[0]));
|
|
||||||
let p1 = attemptOnce(mk(bases[0]));
|
|
||||||
let p2 = (async () => { await this.wait(300); return attemptOnce(mk(bases[1])); })();
|
|
||||||
try {
|
|
||||||
return await Promise.any([p1, p2]);
|
|
||||||
} catch {
|
|
||||||
try { const a = await p1; if (a) return a; } catch {}
|
|
||||||
try { const b = await p2; if (b) return b; } catch {}
|
|
||||||
return null;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async getColorsSafe(url) {
|
buildError({ reason, status, url, videoId, requestId, retryAfterMs, originalError, meta }) {
|
||||||
for (let i = 0; i < 3; i++) {
|
const retryable = this.shouldRetryStatus(status) || reason === "upstream_timeout";
|
||||||
try {
|
const message = this.buildFriendlyMessage(reason, videoId);
|
||||||
const c = await getColors(url);
|
const err = {
|
||||||
if (Array.isArray(c) && c[0] && c[1]) return [c[0].hex(), c[1].hex()];
|
|
||||||
} catch {}
|
|
||||||
await this.wait(this.backoff(i, 120, 4000));
|
|
||||||
}
|
|
||||||
return ["#0ea5e9", "#111827"];
|
|
||||||
}
|
|
||||||
|
|
||||||
async curlGetWithRetry(url, httpHeader, outerSignal) {
|
|
||||||
let lastErr = null;
|
|
||||||
for (let i = 0; i <= 4; i++) {
|
|
||||||
if (outerSignal?.aborted) throw outerSignal.reason || new Error("aborted");
|
|
||||||
try {
|
|
||||||
const res = await curly.get(url, {
|
|
||||||
httpHeader,
|
|
||||||
timeoutMs: 12000,
|
|
||||||
connectTimeoutMs: 6000,
|
|
||||||
});
|
|
||||||
const code = res?.statusCode;
|
|
||||||
if (code >= 200 && code < 300 && res?.data) return res;
|
|
||||||
if (!this.shouldRetryStatus(code)) return res;
|
|
||||||
} catch (e) { lastErr = e; }
|
|
||||||
await this.wait(this.backoff(i, 160, 8000));
|
|
||||||
}
|
|
||||||
if (lastErr) throw lastErr;
|
|
||||||
throw new Error("curlGetWithRetry failed");
|
|
||||||
}
|
|
||||||
|
|
||||||
isvalidvideo(v) {
|
|
||||||
if (v != "assets" && v != "cdn-cgi" && v != "404") return /^([a-zA-Z0-9_-]{11})$/.test(v);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
initError(a, e) { console.error("[LIBPT CORE ERROR] " + a, e); }
|
|
||||||
|
|
||||||
async getYouTubeApiVideo(f, v, contentlang, contentregion) {
|
|
||||||
if (!this.isvalidvideo(v)) return { error: true, reason: "invalid_video_id" };
|
|
||||||
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}®ion=${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.tubeApi}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 { error: true, reason: "not_found_or_unparsable" };
|
|
||||||
|
|
||||||
let p = {};
|
|
||||||
if (f === "true" && vid.authorId) {
|
|
||||||
p = await this.hedgedGetJsonFromBases(bases, `/channels/${vid.authorId}`, `hl=${contentlang}®ion=${contentregion}`, outer.signal) || {};
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!this.checkUnexistingObject(vid)) return { error: true, reason: "missing_author" };
|
|
||||||
|
|
||||||
let fe = { engagement: null };
|
|
||||||
try { fe = await getdislikes(v); } catch {}
|
|
||||||
|
|
||||||
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,
|
|
||||||
channel_uploads: p,
|
|
||||||
engagement: fe.engagement,
|
|
||||||
wiki: "",
|
|
||||||
desc: "",
|
|
||||||
color: c1,
|
|
||||||
color2: c2
|
|
||||||
};
|
|
||||||
|
|
||||||
this.cache.set(v, { result, timestamp: Date.now() });
|
|
||||||
return result;
|
|
||||||
} catch (error) {
|
|
||||||
this.initError("Error getting video", error);
|
|
||||||
return { error: true, reason: "internal_error" };
|
|
||||||
} finally {
|
|
||||||
clearTimeout(outerTimeout);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const pokeTubeApiCore = new InnerTubePokeVidious({
|
|
||||||
tubeApi: "https://inner-api.poketube.fun/api/",
|
|
||||||
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,
|
|
||||||
});
|
|
||||||
|
|
||||||
module.exports = pokeTubeApiCore;
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user