-
-
Notifications
You must be signed in to change notification settings - Fork 242
/
RedactedRequestGenerator.cs
74 lines (60 loc) · 2.46 KB
/
RedactedRequestGenerator.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
using System.Collections.Generic;
using System.Linq;
using NzbDrone.Common.Http;
using NzbDrone.Core.IndexerSearch.Definitions;
namespace NzbDrone.Core.Indexers.Redacted
{
public class RedactedRequestGenerator : IIndexerRequestGenerator
{
private readonly RedactedSettings _settings;
public RedactedRequestGenerator(RedactedSettings settings)
{
_settings = settings;
}
public virtual IndexerPageableRequestChain GetRecentRequests()
{
var pageableRequests = new IndexerPageableRequestChain();
pageableRequests.Add(GetRequest(null));
return pageableRequests;
}
public IndexerPageableRequestChain GetSearchRequests(AlbumSearchCriteria searchCriteria)
{
var pageableRequests = new IndexerPageableRequestChain();
if (searchCriteria.CleanArtistQuery == "VA")
{
pageableRequests.Add(GetRequest($"groupname={searchCriteria.CleanAlbumQuery}"));
}
else
{
pageableRequests.Add(GetRequest($"artistname={searchCriteria.CleanArtistQuery}&groupname={searchCriteria.CleanAlbumQuery}"));
}
return pageableRequests;
}
public IndexerPageableRequestChain GetSearchRequests(ArtistSearchCriteria searchCriteria)
{
var pageableRequests = new IndexerPageableRequestChain();
pageableRequests.Add(GetRequest($"artistname={searchCriteria.CleanArtistQuery}"));
return pageableRequests;
}
private IEnumerable<IndexerRequest> GetRequest(string searchParameters)
{
var requestBuilder = RequestBuilder()
.Resource($"ajax.php?{searchParameters}")
.AddQueryParam("action", "browse")
.AddQueryParam("order_by", "time")
.AddQueryParam("order_way", "desc");
var categories = _settings.Categories.ToList();
if (categories.Any())
{
categories.ForEach(cat => requestBuilder.AddQueryParam($"filter_cat[{cat}]", "1"));
}
yield return new IndexerRequest(requestBuilder.Build());
}
private HttpRequestBuilder RequestBuilder()
{
return new HttpRequestBuilder($"{_settings.BaseUrl.Trim().TrimEnd('/')}")
.Accept(HttpAccept.Json)
.SetHeader("Authorization", _settings.ApiKey);
}
}
}