@serifpersia/dango 3.6.7 → 3.6.9

package/server/dist/trpc/routers/tv.js

@@ -1,551 +0,0 @@
-import { TRPCError } from '@trpc/server';
-import { protectedProcedure, router } from '../index.js';
-import { defineSchema, failed, optStr, readCount, reqId, reqObj, reqStr } from '../validation.js';
-import { getTmdbKey, TMDB_BASE, TMDB_IMAGE } from '../../lib/tmdb.js';
-import { parseJsonBody } from '../../utils/http.utils.js';
-async function resolveTvMeta(mediaType, numericTmdbId, partial) {
-    const meta = { ...partial };
-    if (meta.title && meta.imdbId && meta.year)
-        return meta;
-    try {
-        const tmdbKey = await getTmdbKey();
-        if (tmdbKey) {
-            const tmdbRes = await fetch(`${TMDB_BASE}/${mediaType}/${numericTmdbId}?api_key=${tmdbKey}&append_to_response=external_ids`, { headers: { 'User-Agent': 'Mozilla/5.0' }, signal: AbortSignal.timeout(4000) });
-            if (tmdbRes.ok) {
-                const d = await parseJsonBody(tmdbRes);
-                if (!meta.title)
-                    meta.title = d.title || d.name || '';
-                if (!meta.year)
-                    meta.year = (d.release_date || d.first_air_date || '').split('-')[0] || '';
-                if (!meta.imdbId)
-                    meta.imdbId = d.external_ids?.imdb_id || d.imdb_id || '';
-                if (mediaType === 'tv' && d.number_of_seasons)
-                    meta.totalSeasons = String(d.number_of_seasons);
-            }
-        }
-    }
-    catch {
-        // ignore
-    }
-    return meta;
-}
-function toSearchResult(item, type) {
-    return {
-        id: item.id,
-        title: item.title || item.name || '',
-        year: (item.release_date || item.first_air_date || '').split('-')[0],
-        type,
-        image: item.poster_path ? `${TMDB_IMAGE}/w500${item.poster_path}` : '',
-        backdrop: item.backdrop_path ? `${TMDB_IMAGE}/w780${item.backdrop_path}` : '',
-        overview: item.overview || '',
-        vote_average: item.vote_average || 0,
-        adult: item.adult === true,
-        genre_ids: item.genre_ids || [],
-    };
-}
-const tvSearchInput = () => defineSchema((value) => {
-    const obj = reqObj(value);
-    const out = {};
-    const q = optStr(obj, 'q');
-    if (q !== undefined)
-        out.q = q;
-    if (obj.page !== undefined)
-        out.page = readCount(obj, 'page', 1);
-    const type = optStr(obj, 'type');
-    if (type !== undefined)
-        out.type = type;
-    const genre = optStr(obj, 'genre');
-    if (genre !== undefined)
-        out.genre = genre;
-    const year = optStr(obj, 'year');
-    if (year !== undefined)
-        out.year = year;
-    const sortBy = optStr(obj, 'sortBy');
-    if (sortBy !== undefined)
-        out.sortBy = sortBy;
-    return out;
-});
-const tvTrendingInput = () => defineSchema((value) => {
-    const obj = reqObj(value);
-    const out = {};
-    const mediaType = optStr(obj, 'mediaType');
-    if (mediaType !== undefined)
-        out.mediaType = mediaType;
-    const timeWindow = optStr(obj, 'timeWindow');
-    if (timeWindow !== undefined)
-        out.timeWindow = timeWindow;
-    if (obj.page !== undefined)
-        out.page = readCount(obj, 'page', 1);
-    return out;
-});
-const tvDetailsInput = () => defineSchema((value) => {
-    const obj = reqObj(value);
-    return { type: reqStr(obj, 'type'), id: reqId(obj, 'id') };
-});
-const tvEpisodesInput = () => defineSchema((value) => {
-    const obj = reqObj(value);
-    return { id: reqId(obj, 'id'), season: reqId(obj, 'season') };
-});
-const tvLookupImdbInput = () => defineSchema((value) => ({
-    imdbId: reqStr(reqObj(value), 'imdbId'),
-}));
-const tvSearchImdbInput = () => defineSchema((value) => {
-    const obj = reqObj(value);
-    const out = {};
-    const q = optStr(obj, 'q');
-    if (q !== undefined)
-        out.q = q;
-    return out;
-});
-const tvSubtitlesInput = () => defineSchema((value) => {
-    const obj = reqObj(value);
-    const out = { type: reqStr(obj, 'type'), tmdbId: reqId(obj, 'tmdbId') };
-    const season = optStr(obj, 'season');
-    if (season !== undefined)
-        out.season = season;
-    const episode = optStr(obj, 'episode');
-    if (episode !== undefined)
-        out.episode = episode;
-    return out;
-});
-const tvSourcesInput = () => defineSchema((value) => {
-    const obj = reqObj(value);
-    const out = {
-        provider: reqStr(obj, 'provider'),
-        type: reqStr(obj, 'type'),
-        tmdbId: reqId(obj, 'tmdbId'),
-    };
-    if (obj.season !== undefined)
-        out.season = readCount(obj, 'season', 1);
-    if (obj.episode !== undefined)
-        out.episode = readCount(obj, 'episode', 1);
-    const server = optStr(obj, 'server');
-    if (server !== undefined)
-        out.server = server;
-    const title = optStr(obj, 'title');
-    if (title !== undefined)
-        out.title = title;
-    const year = optStr(obj, 'year');
-    if (year !== undefined)
-        out.year = year;
-    const imdbId = optStr(obj, 'imdbId');
-    if (imdbId !== undefined)
-        out.imdbId = imdbId;
-    const totalSeasons = optStr(obj, 'totalSeasons');
-    if (totalSeasons !== undefined)
-        out.totalSeasons = totalSeasons;
-    return out;
-});
-export const tvRouter = router({
-    search: protectedProcedure.input(tvSearchInput()).query(async ({ ctx, input }) => {
-        const query = input.q || '';
-        const page = input.page || 1;
-        const type = input.type || 'multi';
-        const genre = input.genre || '';
-        const year = input.year || '';
-        const sortBy = input.sortBy || 'popularity.desc';
-        const cacheKey = `tv-search-${query.toLowerCase()}-${page}-${type}-${genre}-${year}-${sortBy}`;
-        const cached = ctx.apiCache.get(cacheKey);
-        if (cached)
-            return cached;
-        const key = await getTmdbKey();
-        if (!key)
-            throw failed('No TMDB API key available');
-        try {
-            let url;
-            if (type === 'tv' || type === 'movie') {
-                if (!query.trim()) {
-                    const params = new URLSearchParams({
-                        api_key: key,
-                        page: String(page),
-                        sort_by: sortBy,
-                    });
-                    if (genre)
-                        params.set('with_genres', genre);
-                    if (year) {
-                        if (type === 'tv')
-                            params.set('first_air_date_year', year);
-                        else
-                            params.set('primary_release_year', year);
-                    }
-                    url = `${TMDB_BASE}/discover/${type}?${params.toString()}`;
-                }
-                else {
-                    const params = new URLSearchParams({
-                        api_key: key,
-                        query: encodeURIComponent(query.trim()),
-                        page: String(page),
-                        include_adult: 'true',
-                    });
-                    url = `${TMDB_BASE}/search/${type}?${params.toString()}`;
-                }
-            }
-            else if (!query.trim()) {
-                const params = new URLSearchParams({
-                    api_key: key,
-                    page: String(page),
-                    sort_by: sortBy,
-                });
-                if (genre)
-                    params.set('with_genres', genre);
-                if (year)
-                    params.set('first_air_date_year', year);
-                url = `${TMDB_BASE}/discover/tv?${params.toString()}`;
-            }
-            else {
-                const params = new URLSearchParams({
-                    api_key: key,
-                    query: encodeURIComponent(query),
-                    page: String(page),
-                    include_adult: 'true',
-                });
-                url = `${TMDB_BASE}/search/multi?${params.toString()}`;
-            }
-            const r = await fetch(url, { headers: { 'User-Agent': 'Mozilla/5.0' } });
-            if (!r.ok)
-                throw failed('TMDB search failed');
-            const d = await parseJsonBody(r);
-            let results;
-            if (type === 'tv' || type === 'movie') {
-                results = (d.results || []).map((item) => toSearchResult(item, type));
-            }
-            else {
-                results = (d.results || [])
-                    .filter((item) => item.media_type !== 'person')
-                    .map((item) => toSearchResult(item, item.media_type || 'tv'));
-            }
-            const payload = {
-                results,
-                total_results: d.total_results || 0,
-                total_pages: d.total_pages || 0,
-                page: d.page || page,
-            };
-            ctx.apiCache.set(cacheKey, payload, 1800);
-            return payload;
-        }
-        catch (e) {
-            if (e instanceof TRPCError)
-                throw e;
-            throw failed(e.message);
-        }
-    }),
-    trending: protectedProcedure.input(tvTrendingInput()).query(async ({ ctx, input }) => {
-        const mediaType = input.mediaType || 'all';
-        const timeWindow = input.timeWindow || 'week';
-        const page = input.page || 1;
-        const cacheKey = `tv-trending-${mediaType}-${timeWindow}-${page}`;
-        const cached = ctx.apiCache.get(cacheKey);
-        if (cached)
-            return cached;
-        const key = await getTmdbKey();
-        if (!key)
-            throw failed('No TMDB API key available');
-        try {
-            const r = await fetch(`${TMDB_BASE}/trending/${mediaType}/${timeWindow}?api_key=${key}&page=${page}`, { headers: { 'User-Agent': 'Mozilla/5.0' } });
-            if (!r.ok)
-                throw failed('TMDB trending failed');
-            const d = await parseJsonBody(r);
-            const results = (d.results || [])
-                .filter((item) => item.media_type !== 'person')
-                .map((item) => toSearchResult(item, item.media_type || 'tv'));
-            const payload = {
-                results,
-                total_results: d.total_results || 0,
-                total_pages: d.total_pages || 0,
-                page: d.page || page,
-            };
-            ctx.apiCache.set(cacheKey, payload, 3600);
-            return payload;
-        }
-        catch (e) {
-            if (e instanceof TRPCError)
-                throw e;
-            throw failed(e.message);
-        }
-    }),
-    genres: protectedProcedure.query(async ({ ctx }) => {
-        const cacheKey = 'tv-genres-list';
-        const cached = ctx.apiCache.get(cacheKey);
-        if (cached)
-            return cached;
-        const key = await getTmdbKey();
-        if (!key)
-            throw failed('No TMDB API key available');
-        try {
-            const [tvRes, movieRes] = await Promise.all([
-                fetch(`${TMDB_BASE}/genre/tv/list?api_key=${key}`, {
-                    headers: { 'User-Agent': 'Mozilla/5.0' },
-                }),
-                fetch(`${TMDB_BASE}/genre/movie/list?api_key=${key}`, {
-                    headers: { 'User-Agent': 'Mozilla/5.0' },
-                }),
-            ]);
-            const tvData = tvRes.ok
-                ? await parseJsonBody(tvRes)
-                : { genres: [] };
-            const movieData = movieRes.ok
-                ? await parseJsonBody(movieRes)
-                : { genres: [] };
-            const payload = {
-                tv: (tvData.genres || []).map((g) => ({
-                    id: g.id,
-                    name: g.name,
-                })),
-                movie: (movieData.genres || []).map((g) => ({
-                    id: g.id,
-                    name: g.name,
-                })),
-            };
-            ctx.apiCache.set(cacheKey, payload, 86400);
-            return payload;
-        }
-        catch (e) {
-            if (e instanceof TRPCError)
-                throw e;
-            throw failed(e.message);
-        }
-    }),
-    details: protectedProcedure.input(tvDetailsInput()).query(async ({ input }) => {
-        const type = input.type;
-        const id = input.id;
-        const key = await getTmdbKey();
-        if (!key)
-            throw failed('No TMDB API key available');
-        try {
-            const r = await fetch(`${TMDB_BASE}/${type}/${id}?api_key=${key}&append_to_response=external_ids`, { headers: { 'User-Agent': 'Mozilla/5.0' } });
-            if (!r.ok)
-                throw failed('TMDB details failed');
-            const d = await parseJsonBody(r);
-            const result = {
-                id: d.id,
-                imdb_id: d.external_ids?.imdb_id || d.imdb_id || null,
-                title: d.title || d.name,
-                overview: d.overview,
-                vote_average: d.vote_average,
-                year: (d.release_date || d.first_air_date || '').split('-')[0],
-                poster: d.poster_path ? `${TMDB_IMAGE}/w500${d.poster_path}` : '',
-                backdrop: d.backdrop_path ? `${TMDB_IMAGE}/original${d.backdrop_path}` : '',
-                adult: d.adult === true,
-            };
-            if (type === 'tv') {
-                result.seasons = (d.seasons || [])
-                    .filter((s) => s.season_number > 0)
-                    .map((s) => ({
-                    season_number: s.season_number,
-                    episode_count: s.episode_count,
-                }));
-                result.number_of_seasons = d.number_of_seasons;
-            }
-            return result;
-        }
-        catch (e) {
-            if (e instanceof TRPCError)
-                throw e;
-            throw failed(e.message);
-        }
-    }),
-    episodes: protectedProcedure.input(tvEpisodesInput()).query(async ({ input }) => {
-        const id = input.id;
-        const season = input.season;
-        const key = await getTmdbKey();
-        if (!key)
-            throw failed('No TMDB API key available');
-        try {
-            const r = await fetch(`${TMDB_BASE}/tv/${id}/season/${season}?api_key=${key}`, {
-                headers: { 'User-Agent': 'Mozilla/5.0' },
-            });
-            if (!r.ok)
-                throw failed('TMDB episodes failed');
-            const d = await parseJsonBody(r);
-            const episodes = (d.episodes || []).map((ep) => ({
-                episode_number: ep.episode_number,
-                name: ep.name,
-                vote_average: ep.vote_average,
-                overview: ep.overview,
-                still_path: ep.still_path,
-            }));
-            return { episodes };
-        }
-        catch (e) {
-            if (e instanceof TRPCError)
-                throw e;
-            throw failed(e.message);
-        }
-    }),
-    lookupImdb: protectedProcedure.input(tvLookupImdbInput()).query(async ({ input }) => {
-        const key = await getTmdbKey();
-        if (!key)
-            return { tmdbId: null, error: 'No TMDB API key available' };
-        try {
-            const r = await fetch(`${TMDB_BASE}/find/${input.imdbId}?api_key=${key}&external_source=imdb_id`, { headers: { 'User-Agent': 'Mozilla/5.0' } });
-            if (!r.ok)
-                return { tmdbId: null, error: `TMDB error ${r.status}` };
-            const d = await parseJsonBody(r);
-            const movie = d.movie_results?.[0];
-            const tv = d.tv_results?.[0];
-            const result = movie || tv;
-            return {
-                tmdbId: result?.id || null,
-                type: movie ? 'movie' : tv ? 'tv' : null,
-                title: result?.title || result?.name || null,
-                year: (result?.release_date || result?.first_air_date || '').substring(0, 4) || null,
-            };
-        }
-        catch (e) {
-            return { tmdbId: null, error: e.message };
-        }
-    }),
-    searchImdb: protectedProcedure.input(tvSearchImdbInput()).query(async ({ ctx, input }) => {
-        const query = input.q || '';
-        if (!query)
-            return [];
-        const cacheKey = `tv-imdb-search-${query.toLowerCase()}`;
-        const cached = ctx.apiCache.get(cacheKey);
-        if (cached)
-            return cached;
-        try {
-            const response = await fetch(`https://v3.sg.media-imdb.com/suggestion/x/${encodeURIComponent(query)}.json`, { headers: { 'User-Agent': 'Mozilla/5.0' } });
-            if (!response.ok)
-                throw failed('IMDB search failed');
-            const data = await parseJsonBody(response);
-            const TV_TYPES = new Set(['tvSeries', 'tvMiniSeries', 'movie']);
-            const matches = (data.d || [])
-                .filter((entry) => Boolean(entry.id && entry.l) && TV_TYPES.has(entry.qid || ''))
-                .slice(0, 10)
-                .map((entry) => ({
-                id: entry.id,
-                title: entry.l,
-                year: entry.y,
-                type: entry.qid,
-                image: entry.i?.imageUrl || '',
-            }));
-            ctx.apiCache.set(cacheKey, matches, 3600);
-            return matches;
-        }
-        catch (e) {
-            if (e instanceof TRPCError)
-                throw e;
-            throw failed(e.message);
-        }
-    }),
-    subtitles: protectedProcedure.input(tvSubtitlesInput()).query(async ({ ctx, input }) => {
-        const type = input.type;
-        const tmdbId = input.tmdbId;
-        const numericTmdbId = parseInt(tmdbId, 10);
-        if (!numericTmdbId)
-            return { subtitles: [] };
-        const mediaType = type === 'movie' ? 'movie' : 'tv';
-        const season = input.season || '1';
-        const episode = input.episode || '1';
-        const cacheKey = mediaType === 'tv'
-            ? `tv-subs-${numericTmdbId}-${season}-${episode}`
-            : `tv-subs-${numericTmdbId}-movie`;
-        const cached = ctx.apiCache.get(cacheKey);
-        if (cached)
-            return cached;
-        try {
-            const searchUrl = mediaType === 'tv'
-                ? `https://subtitles.vidy.st/search?id=${numericTmdbId}&season=${encodeURIComponent(season)}&episode=${encodeURIComponent(episode)}`
-                : `https://subtitles.vidy.st/search?id=${numericTmdbId}`;
-            const r = await fetch(searchUrl, {
-                headers: { 'User-Agent': 'Mozilla/5.0' },
-                signal: AbortSignal.timeout(10000),
-            });
-            if (!r.ok)
-                return { subtitles: [] };
-            const items = (await parseJsonBody(r));
-            const subtitles = (Array.isArray(items) ? items : [])
-                .filter((s) => typeof s.url === 'string' && s.url.startsWith('https://'))
-                .map((s) => ({
-                language: String(s.language || 'en'),
-                label: String(s.display || s.language || 'Unknown'),
-                url: String(s.url),
-                trusted: s.isTrusted === true,
-                hi: s.isHearingImpaired === true,
-                downloads: Number(s.downloadCount) || 0,
-                media: String(s.media || s.release || ''),
-            }))
-                .sort((a, b) => {
-                const aEn = a.language.toLowerCase().startsWith('en') ? 0 : 1;
-                const bEn = b.language.toLowerCase().startsWith('en') ? 0 : 1;
-                if (aEn !== bEn)
-                    return aEn - bEn;
-                if (a.trusted !== b.trusted)
-                    return a.trusted ? -1 : 1;
-                return b.downloads - a.downloads;
-            })
-                .slice(0, 50);
-            const payload = { subtitles };
-            ctx.apiCache.set(cacheKey, payload, 3600);
-            return payload;
-        }
-        catch {
-            return { subtitles: [] };
-        }
-    }),
-    sources: protectedProcedure.input(tvSourcesInput()).query(async ({ ctx, input }) => {
-        const provider = input.provider;
-        const type = input.type;
-        const tmdbId = input.tmdbId;
-        const mediaType = type === 'movie' ? 'movie' : 'tv';
-        const numericTmdbId = parseInt(tmdbId, 10);
-        if (!numericTmdbId)
-            return { sources: [], audioTracks: [], error: 'Bad tmdb id' };
-        const mod = ctx.getTvProvider(String(provider).toLowerCase());
-        if (!mod || typeof mod.getSources !== 'function') {
-            return { sources: [], audioTracks: [], error: `Unknown TV provider ${provider}` };
-        }
-        const season = input.season || 1;
-        const episode = input.episode || 1;
-        const server = input.server;
-        const meta = await resolveTvMeta(mediaType, numericTmdbId, {
-            title: input.title || '',
-            year: input.year || '',
-            imdbId: input.imdbId || '',
-            totalSeasons: input.totalSeasons || '1',
-        });
-        const media = {
-            tmdbId: numericTmdbId,
-            type: mediaType,
-            season,
-            episode,
-            title: meta.title,
-            year: meta.year,
-            imdbId: meta.imdbId,
-            totalSeasons: parseInt(meta.totalSeasons, 10) || 1,
-        };
-        try {
-            const result = await mod.getSources(media, server);
-            if (!result || !Array.isArray(result.sources) || result.sources.length === 0) {
-                return {
-                    provider: mod.name,
-                    server: result?.server || server,
-                    sources: [],
-                    audioTracks: [],
-                    valid: false,
-                    error: 'No sources found',
-                };
-            }
-            return {
-                provider: mod.name,
-                server: result.server || server,
-                servers: result.servers,
-                embedUrl: result.embedUrl,
-                sources: result.sources,
-                audioTracks: result.audioTracks || [],
-                subtitles: result.subtitles || [],
-                referer: result.referer || '',
-                valid: true,
-            };
-        }
-        catch (e) {
-            return {
-                provider: mod.name,
-                sources: [],
-                audioTracks: [],
-                valid: false,
-                error: e.message,
-            };
-        }
-    }),
-});