rarbg: Add IMDB support for tvsearch (#12225) resolves #12127

Jackett · Aug 28, 2021 · 67b96f0 · 67b96f0
1 parent 0065aec
commit 67b96f0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/src/Jackett.Common/Indexers/RarBG.cs b/src/Jackett.Common/Indexers/RarBG.cs
@@ -41,7 +41,7 @@ public class RarBG : BaseWebIndexer
                    {
                        TvSearchParams = new List<TvSearchParam>
                        {
-                           TvSearchParam.Q, TvSearchParam.Season, TvSearchParam.Ep
+                           TvSearchParam.Q, TvSearchParam.Season, TvSearchParam.Ep, TvSearchParam.ImdbId
                        },
                        MovieSearchParams = new List<MovieSearchParam>
                        {
@@ -54,7 +54,8 @@ public class RarBG : BaseWebIndexer
                        BookSearchParams = new List<BookSearchParam>
                        {
                            BookSearchParam.Q
-                       }
+                       },
+                       TvSearchImdbAvailable = true
                    },
                    configService: configService,
                    client: wc,
@@ -230,6 +231,7 @@ private async Task<IEnumerable<ReleaseInfo>> PerformQueryWithRetry(TorznabQuery
         private string BuildSearchUrl(TorznabQuery query)
         {
             var searchString = query.GetQueryString();
+            var episodeSearchString = query.GetEpisodeSearchString();
             var qc = new NameValueCollection
             {
                 { "token", _token },
@@ -240,7 +242,13 @@ private string BuildSearchUrl(TorznabQuery query)
                 { "sort", _sort }
             };
 
-            if (query.ImdbID != null)
+            if (query.IsTVSearch && !string.IsNullOrWhiteSpace(episodeSearchString) && query.ImdbID != null)
+            {
+                qc.Add("mode", "search");
+                qc.Add("search_imdb", query.ImdbID);
+                qc.Add("search_string", episodeSearchString);
+            }
+            else if (query.ImdbID != null)
             {
                 qc.Add("mode", "search");
                 qc.Add("search_imdb", query.ImdbID);

diff --git a/src/Jackett.Common/Models/TorznabCapabilities.cs b/src/Jackett.Common/Models/TorznabCapabilities.cs
@@ -50,7 +50,8 @@ public class TorznabCapabilities
         public bool TvSearchSeasonAvailable => (TvSearchParams.Contains(TvSearchParam.Season));
         public bool TvSearchEpAvailable => (TvSearchParams.Contains(TvSearchParam.Ep));
         //TvSearchImdbAvailable temporarily disabled due to #8107
-        public bool TvSearchImdbAvailable => false; // (TvSearchParams.Contains(TvSearchParam.ImdbId));
+        // Introduce setter so individual trackers can override
+        public bool TvSearchImdbAvailable { get; set; } = false; // (TvSearchParams.Contains(TvSearchParam.ImdbId));
         public bool TvSearchTvdbAvailable => (TvSearchParams.Contains(TvSearchParam.TvdbId));
         public bool TvSearchTvRageAvailable => (TvSearchParams.Contains(TvSearchParam.RId));