2015-05-08 18:50:22 +00:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Net;
|
|
|
|
|
using NLog;
|
|
|
|
|
using NzbDrone.Common.Extensions;
|
|
|
|
|
using CookComputing.XmlRpc;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Download.Clients.RTorrent
|
|
|
|
|
{
|
|
|
|
|
public interface IRTorrentProxy
|
|
|
|
|
{
|
|
|
|
|
string GetVersion(RTorrentSettings settings);
|
|
|
|
|
List<RTorrentTorrent> GetTorrents(RTorrentSettings settings);
|
|
|
|
|
|
|
|
|
|
void AddTorrentFromUrl(string torrentUrl, RTorrentSettings settings);
|
|
|
|
|
void AddTorrentFromFile(string fileName, byte[] fileContent, RTorrentSettings settings);
|
|
|
|
|
void RemoveTorrent(string hash, RTorrentSettings settings);
|
2015-07-08 08:11:33 +00:00
|
|
|
|
void SetTorrentPriority(string hash, RTorrentPriority priority, RTorrentSettings settings);
|
2015-05-08 18:50:22 +00:00
|
|
|
|
void SetTorrentLabel(string hash, string label, RTorrentSettings settings);
|
2015-07-08 08:11:33 +00:00
|
|
|
|
void SetTorrentDownloadDirectory(string hash, string directory, RTorrentSettings settings);
|
2015-05-08 18:50:22 +00:00
|
|
|
|
bool HasHashTorrent(string hash, RTorrentSettings settings);
|
2015-07-08 08:11:33 +00:00
|
|
|
|
void StartTorrent(string hash, RTorrentSettings settings);
|
2015-05-08 18:50:22 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public interface IRTorrent : IXmlRpcProxy
|
|
|
|
|
{
|
2015-05-26 05:35:06 +00:00
|
|
|
|
[XmlRpcMethod("d.multicall2")]
|
|
|
|
|
object[] TorrentMulticall(params string[] parameters);
|
2015-05-08 18:50:22 +00:00
|
|
|
|
|
2015-07-08 08:11:33 +00:00
|
|
|
|
[XmlRpcMethod("load.normal")]
|
|
|
|
|
int LoadUrl(string target, string data);
|
2015-05-08 18:50:22 +00:00
|
|
|
|
|
2015-07-08 08:11:33 +00:00
|
|
|
|
[XmlRpcMethod("load.raw")]
|
2015-05-27 05:01:38 +00:00
|
|
|
|
int LoadBinary(string target, byte[] data);
|
2015-05-08 18:50:22 +00:00
|
|
|
|
|
|
|
|
|
[XmlRpcMethod("d.erase")]
|
|
|
|
|
int Remove(string hash);
|
|
|
|
|
|
2015-05-26 05:35:06 +00:00
|
|
|
|
[XmlRpcMethod("d.custom1.set")]
|
2015-05-08 18:50:22 +00:00
|
|
|
|
string SetLabel(string hash, string label);
|
|
|
|
|
|
2015-05-26 05:35:06 +00:00
|
|
|
|
[XmlRpcMethod("d.priority.set")]
|
2015-05-08 18:50:22 +00:00
|
|
|
|
int SetPriority(string hash, long priority);
|
|
|
|
|
|
2015-07-08 08:11:33 +00:00
|
|
|
|
[XmlRpcMethod("d.directory.set")]
|
|
|
|
|
int SetDirectory(string hash, string directory);
|
|
|
|
|
|
2015-05-26 05:35:06 +00:00
|
|
|
|
[XmlRpcMethod("d.name")]
|
2015-05-08 18:50:22 +00:00
|
|
|
|
string GetName(string hash);
|
|
|
|
|
|
|
|
|
|
[XmlRpcMethod("system.client_version")]
|
|
|
|
|
string GetVersion();
|
2015-07-08 08:11:33 +00:00
|
|
|
|
|
|
|
|
|
[XmlRpcMethod("system.multicall")]
|
|
|
|
|
object[] SystemMulticall(object[] parameters);
|
2015-05-08 18:50:22 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class RTorrentProxy : IRTorrentProxy
|
|
|
|
|
{
|
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
|
|
|
|
public RTorrentProxy(Logger logger)
|
|
|
|
|
{
|
|
|
|
|
_logger = logger;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public string GetVersion(RTorrentSettings settings)
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Executing remote method: system.client_version");
|
|
|
|
|
|
|
|
|
|
var client = BuildClient(settings);
|
|
|
|
|
|
|
|
|
|
var version = client.GetVersion();
|
|
|
|
|
|
|
|
|
|
return version;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public List<RTorrentTorrent> GetTorrents(RTorrentSettings settings)
|
|
|
|
|
{
|
2015-05-26 05:35:06 +00:00
|
|
|
|
_logger.Debug("Executing remote method: d.multicall2");
|
2015-05-08 18:50:22 +00:00
|
|
|
|
|
|
|
|
|
var client = BuildClient(settings);
|
2015-05-26 05:35:06 +00:00
|
|
|
|
var ret = client.TorrentMulticall("", "",
|
|
|
|
|
"d.name=", // string
|
|
|
|
|
"d.hash=", // string
|
|
|
|
|
"d.base_path=", // string
|
|
|
|
|
"d.custom1=", // string (label)
|
|
|
|
|
"d.size_bytes=", // long
|
|
|
|
|
"d.left_bytes=", // long
|
|
|
|
|
"d.down.rate=", // long (in bytes / s)
|
|
|
|
|
"d.ratio=", // long
|
2015-05-08 18:50:22 +00:00
|
|
|
|
"d.is_open=", // long
|
|
|
|
|
"d.is_active=", // long
|
2015-05-26 05:35:06 +00:00
|
|
|
|
"d.complete="); //long
|
2015-05-08 18:50:22 +00:00
|
|
|
|
|
|
|
|
|
var items = new List<RTorrentTorrent>();
|
|
|
|
|
foreach (object[] torrent in ret)
|
|
|
|
|
{
|
2015-10-02 12:35:09 +00:00
|
|
|
|
var labelDecoded = System.Web.HttpUtility.UrlDecode((string) torrent[3]);
|
|
|
|
|
|
2015-05-08 18:50:22 +00:00
|
|
|
|
var item = new RTorrentTorrent();
|
|
|
|
|
item.Name = (string) torrent[0];
|
|
|
|
|
item.Hash = (string) torrent[1];
|
|
|
|
|
item.Path = (string) torrent[2];
|
2015-10-02 12:35:09 +00:00
|
|
|
|
item.Category = labelDecoded;
|
2015-05-08 18:50:22 +00:00
|
|
|
|
item.TotalSize = (long) torrent[4];
|
|
|
|
|
item.RemainingSize = (long) torrent[5];
|
|
|
|
|
item.DownRate = (long) torrent[6];
|
|
|
|
|
item.Ratio = (long) torrent[7];
|
|
|
|
|
item.IsOpen = Convert.ToBoolean((long) torrent[8]);
|
|
|
|
|
item.IsActive = Convert.ToBoolean((long) torrent[9]);
|
|
|
|
|
item.IsFinished = Convert.ToBoolean((long) torrent[10]);
|
|
|
|
|
|
|
|
|
|
items.Add(item);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return items;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void AddTorrentFromUrl(string torrentUrl, RTorrentSettings settings)
|
|
|
|
|
{
|
2015-07-08 08:11:33 +00:00
|
|
|
|
_logger.Debug("Executing remote method: load.normal");
|
2015-05-08 18:50:22 +00:00
|
|
|
|
|
|
|
|
|
var client = BuildClient(settings);
|
|
|
|
|
|
2015-07-08 08:11:33 +00:00
|
|
|
|
var response = client.LoadUrl("", torrentUrl);
|
2015-05-08 18:50:22 +00:00
|
|
|
|
if (response != 0)
|
|
|
|
|
{
|
|
|
|
|
throw new DownloadClientException("Could not add torrent: {0}.", torrentUrl);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2015-10-03 17:45:26 +00:00
|
|
|
|
public void AddTorrentFromFile(string fileName, byte[] fileContent, RTorrentSettings settings)
|
2015-05-08 18:50:22 +00:00
|
|
|
|
{
|
2015-07-08 08:11:33 +00:00
|
|
|
|
_logger.Debug("Executing remote method: load.raw");
|
2015-05-08 18:50:22 +00:00
|
|
|
|
|
|
|
|
|
var client = BuildClient(settings);
|
|
|
|
|
|
2015-05-27 05:01:38 +00:00
|
|
|
|
var response = client.LoadBinary("", fileContent);
|
2015-05-08 18:50:22 +00:00
|
|
|
|
if (response != 0)
|
|
|
|
|
{
|
|
|
|
|
throw new DownloadClientException("Could not add torrent: {0}.", fileName);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void RemoveTorrent(string hash, RTorrentSettings settings)
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Executing remote method: d.erase");
|
|
|
|
|
|
|
|
|
|
var client = BuildClient(settings);
|
|
|
|
|
|
|
|
|
|
var response = client.Remove(hash);
|
|
|
|
|
if (response != 0)
|
|
|
|
|
{
|
|
|
|
|
throw new DownloadClientException("Could not remove torrent: {0}.", hash);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2015-07-08 08:11:33 +00:00
|
|
|
|
public void SetTorrentPriority(string hash, RTorrentPriority priority, RTorrentSettings settings)
|
2015-05-08 18:50:22 +00:00
|
|
|
|
{
|
2015-05-26 05:35:06 +00:00
|
|
|
|
_logger.Debug("Executing remote method: d.priority.set");
|
2015-05-08 18:50:22 +00:00
|
|
|
|
|
|
|
|
|
var client = BuildClient(settings);
|
|
|
|
|
|
|
|
|
|
var response = client.SetPriority(hash, (long) priority);
|
|
|
|
|
if (response != 0)
|
|
|
|
|
{
|
|
|
|
|
throw new DownloadClientException("Could not set priority on torrent: {0}.", hash);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void SetTorrentLabel(string hash, string label, RTorrentSettings settings)
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Executing remote method: d.set_custom1");
|
|
|
|
|
|
2015-10-02 12:35:09 +00:00
|
|
|
|
var labelEncoded = System.Web.HttpUtility.UrlEncode(label);
|
|
|
|
|
|
2015-05-08 18:50:22 +00:00
|
|
|
|
var client = BuildClient(settings);
|
|
|
|
|
|
2015-10-02 12:35:09 +00:00
|
|
|
|
var setLabel = client.SetLabel(hash, labelEncoded);
|
|
|
|
|
if (setLabel != labelEncoded)
|
2015-05-08 18:50:22 +00:00
|
|
|
|
{
|
|
|
|
|
throw new DownloadClientException("Could set label on torrent: {0}.", hash);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2015-07-08 08:11:33 +00:00
|
|
|
|
public void SetTorrentDownloadDirectory(string hash, string directory, RTorrentSettings settings)
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Executing remote method: d.directory.set");
|
|
|
|
|
|
|
|
|
|
var client = BuildClient(settings);
|
|
|
|
|
|
|
|
|
|
var response = client.SetDirectory(hash, directory);
|
|
|
|
|
if (response != 0)
|
|
|
|
|
{
|
|
|
|
|
throw new DownloadClientException("Could not set directory for torrent: {0}.", hash);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public bool HasHashTorrent(string hash, RTorrentSettings settings)
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Executing remote method: d.name");
|
|
|
|
|
|
|
|
|
|
var client = BuildClient(settings);
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
var name = client.GetName(hash);
|
|
|
|
|
if (name.IsNullOrWhiteSpace()) return false;
|
|
|
|
|
bool metaTorrent = name == (hash + ".meta");
|
|
|
|
|
return !metaTorrent;
|
|
|
|
|
}
|
|
|
|
|
catch (Exception)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void StartTorrent(string hash, RTorrentSettings settings)
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Executing remote methods: d.open and d.start");
|
|
|
|
|
|
|
|
|
|
var client = BuildClient(settings);
|
|
|
|
|
|
|
|
|
|
var multicallResponse = client.SystemMulticall(new[]
|
|
|
|
|
{
|
|
|
|
|
new
|
|
|
|
|
{
|
|
|
|
|
methodName = "d.open",
|
|
|
|
|
@params = new[] { hash }
|
|
|
|
|
},
|
|
|
|
|
new
|
|
|
|
|
{
|
|
|
|
|
methodName = "d.start",
|
|
|
|
|
@params = new[] { hash }
|
|
|
|
|
},
|
|
|
|
|
}).SelectMany(c => ((IEnumerable<int>)c));
|
|
|
|
|
|
|
|
|
|
if (multicallResponse.Any(r => r != 0))
|
|
|
|
|
{
|
|
|
|
|
throw new DownloadClientException("Could not start torrent: {0}.", hash);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2015-05-08 18:50:22 +00:00
|
|
|
|
private IRTorrent BuildClient(RTorrentSettings settings)
|
|
|
|
|
{
|
2015-05-27 05:01:38 +00:00
|
|
|
|
var client = XmlRpcProxyGen.Create<IRTorrent>();
|
|
|
|
|
|
|
|
|
|
client.Url = string.Format(@"{0}://{1}:{2}/{3}",
|
2015-05-08 18:50:22 +00:00
|
|
|
|
settings.UseSsl ? "https" : "http",
|
|
|
|
|
settings.Host,
|
|
|
|
|
settings.Port,
|
|
|
|
|
settings.UrlBase);
|
|
|
|
|
|
2015-05-26 05:35:06 +00:00
|
|
|
|
client.EnableCompression = true;
|
|
|
|
|
|
2015-05-08 18:50:22 +00:00
|
|
|
|
if (!settings.Username.IsNullOrWhiteSpace())
|
|
|
|
|
{
|
|
|
|
|
client.Credentials = new NetworkCredential(settings.Username, settings.Password);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return client;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|