You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.9 KiB
52 lines
1.9 KiB
using System.Linq;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.Localization;
|
|
using NzbDrone.Core.Movies;
|
|
using NzbDrone.Core.Movies.Events;
|
|
|
|
namespace NzbDrone.Core.HealthCheck.Checks
|
|
{
|
|
[CheckOn(typeof(MovieUpdatedEvent))]
|
|
[CheckOn(typeof(MoviesDeletedEvent), CheckOnCondition.FailedOnly)]
|
|
[CheckOn(typeof(MovieRefreshCompleteEvent))]
|
|
public class RemovedMovieCheck : HealthCheckBase, ICheckOnCondition<MovieUpdatedEvent>, ICheckOnCondition<MoviesDeletedEvent>
|
|
{
|
|
private readonly IMovieService _movieService;
|
|
|
|
public RemovedMovieCheck(IMovieService movieService, ILocalizationService localizationService)
|
|
: base(localizationService)
|
|
{
|
|
_movieService = movieService;
|
|
}
|
|
|
|
public override HealthCheck Check()
|
|
{
|
|
var deletedMovie = _movieService.GetAllMovies().Where(v => v.Status == MovieStatusType.Deleted).ToList();
|
|
|
|
if (deletedMovie.Empty())
|
|
{
|
|
return new HealthCheck(GetType());
|
|
}
|
|
|
|
var movieText = deletedMovie.Select(s => $"{s.Title} (tmdbid {s.TmdbId})").Join(", ");
|
|
|
|
if (deletedMovie.Count == 1)
|
|
{
|
|
return new HealthCheck(GetType(), HealthCheckResult.Error, string.Format(_localizationService.GetLocalizedString("RemovedMovieCheckSingleMessage"), movieText), "#movie-was-removed-from-tmdb");
|
|
}
|
|
|
|
return new HealthCheck(GetType(), HealthCheckResult.Error, string.Format(_localizationService.GetLocalizedString("RemovedMovieCheckMultipleMessage"), movieText), "#movie-was-removed-from-tmdb");
|
|
}
|
|
|
|
public bool ShouldCheckOnEvent(MoviesDeletedEvent message)
|
|
{
|
|
return message.Movies.Any(m => m.Status == MovieStatusType.Deleted);
|
|
}
|
|
|
|
public bool ShouldCheckOnEvent(MovieUpdatedEvent message)
|
|
{
|
|
return message.Movie.Status == MovieStatusType.Deleted;
|
|
}
|
|
}
|
|
}
|