Slack Improvements

src/NzbDrone.Core/Notifications/Slack/Payloads/SlackPayload.cs

@@ -1,4 +1,4 @@
-using System.Collections.Generic;
+using System.Collections.Generic;
 using Newtonsoft.Json;
 
 namespace NzbDrone.Core.Notifications.Slack.Payloads
@@ -12,6 +12,9 @@ namespace NzbDrone.Core.Notifications.Slack.Payloads
         [JsonProperty("icon_emoji")]
         public string IconEmoji { get; set; }
 
+        [JsonProperty("icon_url")]
+        public string IconUrl { get; set; }
+
         public List<Attachment> Attachments { get; set; }
     }
 }

src/NzbDrone.Core/Notifications/Slack/Slack.cs

@@ -3,6 +3,8 @@ using System.Collections.Generic;
 using FluentValidation.Results;
 using NLog;
 using NzbDrone.Common.Extensions;
+using NzbDrone.Common.Http;
+using NzbDrone.Common.Serializer;
 using NzbDrone.Core.Notifications.Slack.Payloads;
 using NzbDrone.Core.Rest;
 using NzbDrone.Core.Music;
@@ -14,10 +16,12 @@ namespace NzbDrone.Core.Notifications.Slack
 {
     public class Slack : NotificationBase<SlackSettings>
     {
+        private readonly ISlackProxy _proxy;
         private readonly Logger _logger;
 
-        public Slack(Logger logger)
+        public Slack(ISlackProxy proxy, Logger logger)
         {
+            _proxy = proxy;
             _logger = logger;
         }
 
@@ -26,65 +30,51 @@ namespace NzbDrone.Core.Notifications.Slack
 
         public override void OnGrab(GrabMessage message)
         {
-            var payload = new SlackPayload
+            var attachments = new List<Attachment>
-            {
+                              {
-                IconEmoji = Settings.Icon,
+                                  new Attachment
-                Username = Settings.Username,
+                                  {
-                Text = $"Grabbed: {message.Message}",
+                                      Fallback = message.Message,
-                Attachments = new List<Attachment>
+                                      Title = message.Artist.Name,
-                {
+                                      Text = message.Message,
-                    new Attachment
+                                      Color = "warning"
-                    {
+                                  }
-                        Fallback = message.Message,
+                              };
-                        Title = message.Artist.Name,
+            var payload = CreatePayload($"Grabbed: {message.Message}", attachments);
-                        Text = message.Message,
+
-                        Color = "warning"
+            _proxy.SendPayload(payload, Settings);
-                    }
-                }
-            };
-
-            NotifySlack(payload);
         }
 
         public override void OnDownload(DownloadMessage message)
         {
-            var payload = new SlackPayload
+            var attachments = new List<Attachment>
-            {
+                              {
-                IconEmoji = Settings.Icon,
+                                  new Attachment
-                Username = Settings.Username,
+                                  {
-                Text = $"Imported: {message.Message}",
+                                      Fallback = message.Message,
-                Attachments = new List<Attachment>
+                                      Title = message.Artist.Name,
-                {
+                                      Text = message.Message,
-                    new Attachment
+                                      Color = "good"
-                    {
+                                  }
-                        Fallback = message.Message,
+                              };
-                        Title = message.Artist.Name,
+            var payload = CreatePayload($"Imported: {message.Message}", attachments);
-                        Text = message.Message,
+
-                        Color = "good"
+            _proxy.SendPayload(payload, Settings);
-                    }
-                }
-            };
-
-            NotifySlack(payload);
         }
 
         public override void OnRename(Artist artist)
         {
-            var payload = new SlackPayload
+            var attachments = new List<Attachment>
-            {
+                              {
-                IconEmoji = Settings.Icon,
+                                  new Attachment
-                Username = Settings.Username,
+                                  {
-                Text = "Renamed",
+                                      Title = artist.Name,
-                Attachments = new List<Attachment>
+                                  }
-                {
+                              };
-                    new Attachment
+
-                    {
+            var payload = CreatePayload("Renamed", attachments);
-                        Title = artist.Name,
-                    }
-                }
-            };
 
-            NotifySlack(payload);
+            _proxy.SendPayload(payload, Settings);
         }
 
         public override ValidationResult Test()
@@ -101,14 +91,9 @@ namespace NzbDrone.Core.Notifications.Slack
             try
             {
                 var message = $"Test message from Lidarr posted at {DateTime.Now}";
-                var payload = new SlackPayload
+                var payload = CreatePayload(message);
-                {
-                    IconEmoji = Settings.Icon,
-                    Username = Settings.Username,
-                    Text = message
-                };
 
-                NotifySlack(payload);
+                _proxy.SendPayload(payload, Settings);
 
             }
             catch (SlackExeption ex)
@@ -119,24 +104,31 @@ namespace NzbDrone.Core.Notifications.Slack
             return null;
         }
 
-        private void NotifySlack(SlackPayload payload)
+        private SlackPayload CreatePayload(string message, List<Attachment> attachments = null)
         {
-            try
+            var icon = Settings.Icon;
+
+            var payload = new SlackPayload
             {
-                var client = RestClientFactory.BuildClient(Settings.WebHookUrl);
+                Username = Settings.Username,
-                var request = new RestRequest(Method.POST)
+                Text = message,
-                {
+                Attachments = attachments
-                    RequestFormat = DataFormat.Json,
+            };
-                    JsonSerializer = new JsonNetSerializer()
+
-                };
+            if (icon.IsNotNullOrWhiteSpace())
-                request.AddBody(payload);
-                client.ExecuteAndValidate(request);
-            }
-            catch (RestException ex)
             {
-                _logger.Error(ex, "Unable to post payload {0}", payload);
+                // Set the correct icon based on the value
-                throw new SlackExeption("Unable to post payload", ex);
+                if (icon.StartsWith(":") && icon.EndsWith(":"))
+                {
+                    payload.IconEmoji = icon;
+                }
+                else
+                {
+                    payload.IconUrl = icon;
+                }
             }
+
+            return payload;
         }
     }
 }

