From f1d07f74ee05587dc7baf7d55a6504cea18cb273 Mon Sep 17 00:00:00 2001
From: Daniel Martin Gonzalez <danimart1991@outlook.com>
Date: Sun, 27 Feb 2022 10:18:13 +0100
Subject: [PATCH] New: Option to show release group column on series list

---
 .../Series/Index/Table/SeriesIndexHeader.css  |  1 +
 .../src/Series/Index/Table/SeriesIndexRow.css |  1 +
 .../src/Series/Index/Table/SeriesIndexRow.js  | 22 +++++++++++++++-
 frontend/src/Store/Actions/seriesActions.js   | 17 ++++++++++++
 .../src/Store/Actions/seriesIndexActions.js   |  6 +++++
 .../SeriesStats/SeasonStatistics.cs           | 26 ++++++++++++++++++-
 .../SeriesStats/SeriesStatistics.cs           |  1 +
 .../SeriesStats/SeriesStatisticsRepository.cs |  2 +-
 .../SeriesStats/SeriesStatisticsService.cs    |  3 ++-
 .../Series/SeasonStatisticsResource.cs        |  5 +++-
 .../Series/SeriesStatisticsResource.cs        |  4 ++-
 11 files changed, 82 insertions(+), 6 deletions(-)

diff --git a/frontend/src/Series/Index/Table/SeriesIndexHeader.css b/frontend/src/Series/Index/Table/SeriesIndexHeader.css
index 6781ef951..ce9f0f6ed 100644
--- a/frontend/src/Series/Index/Table/SeriesIndexHeader.css
+++ b/frontend/src/Series/Index/Table/SeriesIndexHeader.css
@@ -37,6 +37,7 @@
   flex: 1 0 125px;
 }
 
+.releaseGroups,
 .nextAiring,
 .previousAiring,
 .added,
diff --git a/frontend/src/Series/Index/Table/SeriesIndexRow.css b/frontend/src/Series/Index/Table/SeriesIndexRow.css
index c67a02479..df4b4b9ab 100644
--- a/frontend/src/Series/Index/Table/SeriesIndexRow.css
+++ b/frontend/src/Series/Index/Table/SeriesIndexRow.css
@@ -73,6 +73,7 @@
   flex: 1 0 125px;
 }
 
+.releaseGroups,
 .nextAiring,
 .previousAiring,
 .added,
diff --git a/frontend/src/Series/Index/Table/SeriesIndexRow.js b/frontend/src/Series/Index/Table/SeriesIndexRow.js
index e6161ec9f..2732030ed 100644
--- a/frontend/src/Series/Index/Table/SeriesIndexRow.js
+++ b/frontend/src/Series/Index/Table/SeriesIndexRow.js
@@ -114,6 +114,7 @@ class SeriesIndexRow extends Component {
       episodeCount,
       episodeFileCount,
       totalEpisodeCount,
+      releaseGroups,
       sizeOnDisk
     } = statistics;
 
@@ -413,6 +414,24 @@ class SeriesIndexRow extends Component {
               );
             }
 
