mirror of https://github.com/Radarr/Radarr
74 lines
2.5 KiB
C#
74 lines
2.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using NLog;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Common.Http;
|
|
using NzbDrone.Core.IndexerSearch.Definitions;
|
|
|
|
namespace NzbDrone.Core.Indexers.PassThePopcorn
|
|
{
|
|
public class PassThePopcornRequestGenerator : IIndexerRequestGenerator
|
|
{
|
|
public PassThePopcornSettings Settings { get; set; }
|
|
|
|
public IDictionary<string, string> Cookies { get; set; }
|
|
|
|
public IHttpClient HttpClient { get; set; }
|
|
public Logger Logger { get; set; }
|
|
|
|
public virtual IndexerPageableRequestChain GetRecentRequests()
|
|
{
|
|
var pageableRequests = new IndexerPageableRequestChain();
|
|
|
|
pageableRequests.Add(GetRequest(null));
|
|
|
|
return pageableRequests;
|
|
}
|
|
|
|
public IndexerPageableRequestChain GetSearchRequests(MovieSearchCriteria searchCriteria)
|
|
{
|
|
var pageableRequests = new IndexerPageableRequestChain();
|
|
|
|
if (searchCriteria.Movie.MovieMetadata.Value.ImdbId.IsNotNullOrWhiteSpace())
|
|
{
|
|
pageableRequests.Add(GetRequest(searchCriteria.Movie.MovieMetadata.Value.ImdbId));
|
|
}
|
|
else if (searchCriteria.Movie.Year > 0)
|
|
{
|
|
foreach (var queryTitle in searchCriteria.CleanSceneTitles)
|
|
{
|
|
pageableRequests.Add(GetRequest(string.Format("{0}&year={1}", queryTitle, searchCriteria.Movie.Year)));
|
|
}
|
|
}
|
|
|
|
return pageableRequests;
|
|
}
|
|
|
|
public Func<IDictionary<string, string>> GetCookies { get; set; }
|
|
public Action<IDictionary<string, string>, DateTime?> CookiesUpdater { get; set; }
|
|
|
|
private IEnumerable<IndexerRequest> GetRequest(string searchParameters)
|
|
{
|
|
var request =
|
|
new IndexerRequest(
|
|
$"{Settings.BaseUrl.Trim().TrimEnd('/')}/torrents.php?action=advanced&json=noredirect&searchstr={searchParameters}",
|
|
HttpAccept.Json);
|
|
|
|
request.HttpRequest.Headers["ApiUser"] = Settings.APIUser;
|
|
request.HttpRequest.Headers["ApiKey"] = Settings.APIKey;
|
|
|
|
if (Settings.APIKey.IsNullOrWhiteSpace())
|
|
{
|
|
foreach (var cookie in Cookies)
|
|
{
|
|
request.HttpRequest.Cookies[cookie.Key] = cookie.Value;
|
|
}
|
|
|
|
CookiesUpdater(Cookies, DateTime.Now + TimeSpan.FromDays(30));
|
|
}
|
|
|
|
yield return request;
|
|
}
|
|
}
|
|
}
|