New: Switch to ASPNetCore Kestrel and SignalR

2025-09-17 17:14:18 +02:00 · 2019-10-14 21:21:00 +01:00
parent fdbed91a4e
commit fb5b9c445b
58 changed files with 917 additions and 1298 deletions
--- a/src/NzbDrone.Host/WebHost/WebHostController.cs
+++ b/src/NzbDrone.Host/WebHost/WebHostController.cs
@@ -0,0 +1,140 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using System.Security.Cryptography.X509Certificates;
+using Microsoft.AspNetCore.Builder;
+using Microsoft.AspNetCore.Hosting;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Logging;
+using NLog;
+using NLog.Extensions.Logging;
+using NzbDrone.Common.EnvironmentInfo;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Common.Serializer;
+using NzbDrone.Core.Configuration;
+using Radarr.Host.AccessControl;
+using Radarr.Host.Middleware;
+using LogLevel = Microsoft.Extensions.Logging.LogLevel;
+
+namespace Radarr.Host
+{
+    public class WebHostController : IHostController
+    {
+        private readonly IRuntimeInfo _runtimeInfo;
+        private readonly IConfigFileProvider _configFileProvider;
+        private readonly IFirewallAdapter _firewallAdapter;
+        private readonly IEnumerable<IAspNetCoreMiddleware> _middlewares;
+        private readonly Logger _logger;
+        private IWebHost _host;
+
+        public WebHostController(IRuntimeInfo runtimeInfo,
+                                 IConfigFileProvider configFileProvider,
+                                 IFirewallAdapter firewallAdapter,
+                                 IEnumerable<IAspNetCoreMiddleware> middlewares,
+                                 Logger logger)
+        {
+            _runtimeInfo = runtimeInfo;
+            _configFileProvider = configFileProvider;
+            _firewallAdapter = firewallAdapter;
+            _middlewares = middlewares;
+            _logger = logger;
+        }
+
+        public void StartServer()
+        {
+            if (OsInfo.IsWindows)
+            {
+                if (_runtimeInfo.IsAdmin)
+                {
+                    _firewallAdapter.MakeAccessible();
+                }
+            }
+
+            var bindAddress = _configFileProvider.BindAddress;
+            var enableSsl = _configFileProvider.EnableSsl;
+            var sslCertPath = _configFileProvider.SslCertPath;
+
+            var urls = new List<string>();
+
+            urls.Add(BuildUrl("http", bindAddress, _configFileProvider.Port));
+
+            if (enableSsl && sslCertPath.IsNotNullOrWhiteSpace())
+            {
+                urls.Add(BuildUrl("https", bindAddress, _configFileProvider.SslPort));
+            }
+
+            _host = new WebHostBuilder()
+                .UseUrls(urls.ToArray())
+                .UseKestrel(options =>
+                {
+                    if (enableSsl && sslCertPath.IsNotNullOrWhiteSpace())
+                    {
+                        options.ConfigureHttpsDefaults(configureOptions =>
+                        {
+                            var certificate = new X509Certificate2();
+                            certificate.Import(_configFileProvider.SslCertPath, _configFileProvider.SslCertPassword, X509KeyStorageFlags.DefaultKeySet);
+
+                            configureOptions.ServerCertificate = certificate;
+                        });
+                    }
+                })
+                .ConfigureKestrel(serverOptions =>
+                    {
+                        serverOptions.AllowSynchronousIO = true;
+                    })
+                .ConfigureLogging(logging =>
+                {
+                    logging.AddProvider(new NLogLoggerProvider());
+                    logging.SetMinimumLevel(LogLevel.Warning);
+                })
+                .ConfigureServices(services =>
+                {
+                    services
+                    .AddSignalR()
+                    .AddJsonProtocol(options =>
+                        {
+                            options.PayloadSerializerSettings = Json.GetSerializerSettings();
+                        });
+                })
+                .Configure(app =>
+                {
+                    app.UsePathBase(_configFileProvider.UrlBase);
+                    app.Properties["host.AppName"] = BuildInfo.AppName;
+
+                    foreach (var middleWare in _middlewares.OrderBy(c => c.Order))
+                    {
+                        _logger.Debug("Attaching {0} to host", middleWare.GetType().Name);
+                        middleWare.Attach(app);
+                    }
+                })
+                .UseContentRoot(Directory.GetCurrentDirectory())
+                .Build();
+
+            _logger.Info("Listening on the following URLs:");
+
+            foreach (var url in urls)
+            {
+                _logger.Info("  {0}", url);
+            }
+
+            _host.Start();
+        }
+
+        public async void StopServer()
+        {
+            _logger.Info("Attempting to stop OWIN host");
+
+            await _host.StopAsync(TimeSpan.FromSeconds(5));
+            _host.Dispose();
+            _host = null;
+
+            _logger.Info("Host has stopped");
+        }
+
+        private string BuildUrl(string scheme, string bindAddress, int port)
+        {
+            return $"{scheme}://{bindAddress}:{port}";
+        }
+    }
+}