shpaq/Prowlarr-Prowlarr
1
0
Fork 0
mirror of https://github.com/Prowlarr/Prowlarr.git synced 2025-09-17 17:14:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Steve Adams
2021-08-26 13:22:04 +10:00
committed by GitHub
parent 1d20b9d429
commit 77a76fe5a1
1 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/NzbDrone.Core/Indexers/Definitions/HDBits/HDBitsRequestGenerator.cs
View File

@@ -5,6 +5,7 @@ using NzbDrone.Common.Extensions;
using NzbDrone.Common.Http; using NzbDrone.Common.Http;
using NzbDrone.Common.Serializer; using NzbDrone.Common.Serializer;
using NzbDrone.Core.IndexerSearch.Definitions; using NzbDrone.Core.IndexerSearch.Definitions;
using NzbDrone.Core.Parser;
namespace NzbDrone.Core.Indexers.HDBits namespace NzbDrone.Core.Indexers.HDBits
{ {
@@ -17,27 +18,23 @@ namespace NzbDrone.Core.Indexers.HDBits
{ {
var pageableRequests = new IndexerPageableRequestChain(); var pageableRequests = new IndexerPageableRequestChain();
var query = new TorrentQuery(); var query = new TorrentQuery();
var imdbId = ParseUtil.GetImdbID(searchCriteria.ImdbId).GetValueOrDefault(0);
if (searchCriteria.Categories?.Length > 0) if (searchCriteria.Categories?.Length > 0)
{ {
query.Category = Capabilities.Categories.MapTorznabCapsToTrackers(searchCriteria.Categories).Select(int.Parse).ToArray(); 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; query.Search = searchCriteria.SanitizedSearchTerm;
} }
if (searchCriteria.ImdbId.IsNotNullOrWhiteSpace())
{
var imdbId = int.Parse(searchCriteria.ImdbId.Substring(2));
if (imdbId != 0) if (imdbId != 0)
{ {
query.ImdbInfo = query.ImdbInfo ?? new ImdbInfo(); query.ImdbInfo = query.ImdbInfo ?? new ImdbInfo();
query.ImdbInfo.Id = imdbId; query.ImdbInfo.Id = imdbId;
} }
}
pageableRequests.Add(GetRequest(query)); pageableRequests.Add(GetRequest(query));
@@ -79,13 +76,14 @@ namespace NzbDrone.Core.Indexers.HDBits
var pageableRequests = new IndexerPageableRequestChain(); var pageableRequests = new IndexerPageableRequestChain();
var query = new TorrentQuery(); var query = new TorrentQuery();
var tvdbId = searchCriteria.TvdbId.GetValueOrDefault(0); var tvdbId = searchCriteria.TvdbId.GetValueOrDefault(0);
var imdbId = ParseUtil.GetImdbID(searchCriteria.ImdbId).GetValueOrDefault(0);
if (searchCriteria.Categories?.Length > 0) if (searchCriteria.Categories?.Length > 0)
{ {
query.Category = Capabilities.Categories.MapTorznabCapsToTrackers(searchCriteria.Categories).Select(int.Parse).ToArray(); 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; query.Search = searchCriteria.SanitizedTvSearchString;
} }
@@ -98,6 +96,12 @@ namespace NzbDrone.Core.Indexers.HDBits
query.TvdbInfo.Episode = searchCriteria.Episode; query.TvdbInfo.Episode = searchCriteria.Episode;
} }
if (imdbId != 0)
{
query.ImdbInfo = query.ImdbInfo ?? new ImdbInfo();
query.ImdbInfo.Id = imdbId;
}
pageableRequests.Add(GetRequest(query)); pageableRequests.Add(GetRequest(query));
return pageableRequests; return pageableRequests;