Fixed: VIP Healthcheck not triggered for expired indexers

Prowlarr · Jun 24, 2022 · 86f5768 · 86f5768
1 parent 479e29d
commit 86f5768
Show file tree

Hide file tree

Showing 2 changed files with 66 additions and 17 deletions.
diff --git a/src/NzbDrone.Core/HealthCheck/Checks/IndexerVIPCheck.cs b/src/NzbDrone.Core/HealthCheck/Checks/IndexerVIPCheck.cs
@@ -23,11 +23,10 @@ public IndexerVIPCheck(IIndexerFactory indexerFactory, ILocalizationService loca
 
         public override HealthCheck Check()
         {
-            var enabled = _indexerFactory.Enabled(false);
+            var indexers = _indexerFactory.AllProviders(false);
             var expiringProviders = new List<IIndexer>();
-            var expiredProviders = new List<IIndexer>();
 
-            foreach (var provider in enabled)
+            foreach (var provider in indexers)
             {
                 var settingsType = provider.Definition.Settings.GetType();
                 var vipProp = settingsType.GetProperty("VipExpiration");
@@ -44,11 +43,6 @@ public override HealthCheck Check()
                     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);
@@ -64,15 +58,6 @@ public override HealthCheck Check()
                 "#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());
         }
     }

diff --git a/src/NzbDrone.Core/HealthCheck/Checks/IndexerVIPExpiredCheck.cs b/src/NzbDrone.Core/HealthCheck/Checks/IndexerVIPExpiredCheck.cs
@@ -0,0 +1,64 @@
+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 IndexerVIPExpiredCheck : HealthCheckBase
+    {
+        private readonly IIndexerFactory _indexerFactory;
+
+        public IndexerVIPExpiredCheck(IIndexerFactory indexerFactory, ILocalizationService localizationService)
+            : base(localizationService)
+        {
+            _indexerFactory = indexerFactory;
+        }
+
+        public override HealthCheck Check()
+        {
+            var indexers = _indexerFactory.AllProviders(false);
+            var expiredProviders = new List<IIndexer>();
+
+            foreach (var provider in indexers)
+            {
+                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 (!expiredProviders.Empty())
+            {
+                return new HealthCheck(GetType(),
+                HealthCheckResult.Error,
+                string.Format(_localizationService.GetLocalizedString("IndexerVipCheckExpiredClientMessage"),
+                    string.Join(", ", expiredProviders.Select(v => v.Definition.Name))),
+                "#indexer-vip-expired");
+            }
+
+            return new HealthCheck(GetType());
+        }
+    }
+}