|
|
|
using System;
|
|
|
|
using System.Collections.Generic;
|
|
|
|
using System.Linq;
|
|
|
|
using System.Net.Http;
|
|
|
|
using FluentValidation.Results;
|
|
|
|
using Newtonsoft.Json;
|
|
|
|
using NLog;
|
|
|
|
using NzbDrone.Common.Http;
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Applications.Mylar
|
|
|
|
{
|
|
|
|
public interface IMylarV3Proxy
|
|
|
|
{
|
|
|
|
MylarIndexer AddIndexer(MylarIndexer indexer, MylarSettings settings);
|
|
|
|
List<MylarIndexer> GetIndexers(MylarSettings settings);
|
|
|
|
MylarIndexer GetIndexer(string indexerName, MylarProviderType indexerType, MylarSettings settings);
|
|
|
|
void RemoveIndexer(string indexerName, MylarProviderType indexerType, MylarSettings settings);
|
|
|
|
MylarIndexer UpdateIndexer(MylarIndexer indexer, MylarSettings settings);
|
|
|
|
ValidationFailure TestConnection(MylarSettings settings);
|
|
|
|
}
|
|
|
|
|
|
|
|
public class MylarV1Proxy : IMylarV3Proxy
|
|
|
|
{
|
|
|
|
private readonly IHttpClient _httpClient;
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
|
|
public MylarV1Proxy(IHttpClient httpClient, Logger logger)
|
|
|
|
{
|
|
|
|
_httpClient = httpClient;
|
|
|
|
_logger = logger;
|
|
|
|
}
|
|
|
|
|
|
|
|
public MylarStatus GetStatus(MylarSettings settings)
|
|
|
|
{
|
|
|
|
var request = BuildRequest(settings, "/api", "getVersion", HttpMethod.Get);
|
|
|
|
return Execute<MylarStatus>(request);
|
|
|
|
}
|
|
|
|
|
|
|
|
public List<MylarIndexer> GetIndexers(MylarSettings settings)
|
|
|
|
{
|
|
|
|
var request = BuildRequest(settings, "/api", "listProviders", HttpMethod.Get);
|
|
|
|
|
|
|
|
var response = Execute<MylarIndexerResponse>(request);
|
|
|
|
|
|
|
|
if (!response.Success)
|
|
|
|
{
|
|
|
|
throw new MylarException(string.Format("Mylar Error - Code {0}: {1}", response.Error.Code, response.Error.Message));
|
|
|
|
}
|
|
|
|
|
|
|
|
var indexers = new List<MylarIndexer>();
|
|
|
|
|
|
|
|
var torIndexers = response.Data.Torznabs;
|
|
|
|
torIndexers.ForEach(i => i.Type = MylarProviderType.Torznab);
|
|
|
|
|
|
|
|
var nzbIndexers = response.Data.Newznabs;
|
|
|
|
nzbIndexers.ForEach(i => i.Type = MylarProviderType.Newznab);
|
|
|
|
|
|
|
|
indexers.AddRange(torIndexers);
|
|
|
|
indexers.AddRange(nzbIndexers);
|
|
|
|
indexers.ForEach(i => i.Altername = i.Name);
|
|
|
|
|
|
|
|
return indexers;
|
|
|
|
}
|
|
|
|
|
|
|
|
public MylarIndexer GetIndexer(string indexerName, MylarProviderType indexerType, MylarSettings settings)
|
|
|
|
{
|
|
|
|
var indexers = GetIndexers(settings);
|
|
|
|
|
|
|
|
return indexers.SingleOrDefault(i => i.Name == indexerName && i.Type == indexerType);
|
|
|
|
}
|
|
|
|
|
|
|
|
public void RemoveIndexer(string indexerName, MylarProviderType indexerType, MylarSettings settings)
|
|
|
|
{
|
|
|
|
var parameters = new Dictionary<string, string>
|
|
|
|
{
|
|
|
|
{ "name", indexerName },
|
|
|
|
{ "providertype", indexerType.ToString().ToLower() }
|
|
|
|
};
|
|
|
|
|
|
|
|
var request = BuildRequest(settings, "/api", "delProvider", HttpMethod.Get, parameters);
|
|
|
|
CheckForError(Execute<MylarStatus>(request));
|
|
|
|
}
|
|
|
|
|
|
|
|
public MylarIndexer AddIndexer(MylarIndexer indexer, MylarSettings settings)
|
|
|
|
{
|
|
|
|
var parameters = new Dictionary<string, string>
|
|
|
|
{
|
|
|
|
{ "name", indexer.Name },
|
|
|
|
{ "providertype", indexer.Type.ToString().ToLower() },
|
|
|
|
{ "host", indexer.Host },
|
|
|
|
{ "prov_apikey", indexer.Apikey },
|
|
|
|
{ "enabled", indexer.Enabled.ToString().ToLower() },
|
|
|
|
{ "categories", indexer.Categories }
|
|
|
|
};
|
|
|
|
|
|
|
|
var request = BuildRequest(settings, "/api", "addProvider", HttpMethod.Get, parameters);
|
|
|
|
CheckForError(Execute<MylarStatus>(request));
|
|
|
|
return indexer;
|
|
|
|
}
|
|
|
|
|
|
|
|
public MylarIndexer UpdateIndexer(MylarIndexer indexer, MylarSettings settings)
|
|
|
|
{
|
|
|
|
var parameters = new Dictionary<string, string>
|
|
|
|
{
|
|
|
|
{ "name", indexer.Name },
|
|
|
|
{ "providertype", indexer.Type.ToString().ToLower() },
|
|
|
|
{ "host", indexer.Host },
|
|
|
|
{ "prov_apikey", indexer.Apikey },
|
|
|
|
{ "enabled", indexer.Enabled.ToString().ToLower() },
|
|
|
|
{ "categories", indexer.Categories },
|
|
|
|
{ "altername", indexer.Altername }
|
|
|
|
};
|
|
|
|
|
|
|
|
var request = BuildRequest(settings, "/api", "changeProvider", HttpMethod.Get, parameters);
|
|
|
|
CheckForError(Execute<MylarStatus>(request));
|
|
|
|
return indexer;
|
|
|
|
}
|
|
|
|
|
|
|
|
private void CheckForError(MylarStatus response)
|
|
|
|
{
|
|
|
|
if (!response.Success)
|
|
|
|
{
|
|
|
|
throw new MylarException(string.Format("Mylar Error - Code {0}: {1}", response.Error.Code, response.Error.Message));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
public ValidationFailure TestConnection(MylarSettings settings)
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
var status = GetStatus(settings);
|
|
|
|
|
|
|
|
if (!status.Success)
|
|
|
|
{
|
|
|
|
return new ValidationFailure("ApiKey", status.Error.Message);
|
|
|
|
}
|
|
|
|
|
|
|
|
var indexers = GetIndexers(settings);
|
|
|
|
}
|
|
|
|
catch (HttpException ex)
|
|
|
|
{
|
|
|
|
_logger.Error(ex, "Unable to send test message");
|
|
|
|
return new ValidationFailure("BaseUrl", "Unable to complete application test");
|
|
|
|
}
|
|
|
|
catch (MylarException ex)
|
|
|
|
{
|
|
|
|
_logger.Error(ex, "Connection test failed");
|
|
|
|
return new ValidationFailure("", ex.Message);
|
|
|
|
}
|
|
|
|
catch (Exception ex)
|
|
|
|
{
|
|
|
|
_logger.Error(ex, "Unable to send test message");
|
|
|
|
return new ValidationFailure("", "Unable to send test message");
|
|
|
|
}
|
|
|
|
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
private HttpRequest BuildRequest(MylarSettings settings, string resource, string command, HttpMethod method, Dictionary<string, string> parameters = null)
|
|
|
|
{
|
|
|
|
var baseUrl = settings.BaseUrl.TrimEnd('/');
|
|
|
|
|
|
|
|
var requestBuilder = new HttpRequestBuilder(baseUrl).Resource(resource)
|
|
|
|
.AddQueryParam("cmd", command)
|
|
|
|
.AddQueryParam("apikey", settings.ApiKey);
|
|
|
|
|
|
|
|
if (parameters != null)
|
|
|
|
{
|
|
|
|
foreach (var param in parameters)
|
|
|
|
{
|
|
|
|
requestBuilder.AddQueryParam(param.Key, param.Value);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
var request = requestBuilder.Build();
|
|
|
|
|
|
|
|
request.Headers.ContentType = "application/json";
|
|
|
|
|
|
|
|
request.Method = method;
|
|
|
|
request.AllowAutoRedirect = true;
|
|
|
|
|
|
|
|
return request;
|
|
|
|
}
|
|
|
|
|
|
|
|
private TResource Execute<TResource>(HttpRequest request)
|
|
|
|
where TResource : new()
|
|
|
|
{
|
|
|
|
var response = _httpClient.Execute(request);
|
|
|
|
|
|
|
|
var results = JsonConvert.DeserializeObject<TResource>(response.Content);
|
|
|
|
|
|
|
|
return results;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|