mirror/Sonarr
1
0
Fork 0
mirror of https://github.com/Sonarr/Sonarr synced 2025-02-24 15:11:27 +00:00
Code Issues Releases Wiki Activity

Added support for title query parameter to newznab/torznab, receiving raw series title

Browse source
This commit is contained in:
Taloth Saldono 2020-04-17 00:39:38 +02:00
parent 200aee52f7
commit 05dd17aacb
2 changed files with 83 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
src/NzbDrone.Core.Test/IndexerTests/NewznabTests/NewznabRequestGeneratorFixture.cs
View file

@ -261,6 +261,42 @@ namespace NzbDrone.Core.Test.IndexerTests.NewznabTests
page.Url.Query.Should().Contain("q="); page.Url.Query.Should().Contain("q=");
} }
[Test]
public void should_fallback_to_title()
{
_capabilities.SupportedTvSearchParameters = new[] { "q", "title", "tvdbid", "rid", "season", "ep" };
_capabilities.SupportsAggregateIdSearch = true;
var results = Subject.GetSearchRequests(_singleEpisodeSearchCriteria);
results.Tiers.Should().Be(2);
var pageTier2 = results.GetTier(1).First().First();
pageTier2.Url.Query.Should().NotContain("tvdbid=20");
pageTier2.Url.Query.Should().NotContain("rid=10");
pageTier2.Url.Query.Should().NotContain("q=");
pageTier2.Url.Query.Should().Contain("title=Monkey%20Island");
}
[Test]
public void should_url_encode_title()
{
_capabilities.SupportedTvSearchParameters = new[] { "q", "title", "tvdbid", "rid", "season", "ep" };
_capabilities.SupportsAggregateIdSearch = true;
_singleEpisodeSearchCriteria.SceneTitles[0] = "Elith & Little";
var results = Subject.GetSearchRequests(_singleEpisodeSearchCriteria);
results.Tiers.Should().Be(2);
var pageTier2 = results.GetTier(1).First().First();
pageTier2.Url.Query.Should().NotContain("tvdbid=20");
pageTier2.Url.Query.Should().NotContain("rid=10");
pageTier2.Url.Query.Should().NotContain("q=");
pageTier2.Url.Query.Should().Contain("title=Elith%20%26%20Little");
}
[Test] [Test]
public void should_fallback_to_q() public void should_fallback_to_q()
{ {

56
src/NzbDrone.Core/Indexers/Newznab/NewznabRequestGenerator.cs
View file

@ -47,6 +47,19 @@ namespace NzbDrone.Core.Indexers.Newznab
} }
} }
private bool SupportsTvTitleSearch
{
get
{
var capabilities = _capabilitiesProvider.GetCapabilities(Settings);
return capabilities.SupportedTvSearchParameters != null &&
capabilities.SupportedTvSearchParameters.Contains("title") &&
capabilities.SupportedTvSearchParameters.Contains("season") &&
capabilities.SupportedTvSearchParameters.Contains("ep");
}
}
private bool SupportsTvdbSearch private bool SupportsTvdbSearch
{ {
get get
@ -275,7 +288,19 @@ namespace NzbDrone.Core.Indexers.Newznab
} }
} }
if (SupportsTvSearch)
if (SupportsTvTitleSearch)
{
chain.AddTier();
foreach (var searchTerm in searchCriteria.SceneTitles)
{
chain.Add(GetPagedRequests(MaxPages, Settings.Categories, "tvsearch",
string.Format("&title={0}{1}",
Uri.EscapeDataString(searchTerm),
parameters)));
}
}
else if (SupportsTvSearch)
{ {
chain.AddTier(); chain.AddTier();
foreach (var queryTitle in searchCriteria.QueryTitles) foreach (var queryTitle in searchCriteria.QueryTitles)
@ -290,16 +315,29 @@ namespace NzbDrone.Core.Indexers.Newznab
private void AddSceneTitlePageableRequests(IndexerPageableRequestChain chain, IEnumerable<int> categories, SearchCriteriaBase searchCriteria, Func<SceneMapping, string> parametersFunc) private void AddSceneTitlePageableRequests(IndexerPageableRequestChain chain, IEnumerable<int> categories, SearchCriteriaBase searchCriteria, Func<SceneMapping, string> parametersFunc)
{ {
foreach (var sceneMappingGroup in searchCriteria.SceneMappings.GroupBy(v => v.SceneSeasonNumber)) if (searchCriteria.SceneMappings != null)
{ {
var parameters = parametersFunc(sceneMappingGroup.First()); foreach (var sceneMappingGroup in searchCriteria.SceneMappings.GroupBy(v => v.SceneSeasonNumber))
foreach (var searchTerm in sceneMappingGroup.Select(v => v.SearchTerm).Distinct())
{ {
chain.AddToTier(0, GetPagedRequests(MaxPages, Settings.Categories, "tvsearch", var parameters = parametersFunc(sceneMappingGroup.First());
string.Format("&q={0}{1}",
NewsnabifyTitle(searchTerm), foreach (var searchTerm in sceneMappingGroup.Select(v => v.SearchTerm).Distinct())
parameters))); {
if (SupportsTvTitleSearch)
{
chain.AddToTier(0, GetPagedRequests(MaxPages, Settings.Categories, "tvsearch",
string.Format("&title={0}{1}",
Uri.EscapeDataString(searchTerm),
parameters)));
}
else if (SupportsTvSearch)
{
chain.AddToTier(0, GetPagedRequests(MaxPages, Settings.Categories, "tvsearch",
string.Format("&q={0}{1}",
NewsnabifyTitle(searchTerm),
parameters)));
}
}
} }
} }
} }