using System ;
using System.Collections.Generic ;
using System.IO ;
using System.Linq ;
using System.Net ;
using FluentValidation.Results ;
using NLog ;
using NzbDrone.Common.Disk ;
using NzbDrone.Common.Extensions ;
using NzbDrone.Common.Http ;
using NzbDrone.Core.Blocklisting ;
using NzbDrone.Core.Configuration ;
using NzbDrone.Core.Download.Clients.DownloadStation.Proxies ;
using NzbDrone.Core.Localization ;
using NzbDrone.Core.MediaFiles.TorrentInfo ;
using NzbDrone.Core.Parser.Model ;
using NzbDrone.Core.RemotePathMappings ;
using NzbDrone.Core.ThingiProvider ;
using NzbDrone.Core.Validation ;
namespace NzbDrone.Core.Download.Clients.DownloadStation
{
public class TorrentDownloadStation : TorrentClientBase < DownloadStationSettings >
{
protected readonly IDownloadStationInfoProxy _dsInfoProxy ;
protected readonly IDownloadStationTaskProxySelector _dsTaskProxySelector ;
protected readonly ISharedFolderResolver _sharedFolderResolver ;
protected readonly ISerialNumberProvider _serialNumberProvider ;
protected readonly IFileStationProxy _fileStationProxy ;
public TorrentDownloadStation ( ISharedFolderResolver sharedFolderResolver ,
ISerialNumberProvider serialNumberProvider ,
IFileStationProxy fileStationProxy ,
IDownloadStationInfoProxy dsInfoProxy ,
IDownloadStationTaskProxySelector dsTaskProxySelector ,
ITorrentFileInfoReader torrentFileInfoReader ,
IHttpClient httpClient ,
IConfigService configService ,
IDiskProvider diskProvider ,
IRemotePathMappingService remotePathMappingService ,
ILocalizationService localizationService ,
IBlocklistService blocklistService ,
Logger logger )
: base ( torrentFileInfoReader , httpClient , configService , diskProvider , remotePathMappingService , localizationService , blocklistService , logger )
{
_dsInfoProxy = dsInfoProxy ;
_dsTaskProxySelector = dsTaskProxySelector ;
_fileStationProxy = fileStationProxy ;
_sharedFolderResolver = sharedFolderResolver ;
_serialNumberProvider = serialNumberProvider ;
}
public override string Name = > "Download Station" ;
public override ProviderMessage Message = > new ProviderMessage ( "Radarr is unable to connect to Download Station if 2-Factor Authentication is enabled on your DSM account" , ProviderMessageType . Warning ) ;
private IDownloadStationTaskProxy DsTaskProxy = > _dsTaskProxySelector . GetProxy ( Settings ) ;
protected IEnumerable < DownloadStationTask > GetTasks ( )
{
return DsTaskProxy . GetTasks ( Settings ) . Where ( v = > v . Type . ToLower ( ) = = DownloadStationTaskType . BT . ToString ( ) . ToLower ( ) ) ;
}
public override IEnumerable < DownloadClientItem > GetItems ( )
{
var torrents = GetTasks ( ) ;
var serialNumber = _serialNumberProvider . GetSerialNumber ( Settings ) ;
var items = new List < DownloadClientItem > ( ) ;
foreach ( var torrent in torrents )
{
var outputPath = new OsPath ( $"/{torrent.Additional.Detail[" destination "]}" ) ;
if ( Settings . TvDirectory . IsNotNullOrWhiteSpace ( ) )
{
if ( ! new OsPath ( $"/{Settings.TvDirectory}" ) . Contains ( outputPath ) )
{
continue ;
}
}
else if ( Settings . TvCategory . IsNotNullOrWhiteSpace ( ) )
{
var directories = outputPath . FullPath . Split ( '\\' , '/' ) ;
if ( ! directories . Contains ( Settings . TvCategory ) )
{
continue ;
}
}
var item = new DownloadClientItem ( )
{
Category = Settings . TvCategory ,
DownloadClientInfo = DownloadClientItemClientInfo . FromDownloadClient ( this , false ) ,
DownloadId = CreateDownloadId ( torrent . Id , serialNumber ) ,
Title = torrent . Title ,
TotalSize = torrent . Size ,
RemainingSize = GetRemainingSize ( torrent ) ,
RemainingTime = GetRemainingTime ( torrent ) ,
SeedRatio = GetSeedRatio ( torrent ) ,
Status = GetStatus ( torrent ) ,
Message = GetMessage ( torrent ) ,
CanMoveFiles = IsFinished ( torrent ) ,
CanBeRemoved = IsFinished ( torrent )
} ;
if ( item . Status = = DownloadItemStatus . Completed | | item . Status = = DownloadItemStatus . Failed )
{
item . OutputPath = GetOutputPath ( outputPath , torrent , serialNumber ) ;
}
items . Add ( item ) ;
}
return items ;
}
public override DownloadClientInfo GetStatus ( )
{
try
{
var serialNumber = _serialNumberProvider . GetSerialNumber ( Settings ) ;
// Download station returns the path without the leading `/`, but the leading
// slash is required to get the full path back from download station.
var path = new OsPath ( $"/{GetDownloadDirectory()}" ) ;
var fullPath = _sharedFolderResolver . RemapToFullPath ( path , Settings , serialNumber ) ;
return new DownloadClientInfo
{
IsLocalhost = Settings . Host = = "127.0.0.1" | | Settings . Host = = "localhost" ,
OutputRootFolders = new List < OsPath > { _remotePathMappingService . RemapRemoteToLocal ( Settings . Host , fullPath ) }
} ;
}
catch ( DownloadClientException e )
{
_logger . Debug ( e , "Failed to get config from Download Station" ) ;
throw ;
}
}
public override void RemoveItem ( DownloadClientItem item , bool deleteData )
{
if ( deleteData )
{
DeleteItemData ( item ) ;
}
DsTaskProxy . RemoveTask ( ParseDownloadId ( item . DownloadId ) , Settings ) ;
_logger . Debug ( "{0} removed correctly" , item . DownloadId ) ;
}
protected OsPath GetOutputPath ( OsPath outputPath , DownloadStationTask torrent , string serialNumber )
{
var fullPath = _sharedFolderResolver . RemapToFullPath ( outputPath , Settings , serialNumber ) ;
var remotePath = _remotePathMappingService . RemapRemoteToLocal ( Settings . Host , fullPath ) ;
var finalPath = remotePath + torrent . Title ;
return finalPath ;
}
protected override string AddFromMagnetLink ( RemoteMovie remoteMovie , string hash , string magnetLink )
{
var hashedSerialNumber = _serialNumberProvider . GetSerialNumber ( Settings ) ;
DsTaskProxy . AddTaskFromUrl ( magnetLink , GetDownloadDirectory ( ) , Settings ) ;
var item = GetTasks ( ) . SingleOrDefault ( t = > t . Additional . Detail [ "uri" ] = = magnetLink ) ;
if ( item ! = null )
{
_logger . Debug ( "{0} added correctly" , remoteMovie ) ;
return CreateDownloadId ( item . Id , hashedSerialNumber ) ;
}
_logger . Debug ( "No such task {0} in Download Station" , magnetLink ) ;
throw new DownloadClientException ( "Failed to add magnet task to Download Station" ) ;
}
protected override string AddFromTorrentFile ( RemoteMovie remoteMovie , string hash , string filename , byte [ ] fileContent )
{
var hashedSerialNumber = _serialNumberProvider . GetSerialNumber ( Settings ) ;
DsTaskProxy . AddTaskFromData ( fileContent , filename , GetDownloadDirectory ( ) , Settings ) ;
var items = GetTasks ( ) . Where ( t = > t . Additional . Detail [ "uri" ] = = Path . GetFileNameWithoutExtension ( filename ) ) ;
var item = items . SingleOrDefault ( ) ;
if ( item ! = null )
{
_logger . Debug ( "{0} added correctly" , remoteMovie ) ;
return CreateDownloadId ( item . Id , hashedSerialNumber ) ;
}
_logger . Debug ( "No such task {0} in Download Station" , filename ) ;
throw new DownloadClientException ( "Failed to add torrent task to Download Station" ) ;
}
protected override void Test ( List < ValidationFailure > failures )
{
failures . AddIfNotNull ( TestConnection ( ) ) ;
if ( failures . HasErrors ( ) )
{
return ;
}
failures . AddIfNotNull ( TestOutputPath ( ) ) ;
failures . AddIfNotNull ( TestGetTorrents ( ) ) ;
}
protected bool IsFinished ( DownloadStationTask torrent )
{
return torrent . Status = = DownloadStationTaskStatus . Finished ;
}
protected string GetMessage ( DownloadStationTask torrent )
{
if ( torrent . StatusExtra ! = null )
{
if ( torrent . Status = = DownloadStationTaskStatus . Extracting )
{
return $"Extracting: {int.Parse(torrent.StatusExtra[" unzip_progress "])}%" ;
}
if ( torrent . Status = = DownloadStationTaskStatus . Error )
{
return torrent . StatusExtra [ "error_detail" ] ;
}
}
return null ;
}
protected DownloadItemStatus GetStatus ( DownloadStationTask torrent )
{
switch ( torrent . Status )
{
case DownloadStationTaskStatus . Unknown :
case DownloadStationTaskStatus . Waiting :
case DownloadStationTaskStatus . FilehostingWaiting :
return torrent . Size = = 0 | | GetRemainingSize ( torrent ) > 0 ? DownloadItemStatus . Queued : DownloadItemStatus . Completed ;
case DownloadStationTaskStatus . Paused :
return DownloadItemStatus . Paused ;
case DownloadStationTaskStatus . Finished :
case DownloadStationTaskStatus . Seeding :
return DownloadItemStatus . Completed ;
case DownloadStationTaskStatus . Error :
return DownloadItemStatus . Failed ;
}
return DownloadItemStatus . Downloading ;
}
protected long GetRemainingSize ( DownloadStationTask torrent )
{
var downloadedString = torrent . Additional . Transfer [ "size_downloaded" ] ;
if ( downloadedString . IsNullOrWhiteSpace ( ) | | ! long . TryParse ( downloadedString , out var downloadedSize ) )
{
_logger . Debug ( "Torrent {0} has invalid size_downloaded: {1}" , torrent . Title , downloadedString ) ;
downloadedSize = 0 ;
}
return torrent . Size - Math . Max ( 0 , downloadedSize ) ;
}
protected TimeSpan ? GetRemainingTime ( DownloadStationTask torrent )
{
var speedString = torrent . Additional . Transfer [ "speed_download" ] ;
if ( speedString . IsNullOrWhiteSpace ( ) | | ! long . TryParse ( speedString , out var downloadSpeed ) )
{
_logger . Debug ( "Torrent {0} has invalid speed_download: {1}" , torrent . Title , speedString ) ;
downloadSpeed = 0 ;
}
if ( downloadSpeed < = 0 )
{
return null ;
}
var remainingSize = GetRemainingSize ( torrent ) ;
return TimeSpan . FromSeconds ( remainingSize / downloadSpeed ) ;
}
protected double? GetSeedRatio ( DownloadStationTask torrent )
{
var downloaded = torrent . Additional . Transfer [ "size_downloaded" ] . ParseInt64 ( ) ;
var uploaded = torrent . Additional . Transfer [ "size_uploaded" ] . ParseInt64 ( ) ;
if ( downloaded . HasValue & & uploaded . HasValue )
{
return downloaded < = 0 ? 0 : ( double ) uploaded . Value / downloaded . Value ;
}
return null ;
}
protected ValidationFailure TestOutputPath ( )
{
try
{
var downloadDir = GetDownloadDirectory ( ) ;
if ( downloadDir = = null )
{
return new NzbDroneValidationFailure ( nameof ( Settings . TvDirectory ) , _localizationService . GetLocalizedString ( "DownloadClientDownloadStationValidationNoDefaultDestination" ) )
{
DetailedDescription = _localizationService . GetLocalizedString ( "DownloadClientDownloadStationValidationNoDefaultDestinationDetail" , new Dictionary < string , object > { { "username" , Settings . Username } } )
} ;
}
var sharedFolder = downloadDir . Split ( '\\' , '/' ) [ 0 ] ;
var fieldName = Settings . TvDirectory . IsNotNullOrWhiteSpace ( ) ? nameof ( Settings . TvDirectory ) : nameof ( Settings . TvCategory ) ;
var folderInfo = _fileStationProxy . GetInfoFileOrDirectory ( $"/{downloadDir}" , Settings ) ;
if ( folderInfo . Additional = = null )
{
return new NzbDroneValidationFailure ( fieldName , _localizationService . GetLocalizedString ( "DownloadClientDownloadStationValidationSharedFolderMissing" ) )
{
DetailedDescription = _localizationService . GetLocalizedString ( "DownloadClientDownloadStationValidationSharedFolderMissingDetail" , new Dictionary < string , object > { { "sharedFolder" , sharedFolder } } )
} ;
}
if ( ! folderInfo . IsDir )
{
return new NzbDroneValidationFailure ( fieldName , _localizationService . GetLocalizedString ( "DownloadClientDownloadStationValidationFolderMissing" ) )
{
DetailedDescription = _localizationService . GetLocalizedString ( "DownloadClientDownloadStationValidationFolderMissingDetail" , new Dictionary < string , object > { { "downloadDir" , downloadDir } , { "sharedFolder" , sharedFolder } } )
} ;
}
return null ;
}
catch ( DownloadClientAuthenticationException ex )
{
// User could not have permission to access to downloadstation
_logger . Error ( ex , ex . Message ) ;
return new NzbDroneValidationFailure ( string . Empty , ex . Message ) ;
}
catch ( Exception ex )
{
_logger . Error ( ex , "Error testing Torrent Download Station" ) ;
return new NzbDroneValidationFailure ( string . Empty , _localizationService . GetLocalizedString ( "DownloadClientValidationUnknownException" , new Dictionary < string , object > { { "exception" , ex . Message } } ) ) ;
}
}
protected ValidationFailure TestConnection ( )
{
try
{
return ValidateVersion ( ) ;
}
catch ( DownloadClientAuthenticationException ex )
{
_logger . Error ( ex , ex . Message ) ;
return new NzbDroneValidationFailure ( "Username" , "Authentication failure" )
{
DetailedDescription = $"Please verify your username and password. Also verify if the host running Radarr isn't blocked from accessing {Name} by WhiteList limitations in the {Name} configuration."
} ;
}
catch ( WebException ex )
{
_logger . Error ( ex , "Unable to connect to Torrent Download Station" ) ;
if ( ex . Status = = WebExceptionStatus . ConnectFailure )
{
return new NzbDroneValidationFailure ( "Host" , "Unable to connect" )
{
DetailedDescription = "Please verify the hostname and port."
} ;
}
return new NzbDroneValidationFailure ( string . Empty , $"Unknown exception: {ex.Message}" ) ;
}
catch ( Exception ex )
{
_logger . Error ( ex , "Error testing Torrent Download Station" ) ;
return new NzbDroneValidationFailure ( "Host" , "Unable to connect to Torrent Download Station" )
{
DetailedDescription = ex . Message
} ;
}
}
protected ValidationFailure ValidateVersion ( )
{
var info = DsTaskProxy . GetApiInfo ( Settings ) ;
_logger . Debug ( "Download Station api version information: Min {0} - Max {1}" , info . MinVersion , info . MaxVersion ) ;
if ( info . MinVersion > 2 | | info . MaxVersion < 2 )
{
return new ValidationFailure ( string . Empty , $"Download Station API version not supported, should be at least 2. It supports from {info.MinVersion} to {info.MaxVersion}" ) ;
}
return null ;
}
protected ValidationFailure TestGetTorrents ( )
{
try
{
GetItems ( ) ;
return null ;
}
catch ( Exception ex )
{
return new NzbDroneValidationFailure ( string . Empty , $"Failed to get the list of torrents: {ex.Message}" ) ;
}
}
protected string ParseDownloadId ( string id )
{
return id . Split ( ':' ) [ 1 ] ;
}
protected string CreateDownloadId ( string id , string hashedSerialNumber )
{
return $"{hashedSerialNumber}:{id}" ;
}
protected string GetDefaultDir ( )
{
var config = _dsInfoProxy . GetConfig ( Settings ) ;
var path = config [ "default_destination" ] as string ;
return path ;
}
protected string GetDownloadDirectory ( )
{
if ( Settings . TvDirectory . IsNotNullOrWhiteSpace ( ) )
{
return Settings . TvDirectory . TrimStart ( '/' ) ;
}
var destDir = GetDefaultDir ( ) ;
if ( destDir . IsNotNullOrWhiteSpace ( ) & & Settings . TvCategory . IsNotNullOrWhiteSpace ( ) )
{
return $"{destDir.TrimEnd('/')}/{Settings.TvCategory}" ;
}
return destDir . TrimEnd ( '/' ) ;
}
}
}