New: HDBits to parse IMDB using parser utils (#454)

Prowlarr · Aug 26, 2021 · 77a76fe · 77a76fe
1 parent 1d20b9d
commit 77a76fe
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/src/NzbDrone.Core/Indexers/Definitions/HDBits/HDBitsRequestGenerator.cs b/src/NzbDrone.Core/Indexers/Definitions/HDBits/HDBitsRequestGenerator.cs
@@ -5,6 +5,7 @@
 using NzbDrone.Common.Http;
 using NzbDrone.Common.Serializer;
 using NzbDrone.Core.IndexerSearch.Definitions;
+using NzbDrone.Core.Parser;
 
 namespace NzbDrone.Core.Indexers.HDBits
 {
@@ -17,26 +18,22 @@ public virtual IndexerPageableRequestChain GetSearchRequests(MovieSearchCriteria
         {
             var pageableRequests = new IndexerPageableRequestChain();
             var query = new TorrentQuery();
+            var imdbId = ParseUtil.GetImdbID(searchCriteria.ImdbId).GetValueOrDefault(0);
 
             if (searchCriteria.Categories?.Length > 0)
             {
                 query.Category = Capabilities.Categories.MapTorznabCapsToTrackers(searchCriteria.Categories).Select(int.Parse).ToArray();
             }
 
-            if (searchCriteria.ImdbId.IsNullOrWhiteSpace() && searchCriteria.SearchTerm.IsNotNullOrWhiteSpace())
+            if (imdbId == 0 && searchCriteria.SearchTerm.IsNotNullOrWhiteSpace())
             {
                 query.Search = searchCriteria.SanitizedSearchTerm;
             }
 
-            if (searchCriteria.ImdbId.IsNotNullOrWhiteSpace())
+            if (imdbId != 0)
             {
-                var imdbId = int.Parse(searchCriteria.ImdbId.Substring(2));
-
-                if (imdbId != 0)
-                {
-                    query.ImdbInfo = query.ImdbInfo ?? new ImdbInfo();
-                    query.ImdbInfo.Id = imdbId;
-                }
+                query.ImdbInfo = query.ImdbInfo ?? new ImdbInfo();
+                query.ImdbInfo.Id = imdbId;
             }
 
             pageableRequests.Add(GetRequest(query));
@@ -79,13 +76,14 @@ public IndexerPageableRequestChain GetSearchRequests(TvSearchCriteria searchCrit
             var pageableRequests = new IndexerPageableRequestChain();
             var query = new TorrentQuery();
             var tvdbId = searchCriteria.TvdbId.GetValueOrDefault(0);
+            var imdbId = ParseUtil.GetImdbID(searchCriteria.ImdbId).GetValueOrDefault(0);
 
             if (searchCriteria.Categories?.Length > 0)
             {
                 query.Category = Capabilities.Categories.MapTorznabCapsToTrackers(searchCriteria.Categories).Select(int.Parse).ToArray();
             }
 
-            if (tvdbId == 0 && searchCriteria.SearchTerm.IsNotNullOrWhiteSpace())
+            if (tvdbId == 0 && imdbId == 0 && searchCriteria.SearchTerm.IsNotNullOrWhiteSpace())
             {
                 query.Search = searchCriteria.SanitizedTvSearchString;
             }
@@ -98,6 +96,12 @@ public IndexerPageableRequestChain GetSearchRequests(TvSearchCriteria searchCrit
                 query.TvdbInfo.Episode = searchCriteria.Episode;
             }
 
+            if (imdbId != 0)
+            {
+                query.ImdbInfo = query.ImdbInfo ?? new ImdbInfo();
+                query.ImdbInfo.Id = imdbId;
+            }
+
             pageableRequests.Add(GetRequest(query));
 
             return pageableRequests;