diff --git a/seasoned_api/src/tmdb/convertTmdbToSeasoned.js b/seasoned_api/src/tmdb/convertTmdbToSeasoned.js index 739e2d0..1643e8d 100644 --- a/seasoned_api/src/tmdb/convertTmdbToSeasoned.js +++ b/seasoned_api/src/tmdb/convertTmdbToSeasoned.js @@ -53,4 +53,5 @@ function convertTmdbToSeasoned(tmdbObject, strictType=undefined) { } } -module.exports = convertTmdbToSeasoned; \ No newline at end of file +module.exports = convertTmdbToSeasoned; + diff --git a/seasoned_api/src/tmdb/tmdb.js b/seasoned_api/src/tmdb/tmdb.js index 701d2d1..d876287 100644 --- a/seasoned_api/src/tmdb/tmdb.js +++ b/seasoned_api/src/tmdb/tmdb.js @@ -31,7 +31,13 @@ class TMDB { // Here we convert the filtered result from the tmdb api to seaonsed objects let seasonedItems = filteredTmdbItems.map((tmdbItem) => { - return convertTmdbToSeasoned(tmdbItem, type); + + if (type === 'movie') + return convertTmdbToSeasoned(tmdbItem, 'movie'); + else if (type === 'show') + return convertTmdbToSeasoned(tmdbItem, 'show'); + else + return convertTmdbToSeasoned(tmdbItem); }); // TODO add page number if results are larger than 20 @@ -294,4 +300,4 @@ class TMDB { } } -module.exports = TMDB; \ No newline at end of file +module.exports = TMDB;