parent
6d9b028814
commit
fa7647135b
@ -0,0 +1,73 @@
|
|||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Linq;
|
||||||
|
using NzbDrone.Common.Extensions;
|
||||||
|
using NzbDrone.Core.Indexers;
|
||||||
|
using NzbDrone.Core.Indexers.Newznab;
|
||||||
|
using NzbDrone.Core.Localization;
|
||||||
|
using NzbDrone.Core.ThingiProvider.Events;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.HealthCheck.Checks
|
||||||
|
{
|
||||||
|
[CheckOn(typeof(ProviderAddedEvent<IIndexer>))]
|
||||||
|
[CheckOn(typeof(ProviderUpdatedEvent<IIndexer>))]
|
||||||
|
[CheckOn(typeof(ProviderDeletedEvent<IIndexer>))]
|
||||||
|
public class NewznabVIPCheck : HealthCheckBase
|
||||||
|
{
|
||||||
|
private readonly IIndexerFactory _indexerFactory;
|
||||||
|
|
||||||
|
public NewznabVIPCheck(IIndexerFactory indexerFactory, ILocalizationService localizationService)
|
||||||
|
: base(localizationService)
|
||||||
|
{
|
||||||
|
_indexerFactory = indexerFactory;
|
||||||
|
}
|
||||||
|
|
||||||
|
public override HealthCheck Check()
|
||||||
|
{
|
||||||
|
var enabled = _indexerFactory.Enabled(false);
|
||||||
|
var newznabProviders = enabled.Where(i => i.Definition.Implementation == typeof(Newznab).Name);
|
||||||
|
var expiringProviders = new List<IIndexer>();
|
||||||
|
var expiredProviders = new List<IIndexer>();
|
||||||
|
|
||||||
|
foreach (var provider in newznabProviders)
|
||||||
|
{
|
||||||
|
var expiration = ((NewznabSettings)provider.Definition.Settings).VipExpiration;
|
||||||
|
|
||||||
|
if (expiration.IsNullOrWhiteSpace())
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (DateTime.Parse(expiration).Before(DateTime.Now))
|
||||||
|
{
|
||||||
|
expiredProviders.Add(provider);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (DateTime.Parse(expiration).Between(DateTime.Now, DateTime.Now.AddDays(7)))
|
||||||
|
{
|
||||||
|
expiringProviders.Add(provider);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!expiringProviders.Empty())
|
||||||
|
{
|
||||||
|
return new HealthCheck(GetType(),
|
||||||
|
HealthCheckResult.Warning,
|
||||||
|
string.Format(_localizationService.GetLocalizedString("NewznabVipCheckExpiringClientMessage"),
|
||||||
|
string.Join(", ", expiringProviders.Select(v => v.Definition.Name))),
|
||||||
|
"#newznab-vip-expiring");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!expiredProviders.Empty())
|
||||||
|
{
|
||||||
|
return new HealthCheck(GetType(),
|
||||||
|
HealthCheckResult.Warning,
|
||||||
|
string.Format(_localizationService.GetLocalizedString("NewznabVipCheckExpiredClientMessage"),
|
||||||
|
string.Join(", ", expiredProviders.Select(v => v.Definition.Name))),
|
||||||
|
"#newznab-vip-expired");
|
||||||
|
}
|
||||||
|
|
||||||
|
return new HealthCheck(GetType());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in new issue