src/NzbDrone.Core/Notifications/Slack/SlackProxy.cs

@@ -0,0 +1,46 @@
+using NLog;
+using NzbDrone.Common.Http;
+using NzbDrone.Common.Serializer;
+using NzbDrone.Core.Notifications.Slack.Payloads;
+using NzbDrone.Core.Rest;
+
+namespace NzbDrone.Core.Notifications.Slack
+{
+    public interface ISlackProxy
+    {
+        void SendPayload(SlackPayload payload, SlackSettings settings);
+    }
+
+    public class SlackProxy : ISlackProxy
+    {
+        private readonly IHttpClient _httpClient;
+        private readonly Logger _logger;
+
+        public SlackProxy(IHttpClient httpClient, Logger logger)
+        {
+            _httpClient = httpClient;
+            _logger = logger;
+        }
+
+        public void SendPayload(SlackPayload payload, SlackSettings settings)
+        {
+            try
+            {
+                var request = new HttpRequestBuilder(settings.WebHookUrl)
+                    .Accept(HttpAccept.Json)
+                    .Build();
+
+                request.Method = HttpMethod.POST;
+                request.Headers.ContentType = "application/json";
+                request.SetContent(payload.ToJson());
+
+                _httpClient.Execute(request);
+            }
+            catch (RestException ex)
+            {
+                _logger.Error(ex, "Unable to post payload {0}", payload);
+                throw new SlackExeption("Unable to post payload", ex);
+            }
+        }
+    }
+}

src/NzbDrone.Core/Notifications/Slack/SlackSettings.cs

@@ -1,4 +1,4 @@
-using FluentValidation;
+using FluentValidation;
 using NzbDrone.Core.Annotations;
 using NzbDrone.Core.ThingiProvider;
 using NzbDrone.Core.Validation;
@@ -24,7 +24,7 @@ namespace NzbDrone.Core.Notifications.Slack
         [FieldDefinition(1, Label = "Username", HelpText = "Choose the username that this integration will post as", Type = FieldType.Textbox)]
         public string Username { get; set; }
 
-        [FieldDefinition(2, Label = "Icon", HelpText = "Change the icon that is used for messages from this integration", Type = FieldType.Textbox, HelpLink = "http://www.emoji-cheat-sheet.com/")]
+        [FieldDefinition(2, Label = "Icon", HelpText = "Change the icon that is used for messages from this integration (Emoji or URL)", Type = FieldType.Textbox, HelpLink = "http://www.emoji-cheat-sheet.com/")]
         public string Icon { get; set; }
 
         public NzbDroneValidationResult Validate()

src/NzbDrone.Core/NzbDrone.Core.csproj

@@ -804,6 +804,7 @@
     <Compile Include="Notifications\Slack\Payloads\SlackPayload.cs" />
     <Compile Include="Notifications\Slack\Slack.cs" />
     <Compile Include="Notifications\Slack\SlackExeption.cs" />
+    <Compile Include="Notifications\Slack\SlackProxy.cs" />
     <Compile Include="Notifications\Slack\SlackSettings.cs" />
     <Compile Include="Notifications\Synology\SynologyException.cs" />
     <Compile Include="Notifications\Synology\SynologyIndexer.cs" />