New: Make VIP Check Generic

2026-04-22 22:34:53 -04:00 · 2021-08-23 22:35:28 -04:00
parent c72222a696
commit 03f821f484
2 changed files with 18 additions and 12 deletions
@@ -0,0 +1,79 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Core.Indexers;
+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 IndexerVIPCheck : HealthCheckBase
+    {
+        private readonly IIndexerFactory _indexerFactory;
+
+        public IndexerVIPCheck(IIndexerFactory indexerFactory, ILocalizationService localizationService)
+            : base(localizationService)
+        {
+            _indexerFactory = indexerFactory;
+        }
+
+        public override HealthCheck Check()
+        {
+            var enabled = _indexerFactory.Enabled(false);
+            var expiringProviders = new List<IIndexer>();
+            var expiredProviders = new List<IIndexer>();
+
+            foreach (var provider in enabled)
+            {
+                var settingsType = provider.Definition.Settings.GetType();
+                var vipProp = settingsType.GetProperty("VipExpiration");
+
+                if (vipProp == null)
+                {
+                    continue;
+                }
+
+                var expiration = (string)vipProp.GetValue(provider.Definition.Settings);
+
+                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("IndexerVipCheckExpiringClientMessage"),
+                    string.Join(", ", expiringProviders.Select(v => v.Definition.Name))),
+                "#indexer-vip-expiring");
+            }
+
+            if (!expiredProviders.Empty())
+            {
+                return new HealthCheck(GetType(),
+                HealthCheckResult.Warning,
+                string.Format(_localizationService.GetLocalizedString("IndexerVipCheckExpiredClientMessage"),
+                    string.Join(", ", expiredProviders.Select(v => v.Definition.Name))),
+                "#indexer-vip-expired");
+            }
+
+            return new HealthCheck(GetType());
+        }
+    }
+}