From 0be449033f271b81a2d3314d1d7ee110ac9727e4 Mon Sep 17 00:00:00 2001
From: Qstick <qstick@gmail.com>
Date: Sat, 27 Jan 2024 20:24:56 -0600
Subject: [PATCH] New: Trending and Popular Movies in Discovery

---
 frontend/src/Components/ImportListList.js     |  2 +-
 .../Overview/DiscoverMovieOverview.js         | 24 ++++++++++
 .../Table/DiscoverMovieHeader.css             |  4 +-
 .../Table/DiscoverMovieHeader.css.d.ts        |  2 +
 .../Table/DiscoverMovieHeader.js              | 34 +++++++++++++
 .../DiscoverMovie/Table/DiscoverMovieRow.css  |  9 +++-
 .../Table/DiscoverMovieRow.css.d.ts           |  3 ++
 .../DiscoverMovie/Table/DiscoverMovieRow.js   | 45 +++++++++++++++++
 frontend/src/Helpers/Props/icons.js           |  2 +
 .../src/Store/Actions/discoverMovieActions.js | 48 +++++++++++++++++++
 src/NzbDrone.Core/Localization/Core/en.json   |  2 +
 .../MetadataSource/IProvideMovieInfo.cs       |  2 +
 .../MetadataSource/SkyHook/SkyHookProxy.cs    | 28 +++++++++++
 .../ImportLists/ImportListMoviesController.cs | 19 +++++++-
 .../ImportLists/ImportListMoviesResource.cs   |  2 +
 15 files changed, 221 insertions(+), 5 deletions(-)

