feat: Add rating vote count filters

jorenn92 · Jan 3, 2025 · 05617e1 · 05617e1
1 parent 777f206
commit 05617e1
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 0 deletions.
diff --git a/server/src/modules/rules/constants/rules.constants.ts b/server/src/modules/rules/constants/rules.constants.ts
@@ -443,20 +443,41 @@ export class RuleConstants {
           mediaType: MediaType.MOVIE,
           type: RuleType.NUMBER,
         },
+        {
+          id: 17,
+          name: 'rottenTomatoesRatingVotes',
+          humanName: 'Rotten Tomatoes rating vote count',
+          mediaType: MediaType.MOVIE,
+          type: RuleType.NUMBER,
+        },
         {
           id: 15,
           name: 'traktRating',
           humanName: 'Trakt rating (scale 0-10)',
           mediaType: MediaType.MOVIE,
           type: RuleType.NUMBER,
         },
+        {
+          id: 18,
+          name: 'traktRatingVotes',
+          humanName: 'Trakt rating vote count',
+          mediaType: MediaType.MOVIE,
+          type: RuleType.NUMBER,
+        },
         {
           id: 16,
           name: 'imdbRating',
           humanName: 'IMDb rating (scale 0-10)',
           mediaType: MediaType.MOVIE,
           type: RuleType.NUMBER,
         },
+        {
+          id: 19,
+          name: 'imdbRatingVotes',
+          humanName: 'IMDb rating vote count',
+          mediaType: MediaType.MOVIE,
+          type: RuleType.NUMBER,
+        },
       ],
     },
     {
@@ -620,6 +641,13 @@ export class RuleConstants {
           mediaType: MediaType.SHOW,
           type: RuleType.NUMBER,
         },
+        {
+          id: 20,
+          name: 'ratingVotes',
+          humanName: 'Show rating (IMDb) vote count',
+          mediaType: MediaType.SHOW,
+          type: RuleType.NUMBER,
+        },
       ],
     },
     {

diff --git a/server/src/modules/rules/getter/radarr-getter.service.ts b/server/src/modules/rules/getter/radarr-getter.service.ts
@@ -135,12 +135,21 @@ export class RadarrGetterService {
           case 'rottenTomatoesRating': {
             return movieResponse.ratings.rottenTomatoes?.value ?? null;
           }
+          case 'rottenTomatoesRatingVotes': {
+            return movieResponse.ratings.rottenTomatoes?.votes ?? null;
+          }
           case 'traktRating': {
             return movieResponse.ratings.trakt?.value ?? null;
           }
+          case 'traktRatingVotes': {
+            return movieResponse.ratings.trakt?.votes ?? null;
+          }
           case 'imdbRating': {
             return movieResponse.ratings.imdb?.value ?? null;
           }
+          case 'imdbRatingVotes': {
+            return movieResponse.ratings.imdb?.votes ?? null;
+          }
         }
       } else {
         this.logger.debug(

diff --git a/server/src/modules/rules/getter/sonarr-getter.service.ts b/server/src/modules/rules/getter/sonarr-getter.service.ts
@@ -300,6 +300,9 @@ export class SonarrGetterService {
             case 'rating': {
               return showResponse.ratings?.value ?? null;
             }
+            case 'ratingVotes': {
+              return showResponse.ratings?.votes ?? null;
+            }
           }
         } else return null;
       } else {