mirror of https://github.com/Radarr/Radarr
85 lines
3.3 KiB
C#
85 lines
3.3 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NzbDrone.Common.Disk;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.ImportLists;
|
|
using NzbDrone.Core.Localization;
|
|
using NzbDrone.Core.MediaFiles.Events;
|
|
using NzbDrone.Core.Movies.Events;
|
|
using NzbDrone.Core.ThingiProvider.Events;
|
|
|
|
namespace NzbDrone.Core.HealthCheck.Checks
|
|
{
|
|
[CheckOn(typeof(ProviderUpdatedEvent<IImportList>))]
|
|
[CheckOn(typeof(MoviesDeletedEvent))]
|
|
[CheckOn(typeof(MovieMovedEvent))]
|
|
[CheckOn(typeof(MovieFileImportedEvent), CheckOnCondition.FailedOnly)]
|
|
[CheckOn(typeof(MovieImportFailedEvent), CheckOnCondition.SuccessfulOnly)]
|
|
public class ImportListRootFolderCheck : HealthCheckBase
|
|
{
|
|
private readonly IImportListFactory _importListFactory;
|
|
private readonly IDiskProvider _diskProvider;
|
|
|
|
public ImportListRootFolderCheck(IImportListFactory importListFactory, IDiskProvider diskProvider, ILocalizationService localizationService)
|
|
: base(localizationService)
|
|
{
|
|
_importListFactory = importListFactory;
|
|
_diskProvider = diskProvider;
|
|
}
|
|
|
|
public override HealthCheck Check()
|
|
{
|
|
var importLists = _importListFactory.All();
|
|
var missingRootFolders = new Dictionary<string, List<ImportListDefinition>>();
|
|
|
|
foreach (var importList in importLists)
|
|
{
|
|
var rootFolderPath = importList.RootFolderPath;
|
|
|
|
if (missingRootFolders.ContainsKey(rootFolderPath))
|
|
{
|
|
missingRootFolders[rootFolderPath].Add(importList);
|
|
|
|
continue;
|
|
}
|
|
|
|
if (rootFolderPath.IsNullOrWhiteSpace() || !_diskProvider.FolderExists(rootFolderPath))
|
|
{
|
|
missingRootFolders.Add(rootFolderPath, new List<ImportListDefinition> { importList });
|
|
}
|
|
}
|
|
|
|
if (missingRootFolders.Any())
|
|
{
|
|
if (missingRootFolders.Count == 1)
|
|
{
|
|
var missingRootFolder = missingRootFolders.First();
|
|
|
|
return new HealthCheck(GetType(),
|
|
HealthCheckResult.Error,
|
|
_localizationService.GetLocalizedString("ImportListMissingRoot", new Dictionary<string, object>
|
|
{
|
|
{ "rootFolderInfo", FormatRootFolder(missingRootFolder.Key, missingRootFolder.Value) }
|
|
}),
|
|
"#import-list-missing-root-folder");
|
|
}
|
|
|
|
return new HealthCheck(GetType(),
|
|
HealthCheckResult.Error,
|
|
_localizationService.GetLocalizedString("ImportListMultipleMissingRoots", new Dictionary<string, object>
|
|
{
|
|
{ "rootFoldersInfo", string.Join(" | ", missingRootFolders.Select(m => FormatRootFolder(m.Key, m.Value))) }
|
|
}),
|
|
"#import-list-missing-root-folder");
|
|
}
|
|
|
|
return new HealthCheck(GetType());
|
|
}
|
|
|
|
private string FormatRootFolder(string rootFolderPath, List<ImportListDefinition> importLists)
|
|
{
|
|
return $"{rootFolderPath} ({string.Join(", ", importLists.Select(l => l.Name))})";
|
|
}
|
|
}
|
|
}
|