From f85235d21428b276229932524b13c9fd078b78d6 Mon Sep 17 00:00:00 2001
From: Robin Dadswell <19610103+RobinDadswell@users.noreply.github.com>
Date: Thu, 20 May 2021 12:04:37 +0100
Subject: [PATCH] Show User Agent in System->Tasks for externally triggered
 commands (#63)

(cherry picked from commit fe8f319f7bfdadb7218b6313ada6cae1d2a35ad8)

Co-authored-by: Taloth Saldono <Taloth@users.noreply.github.com>
---
 .../src/System/Tasks/Queued/QueuedTaskRow.css |  9 ++++++++
 .../src/System/Tasks/Queued/QueuedTaskRow.js  | 15 ++++++++++++-
 src/NzbDrone.Common/Http/UserAgentParser.cs   | 21 +++++++++++++++++++
 .../Messaging/Commands/Command.cs             |  2 ++
 .../Commands/CommandController.cs             |  1 +
 .../Commands/CommandResource.cs               |  5 +++++
 6 files changed, 52 insertions(+), 1 deletion(-)
 create mode 100644 src/NzbDrone.Common/Http/UserAgentParser.cs

diff --git a/frontend/src/System/Tasks/Queued/QueuedTaskRow.css b/frontend/src/System/Tasks/Queued/QueuedTaskRow.css
index 6e38929c9..034804711 100644
--- a/frontend/src/System/Tasks/Queued/QueuedTaskRow.css
+++ b/frontend/src/System/Tasks/Queued/QueuedTaskRow.css
@@ -10,6 +10,15 @@
   width: 100%;
 }
 
+.commandName {
+  display: inline-block;
+  min-width: 220px;
+}
+
+.userAgent {
+  color: #b0b0b0;
+}
+
 .queued,
 .started,
 .ended {
diff --git a/frontend/src/System/Tasks/Queued/QueuedTaskRow.js b/frontend/src/System/Tasks/Queued/QueuedTaskRow.js
index cfe85573c..b19525326 100644
--- a/frontend/src/System/Tasks/Queued/QueuedTaskRow.js
+++ b/frontend/src/System/Tasks/Queued/QueuedTaskRow.js
@@ -157,6 +157,7 @@ class QueuedTaskRow extends Component {
       status,
       duration,
       message,
+      clientUserAgent,
       longDateFormat,
       timeFormat,
       onCancelPress
@@ -192,7 +193,18 @@ class QueuedTaskRow extends Component {
           </span>
         </TableRowCell>
 
-        <TableRowCell>{commandName}</TableRowCell>
+        <TableRowCell>
+          <span className={styles.commandName}>
+            {commandName}
+          </span>
+          {
+            clientUserAgent ?
+              <span className={styles.userAgent} title="User-Agent provided by the app that called the API">
+                from: {clientUserAgent}
+              </span> :
+              null
+          }
+        </TableRowCell>
 
         <TableRowCell
           className={styles.queued}
@@ -256,6 +268,7 @@ QueuedTaskRow.propTypes = {
   status: PropTypes.string.isRequired,
   duration: PropTypes.string,
   message: PropTypes.string,
+  clientUserAgent: PropTypes.string,
   showRelativeDates: PropTypes.bool.isRequired,
   shortDateFormat: PropTypes.string.isRequired,
   longDateFormat: PropTypes.string.isRequired,
diff --git a/src/NzbDrone.Common/Http/UserAgentParser.cs b/src/NzbDrone.Common/Http/UserAgentParser.cs
new file mode 100644
index 000000000..2c382236a
--- /dev/null
+++ b/src/NzbDrone.Common/Http/UserAgentParser.cs
@@ -0,0 +1,21 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace NzbDrone.Common.Http
+{
+    public static class UserAgentParser
+    {
+        public static string SimplifyUserAgent(string userAgent)
+        {
+            if (userAgent == null || userAgent.StartsWith("Mozilla/5.0"))
+            {
+                return null;
+            }
+
+            return userAgent;
+        }
+    }
+}
diff --git a/src/NzbDrone.Core/Messaging/Commands/Command.cs b/src/NzbDrone.Core/Messaging/Commands/Command.cs
index aa3ffa685..020b5ec64 100644
--- a/src/NzbDrone.Core/Messaging/Commands/Command.cs
+++ b/src/NzbDrone.Core/Messaging/Commands/Command.cs
@@ -34,6 +34,8 @@ namespace NzbDrone.Core.Messaging.Commands
         public CommandTrigger Trigger { get; set; }
         public bool SuppressMessages { get; set; }
 
+        public string ClientUserAgent { get; set; }
+
         public Command()
         {
             Name = GetType().Name.Replace("Command", "");
diff --git a/src/Prowlarr.Api.V1/Commands/CommandController.cs b/src/Prowlarr.Api.V1/Commands/CommandController.cs
index 5980a7a4d..a375b7966 100644
--- a/src/Prowlarr.Api.V1/Commands/CommandController.cs
+++ b/src/Prowlarr.Api.V1/Commands/CommandController.cs
@@ -62,6 +62,7 @@ namespace Prowlarr.Api.V1.Commands
             command.Trigger = CommandTrigger.Manual;
             command.SuppressMessages = !command.SendUpdatesToClient;
             command.SendUpdatesToClient = true;
+            command.ClientUserAgent = Request.Headers["User-Agent"];
 
             var trackedCommand = _commandQueueManager.Push(command, CommandPriority.Normal, CommandTrigger.Manual);
             return Created(trackedCommand.Id);
diff --git a/src/Prowlarr.Api.V1/Commands/CommandResource.cs b/src/Prowlarr.Api.V1/Commands/CommandResource.cs
index 589f50eac..371091a8d 100644
--- a/src/Prowlarr.Api.V1/Commands/CommandResource.cs
+++ b/src/Prowlarr.Api.V1/Commands/CommandResource.cs
@@ -3,6 +3,7 @@ using System.Collections.Generic;
 using System.Linq;
 using System.Text.Json.Serialization;
 using NzbDrone.Common.Extensions;
+using NzbDrone.Common.Http;
 using NzbDrone.Core.Messaging.Commands;
 using Prowlarr.Http.REST;
 
@@ -23,6 +24,8 @@ namespace Prowlarr.Api.V1.Commands
         public string Exception { get; set; }
         public CommandTrigger Trigger { get; set; }
 
+        public string ClientUserAgent { get; set; }
+
         [JsonIgnore]
         public string CompletionMessage { get; set; }
 
@@ -106,6 +109,8 @@ namespace Prowlarr.Api.V1.Commands
                 Exception = model.Exception,
                 Trigger = model.Trigger,
 
+                ClientUserAgent = UserAgentParser.SimplifyUserAgent(model.Body.ClientUserAgent),
+
                 CompletionMessage = model.Body.CompletionMessage,
                 LastExecutionTime = model.Body.LastExecutionTime
             };