let axios = require('axios');
let BaseClient = require('./base.client');
/**
* The Movie Database API Client.
* @extends BaseClient
*/
class MovieDBClient extends BaseClient{
/**
* @param {string} apiKey The movie db api key ([Ref]{@link https://developers.themoviedb.org/3/getting-started/introduction})
* @param {string} baseUrl Base url for movie db api
*/
constructor(apiKey, baseUrl) {
super(apiKey, baseUrl);
}
/**
* Get a movie db client instance.
* @param {string} apiKey The movie db api key
* @returns {MovieDBClient}
*/
static getInstance(apiKey) {
return new MovieDBClient(apiKey, 'https://api.themoviedb.org/3/');
}
get authParam() {
return 'api_key';
}
get requestKeyMap() {
return {
title: 'query',
query: 'query',
page: 'page',
year: 'primary_release_year'
};
}
/**
* Get a media(movie/series) by looking up using different params (e.g. title, imdb id, year etc.)
* @param {object} options The options based on which the movie will be fetched.
* @example
* get({title: 'Saw', year: 2004, type: 'movie'})
* @example
* get({imdbID: 'tt12444'})
* @returns {Promise}
* @fulfil {object} - The movie/series object.
* @reject {Error} - The error with an appropriate `message`.
*/
get(options) {
if (!options.imdbID) {
if (!options.title || !options.year || !options.type) {
throw new Error('Either one of the format of data must be provided: imdbID | (title, year, type)');
}
}
let url, responseTransformer, reqOptions = {};
if (options.imdbID) {
url = this._getURL('imdbID') + '/' + options.imdbID;
reqOptions['external_source'] = 'imdb_id';
responseTransformer = (data) => {
let nonEmptyResultsKey = Object.keys(data).filter((k) => {
return data[k].length > 0;
});
if (nonEmptyResultsKey.length > 0) {
return data[nonEmptyResultsKey[0]];
} else {
return [];
}
};
} else {
options.type = options.type || 'movie';
reqOptions = this._translateIncomingRequestOptions(options);
reqOptions['include_adult'] = false;
reqOptions['page'] = 1;
url = this._getURL(options.type);
if (options.type === 'series') {
reqOptions['first_air_date_year'] = options.year;
delete reqOptions['primary_release_year'];
}
let mediaType = options.type === 'series' ? 'tv' : 'movie';
responseTransformer = (data) => {
return this._findMatchByTitleAndYear(
data.results,
options.title,
options.year,
mediaType,
true
);
};
}
return this._makeHTTPGET(
url,
reqOptions,
null,
responseTransformer
).then((resp) => {
if (resp.data.length) {
// fetch the details
return this.getDetails(resp.data[0].id, this._guessMediaType(resp.data[0]));
} else {
return {};
}
});
}
/**
* Search for a movie/series.
* @param {object} options The options based on which the search will be conducted.
* @example
* search({query: 'Saw', year: 2004, type: 'movie'})
* @returns {Promise}
* @fulfil {object} - The movie/series object.
* @reject {Error} - The error with an appropriate `message`.
*/
search(options) {
if (!options.query) {
throw new Error('Query must be given');
}
options.type = options.type || 'movie';
let reqOptions = this._translateIncomingRequestOptions(options);
reqOptions['include_adult'] = false;
let url = this._getURL('search');
let responseTransformer = (data) => {
return {
currentPage: options.page,
totalPages: data['total_pages'],
numFound: data['total_results'],
results: data.results.map((result) => {
return this._translateResponseObject(result, result['media_type'])
})
};
};
return this._makeHTTPGET(
url,
reqOptions,
null,
responseTransformer
).then((resp) => {
return resp.data;
});
}
/**
* Get a media(movie/series) by title and year
* @param {string} title Title of the media (movie/series)
* @param {number} year The year media type(movie/series) was released (Optional).
* @param {string} [type='movie'] Type of the media; movie/series (Optional).
* @returns {Promise}
* @fulfil {object} - The movie/series object.
* @reject {Error} - The error with an appropriate `message`.
*/
getByTitleAndYear(title, year, type = 'movie') {
return this.get({title: title, year: year, type: type});
}
/**
* Get a media(movie/series) by IMDB ID
* @param {string} imdbID IMDB ID of the media
* @returns {Promise}
* @fulfil {object} - The movie/series object.
* @reject {Error} - The error with an appropriate `message`.
*/
getByIMDBId(imdbID) {
return this.get({imdbID: imdbID});
}
/**
* Get details of a movie or series with given tmdb id
* @param {string} tmdbID The Movie DB ID of the media
* @param {string} type The media type; movie/series
* @returns {Promise}
* @fulfil {object} - The movie/series object.
* @reject {Error} - The error with an appropriate `message`.
*/
getDetails(tmdbID, type) {
let mediaType = type === 'series' ? 'tv' : 'movie';
let detailsUrl = this._getURL(type + 'Details') + '/' + tmdbID;
return this._makeHTTPGET(detailsUrl, null, null, null)
.then((resp) => {
return this._translateResponseObject(resp.data, mediaType);
});
}
_getURL(identifier) {
return {
movie: 'search/movie',
series: 'search/tv',
search: 'search/multi',
movieDetails: 'movie',
seriesDetails: 'tv',
genreList: 'genre/movie/list',
imdbID: 'find'
}[identifier];
}
_findMatchByTitleAndYear(movieResults, qTitle, qYear, mediaType, exact=false) {
return movieResults.filter((movieResult) => {
let titleKey = mediaType === 'tv' ? 'name' : 'title';
let dateKey = mediaType === 'tv' ? 'first_air_date' : 'release_date';
let title = movieResult[titleKey];
let year = movieResult[dateKey].match(/(\d{4})-\d{2}-\d{2}/)[1];
let normalizedTitle = title.toLowerCase().replace(/_|-|\.|:/, ' ');
let normalizedQueryTitle = qTitle.toLowerCase();
return exact ? (normalizedQueryTitle === normalizedTitle && year === qYear)
: (normalizedTitle.includes(normalizedQueryTitle) || (qYear && year == qYear));
});
}
_guessMediaType(response) {
return response.name ? 'series' : (response.title ? 'movie' : '');
}
_translateResponseObject(responseObj, mediaType) {
let response = {};
let titleKey = mediaType === 'tv' ? 'name' : 'title';
let dateKey = mediaType === 'tv' ? 'first_air_date' : 'release_date';
response['id'] = responseObj['id'];
response['imdbID'] = responseObj['imdb_id'];
response['title'] = responseObj[titleKey];
response['year'] = responseObj[dateKey].match(/(\d{4})-\d{2}-\d{2}/)[1];
response['summary'] = responseObj['overview'];
response['poster'] = 'http://image.tmdb.org/t/p/original' + responseObj['poster_path'];
response['genres'] = responseObj['genres'] || responseObj['genre_ids'];
response['rating'] = responseObj['vote_average'];
return response;
}
}
module.exports = MovieDBClient;