2017-10-08 06:11:36 +00:00
|
|
|
using System.Linq;
|
2014-05-26 05:14:15 +00:00
|
|
|
using NzbDrone.Common.Disk;
|
2018-04-09 02:57:09 +00:00
|
|
|
using NzbDrone.Core.ImportLists;
|
2017-12-31 20:24:41 +00:00
|
|
|
using NzbDrone.Core.MediaFiles.Events;
|
2017-10-08 06:11:36 +00:00
|
|
|
using NzbDrone.Core.Music;
|
2017-10-27 03:21:06 +00:00
|
|
|
using NzbDrone.Core.Music.Events;
|
2020-12-13 02:28:59 +00:00
|
|
|
using NzbDrone.Core.RootFolders;
|
2014-05-26 05:14:15 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.HealthCheck.Checks
|
|
|
|
{
|
2017-10-27 03:21:06 +00:00
|
|
|
[CheckOn(typeof(ArtistDeletedEvent))]
|
|
|
|
[CheckOn(typeof(ArtistMovedEvent))]
|
2017-12-31 20:24:41 +00:00
|
|
|
[CheckOn(typeof(TrackImportedEvent), CheckOnCondition.FailedOnly)]
|
|
|
|
[CheckOn(typeof(TrackImportFailedEvent), CheckOnCondition.SuccessfulOnly)]
|
2014-05-26 05:14:15 +00:00
|
|
|
public class RootFolderCheck : HealthCheckBase
|
|
|
|
{
|
2017-10-08 06:11:36 +00:00
|
|
|
private readonly IArtistService _artistService;
|
2018-04-09 02:57:09 +00:00
|
|
|
private readonly IImportListFactory _importListFactory;
|
2014-05-26 05:14:15 +00:00
|
|
|
private readonly IDiskProvider _diskProvider;
|
2020-12-13 02:28:59 +00:00
|
|
|
private readonly IRootFolderService _rootFolderService;
|
2014-05-26 05:14:15 +00:00
|
|
|
|
2020-12-13 02:28:59 +00:00
|
|
|
public RootFolderCheck(IArtistService artistService, IImportListFactory importListFactory, IDiskProvider diskProvider, IRootFolderService rootFolderService)
|
2014-05-26 05:14:15 +00:00
|
|
|
{
|
2017-10-08 06:11:36 +00:00
|
|
|
_artistService = artistService;
|
2018-04-09 02:57:09 +00:00
|
|
|
_importListFactory = importListFactory;
|
2014-05-26 05:14:15 +00:00
|
|
|
_diskProvider = diskProvider;
|
2020-12-13 02:28:59 +00:00
|
|
|
_rootFolderService = rootFolderService;
|
2014-05-26 05:14:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public override HealthCheck Check()
|
|
|
|
{
|
2020-12-13 02:28:59 +00:00
|
|
|
var rootFolders = _artistService.GetAllArtists()
|
|
|
|
.Select(s => _rootFolderService.GetBestRootFolderPath(s.Path))
|
|
|
|
.Distinct();
|
|
|
|
|
|
|
|
var missingRootFolders = rootFolders.Where(s => !_diskProvider.FolderExists(s))
|
|
|
|
.ToList();
|
2014-05-26 05:14:15 +00:00
|
|
|
|
2018-04-09 02:57:09 +00:00
|
|
|
missingRootFolders.AddRange(_importListFactory.All()
|
|
|
|
.Select(s => s.RootFolderPath)
|
|
|
|
.Distinct()
|
|
|
|
.Where(s => !_diskProvider.FolderExists(s))
|
|
|
|
.ToList());
|
|
|
|
|
|
|
|
missingRootFolders = missingRootFolders.Distinct().ToList();
|
|
|
|
|
2014-05-26 05:14:15 +00:00
|
|
|
if (missingRootFolders.Any())
|
|
|
|
{
|
|
|
|
if (missingRootFolders.Count == 1)
|
|
|
|
{
|
2014-06-03 21:04:28 +00:00
|
|
|
return new HealthCheck(GetType(), HealthCheckResult.Error, "Missing root folder: " + missingRootFolders.First(), "#missing-root-folder");
|
2014-05-26 05:14:15 +00:00
|
|
|
}
|
|
|
|
|
2015-10-03 17:45:26 +00:00
|
|
|
var message = string.Format("Multiple root folders are missing: {0}", string.Join(" | ", missingRootFolders));
|
2014-06-03 21:04:28 +00:00
|
|
|
return new HealthCheck(GetType(), HealthCheckResult.Error, message, "#missing-root-folder");
|
2014-05-26 05:14:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return new HealthCheck(GetType());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|