mirror of https://github.com/Radarr/Radarr
61 lines
2.3 KiB
C#
61 lines
2.3 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Core.Download;
|
|
using NzbDrone.Core.Localization;
|
|
using NzbDrone.Core.ThingiProvider.Events;
|
|
|
|
namespace NzbDrone.Core.HealthCheck.Checks
|
|
{
|
|
[CheckOn(typeof(ProviderAddedEvent<IDownloadClient>))]
|
|
[CheckOn(typeof(ProviderUpdatedEvent<IDownloadClient>))]
|
|
[CheckOn(typeof(ProviderDeletedEvent<IDownloadClient>))]
|
|
[CheckOn(typeof(ProviderStatusChangedEvent<IDownloadClient>))]
|
|
public class DownloadClientCheck : HealthCheckBase
|
|
{
|
|
private readonly IProvideDownloadClient _downloadClientProvider;
|
|
private readonly Logger _logger;
|
|
|
|
public DownloadClientCheck(IProvideDownloadClient downloadClientProvider, ILocalizationService localizationService, Logger logger)
|
|
: base(localizationService)
|
|
{
|
|
_downloadClientProvider = downloadClientProvider;
|
|
_logger = logger;
|
|
}
|
|
|
|
public override HealthCheck Check()
|
|
{
|
|
var downloadClients = _downloadClientProvider.GetDownloadClients().ToList();
|
|
|
|
if (!downloadClients.Any())
|
|
{
|
|
return new HealthCheck(GetType(), HealthCheckResult.Warning, _localizationService.GetLocalizedString("DownloadClientCheckNoneAvailableMessage"), "#no-download-client-is-available");
|
|
}
|
|
|
|
foreach (var downloadClient in downloadClients)
|
|
{
|
|
try
|
|
{
|
|
downloadClient.GetItems();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.Debug(ex, "Unable to communicate with {0}", downloadClient.Definition.Name);
|
|
|
|
return new HealthCheck(GetType(),
|
|
HealthCheckResult.Error,
|
|
_localizationService.GetLocalizedString("DownloadClientCheckUnableToCommunicateMessage", new Dictionary<string, object>
|
|
{
|
|
{ "downloadClientName", downloadClient.Definition.Name },
|
|
{ "errorMessage", ex.Message }
|
|
}),
|
|
"#unable-to-communicate-with-download-client");
|
|
}
|
|
}
|
|
|
|
return new HealthCheck(GetType());
|
|
}
|
|
}
|
|
}
|