|
|
|
@ -1,101 +0,0 @@
|
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using NzbDrone.Common.Disk;
|
|
|
|
|
using NzbDrone.Core.Configuration;
|
|
|
|
|
using NzbDrone.Core.Configuration.Events;
|
|
|
|
|
using NzbDrone.Core.Download;
|
|
|
|
|
using NzbDrone.Core.Download.Clients;
|
|
|
|
|
using NzbDrone.Core.Download.Clients.Nzbget;
|
|
|
|
|
using NzbDrone.Core.Download.Clients.Sabnzbd;
|
|
|
|
|
using NzbDrone.Core.ThingiProvider.Events;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.HealthCheck.Checks
|
|
|
|
|
{
|
|
|
|
|
[CheckOn(typeof(ProviderUpdatedEvent<IDownloadClient>))]
|
|
|
|
|
[CheckOn(typeof(ProviderDeletedEvent<IDownloadClient>))]
|
|
|
|
|
[CheckOn(typeof(ConfigSavedEvent))]
|
|
|
|
|
public class ImportMechanismCheck : HealthCheckBase
|
|
|
|
|
{
|
|
|
|
|
private readonly IConfigService _configService;
|
|
|
|
|
private readonly IProvideDownloadClient _provideDownloadClient;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
public ImportMechanismCheck(IConfigService configService, IProvideDownloadClient provideDownloadClient)
|
|
|
|
|
{
|
|
|
|
|
_configService = configService;
|
|
|
|
|
_provideDownloadClient = provideDownloadClient;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public override HealthCheck Check()
|
|
|
|
|
{
|
|
|
|
|
var droneFactoryFolder = new OsPath(_configService.DownloadedEpisodesFolder);
|
|
|
|
|
List<ImportMechanismCheckStatus> downloadClients;
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
downloadClients = _provideDownloadClient.GetDownloadClients().Select(v => new ImportMechanismCheckStatus
|
|
|
|
|
{
|
|
|
|
|
DownloadClient = v,
|
|
|
|
|
Status = v.GetStatus()
|
|
|
|
|
}).ToList();
|
|
|
|
|
}
|
|
|
|
|
catch (Exception)
|
|
|
|
|
{
|
|
|
|
|
// One or more download clients failed, assume the health is okay and verify later
|
|
|
|
|
return new HealthCheck(GetType());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var downloadClientIsLocalHost = downloadClients.All(v => v.Status.IsLocalhost);
|
|
|
|
|
var downloadClientOutputInDroneFactory = !droneFactoryFolder.IsEmpty &&
|
|
|
|
|
downloadClients.Any(v => v.Status.OutputRootFolders != null &&
|
|
|
|
|
v.Status.OutputRootFolders.Any(droneFactoryFolder.Contains));
|
|
|
|
|
|
|
|
|
|
if (!_configService.IsDefined("EnableCompletedDownloadHandling"))
|
|
|
|
|
{
|
|
|
|
|
// Migration helper logic
|
|
|
|
|
if (!downloadClientIsLocalHost)
|
|
|
|
|
{
|
|
|
|
|
return new HealthCheck(GetType(), HealthCheckResult.Warning, "Enable Completed Download Handling if possible (Multi-Computer unsupported)", "Migrating-to-Completed-Download-Handling#Unsupported-download-client-on-different-computer");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (downloadClients.All(v => v.DownloadClient is Sabnzbd))
|
|
|
|
|
{
|
|
|
|
|
// With Sabnzbd we can check if the category should be changed.
|
|
|
|
|
if (downloadClientOutputInDroneFactory)
|
|
|
|
|
{
|
|
|
|
|
return new HealthCheck(GetType(), HealthCheckResult.Warning, "Enable Completed Download Handling if possible (Sabnzbd - Conflicting Category)", "Migrating-to-Completed-Download-Handling#sabnzbd-conflicting-download-client-category");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return new HealthCheck(GetType(), HealthCheckResult.Warning, "Enable Completed Download Handling if possible (Sabnzbd)", "Migrating-to-Completed-Download-Handling#sabnzbd-enable-completed-download-handling");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (downloadClients.All(v => v.DownloadClient is Nzbget))
|
|
|
|
|
{
|
|
|
|
|
// With Nzbget we can check if the category should be changed.
|
|
|
|
|
if (downloadClientOutputInDroneFactory)
|
|
|
|
|
{
|
|
|
|
|
return new HealthCheck(GetType(), HealthCheckResult.Warning, "Enable Completed Download Handling if possible (Nzbget - Conflicting Category)", "Migrating-to-Completed-Download-Handling#nzbget-conflicting-download-client-category");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return new HealthCheck(GetType(), HealthCheckResult.Warning, "Enable Completed Download Handling if possible (Nzbget)", "Migrating-to-Completed-Download-Handling#nzbget-enable-completed-download-handling");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return new HealthCheck(GetType(), HealthCheckResult.Warning, "Enable Completed Download Handling if possible", "Migrating-to-Completed-Download-Handling");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (!_configService.EnableCompletedDownloadHandling && droneFactoryFolder.IsEmpty)
|
|
|
|
|
{
|
|
|
|
|
return new HealthCheck(GetType(), HealthCheckResult.Warning, "Enable Completed Download Handling or configure Drone factory");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return new HealthCheck(GetType());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class ImportMechanismCheckStatus
|
|
|
|
|
{
|
|
|
|
|
public IDownloadClient DownloadClient { get; set; }
|
|
|
|
|
public DownloadClientInfo Status { get; set; }
|
|
|
|
|
}
|
|
|
|
|
}
|