|
|
|
@ -9,46 +9,37 @@ namespace NzbDrone.Core.Indexers.Nyaa
|
|
|
|
|
{
|
|
|
|
|
public NyaaSettings Settings { get; set; }
|
|
|
|
|
|
|
|
|
|
public int MaxPages { get; set; }
|
|
|
|
|
public int PageSize { get; set; }
|
|
|
|
|
|
|
|
|
|
public NyaaRequestGenerator()
|
|
|
|
|
public virtual IndexerPageableRequestChain GetRecentRequests()
|
|
|
|
|
{
|
|
|
|
|
MaxPages = 30;
|
|
|
|
|
PageSize = 100;
|
|
|
|
|
var pageableRequests = new IndexerPageableRequestChain();
|
|
|
|
|
|
|
|
|
|
pageableRequests.Add(GetPagedRequests(null));
|
|
|
|
|
|
|
|
|
|
return pageableRequests;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public virtual IndexerPageableRequestChain GetRecentRequests()
|
|
|
|
|
public IndexerPageableRequestChain GetSearchRequests(MovieSearchCriteria searchCriteria)
|
|
|
|
|
{
|
|
|
|
|
var pageableRequests = new IndexerPageableRequestChain();
|
|
|
|
|
|
|
|
|
|
pageableRequests.Add(GetPagedRequests(MaxPages, null));
|
|
|
|
|
foreach (var queryTitle in searchCriteria.SceneTitles)
|
|
|
|
|
{
|
|
|
|
|
pageableRequests.Add(GetPagedRequests(PrepareQuery($"{queryTitle} {searchCriteria.Movie.Year}")));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return pageableRequests;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private IEnumerable<IndexerRequest> GetPagedRequests(int maxPages, string term)
|
|
|
|
|
private IEnumerable<IndexerRequest> GetPagedRequests(string term)
|
|
|
|
|
{
|
|
|
|
|
var baseUrl = string.Format("{0}/?page=rss{1}", Settings.BaseUrl.TrimEnd('/'), Settings.AdditionalParameters);
|
|
|
|
|
var baseUrl = $"{Settings.BaseUrl.TrimEnd('/')}/?page=rss{Settings.AdditionalParameters}";
|
|
|
|
|
|
|
|
|
|
if (term != null)
|
|
|
|
|
{
|
|
|
|
|
baseUrl += "&term=" + term;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (PageSize == 0)
|
|
|
|
|
{
|
|
|
|
|
yield return new IndexerRequest(baseUrl, HttpAccept.Rss);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
yield return new IndexerRequest(baseUrl, HttpAccept.Rss);
|
|
|
|
|
|
|
|
|
|
for (var page = 1; page < maxPages; page++)
|
|
|
|
|
{
|
|
|
|
|
yield return new IndexerRequest($"{baseUrl}&offset={page + 1}", HttpAccept.Rss);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
yield return new IndexerRequest(baseUrl, HttpAccept.Rss);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private string PrepareQuery(string query)
|
|
|
|
@ -56,18 +47,6 @@ namespace NzbDrone.Core.Indexers.Nyaa
|
|
|
|
|
return query.Replace(' ', '+');
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public IndexerPageableRequestChain GetSearchRequests(MovieSearchCriteria searchCriteria)
|
|
|
|
|
{
|
|
|
|
|
var pageableRequests = new IndexerPageableRequestChain();
|
|
|
|
|
|
|
|
|
|
foreach (var queryTitle in searchCriteria.SceneTitles)
|
|
|
|
|
{
|
|
|
|
|
pageableRequests.Add(GetPagedRequests(MaxPages, PrepareQuery(string.Format("{0} {1}", queryTitle, searchCriteria.Movie.Year))));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return pageableRequests;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public Func<IDictionary<string, string>> GetCookies { get; set; }
|
|
|
|
|
public Action<IDictionary<string, string>, DateTime?> CookiesUpdater { get; set; }
|
|
|
|
|
}
|
|
|
|
|