From fe8f319f7bfdadb7218b6313ada6cae1d2a35ad8 Mon Sep 17 00:00:00 2001
From: Taloth Saldono <Taloth@users.noreply.github.com>
Date: Sun, 16 May 2021 12:41:53 +0200
Subject: [PATCH] Show User Agent in System->Tasks for externally triggered
 commands

---
 .../src/System/Tasks/Queued/QueuedTaskRow.css |  9 ++++++++
 .../src/System/Tasks/Queued/QueuedTaskRow.js  | 13 +++++++++++-
 src/NzbDrone.Api/Commands/CommandResource.cs  |  5 +++++
 src/NzbDrone.Common/Http/UserAgentParser.cs   | 21 +++++++++++++++++++
 .../Messaging/Commands/Command.cs             |  2 ++
 src/Sonarr.Api.V3/Commands/CommandModule.cs   |  2 ++
 src/Sonarr.Api.V3/Commands/CommandResource.cs |  5 +++++
 7 files changed, 56 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 12393cb96..1aac3ef21 100644
--- a/frontend/src/System/Tasks/Queued/QueuedTaskRow.js
+++ b/frontend/src/System/Tasks/Queued/QueuedTaskRow.js
@@ -156,6 +156,7 @@ class QueuedTaskRow extends Component {
       status,
       duration,
       message,
+      clientUserAgent,
       longDateFormat,
       timeFormat,
       onCancelPress
@@ -191,7 +192,17 @@ 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>
+          }
+        </TableRowCell>
 
         <TableRowCell
           className={styles.queued}
diff --git a/src/NzbDrone.Api/Commands/CommandResource.cs b/src/NzbDrone.Api/Commands/CommandResource.cs
index 4180851f4..03620b5f3 100644
--- a/src/NzbDrone.Api/Commands/CommandResource.cs
+++ b/src/NzbDrone.Api/Commands/CommandResource.cs
@@ -4,6 +4,7 @@ using System.Linq;
 using Newtonsoft.Json;
 using Sonarr.Http.REST;
 using NzbDrone.Core.Messaging.Commands;
+using NzbDrone.Common.Http;
 
 namespace NzbDrone.Api.Commands
 {
@@ -21,6 +22,8 @@ namespace NzbDrone.Api.Commands
         public string Exception { get; set; }
         public CommandTrigger Trigger { get; set; }
 
+        public string ClientUserAgent { get; set; }
+
         [JsonIgnore]
         public string CompletionMessage { get; set; }
 
@@ -117,6 +120,8 @@ namespace NzbDrone.Api.Commands
                 Exception = model.Exception,
                 Trigger = model.Trigger,
 
+                ClientUserAgent = UserAgentParser.SimplifyUserAgent(model.Body.ClientUserAgent),
+
                 CompletionMessage = model.Body.CompletionMessage,
                 LastExecutionTime = model.Body.LastExecutionTime
             };
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 001143034..8ca57519f 100644
--- a/src/NzbDrone.Core/Messaging/Commands/Command.cs
+++ b/src/NzbDrone.Core/Messaging/Commands/Command.cs
@@ -29,6 +29,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/Sonarr.Api.V3/Commands/CommandModule.cs b/src/Sonarr.Api.V3/Commands/CommandModule.cs
index 6644fd395..a9e89e10b 100644
--- a/src/Sonarr.Api.V3/Commands/CommandModule.cs
+++ b/src/Sonarr.Api.V3/Commands/CommandModule.cs
@@ -57,6 +57,8 @@ namespace Sonarr.Api.V3.Commands
             command.SuppressMessages = !command.SendUpdatesToClient;
             command.SendUpdatesToClient = true;
 
+            command.ClientUserAgent = Request.Headers.UserAgent;
+
             var trackedCommand = _commandQueueManager.Push(command, CommandPriority.Normal, CommandTrigger.Manual);
             return trackedCommand.Id;
         }
diff --git a/src/Sonarr.Api.V3/Commands/CommandResource.cs b/src/Sonarr.Api.V3/Commands/CommandResource.cs
index cbec03a7f..cef9722f0 100644
--- a/src/Sonarr.Api.V3/Commands/CommandResource.cs
+++ b/src/Sonarr.Api.V3/Commands/CommandResource.cs
@@ -3,6 +3,7 @@ using System.Collections.Generic;
 using System.Linq;
 using Newtonsoft.Json;
 using NzbDrone.Common.Extensions;
+using NzbDrone.Common.Http;
 using NzbDrone.Core.Messaging.Commands;
 using Sonarr.Http.REST;
 
@@ -23,6 +24,8 @@ namespace Sonarr.Api.V3.Commands
         public string Exception { get; set; }
         public CommandTrigger Trigger { get; set; }
 
+        public string ClientUserAgent { get; set; }
+
         [JsonIgnore]
         public string CompletionMessage { get; set; }
 
@@ -89,6 +92,8 @@ namespace Sonarr.Api.V3.Commands
                 Exception = model.Exception,
                 Trigger = model.Trigger,
 
+                ClientUserAgent = UserAgentParser.SimplifyUserAgent(model.Body.ClientUserAgent),
+
                 CompletionMessage = model.Body.CompletionMessage,
                 LastExecutionTime = model.Body.LastExecutionTime
             };