diff --git a/seasoned_api/src/tmdb/tmdb.js b/seasoned_api/src/tmdb/tmdb.js index 3657679..0557a2f 100644 --- a/seasoned_api/src/tmdb/tmdb.js +++ b/seasoned_api/src/tmdb/tmdb.js @@ -31,7 +31,12 @@ class TMDB { // Here we convert the filtered result from the tmdb api to seaonsed objects let seasonedItems = filteredTmdbItems.map((tmdbItem) => { - return convertTmdbToSeasoned(tmdbItem); + if (type === 'movie') + return convertTmdbToSeasoned(tmdbItem, 'movie'); + else if (type === 'show') + return convertTmdbToSeaosned(tmdbItem, 'show'); + else + return convertTmdbToSeasoned(tmdbItem); }); // TODO add page number if results are larger than 20 @@ -294,4 +299,4 @@ class TMDB { } } -module.exports = TMDB; \ No newline at end of file +module.exports = TMDB;