New: Use ASP.NET Core instead of Nancy

2026-04-19 21:46:50 -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,119 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using Microsoft.AspNetCore.Mvc;
+using NzbDrone.Common;
+using NzbDrone.Common.Serializer;
+using NzbDrone.Common.TPL;
+using NzbDrone.Core.Datastore.Events;
+using NzbDrone.Core.Messaging.Commands;
+using NzbDrone.Core.Messaging.Events;
+using NzbDrone.Core.ProgressMessaging;
+using NzbDrone.SignalR;
+using Radarr.Http;
+using Radarr.Http.REST;
+using Radarr.Http.REST.Attributes;
+using Radarr.Http.Validation;
+
+namespace Radarr.Api.V3.Commands
+{
+    [V3ApiController]
+    public class CommandController : RestControllerWithSignalR<CommandResource, CommandModel>, IHandle<CommandUpdatedEvent>
+    {
+        private readonly IManageCommandQueue _commandQueueManager;
+        private readonly IServiceFactory _serviceFactory;
+        private readonly Debouncer _debouncer;
+        private readonly Dictionary<int, CommandResource> _pendingUpdates;
+
+        public CommandController(IManageCommandQueue commandQueueManager,
+                             IBroadcastSignalRMessage signalRBroadcaster,
+                             IServiceFactory serviceFactory)
+            : base(signalRBroadcaster)
+        {
+            _commandQueueManager = commandQueueManager;
+            _serviceFactory = serviceFactory;
+
+            _debouncer = new Debouncer(SendUpdates, TimeSpan.FromSeconds(0.1));
+            _pendingUpdates = new Dictionary<int, CommandResource>();
+
+            PostValidator.RuleFor(c => c.Name).NotBlank();
+        }
+
+        public override CommandResource GetResourceById(int id)
+        {
+            return _commandQueueManager.Get(id).ToResource();
+        }
+
+        [RestPostById]
+        public ActionResult<CommandResource> StartCommand(CommandResource commandResource)
+        {
+            var commandType =
+                _serviceFactory.GetImplementations(typeof(Command))
+                               .Single(c => c.Name.Replace("Command", "")
+                                             .Equals(commandResource.Name, StringComparison.InvariantCultureIgnoreCase));
+
+            Request.Body.Seek(0, SeekOrigin.Begin);
+            using (var reader = new StreamReader(Request.Body))
+            {
+                var body = reader.ReadToEnd();
+
+                dynamic command = STJson.Deserialize(body, commandType);
+
+                command.Trigger = CommandTrigger.Manual;
+                command.SuppressMessages = !command.SendUpdatesToClient;
+                command.SendUpdatesToClient = true;
+                command.ClientUserAgent = Request.Headers["UserAgent"];
+
+                var trackedCommand = _commandQueueManager.Push(command, CommandPriority.Normal, CommandTrigger.Manual);
+                return Created(trackedCommand.Id);
+            }
+        }
+
+        [HttpGet]
+        public List<CommandResource> GetStartedCommands()
+        {
+            return _commandQueueManager.All().ToResource();
+        }
+
+        [RestDeleteById]
+        public void CancelCommand(int id)
+        {
+            _commandQueueManager.Cancel(id);
+        }
+
+        [NonAction]
+        public void Handle(CommandUpdatedEvent message)
+        {
+            if (message.Command.Body.SendUpdatesToClient)
+            {
+                lock (_pendingUpdates)
+                {
+                    _pendingUpdates[message.Command.Id] = message.Command.ToResource();
+                }
+
+                _debouncer.Execute();
+            }
+        }
+
+        private void SendUpdates()
+        {
+            lock (_pendingUpdates)
+            {
+                var pendingUpdates = _pendingUpdates.Values.ToArray();
+                _pendingUpdates.Clear();
+
+                foreach (var pendingUpdate in pendingUpdates)
+                {
+                    BroadcastResourceChange(ModelAction.Updated, pendingUpdate);
+
+                    if (pendingUpdate.Name == typeof(MessagingCleanupCommand).Name.Replace("Command", "") &&
+                        pendingUpdate.Status == CommandStatus.Completed)
+                    {
+                        BroadcastResourceChange(ModelAction.Sync);
+                    }
+                }
+            }
+        }
+    }
+}