You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
3.1 KiB
98 lines
3.1 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Common.Http;
|
|
using NzbDrone.Common.Serializer;
|
|
using NzbDrone.Core.IndexerSearch.Definitions;
|
|
|
|
namespace NzbDrone.Core.Indexers.HDBits
|
|
{
|
|
public class HDBitsRequestGenerator : IIndexerRequestGenerator
|
|
{
|
|
public HDBitsSettings Settings { get; set; }
|
|
|
|
public virtual IndexerPageableRequestChain GetSearchRequests(MovieSearchCriteria searchCriteria)
|
|
{
|
|
var pageableRequests = new IndexerPageableRequestChain();
|
|
var query = new TorrentQuery();
|
|
|
|
if (TryAddSearchParameters(query, searchCriteria))
|
|
{
|
|
pageableRequests.Add(GetRequest(query));
|
|
}
|
|
|
|
return pageableRequests;
|
|
}
|
|
|
|
private bool TryAddSearchParameters(TorrentQuery query, MovieSearchCriteria searchCriteria)
|
|
{
|
|
if (searchCriteria.ImdbId.IsNullOrWhiteSpace())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
var imdbId = int.Parse(searchCriteria.ImdbId.Substring(2));
|
|
|
|
if (imdbId != 0)
|
|
{
|
|
query.ImdbInfo = query.ImdbInfo ?? new ImdbInfo();
|
|
query.ImdbInfo.Id = imdbId;
|
|
|
|
//TODO Map Categories
|
|
query.Category = searchCriteria.Categories;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public Func<IDictionary<string, string>> GetCookies { get; set; }
|
|
public Action<IDictionary<string, string>, DateTime?> CookiesUpdater { get; set; }
|
|
|
|
private IEnumerable<IndexerRequest> GetRequest(TorrentQuery query)
|
|
{
|
|
var request = new HttpRequestBuilder(Settings.BaseUrl)
|
|
.Resource("/api/torrents")
|
|
.Build();
|
|
|
|
request.Method = HttpMethod.POST;
|
|
const string appJson = "application/json";
|
|
request.Headers.Accept = appJson;
|
|
request.Headers.ContentType = appJson;
|
|
|
|
query.Username = Settings.Username;
|
|
query.Passkey = Settings.ApiKey;
|
|
|
|
//TODO Add from searchCriteria
|
|
query.Category = query.Category.ToArray();
|
|
query.Codec = Settings.Codecs.ToArray();
|
|
query.Medium = Settings.Mediums.ToArray();
|
|
|
|
request.SetContent(query.ToJson());
|
|
|
|
yield return new IndexerRequest(request);
|
|
}
|
|
|
|
public IndexerPageableRequestChain GetSearchRequests(MusicSearchCriteria searchCriteria)
|
|
{
|
|
return new IndexerPageableRequestChain();
|
|
}
|
|
|
|
public IndexerPageableRequestChain GetSearchRequests(TvSearchCriteria searchCriteria)
|
|
{
|
|
return new IndexerPageableRequestChain();
|
|
}
|
|
|
|
public IndexerPageableRequestChain GetSearchRequests(BookSearchCriteria searchCriteria)
|
|
{
|
|
return new IndexerPageableRequestChain();
|
|
}
|
|
|
|
public IndexerPageableRequestChain GetSearchRequests(BasicSearchCriteria searchCriteria)
|
|
{
|
|
return new IndexerPageableRequestChain();
|
|
}
|
|
}
|
|
}
|