128 lines
4.2 KiB
TypeScript
128 lines
4.2 KiB
TypeScript
import { Mutex } from 'async-mutex';
|
|
import { NotFoundError } from '../error';
|
|
import { Context } from '../types';
|
|
import { envGetRequired } from './env';
|
|
import { CustomRequestConfig, Fetcher } from './Fetcher';
|
|
import { ImdbId, TmdbId } from './id';
|
|
|
|
interface FindResponsePartial {
|
|
movie_results: {
|
|
id: number;
|
|
}[];
|
|
tv_results: {
|
|
id: number;
|
|
}[];
|
|
}
|
|
|
|
interface ExternalIdsResponsePartial {
|
|
imdb_id: string;
|
|
}
|
|
|
|
interface MovieDetailsResponsePartial {
|
|
original_title: string;
|
|
release_date: string;
|
|
title: string;
|
|
}
|
|
|
|
interface TvDetailsResponsePartial {
|
|
first_air_date: string;
|
|
name: string;
|
|
original_name: string;
|
|
}
|
|
|
|
const mutexes = new Map<string, Mutex>();
|
|
const tmdbFetch = async (ctx: Context, fetcher: Fetcher, path: string, searchParams?: Record<string, string | undefined>): Promise<unknown> => {
|
|
const config: CustomRequestConfig = {
|
|
headers: {
|
|
'Authorization': 'Bearer ' + envGetRequired('TMDB_ACCESS_TOKEN'),
|
|
'Content-Type': 'application/json',
|
|
},
|
|
queueLimit: 50,
|
|
};
|
|
|
|
const url = new URL(`https://api.themoviedb.org/3${path}`);
|
|
|
|
Object.entries(searchParams ?? {}).forEach(([name, value]) => {
|
|
if (value) {
|
|
url.searchParams.set(name, value);
|
|
}
|
|
});
|
|
|
|
let mutex = mutexes.get(url.href);
|
|
if (!mutex) {
|
|
mutex = new Mutex();
|
|
mutexes.set(url.href, mutex);
|
|
}
|
|
|
|
const data = await mutex.runExclusive(async () => {
|
|
return await fetcher.json(ctx, url, config);
|
|
});
|
|
|
|
if (!mutex.isLocked()) {
|
|
mutexes.delete(url.href);
|
|
}
|
|
|
|
return data;
|
|
};
|
|
|
|
const imdbTmdbMap = new Map<string, number>();
|
|
export const getTmdbIdFromImdbId = async (ctx: Context, fetcher: Fetcher, imdbId: ImdbId): Promise<TmdbId> => {
|
|
// Manual mismatch fixes
|
|
if (imdbId.id === 'tt13207736' && imdbId.season === 2) {
|
|
// Monsters: The Lyle and Erik Menendez Story (2024)
|
|
return new TmdbId(225634, imdbId.season - 1, imdbId.episode);
|
|
}
|
|
if (imdbId.id === 'tt13207736' && imdbId.season === 3) {
|
|
// Monster: The Ed Gein Story (2025)
|
|
return new TmdbId(286801, imdbId.season - 2, imdbId.episode);
|
|
}
|
|
|
|
if (imdbTmdbMap.has(imdbId.id)) {
|
|
return new TmdbId(imdbTmdbMap.get(imdbId.id) as number, imdbId.season, imdbId.episode);
|
|
}
|
|
|
|
const response = await tmdbFetch(ctx, fetcher, `/find/${imdbId.id}?external_source=imdb_id`) as FindResponsePartial;
|
|
|
|
const id = (imdbId.season ? response.tv_results[0] : response.movie_results[0])?.id;
|
|
|
|
if (!id) {
|
|
throw new NotFoundError(`Could not get TMDB ID of IMDb ID "${imdbId.id}"`);
|
|
}
|
|
|
|
imdbTmdbMap.set(imdbId.id, id);
|
|
return new TmdbId(id, imdbId.season, imdbId.episode);
|
|
};
|
|
|
|
const tmdbImdbMap = new Map<number, string>();
|
|
export const getImdbIdFromTmdbId = async (ctx: Context, fetcher: Fetcher, tmdbId: TmdbId): Promise<ImdbId> => {
|
|
if (tmdbImdbMap.has(tmdbId.id)) {
|
|
return new ImdbId(tmdbImdbMap.get(tmdbId.id) as string, tmdbId.season, tmdbId.episode);
|
|
}
|
|
|
|
const type = tmdbId.season ? 'tv' : 'movie';
|
|
|
|
const response = await tmdbFetch(ctx, fetcher, `/${type}/${tmdbId.id}/external_ids`) as ExternalIdsResponsePartial;
|
|
|
|
tmdbImdbMap.set(tmdbId.id, response.imdb_id);
|
|
return new ImdbId(response.imdb_id, tmdbId.season, tmdbId.episode);
|
|
};
|
|
|
|
const getTmdbMovieDetails = async (ctx: Context, fetcher: Fetcher, tmdbId: TmdbId, language?: string): Promise<MovieDetailsResponsePartial> => {
|
|
return await tmdbFetch(ctx, fetcher, `/movie/${tmdbId.id}`, { language }) as MovieDetailsResponsePartial;
|
|
};
|
|
|
|
const getTmdbTvDetails = async (ctx: Context, fetcher: Fetcher, tmdbId: TmdbId, language?: string): Promise<TvDetailsResponsePartial> => {
|
|
return await tmdbFetch(ctx, fetcher, `/tv/${tmdbId.id}`, { language }) as TvDetailsResponsePartial;
|
|
};
|
|
|
|
export const getTmdbNameAndYear = async (ctx: Context, fetcher: Fetcher, tmdbId: TmdbId, language?: string): Promise<[string, number, string]> => {
|
|
if (tmdbId.season) {
|
|
const tmdbDetails = await getTmdbTvDetails(ctx, fetcher, tmdbId, language);
|
|
|
|
return [tmdbDetails.name, (new Date(tmdbDetails.first_air_date)).getFullYear(), tmdbDetails.original_name];
|
|
}
|
|
|
|
const tmdbDetails = await getTmdbMovieDetails(ctx, fetcher, tmdbId, language);
|
|
|
|
return [tmdbDetails.title, (new Date(tmdbDetails.release_date)).getFullYear(), tmdbDetails.original_title];
|
|
};
|