2021-03-30 07:58:36 +00:00
|
|
|
using System;
|
|
|
|
using System.Linq;
|
2023-01-17 04:38:05 +00:00
|
|
|
using System.Net.Http;
|
2021-03-30 07:58:36 +00:00
|
|
|
using NLog;
|
2021-05-08 23:43:56 +00:00
|
|
|
using NzbDrone.Common.Extensions;
|
2021-03-30 07:58:36 +00:00
|
|
|
using NzbDrone.Core.Datastore.Events;
|
|
|
|
using NzbDrone.Core.Download;
|
|
|
|
using NzbDrone.Core.Download.Clients;
|
|
|
|
using NzbDrone.Core.Localization;
|
|
|
|
using NzbDrone.Core.RemotePathMappings;
|
|
|
|
using NzbDrone.Core.RootFolders;
|
|
|
|
using NzbDrone.Core.ThingiProvider.Events;
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.HealthCheck.Checks
|
|
|
|
{
|
|
|
|
[CheckOn(typeof(ProviderAddedEvent<IDownloadClient>))]
|
|
|
|
[CheckOn(typeof(ProviderUpdatedEvent<IDownloadClient>))]
|
|
|
|
[CheckOn(typeof(ProviderDeletedEvent<IDownloadClient>))]
|
|
|
|
[CheckOn(typeof(ModelEvent<RootFolder>))]
|
|
|
|
[CheckOn(typeof(ModelEvent<RemotePathMapping>))]
|
|
|
|
|
|
|
|
public class DownloadClientRootFolderCheck : HealthCheckBase, IProvideHealthCheck
|
|
|
|
{
|
|
|
|
private readonly IProvideDownloadClient _downloadClientProvider;
|
|
|
|
private readonly IRootFolderService _rootFolderService;
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
|
|
public DownloadClientRootFolderCheck(IProvideDownloadClient downloadClientProvider,
|
|
|
|
IRootFolderService rootFolderService,
|
|
|
|
Logger logger,
|
|
|
|
ILocalizationService localizationService)
|
|
|
|
: base(localizationService)
|
|
|
|
{
|
|
|
|
_downloadClientProvider = downloadClientProvider;
|
|
|
|
_rootFolderService = rootFolderService;
|
|
|
|
_logger = logger;
|
|
|
|
}
|
|
|
|
|
|
|
|
public override HealthCheck Check()
|
|
|
|
{
|
2023-01-17 04:38:05 +00:00
|
|
|
// Only check clients not in failure status, those get another message
|
|
|
|
var clients = _downloadClientProvider.GetDownloadClients(true);
|
|
|
|
|
2021-03-30 07:58:36 +00:00
|
|
|
var rootFolders = _rootFolderService.All();
|
|
|
|
|
|
|
|
foreach (var client in clients)
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
var status = client.GetStatus();
|
|
|
|
var folders = status.OutputRootFolders;
|
2021-05-08 23:43:56 +00:00
|
|
|
foreach (var folder in folders)
|
2021-03-30 07:58:36 +00:00
|
|
|
{
|
2021-05-08 23:43:56 +00:00
|
|
|
if (rootFolders.Any(r => r.Path.PathEquals(folder.FullPath)))
|
2021-03-30 07:58:36 +00:00
|
|
|
{
|
2021-06-05 19:26:09 +00:00
|
|
|
return new HealthCheck(GetType(), HealthCheckResult.Warning, string.Format(_localizationService.GetLocalizedString("DownloadClientCheckDownloadingToRoot"), client.Definition.Name, folder.FullPath), "#downloads-in-root-folder");
|
2021-03-30 07:58:36 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
catch (DownloadClientException ex)
|
|
|
|
{
|
|
|
|
_logger.Debug(ex, "Unable to communicate with {0}", client.Definition.Name);
|
|
|
|
}
|
2023-01-17 04:38:05 +00:00
|
|
|
catch (HttpRequestException ex)
|
|
|
|
{
|
|
|
|
_logger.Debug(ex, "Unable to communicate with {0}", client.Definition.Name);
|
|
|
|
}
|
2021-03-30 07:58:36 +00:00
|
|
|
catch (Exception ex)
|
|
|
|
{
|
|
|
|
_logger.Error(ex, "Unknown error occured in DownloadClientRootFolderCheck HealthCheck");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return new HealthCheck(GetType());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|