From cae5badee09c1fbed2b04282a2bb4b1666add6b4 Mon Sep 17 00:00:00 2001
From: Qstick <qstick@gmail.com>
Date: Sun, 17 Oct 2021 11:41:31 -0500
Subject: [PATCH] New: Support server notifications

(cherry picked from commit f5f0dd6fae5bc9f308506d56be42ac9a4be908e7)

Closes Radarr #5393

[common]
---
 .../HealthCheck/HealthCheckServiceFixture.cs  |  4 ++
 .../HealthCheck/HealthCheckService.cs         |  5 ++
 .../ServerSideNotificationService.cs          | 65 +++++++++++++++++++
 3 files changed, 74 insertions(+)
 create mode 100644 src/NzbDrone.Core/HealthCheck/ServerSideNotificationService.cs

diff --git a/src/NzbDrone.Core.Test/HealthCheck/HealthCheckServiceFixture.cs b/src/NzbDrone.Core.Test/HealthCheck/HealthCheckServiceFixture.cs
index fcf9d58c8..74802a39e 100644
--- a/src/NzbDrone.Core.Test/HealthCheck/HealthCheckServiceFixture.cs
+++ b/src/NzbDrone.Core.Test/HealthCheck/HealthCheckServiceFixture.cs
@@ -19,6 +19,10 @@ namespace NzbDrone.Core.Test.HealthCheck
 
             Mocker.SetConstant<IEnumerable<IProvideHealthCheck>>(new[] { _healthCheck });
             Mocker.SetConstant<ICacheManager>(Mocker.Resolve<CacheManager>());
+
+            Mocker.GetMock<IServerSideNotificationService>()
+                .Setup(v => v.GetServerChecks())
+                .Returns(new List<Core.HealthCheck.HealthCheck>());
         }
 
         [Test]
diff --git a/src/NzbDrone.Core/HealthCheck/HealthCheckService.cs b/src/NzbDrone.Core/HealthCheck/HealthCheckService.cs
index 24b112207..679044083 100644
--- a/src/NzbDrone.Core/HealthCheck/HealthCheckService.cs
+++ b/src/NzbDrone.Core/HealthCheck/HealthCheckService.cs
@@ -25,6 +25,7 @@ namespace NzbDrone.Core.HealthCheck
         private readonly IProvideHealthCheck[] _startupHealthChecks;
         private readonly IProvideHealthCheck[] _scheduledHealthChecks;
         private readonly Dictionary<Type, IEventDrivenHealthCheck[]> _eventDrivenHealthChecks;
+        private readonly IServerSideNotificationService _serverSideNotificationService;
         private readonly IEventAggregator _eventAggregator;
         private readonly ICacheManager _cacheManager;
         private readonly Logger _logger;
@@ -32,11 +33,13 @@ namespace NzbDrone.Core.HealthCheck
         private readonly ICached<HealthCheck> _healthCheckResults;
 
         public HealthCheckService(IEnumerable<IProvideHealthCheck> healthChecks,
+                                  IServerSideNotificationService serverSideNotificationService,
                                   IEventAggregator eventAggregator,
                                   ICacheManager cacheManager,
                                   Logger logger)
         {
             _healthChecks = healthChecks.ToArray();
+            _serverSideNotificationService = serverSideNotificationService;
             _eventAggregator = eventAggregator;
             _cacheManager = cacheManager;
             _logger = logger;
@@ -72,6 +75,8 @@ namespace NzbDrone.Core.HealthCheck
             var results = healthChecks.Select(c => c.Check())
                                        .ToList();
 
+            results.AddRange(_serverSideNotificationService.GetServerChecks());
+
             foreach (var result in results)
             {
                 if (result.Type == HealthCheckResult.Ok)
diff --git a/src/NzbDrone.Core/HealthCheck/ServerSideNotificationService.cs b/src/NzbDrone.Core/HealthCheck/ServerSideNotificationService.cs
new file mode 100644
index 000000000..fe14e4b86
--- /dev/null
+++ b/src/NzbDrone.Core/HealthCheck/ServerSideNotificationService.cs
@@ -0,0 +1,65 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Runtime.InteropServices;
+using NLog;
+using NzbDrone.Common.Cloud;
+using NzbDrone.Common.EnvironmentInfo;
+using NzbDrone.Common.Http;
+using NzbDrone.Common.Serializer;
+using NzbDrone.Core.Configuration;
+
+namespace NzbDrone.Core.HealthCheck
+{
+    public interface IServerSideNotificationService
+    {
+        public List<HealthCheck> GetServerChecks();
+    }
+
+    public class ServerSideNotificationService : IServerSideNotificationService
+    {
+        private readonly IHttpClient _client;
+        private readonly IConfigFileProvider _configFileProvider;
+        private readonly IHttpRequestBuilderFactory _cloudRequestBuilder;
+        private readonly Logger _logger;
+
+        public ServerSideNotificationService(IHttpClient client, IConfigFileProvider configFileProvider, IRadarrCloudRequestBuilder cloudRequestBuilder, Logger logger)
+        {
+            _client = client;
+            _configFileProvider = configFileProvider;
+            _cloudRequestBuilder = cloudRequestBuilder.Services;
+            _logger = logger;
+        }
+
+        public List<HealthCheck> GetServerChecks()
+        {
+            var request = _cloudRequestBuilder.Create()
+                                              .Resource("/notification")
+                                              .AddQueryParam("version", BuildInfo.Version)
+                                              .AddQueryParam("os", OsInfo.Os.ToString().ToLowerInvariant())
+                                              .AddQueryParam("arch", RuntimeInformation.OSArchitecture)
+                                              .AddQueryParam("runtime", PlatformInfo.Platform.ToString().ToLowerInvariant())
+                                              .AddQueryParam("branch", _configFileProvider.Branch)
+                                              .Build();
+            try
+            {
+                _logger.Trace("Getting server side health notifications");
+                var response = _client.Execute(request);
+                var result = Json.Deserialize<List<ServerNotificationResponse>>(response.Content);
+                return result.Select(x => new HealthCheck(GetType(), x.Type, x.Message, x.WikiUrl)).ToList();
+            }
+            catch (Exception ex)
+            {
+                _logger.Error(ex, "Failed to retrieve server notifications");
+                return new List<HealthCheck>();
+            }
+        }
+    }
+
+    public class ServerNotificationResponse
+    {
+        public HealthCheckResult Type { get; set; }
+        public string Message { get; set; }
+        public string WikiUrl { get; set; }
+    }
+}