Files
sdk/test/index.js
2026-11-07 10:42:12 -08:00

263 lines
9.1 KiB
JavaScript

// usage: npm test
// Configure the variables below to customize your testing environment
import fs, { truncate } from "fs";
import path from "path";
import VylaSDK from "../src/sdk.js";
import dotenv from "dotenv";
const CONFIG = {
DEBUG: true, // Set to true to print outgoing requests, headers, formats, and responses
MOVIE_ID: "936075", // Movie TMDB ID to test
TV_ID: "37854", // TV TMDB ID to test, for anime I recommend using a random ID like 37854
TV_SEASON: "1", // TV Season to test
TV_EPISODE: "1", // TV Episode to test
STREAM_SOURCE: "", // Source key to test
TEST_SUBTITLE: false,
TEST_DOWNLOAD: false,
TEST_MOVIE: true,
TEST_TV: true,
TEST_MANGA: false,
MANGA_QUERY: "Steel Ball Run",
EXTRA_SDK_PARAMS: {
// penguManifest: ""
}
};
dotenv.config({
debug: CONFIG.DEBUG
});
// debug
if (CONFIG.DEBUG && !process.argv.includes("all")) {
const originalFetch = globalThis.fetch;
const debugFile = path.resolve("test/debug-output.json");
fs.mkdirSync(path.dirname(debugFile), { recursive: true });
const logs = [];
const save = () => {
fs.writeFileSync(debugFile, JSON.stringify(logs, null, 2));
};
globalThis.fetch = async (input, init) => {
const url = typeof input === "string" ? input : input?.url || String(input);
const request = {
timestamp: new Date().toISOString(),
request: {
url,
method: init?.method || "GET",
headers: init?.headers || {},
body: init?.body || null
}
};
console.log("\nRequest:");
console.log(`URL: ${url}`);
console.log(`Method: ${request.request.method}`);
if (init?.headers) console.log("Headers:", JSON.stringify(init.headers, null, 2));
if (init?.body) console.log("Body:", init.body);
try {
const response = await originalFetch(input, init);
request.response = {
status: response.status,
statusText: response.statusText
};
console.log("\nResponse:");
console.log(`URL: ${url}`);
console.log(`Status: ${response.status} ${response.statusText}`);
try {
const clone = response.clone();
const text = await clone.text();
try {
request.response.format = "json";
request.response.body = JSON.parse(text);
console.log("Format: JSON");
console.log(JSON.stringify(request.response.body, null, 2));
} catch {
request.response.format = "text";
request.response.body = text;
console.log("Format: Text/HTML");
console.log(text.substring(0, 800) + (text.length > 800 ? "..." : ""));
}
} catch (err) {
request.response.body = `<Failed to read response body: ${err.message}>`;
}
logs.push(request);
save();
return response;
} catch (err) {
request.error = err.message;
logs.push(request);
save();
console.log(`URL: ${url}`);
console.log(`Error: ${err.message}`);
throw err;
}
};
}
const sdk = new VylaSDK({
tmdbApiKey: process.env.TMDB_API_KEY,
...CONFIG.EXTRA_SDK_PARAMS
});
if (process.argv.includes("all")) {
const sources = sdk.getSources(true);
console.log(`Testing ${sources.length} sources in parallel...\n`);
const results = {};
const failed = [];
await Promise.all(
sources.map(async (cfg) => {
try {
const res = await sdk.probeSource(cfg.key);
results[cfg.key] = res;
if (res?.ok) {
console.log(`[PASS] ${cfg.key} (${res.ms}ms)`);
} else {
failed.push(cfg.key);
console.log(`[FAIL] ${cfg.key} (${res?.ms ?? 0}ms)`);
}
} catch (err) {
results[cfg.key] = { ok: false, ms: null, error: err.message };
failed.push(cfg.key);
console.log(`[FAIL] ${cfg.key} (${err.message})`);
}
})
);
console.log(`\nFinished testing. Total: ${sources.length}, Passed: ${sources.length - failed.length}, Failed: ${failed.length}`);
console.log("Sources that returned nothing:", failed);
const debugFile = path.resolve("test/debug-output.json");
fs.mkdirSync(path.dirname(debugFile), { recursive: true });
fs.writeFileSync(debugFile, JSON.stringify({
sourcesThatReturnedNothing: failed,
results
}, null, 2));
process.exit(0);
}
console.log("Starting...");
if (CONFIG.TEST_SUBTITLE && CONFIG.TEST_TV) {
try {
console.log(`\nTV Subtitles for ID: ${CONFIG.TV_ID} (S${CONFIG.TV_SEASON}E${CONFIG.TV_EPISODE})`);
const tvSubs = await sdk.getSubtitles(CONFIG.TV_ID, CONFIG.TV_SEASON, CONFIG.TV_EPISODE);
console.log("TV Subtitles Count:", tvSubs.length);
console.log("First TV Subtitle Sample:", tvSubs[0] || "None");
} catch (err) {
console.error("TV Subtitles Error:", err.message);
}
} else {
console.log("\nSkipped Subtitles Test");
}
if (CONFIG.TEST_SUBTITLE && CONFIG.TEST_MOVIE) {
try {
console.log(`\nMovie Subtitles for ID: ${CONFIG.MOVIE_ID}`);
const movieSubs = await sdk.getSubtitles(CONFIG.MOVIE_ID);
console.log("Movie Subtitles Count:", movieSubs.length);
console.log("First Subtitle Sample:", movieSubs[0] || "None");
} catch (err) {
console.error("Movie Subtitles Error:", err.message);
}
} else {
console.log("\nSkipped Subtitles Test");
}
if (CONFIG.TEST_DOWNLOAD && CONFIG.TEST_MOVIE) {
try {
console.log(`\nMovie Downloads for ID: ${CONFIG.MOVIE_ID}`);
const movieDownloads = await sdk.getDownloads(CONFIG.MOVIE_ID);
console.log("Movie Downloads:", movieDownloads);
} catch (err) {
console.error("Movie Downloads Error:", err.message);
}
} else {
console.log("\nSkipped Downloads Test");
}
if (CONFIG.TEST_DOWNLOAD && CONFIG.TEST_TV) {
try {
console.log(`\nTV Downloads for ID: ${CONFIG.TV_ID} (S${CONFIG.TV_SEASON}E${CONFIG.TV_EPISODE})`);
const tvDownloads = await sdk.getDownloads(CONFIG.TV_ID, CONFIG.TV_SEASON, CONFIG.TV_EPISODE);
console.log("TV Downloads:", tvDownloads);
} catch (err) {
console.error("TV Downloads Error:", err.message);
}
} else {
console.log("\nSkipped Downloads Test");
}
if (CONFIG.TEST_MOVIE) {
try {
console.log(`\nMovie Stream from "${CONFIG.STREAM_SOURCE}" for ID: ${CONFIG.MOVIE_ID}`);
const movieResult = await sdk.getStream(CONFIG.STREAM_SOURCE, CONFIG.MOVIE_ID);
console.log("Movie Stream:", movieResult);
} catch (err) {
console.error("Movie Stream Error:", err.message);
}
} else {
console.log("\nSkipped Stream Test");
}
if (CONFIG.TEST_TV) {
try {
console.log(`\nTV Stream from "${CONFIG.STREAM_SOURCE}" for ID: ${CONFIG.TV_ID} (S${CONFIG.TV_SEASON}E${CONFIG.TV_EPISODE})`);
const tvResult = await sdk.getStream(CONFIG.STREAM_SOURCE, CONFIG.TV_ID, CONFIG.TV_SEASON, CONFIG.TV_EPISODE);
console.log("TV Stream:", tvResult);
} catch (err) {
console.error("TV Stream Error:", err.message);
}
} else {
console.log("\nSkipped Stream Test");
}
if (CONFIG.TEST_MANGA) {
try {
console.log(`\nSearching Manga for: "${CONFIG.MANGA_QUERY}"`);
const searchRes = await sdk.searchManga(CONFIG.MANGA_QUERY, 1, 5);
console.log("Manga Search Found:", searchRes.total);
console.log("Top Hit:", searchRes.results[0]);
if (searchRes.results.length > 0) {
const mangaId = searchRes.results[0].id;
console.log(`\nFetching Manga Details for ID: ${mangaId}`);
const mangaDetails = await sdk.getManga(mangaId);
console.log("Manga Title:", mangaDetails.title);
console.log("Total Chapters:", mangaDetails.totalChapters);
const firstChapter = mangaDetails.chapters[mangaDetails.chapters.length - 1] || mangaDetails.chapters[0];
if (firstChapter) {
console.log(`\nFetching Pages for Chapter: ${firstChapter.title} (${firstChapter.id})`);
const chapterPages = await sdk.getMangaChapter(mangaId, firstChapter.id);
console.log("Page Count:", chapterPages.pages.length);
console.log("Sample First Page:", chapterPages.pages[0]);
}
}
} catch (err) {
console.error("Manga Test Error:", err.message);
}
} else {
console.log("\nSkipped Manga Test");
}