Skip to content

Commit

Permalink
Fixed: (SceneHD) Category filtering
Browse files Browse the repository at this point in the history
Fixes #2028
  • Loading branch information
mynameisbogdan committed Mar 19, 2024
1 parent d279c97 commit 680bf46
Showing 1 changed file with 21 additions and 17 deletions.
38 changes: 21 additions & 17 deletions src/NzbDrone.Core/Indexers/Definitions/SceneHD.cs
Expand Up @@ -38,7 +38,7 @@ public SceneHD(IIndexerHttpClient httpClient, IEventAggregator eventAggregator,

public override IIndexerRequestGenerator GetRequestGenerator()
{
return new SceneHDRequestGenerator() { Settings = Settings, Capabilities = Capabilities };
return new SceneHDRequestGenerator(Settings, Capabilities);
}

public override IParseIndexerResponse GetParser()
Expand Down Expand Up @@ -88,38 +88,42 @@ private IndexerCapabilities SetCapabilities()

public class SceneHDRequestGenerator : IIndexerRequestGenerator
{
public SceneHDSettings Settings { get; set; }
public IndexerCapabilities Capabilities { get; set; }
public string BaseUrl { get; set; }
private readonly SceneHDSettings _settings;
private readonly IndexerCapabilities _capabilities;

public SceneHDRequestGenerator(SceneHDSettings settings, IndexerCapabilities capabilities)
{
_settings = settings;
_capabilities = capabilities;
}

private IEnumerable<IndexerRequest> GetPagedRequests(string term, int[] categories, string imdbId = null)
{
var search = new[] { imdbId, term };

var qc = new NameValueCollection
var parameters = new NameValueCollection
{
{ "api", "" },
{ "passkey", Settings.Passkey },
{ "passkey", _settings.Passkey },
{ "search", string.Join(" ", search.Where(s => s.IsNotNullOrWhiteSpace())) },
{ "search", string.Join(" ", search.Where(s => s.IsNotNullOrWhiteSpace())) }
};

foreach (var cat in Capabilities.Categories.MapTorznabCapsToTrackers(categories))
if (categories?.Length > 0)
{
qc.Add("categories[" + cat + "]", "1");
parameters.Add("cat", _capabilities.Categories.MapTorznabCapsToTrackers(categories).Distinct().Join(","));
}

var searchUrl = string.Format("{0}/browse.php?{1}", Settings.BaseUrl.TrimEnd('/'), qc.GetQueryString());

var request = new IndexerRequest(searchUrl, HttpAccept.Json);
var searchUrl = $"{_settings.BaseUrl.TrimEnd('/')}/browse.php?{parameters.GetQueryString()}";

yield return request;
yield return new IndexerRequest(searchUrl, HttpAccept.Json);
}

public IndexerPageableRequestChain GetSearchRequests(MovieSearchCriteria searchCriteria)
{
var pageableRequests = new IndexerPageableRequestChain();

pageableRequests.Add(GetPagedRequests(string.Format("{0}", searchCriteria.SanitizedSearchTerm), searchCriteria.Categories, searchCriteria.FullImdbId));
pageableRequests.Add(GetPagedRequests($"{searchCriteria.SanitizedSearchTerm}", searchCriteria.Categories, searchCriteria.FullImdbId));

return pageableRequests;
}
Expand All @@ -128,7 +132,7 @@ public IndexerPageableRequestChain GetSearchRequests(MusicSearchCriteria searchC
{
var pageableRequests = new IndexerPageableRequestChain();

pageableRequests.Add(GetPagedRequests(string.Format("{0}", searchCriteria.SanitizedSearchTerm), searchCriteria.Categories));
pageableRequests.Add(GetPagedRequests($"{searchCriteria.SanitizedSearchTerm}", searchCriteria.Categories));

return pageableRequests;
}
Expand All @@ -137,7 +141,7 @@ public IndexerPageableRequestChain GetSearchRequests(TvSearchCriteria searchCrit
{
var pageableRequests = new IndexerPageableRequestChain();

pageableRequests.Add(GetPagedRequests(string.Format("{0}", searchCriteria.SanitizedTvSearchString), searchCriteria.Categories, searchCriteria.FullImdbId));
pageableRequests.Add(GetPagedRequests($"{searchCriteria.SanitizedTvSearchString}", searchCriteria.Categories, searchCriteria.FullImdbId));

return pageableRequests;
}
Expand All @@ -146,7 +150,7 @@ public IndexerPageableRequestChain GetSearchRequests(BookSearchCriteria searchCr
{
var pageableRequests = new IndexerPageableRequestChain();

pageableRequests.Add(GetPagedRequests(string.Format("{0}", searchCriteria.SanitizedSearchTerm), searchCriteria.Categories));
pageableRequests.Add(GetPagedRequests($"{searchCriteria.SanitizedSearchTerm}", searchCriteria.Categories));

return pageableRequests;
}
Expand All @@ -155,7 +159,7 @@ public IndexerPageableRequestChain GetSearchRequests(BasicSearchCriteria searchC
{
var pageableRequests = new IndexerPageableRequestChain();

pageableRequests.Add(GetPagedRequests(string.Format("{0}", searchCriteria.SanitizedSearchTerm), searchCriteria.Categories));
pageableRequests.Add(GetPagedRequests($"{searchCriteria.SanitizedSearchTerm}", searchCriteria.Categories));

return pageableRequests;
}
Expand Down

0 comments on commit 680bf46

Please sign in to comment.