+            if (name === 'releaseGroups') {
+              const joinedReleaseGroups = releaseGroups.join(', ');
+              const truncatedReleaseGroups = releaseGroups.length > 3 ?
+                `${releaseGroups.slice(0, 3).join(', ')}...` :
+                joinedReleaseGroups;
+
+              return (
+                <VirtualTableRowCell
+                  key={name}
+                  className={styles[name]}
+                >
+                  <span title={joinedReleaseGroups}>
+                    {truncatedReleaseGroups}
+                  </span>
+                </VirtualTableRowCell>
+              );
+            }
+
             if (name === 'tags') {
               return (
                 <VirtualTableRowCell
@@ -534,7 +553,8 @@ SeriesIndexRow.defaultProps = {
     seasonCount: 0,
     episodeCount: 0,
     episodeFileCount: 0,
-    totalEpisodeCount: 0
+    totalEpisodeCount: 0,
+    releaseGroups: []
   },
   genres: [],
   tags: []
diff --git a/frontend/src/Store/Actions/seriesActions.js b/frontend/src/Store/Actions/seriesActions.js
index 5b66e1472..62660c9f5 100644
--- a/frontend/src/Store/Actions/seriesActions.js
+++ b/frontend/src/Store/Actions/seriesActions.js
@@ -131,6 +131,18 @@ export const filterPredicates = {
     return predicate(item.ratings.value * 10, filterValue);
   },
 
+  releaseGroups: function(item, filterValue, type) {
+    const { statistics = {} } = item;
+
+    const {
+      releaseGroups = []
+    } = statistics;
+
+    const predicate = filterTypePredicates[type];
+
+    return predicate(releaseGroups, filterValue);
+  },
+
   seasonCount: function(item, filterValue, type) {
     const predicate = filterTypePredicates[type];
     const seasonCount = item.statistics ? item.statistics.seasonCount : 0;
@@ -261,6 +273,11 @@ export const filterBuilderProps = [
       return tagList.sort(sortByName);
     }
   },
+  {
+    name: 'releaseGroups',
+    label: 'Release Groups',
+    type: filterBuilderTypes.ARRAY
+  },
   {
     name: 'ratings',
     label: 'Rating',
diff --git a/frontend/src/Store/Actions/seriesIndexActions.js b/frontend/src/Store/Actions/seriesIndexActions.js
index ffdab4bfa..a56e07e09 100644
--- a/frontend/src/Store/Actions/seriesIndexActions.js
+++ b/frontend/src/Store/Actions/seriesIndexActions.js
@@ -167,6 +167,12 @@ export const defaultState = {
       isSortable: false,
       isVisible: false
     },
+    {
+      name: 'releaseGroups',
+      label: 'Release Groups',
+      isSortable: false,
+      isVisible: false
+    },
     {
       name: 'tags',
       label: 'Tags',
diff --git a/src/NzbDrone.Core/SeriesStats/SeasonStatistics.cs b/src/NzbDrone.Core/SeriesStats/SeasonStatistics.cs
index 9542b0d0a..1b6c5c001 100644
--- a/src/NzbDrone.Core/SeriesStats/SeasonStatistics.cs
+++ b/src/NzbDrone.Core/SeriesStats/SeasonStatistics.cs
@@ -1,5 +1,8 @@
-using System;
+using NzbDrone.Common.Extensions;
 using NzbDrone.Core.Datastore;
+using System;
+using System.Collections.Generic;
+using System.Linq;
 
 namespace NzbDrone.Core.SeriesStats
 {
@@ -14,6 +17,7 @@ namespace NzbDrone.Core.SeriesStats
         public int AvailableEpisodeCount { get; set; }
         public int TotalEpisodeCount { get; set; }
         public long SizeOnDisk { get; set; }
+        public string ReleaseGroupsString { get; set; }
 
         public DateTime? NextAiring
         {
@@ -54,5 +58,25 @@ namespace NzbDrone.Core.SeriesStats
                 return previousAiring;
             }
         }
+
+        public List<string> ReleaseGroups
+        {
+            get
+            {
+                var releasegroups = new List<string>();
+
+                if (ReleaseGroupsString.IsNotNullOrWhiteSpace())
+                {
+                    releasegroups = ReleaseGroupsString
+                        .Split('|')
+                        .Distinct()
+                        .Where(rg => rg.IsNotNullOrWhiteSpace())
+                        .OrderBy(rg => rg)
+                        .ToList();
+                }
+
+                return releasegroups;
+            }
+        }
     }
 }
diff --git a/src/NzbDrone.Core/SeriesStats/SeriesStatistics.cs b/src/NzbDrone.Core/SeriesStats/SeriesStatistics.cs
index 363d1f346..44f477c0b 100644
--- a/src/NzbDrone.Core/SeriesStats/SeriesStatistics.cs
+++ b/src/NzbDrone.Core/SeriesStats/SeriesStatistics.cs
@@ -13,6 +13,7 @@ namespace NzbDrone.Core.SeriesStats
         public int EpisodeCount { get; set; }
         public int TotalEpisodeCount { get; set; }
         public long SizeOnDisk { get; set; }
+        public List<string> ReleaseGroups { get; set; }
         public List<SeasonStatistics> SeasonStatistics { get; set; }
 
         public DateTime? NextAiring
diff --git a/src/NzbDrone.Core/SeriesStats/SeriesStatisticsRepository.cs b/src/NzbDrone.Core/SeriesStats/SeriesStatisticsRepository.cs
index 699531198..469dfd2f0 100644
--- a/src/NzbDrone.Core/SeriesStats/SeriesStatisticsRepository.cs
+++ b/src/NzbDrone.Core/SeriesStats/SeriesStatisticsRepository.cs
@@ -53,7 +53,7 @@ namespace NzbDrone.Core.SeriesStats
 
         private string GetSelectClause(bool series)
         {
-            return @"SELECT Episodes.*, SUM(EpisodeFiles.Size) as SizeOnDisk FROM
+            return @"SELECT Episodes.*, SUM(EpisodeFiles.Size) as SizeOnDisk, GROUP_CONCAT(EpisodeFiles.ReleaseGroup, '|') AS ReleaseGroupsString FROM
                      (SELECT
                      Episodes.SeriesId,
                      Episodes.SeasonNumber,
diff --git a/src/NzbDrone.Core/SeriesStats/SeriesStatisticsService.cs b/src/NzbDrone.Core/SeriesStats/SeriesStatisticsService.cs
index b273f84ce..2c1900f9a 100644
--- a/src/NzbDrone.Core/SeriesStats/SeriesStatisticsService.cs
+++ b/src/NzbDrone.Core/SeriesStats/SeriesStatisticsService.cs
@@ -43,7 +43,8 @@ namespace NzbDrone.Core.SeriesStats
                                        EpisodeFileCount = seasonStatistics.Sum(s => s.EpisodeFileCount),
                                        EpisodeCount = seasonStatistics.Sum(s => s.EpisodeCount),
                                        TotalEpisodeCount = seasonStatistics.Sum(s => s.TotalEpisodeCount),
-                                       SizeOnDisk = seasonStatistics.Sum(s => s.SizeOnDisk)
+                                       SizeOnDisk = seasonStatistics.Sum(s => s.SizeOnDisk),
+                                       ReleaseGroups = seasonStatistics.SelectMany(s => s.ReleaseGroups).Distinct().ToList()
                                    };
 
             var nextAiring = seasonStatistics.Where(s => s.NextAiring != null)
diff --git a/src/Sonarr.Api.V3/Series/SeasonStatisticsResource.cs b/src/Sonarr.Api.V3/Series/SeasonStatisticsResource.cs
index 12ee9434f..e482cd12a 100644
--- a/src/Sonarr.Api.V3/Series/SeasonStatisticsResource.cs
+++ b/src/Sonarr.Api.V3/Series/SeasonStatisticsResource.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using NzbDrone.Core.SeriesStats;
 
 namespace Sonarr.Api.V3.Series
@@ -11,6 +12,7 @@ namespace Sonarr.Api.V3.Series
         public int EpisodeCount { get; set; }
         public int TotalEpisodeCount { get; set; }
         public long SizeOnDisk { get; set; }
+        public List<string> ReleaseGroups { get; set; }
 
         public decimal PercentOfEpisodes
         {
@@ -36,7 +38,8 @@ namespace Sonarr.Api.V3.Series
                 EpisodeFileCount = model.EpisodeFileCount,
                 EpisodeCount = model.EpisodeCount,
                 TotalEpisodeCount = model.TotalEpisodeCount,
-                SizeOnDisk = model.SizeOnDisk
+                SizeOnDisk = model.SizeOnDisk,
+                ReleaseGroups = model.ReleaseGroups
             };
         }
     }
diff --git a/src/Sonarr.Api.V3/Series/SeriesStatisticsResource.cs b/src/Sonarr.Api.V3/Series/SeriesStatisticsResource.cs
index a2c797b95..dcdecbdf2 100644
--- a/src/Sonarr.Api.V3/Series/SeriesStatisticsResource.cs
+++ b/src/Sonarr.Api.V3/Series/SeriesStatisticsResource.cs
@@ -11,6 +11,7 @@ namespace Sonarr.Api.V3.Series
         public int EpisodeCount { get; set; }
         public int TotalEpisodeCount { get; set; }
         public long SizeOnDisk { get; set; }
+        public List<string> ReleaseGroups { get; set; }
 
         public decimal PercentOfEpisodes
         {
@@ -37,7 +38,8 @@ namespace Sonarr.Api.V3.Series
                 EpisodeFileCount = model.EpisodeFileCount,
                 EpisodeCount = model.EpisodeCount,
                 TotalEpisodeCount = model.TotalEpisodeCount,
-                SizeOnDisk = model.SizeOnDisk
+                SizeOnDisk = model.SizeOnDisk,
+                ReleaseGroups = model.ReleaseGroups
             };
         }
     }