|
|
|
|
@ -0,0 +1,88 @@
|
|
|
|
|
using System;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using NzbDrone.Common.Serializer;
|
|
|
|
|
using NzbDrone.Core.Configuration;
|
|
|
|
|
using RestSharp;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Download.Clients.Sabnzbd
|
|
|
|
|
{
|
|
|
|
|
public interface ISabCommunicationProxy
|
|
|
|
|
{
|
|
|
|
|
string DownloadNzb(Stream nzb, string name, string category, int priority);
|
|
|
|
|
string ProcessRequest(IRestRequest restRequest, string action);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class SabCommunicationProxy : ISabCommunicationProxy
|
|
|
|
|
{
|
|
|
|
|
private readonly IConfigService _configService;
|
|
|
|
|
|
|
|
|
|
public SabCommunicationProxy(IConfigService configService)
|
|
|
|
|
{
|
|
|
|
|
_configService = configService;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public string DownloadNzb(Stream nzb, string title, string category, int priority)
|
|
|
|
|
{
|
|
|
|
|
var request = new RestRequest(Method.POST);
|
|
|
|
|
var action = String.Format("mode=addfile&cat={0}&priority={1}", category, priority);
|
|
|
|
|
|
|
|
|
|
request.AddFile("name", ReadFully(nzb), title, "application/x-nzb");
|
|
|
|
|
|
|
|
|
|
return ProcessRequest(request, action);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public string ProcessRequest(IRestRequest restRequest, string action)
|
|
|
|
|
{
|
|
|
|
|
var client = BuildClient(action);
|
|
|
|
|
var response = client.Execute(restRequest);
|
|
|
|
|
|
|
|
|
|
CheckForError(response);
|
|
|
|
|
|
|
|
|
|
return response.Content;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private IRestClient BuildClient(string action)
|
|
|
|
|
{
|
|
|
|
|
var protocol = _configService.SabUseSsl ? "https" : "http";
|
|
|
|
|
|
|
|
|
|
var url = string.Format(@"{0}://{1}:{2}/api?{3}&apikey={4}&ma_username={5}&ma_password={6}&output=json",
|
|
|
|
|
protocol,
|
|
|
|
|
_configService.SabHost,
|
|
|
|
|
_configService.SabPort,
|
|
|
|
|
action,
|
|
|
|
|
_configService.SabApiKey,
|
|
|
|
|
_configService.SabUsername,
|
|
|
|
|
_configService.SabPassword);
|
|
|
|
|
|
|
|
|
|
return new RestClient(url);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void CheckForError(IRestResponse response)
|
|
|
|
|
{
|
|
|
|
|
if (response.ResponseStatus != ResponseStatus.Completed)
|
|
|
|
|
{
|
|
|
|
|
throw new ApplicationException("Unable to connect to SABnzbd, please check your settings");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var result = Json.Deserialize<SabJsonError>(response.Content);
|
|
|
|
|
|
|
|
|
|
if (result.Status != null && result.Status.Equals("false", StringComparison.InvariantCultureIgnoreCase))
|
|
|
|
|
throw new ApplicationException(result.Error);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//TODO: Find a better home for this
|
|
|
|
|
private byte[] ReadFully(Stream input)
|
|
|
|
|
{
|
|
|
|
|
byte[] buffer = new byte[16 * 1024];
|
|
|
|
|
using (MemoryStream ms = new MemoryStream())
|
|
|
|
|
{
|
|
|
|
|
int read;
|
|
|
|
|
while ((read = input.Read(buffer, 0, buffer.Length)) > 0)
|
|
|
|
|
{
|
|
|
|
|
ms.Write(buffer, 0, read);
|
|
|
|
|
}
|
|
|
|
|
return ms.ToArray();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
Mark McDowall
11 years ago