Fix capitalization error of LidarrRestModuleWithSignalR.cs. Resolves #122 (#123)

2026-04-20 21:54:25 -04:00 · 2017-11-15 22:06:03 -05:00
parent 34ca4bf150
commit 1fa141d6c6
1 changed files with 0 additions and 0 deletions
@@ -0,0 +1,71 @@
+using NzbDrone.Core.Datastore;
+using NzbDrone.Core.Datastore.Events;
+using NzbDrone.Core.Messaging.Events;
+using NzbDrone.SignalR;
+using Lidarr.Http.REST;
+
+namespace Lidarr.Http
+{
+    public abstract class LidarrRestModuleWithSignalR<TResource, TModel> : LidarrRestModule<TResource>, IHandle<ModelEvent<TModel>>
+        where TResource : RestResource, new()
+        where TModel : ModelBase, new()
+    {
+        private readonly IBroadcastSignalRMessage _signalRBroadcaster;
+
+        protected LidarrRestModuleWithSignalR(IBroadcastSignalRMessage signalRBroadcaster)
+        {
+            _signalRBroadcaster = signalRBroadcaster;
+        }
+
+        protected LidarrRestModuleWithSignalR(IBroadcastSignalRMessage signalRBroadcaster, string resource)
+            : base(resource)
+        {
+            _signalRBroadcaster = signalRBroadcaster;
+        }
+
+        public void Handle(ModelEvent<TModel> message)
+        {
+            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)
+        {
+            var resource = GetResourceById(id);
+            BroadcastResourceChange(action, resource);
+        }
+
+
+        protected void BroadcastResourceChange(ModelAction action, TResource resource)
+        {
+            var signalRMessage = new SignalRMessage
+            {
+                Name = Resource,
+                Body = new ResourceChangeMessage<TResource>(resource, action),
+                Action = action
+            };
+
+            _signalRBroadcaster.BroadcastMessage(signalRMessage);
+        }
+
+     
+        protected void BroadcastResourceChange(ModelAction action)
+        {
+            if (GetType().Namespace.Contains("V1"))
+            {
+                var signalRMessage = new SignalRMessage
+                {
+                    Name = Resource,
+                    Body = new ResourceChangeMessage<TResource>(action),
+                    Action = action
+                };
+
+                _signalRBroadcaster.BroadcastMessage(signalRMessage);
+            }
+        }
+    }
+}