diff --git a/frontend/src/Components/ImportListList.js b/frontend/src/Components/ImportListList.js
index 7695b26d3a..c7fb393f51 100644
--- a/frontend/src/Components/ImportListList.js
+++ b/frontend/src/Components/ImportListList.js
@@ -19,7 +19,7 @@ function ImportListList({ lists, importListList }) {
           return (
             <Label
               key={list.id}
-              kind={kinds.INFO}
+              kind={kinds.SUCCESS}
               size={sizes.MEDIUM}
             >
               {list.name}
diff --git a/frontend/src/DiscoverMovie/Overview/DiscoverMovieOverview.js b/frontend/src/DiscoverMovie/Overview/DiscoverMovieOverview.js
index 5da6c840bb..dbc2714161 100644
--- a/frontend/src/DiscoverMovie/Overview/DiscoverMovieOverview.js
+++ b/frontend/src/DiscoverMovie/Overview/DiscoverMovieOverview.js
@@ -97,6 +97,8 @@ class DiscoverMovieOverview extends Component {
       isExisting,
       isExcluded,
       isRecommendation,
+      isPopular,
+      isTrending,
       isSelected,
       overviewOptions,
       ...otherProps
@@ -214,6 +216,26 @@ class DiscoverMovieOverview extends Component {
                   null
               }
 
+              {
+                isPopular ?
+                  <Label
+                    kind={kinds.INFO}
+                  >
+                    {translate('Popular')}
+                  </Label> :
+                  null
+              }
+
+              {
+                isTrending ?
+                  <Label
+                    kind={kinds.INFO}
+                  >
+                    {translate('Trending')}
+                  </Label> :
+                  null
+              }
+
               <ImportListListConnector
                 lists={lists}
               />
@@ -283,6 +305,8 @@ DiscoverMovieOverview.propTypes = {
   isExisting: PropTypes.bool.isRequired,
   isExcluded: PropTypes.bool.isRequired,
   isRecommendation: PropTypes.bool.isRequired,
+  isPopular: PropTypes.bool.isRequired,
+  isTrending: PropTypes.bool.isRequired,
   isSelected: PropTypes.bool,
   lists: PropTypes.arrayOf(PropTypes.number).isRequired,
   onSelectedChange: PropTypes.func.isRequired
diff --git a/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.css b/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.css
index c543d231d1..3afdebe475 100644
--- a/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.css
+++ b/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.css
@@ -57,10 +57,12 @@
   flex: 0 0 115px;
 }
 
+.isTrending,
+.isPopular,
 .isRecommendation {
   composes: headerCell from '~Components/Table/VirtualTableHeaderCell.css';
 
-  flex: 0 0 50px;
+  flex: 0 0 30px;
 }
 
 .actions {
diff --git a/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.css.d.ts b/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.css.d.ts
index fd8bbf0f89..254a1cfb50 100644
--- a/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.css.d.ts
+++ b/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.css.d.ts
@@ -7,7 +7,9 @@ interface CssExports {
   'digitalRelease': string;
   'genres': string;
   'inCinemas': string;
+  'isPopular': string;
   'isRecommendation': string;
+  'isTrending': string;
   'lists': string;
   'originalLanguage': string;
   'physicalRelease': string;
diff --git a/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.js b/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.js
index 8b5a5298fc..fdde789b0c 100644
--- a/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.js
+++ b/frontend/src/DiscoverMovie/Table/DiscoverMovieHeader.js
@@ -103,6 +103,40 @@ class DiscoverMovieHeader extends Component {
               );
             }
 
+            if (name === 'isTrending') {
+              return (
+                <VirtualTableHeaderCell
+                  key={name}
+                  className={styles[name]}
+                  name={name}
+                  isSortable={true}
+                  {...otherProps}
+                >
+                  <Icon
+                    name={icons.TRENDING}
+                    size={12}
+                  />
+                </VirtualTableHeaderCell>
+              );
+            }
+
+            if (name === 'isPopular') {
+              return (
+                <VirtualTableHeaderCell
+                  key={name}
+                  className={styles[name]}
+                  name={name}
+                  isSortable={true}
+                  {...otherProps}
+                >
+                  <Icon
+                    name={icons.POPULAR}
+                    size={12}
+                  />
+                </VirtualTableHeaderCell>
+              );
+            }
+
             return (
               <VirtualTableHeaderCell
                 key={name}
diff --git a/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.css b/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.css
index 3e2ebd2e79..f41b5e4ccd 100644
--- a/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.css
+++ b/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.css
@@ -76,10 +76,12 @@
   flex: 1 0 110px;
 }
 
+.isTrending,
+.isPopular,
 .isRecommendation {
   composes: cell;
 
-  flex: 0 0 50px;
+  flex: 0 0 30px;
 }
 
 .actions {
@@ -95,6 +97,11 @@
   margin-top: 0;
 }
 
+.statusIcon {
+  width: 20px !important;
+  text-align: center;
+}
+
 .externalLinks {
   margin-right: 0.5em;
 }
diff --git a/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.css.d.ts b/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.css.d.ts
index 8b7ec2014e..7ecc09cedb 100644
--- a/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.css.d.ts
+++ b/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.css.d.ts
@@ -12,7 +12,9 @@ interface CssExports {
   'externalLinks': string;
   'genres': string;
   'inCinemas': string;
+  'isPopular': string;
   'isRecommendation': string;
+  'isTrending': string;
   'lists': string;
   'originalLanguage': string;
   'physicalRelease': string;
@@ -21,6 +23,7 @@ interface CssExports {
   'runtime': string;
   'sortTitle': string;
   'status': string;
+  'statusIcon': string;
   'studio': string;
 }
 export const cssExports: CssExports;
diff --git a/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.js b/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.js
index 9eba818500..6541ec2389 100644
--- a/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.js
+++ b/frontend/src/DiscoverMovie/Table/DiscoverMovieRow.js
@@ -82,6 +82,8 @@ class DiscoverMovieRow extends Component {
       isExisting,
       isExcluded,
       isRecommendation,
+      isTrending,
+      isPopular,
       isSelected,
       lists,
       onSelectedChange
@@ -305,6 +307,7 @@ class DiscoverMovieRow extends Component {
                   {
                     isRecommendation ?
                       <Icon
+                        className={styles.statusIcon}
                         name={icons.RECOMMENDED}
                         size={12}
                         title={translate('MovieIsRecommend')}
@@ -315,6 +318,46 @@ class DiscoverMovieRow extends Component {
               );
             }
 
+            if (name === 'isTrending') {
+              return (
+                <VirtualTableRowCell
+                  key={name}
+                  className={styles[name]}
+                >
+                  {
+                    isTrending ?
+                      <Icon
+                        className={styles.statusIcon}
+                        name={icons.TRENDING}
+                        size={12}
+                        title={translate('MovieIsTrending')}
+                      /> :
+                      null
+                  }
+                </VirtualTableRowCell>
+              );
+            }
+
+            if (name === 'isPopular') {
+              return (
+                <VirtualTableRowCell
+                  key={name}
+                  className={styles[name]}
+                >
+                  {
+                    isPopular ?
+                      <Icon
+                        className={styles.statusIcon}
+                        name={icons.POPULAR}
+                        size={12}
+                        title={translate('MovieIsPopular')}
+                      /> :
+                      null
+                  }
+                </VirtualTableRowCell>
+              );
+            }
+
             if (name === 'actions') {
               return (
                 <VirtualTableRowCell
@@ -404,6 +447,8 @@ DiscoverMovieRow.propTypes = {
   isExcluded: PropTypes.bool.isRequired,
   isSelected: PropTypes.bool,
   isRecommendation: PropTypes.bool.isRequired,
+  isPopular: PropTypes.bool.isRequired,
+  isTrending: PropTypes.bool.isRequired,
   lists: PropTypes.arrayOf(PropTypes.number).isRequired,
   onSelectedChange: PropTypes.func.isRequired
 };
diff --git a/frontend/src/Helpers/Props/icons.js b/frontend/src/Helpers/Props/icons.js
index d404e8db34..81db4866b0 100644
--- a/frontend/src/Helpers/Props/icons.js
+++ b/frontend/src/Helpers/Props/icons.js
@@ -23,6 +23,7 @@ import {
 import {
   faArrowCircleLeft as fasArrowCircleLeft,
   faArrowCircleRight as fasArrowCircleRight,
+  faArrowTrendUp as fasArrowTrendUp,
   faAsterisk as fasAsterisk,
   faBackward as fasBackward,
   faBan as fasBan,
@@ -233,6 +234,7 @@ export const TAGS = fasTags;
 export const TBA = fasQuestionCircle;
 export const TEST = fasVial;
 export const TRANSLATE = fasLanguage;
+export const TRENDING = fasArrowTrendUp;
 export const UNGROUP = farObjectUngroup;
 export const UNKNOWN = fasQuestion;
 export const UNMONITORED = farBookmark;
diff --git a/frontend/src/Store/Actions/discoverMovieActions.js b/frontend/src/Store/Actions/discoverMovieActions.js
index 9d0921a8ee..92184a9fcb 100644
--- a/frontend/src/Store/Actions/discoverMovieActions.js
+++ b/frontend/src/Store/Actions/discoverMovieActions.js
@@ -88,6 +88,20 @@ export const defaultState = {
       isVisible: true,
       isModifiable: false
     },
+    {
+      name: 'isTrending',
+      columnLabel: 'Trending',
+      isSortable: true,
+      isVisible: true,
+      isModifiable: false
+    },
+    {
+      name: 'isPopular',
+      columnLabel: 'Popular',
+      isSortable: true,
+      isVisible: true,
+      isModifiable: false
+    },
     {
       name: 'sortTitle',
       label: () => translate('MovieTitle'),
@@ -267,6 +281,28 @@ export const defaultState = {
       label: () => translate('All'),
       filters: []
     },
+    {
+      key: 'popular',
+      label: 'Popular',
+      filters: [
+        {
+          key: 'isPopular',
+          value: true,
+          type: filterTypes.EQUAL
+        }
+      ]
+    },
+    {
+      key: 'trending',
+      label: 'Trending',
+      filters: [
+        {
+          key: 'isTrending',
+          value: true,
+          type: filterTypes.EQUAL
+        }
+      ]
+    },
     {
       key: 'newNotExcluded',
       label: 'New Non-Excluded',
@@ -456,6 +492,18 @@ export const defaultState = {
       label: 'Recommended',
       type: filterBuilderTypes.EXACT,
       valueType: filterBuilderValueTypes.BOOL
+    },
+    {
+      name: 'isTrending',
+      label: 'Trending',
+      type: filterBuilderTypes.EXACT,
+      valueType: filterBuilderValueTypes.BOOL
+    },
+    {
+      name: 'isPopular',
+      label: 'Popular',
+      type: filterBuilderTypes.EXACT,
+      valueType: filterBuilderValueTypes.BOOL
     }
   ]
 };
diff --git a/src/NzbDrone.Core/Localization/Core/en.json b/src/NzbDrone.Core/Localization/Core/en.json
index 32e84a4a2d..11a34da918 100644
--- a/src/NzbDrone.Core/Localization/Core/en.json
+++ b/src/NzbDrone.Core/Localization/Core/en.json
@@ -877,7 +877,9 @@
   "MovieIsDownloading": "Movie is downloading",
   "MovieIsMonitored": "Movie is monitored",
   "MovieIsOnImportExclusionList": "Movie is on Import Exclusion List",
+  "MovieIsPopular": "Movie is Popular on TMDb",
   "MovieIsRecommend": "Movie is recommended based on recent addition",
+  "MovieIsTrending": "Movie is Trending on TMDb",
   "MovieIsUnmonitored": "Movie is unmonitored",
   "MovieMatchType": "Movie Match Type",
   "MovieNaming": "Movie Naming",
diff --git a/src/NzbDrone.Core/MetadataSource/IProvideMovieInfo.cs b/src/NzbDrone.Core/MetadataSource/IProvideMovieInfo.cs
index b2323f7b63..da17aad04c 100644
--- a/src/NzbDrone.Core/MetadataSource/IProvideMovieInfo.cs
+++ b/src/NzbDrone.Core/MetadataSource/IProvideMovieInfo.cs
@@ -12,6 +12,8 @@ public interface IProvideMovieInfo
         Tuple<MovieMetadata, List<Credit>> GetMovieInfo(int tmdbId);
         MovieCollection GetCollectionInfo(int tmdbId);
         List<MovieMetadata> GetBulkMovieInfo(List<int> tmdbIds);
+        List<MovieMetadata> GetTrendingMovies();
+        List<MovieMetadata> GetPopularMovies();
 
         HashSet<int> GetChangedMovies(DateTime startTime);
     }
diff --git a/src/NzbDrone.Core/MetadataSource/SkyHook/SkyHookProxy.cs b/src/NzbDrone.Core/MetadataSource/SkyHook/SkyHookProxy.cs
index d27823c079..49ac659eaa 100644
--- a/src/NzbDrone.Core/MetadataSource/SkyHook/SkyHookProxy.cs
+++ b/src/NzbDrone.Core/MetadataSource/SkyHook/SkyHookProxy.cs
@@ -70,6 +70,34 @@ public HashSet<int> GetChangedMovies(DateTime startTime)
             return new HashSet<int>(response.Resource);
         }
 
+        public List<MovieMetadata> GetTrendingMovies()
+        {
+            var request = _radarrMetadata.Create()
+                .SetSegment("route", "list/tmdb/trending")
+                .Build();
+
+            request.AllowAutoRedirect = true;
+            request.SuppressHttpError = true;
+
+            var response = _httpClient.Get<List<MovieResource>>(request);
+
+            return response.Resource.DistinctBy(x => x.TmdbId).Select(MapMovie).ToList();
+        }
+
+        public List<MovieMetadata> GetPopularMovies()
+        {
+            var request = _radarrMetadata.Create()
+                .SetSegment("route", "list/tmdb/popular")
+                .Build();
+
+            request.AllowAutoRedirect = true;
+            request.SuppressHttpError = true;
+
+            var response = _httpClient.Get<List<MovieResource>>(request);
+
+            return response.Resource.DistinctBy(x => x.TmdbId).Select(MapMovie).ToList();
+        }
+
         public Tuple<MovieMetadata, List<Credit>> GetMovieInfo(int tmdbId)
         {
             var httpRequest = _radarrMetadata.Create()
diff --git a/src/Radarr.Api.V3/ImportLists/ImportListMoviesController.cs b/src/Radarr.Api.V3/ImportLists/ImportListMoviesController.cs
index 42188bdf89..678892d783 100644
--- a/src/Radarr.Api.V3/ImportLists/ImportListMoviesController.cs
+++ b/src/Radarr.Api.V3/ImportLists/ImportListMoviesController.cs
@@ -54,7 +54,7 @@ public ImportListMoviesController(IMovieService movieService,
         }
 
         [HttpGet]
-        public object GetDiscoverMovies(bool includeRecommendations = false)
+        public object GetDiscoverMovies(bool includeRecommendations = false, bool includeTrending = false, bool includePopular = false)
         {
             var movieLanguage = (Language)_configService.MovieInfoLanguage;
 
@@ -77,6 +77,17 @@ public object GetDiscoverMovies(bool includeRecommendations = false)
                 realResults.ForEach(x => x.IsRecommendation = true);
             }
 
+            // Add TMDB Trending
+            var trendingResults = _movieInfo.GetTrendingMovies();
+
+            realResults.AddRange(MapToResource(trendingResults.Select(m => new Movie { MovieMetadata = m }).Where(x => x != null), movieLanguage, true));
+
+            // Add TMDB Popular
+            var popularResults = _movieInfo.GetPopularMovies();
+
+            realResults.AddRange(MapToResource(popularResults.Select(m => new Movie { MovieMetadata = m }).Where(x => x != null), movieLanguage, false, true));
+
+            // Add List Movies
             var listMovies = MapToResource(_listMovieService.GetAllForLists(_importListFactory.Enabled().Select(x => x.Definition.Id).ToList()), movieLanguage).ToList();
 
             realResults.AddRange(listMovies);
@@ -92,6 +103,8 @@ public object GetDiscoverMovies(bool includeRecommendations = false)
                 movie.IsExcluded = listExclusions.Any(e => e.TmdbId == movie.TmdbId);
                 movie.IsExisting = existingTmdbIds.Any(e => e == movie.TmdbId);
                 movie.IsRecommendation = x.Any(m => m.IsRecommendation);
+                movie.IsPopular = x.Any(m => m.IsPopular);
+                movie.IsTrending = x.Any(m => m.IsTrending);
 
                 return movie;
             }).ToList();
@@ -107,7 +120,7 @@ public object AddMovies([FromBody] List<MovieResource> resource)
             return _addMovieService.AddMovies(newMovies, true).ToResource(0);
         }
 
-        private IEnumerable<ImportListMoviesResource> MapToResource(IEnumerable<Movie> movies, Language language)
+        private IEnumerable<ImportListMoviesResource> MapToResource(IEnumerable<Movie> movies, Language language, bool isTrending = false, bool isPopular = false)
         {
             // Avoid calling for naming spec on every movie in filenamebuilder
             var namingConfig = _namingService.GetConfig();
@@ -127,6 +140,8 @@ private IEnumerable<ImportListMoviesResource> MapToResource(IEnumerable<Movie> m
                 resource.Title = translation?.Title ?? resource.Title;
                 resource.Overview = translation?.Overview ?? resource.Overview;
                 resource.Folder = _fileNameBuilder.GetMovieFolder(currentMovie, namingConfig);
+                resource.IsTrending = isTrending;
+                resource.IsPopular = isPopular;
 
                 yield return resource;
             }
diff --git a/src/Radarr.Api.V3/ImportLists/ImportListMoviesResource.cs b/src/Radarr.Api.V3/ImportLists/ImportListMoviesResource.cs
index c90f8dafd6..4aefbfed25 100644
--- a/src/Radarr.Api.V3/ImportLists/ImportListMoviesResource.cs
+++ b/src/Radarr.Api.V3/ImportLists/ImportListMoviesResource.cs
@@ -42,6 +42,8 @@ public ImportListMoviesResource()
         public MovieCollection Collection { get; set; }
         public bool IsExcluded { get; set; }
         public bool IsExisting { get; set; }
+        public bool IsTrending { get; set; }
+        public bool IsPopular { get; set; }
 
         public bool IsRecommendation { get; set; }
         public HashSet<int> Lists { get; set; }