New: Use ASP.NET Core instead of Nancy

2026-04-20 21:55:03 -04:00 · 2021-10-21 21:04:19 +01:00
parent c14ef7bee7
commit 2d53ec24f8
160 changed files with 2866 additions and 3657 deletions
@@ -0,0 +1,106 @@
+using System.Reflection;
+using Microsoft.AspNetCore.Mvc;
+using NzbDrone.Core.Datastore;
+using NzbDrone.Core.Datastore.Events;
+using NzbDrone.Core.Messaging.Events;
+using NzbDrone.SignalR;
+
+namespace Radarr.Http.REST
+{
+    public abstract class RestControllerWithSignalR<TResource, TModel> : RestController<TResource>, IHandle<ModelEvent<TModel>>
+        where TResource : RestResource, new()
+        where TModel : ModelBase, new()
+    {
+        protected string Resource { get; }
+        private readonly IBroadcastSignalRMessage _signalRBroadcaster;
+
+        protected RestControllerWithSignalR(IBroadcastSignalRMessage signalRBroadcaster)
+        {
+            _signalRBroadcaster = signalRBroadcaster;
+
+            var apiAttribute = GetType().GetCustomAttribute<VersionedApiControllerAttribute>();
+            if (apiAttribute != null && apiAttribute.Resource != VersionedApiControllerAttribute.CONTROLLER_RESOURCE)
+            {
+                Resource = apiAttribute.Resource;
+            }
+            else
+            {
+                Resource = new TResource().ResourceName.Trim('/');
+            }
+        }
+
+        [NonAction]
+        public void Handle(ModelEvent<TModel> message)
+        {
+            if (!_signalRBroadcaster.IsConnected)
+            {
+                return;
+            }
+
+            if (message.Action == ModelAction.Deleted || message.Action == ModelAction.Sync)
+            {
+                BroadcastResourceChange(message.Action);
+            }
+
+            BroadcastResourceChange(message.Action, message.Model.Id);
+        }
+
+        protected void BroadcastResourceChange(ModelAction action, int id)
+        {
+            if (!_signalRBroadcaster.IsConnected)
+            {
+                return;
+            }
+
+            if (action == ModelAction.Deleted)
+            {
+                BroadcastResourceChange(action, new TResource { Id = id });
+            }
+            else
+            {
+                var resource = GetResourceById(id);
+                BroadcastResourceChange(action, resource);
+            }
+        }
+
+        protected void BroadcastResourceChange(ModelAction action, TResource resource)
+        {
+            if (!_signalRBroadcaster.IsConnected)
+            {
+                return;
+            }
+
+            if (GetType().Namespace.Contains("V3"))
+            {
+                var signalRMessage = new SignalRMessage
+                {
+                    Name = Resource,
+                    Body = new ResourceChangeMessage<TResource>(resource, action),
+                    Action = action
+                };
+
+                _signalRBroadcaster.BroadcastMessage(signalRMessage);
+            }
+        }
+
+        protected void BroadcastResourceChange(ModelAction action)
+        {
+            if (!_signalRBroadcaster.IsConnected)
+            {
+                return;
+            }
+
+            if (GetType().Namespace.Contains("V3"))
+            {
+                var signalRMessage = new SignalRMessage
+                {
+                    Name = Resource,
+                    Body = new ResourceChangeMessage<TResource>(action),
+                    Action = action
+                };
+
+                _signalRBroadcaster.BroadcastMessage(signalRMessage);
+            }
+        }
+    }
+}