using System; using System.Threading.Tasks; using Microsoft.Extensions.Logging; using Ombi.Api.Slack; using Ombi.Api.Slack.Models; using Ombi.Core.Settings; using Ombi.Helpers; using Ombi.Notifications.Models; using Ombi.Settings.Settings.Models; using Ombi.Settings.Settings.Models.Notifications; using Ombi.Store.Entities; using Ombi.Store.Repository; using Ombi.Store.Repository.Requests; namespace Ombi.Notifications.Agents { public class SlackNotification : BaseNotification, ISlackNotification { public SlackNotification(ISlackApi api, ISettingsService sn, ILogger log, INotificationTemplatesRepository r, IMovieRequestRepository m, ITvRequestRepository t, ISettingsService s, IRepository sub, IMusicRequestRepository music, IRepository userPref) : base(sn, r, m, t, s, log, sub, music, userPref) { Api = api; Logger = log; } public override string NotificationName => "SlackNotification"; private ISlackApi Api { get; } private ILogger Logger { get; } protected override bool ValidateConfiguration(SlackNotificationSettings settings) { if (!settings.Enabled) { return false; } if (string.IsNullOrEmpty(settings.WebhookUrl)) { return false; } try { var a = settings.Token; var b = settings.Team; var c = settings.Service; } catch (IndexOutOfRangeException) { return false; } return true; } protected override async Task NewRequest(NotificationOptions model, SlackNotificationSettings settings) { await Run(model, settings, NotificationType.NewRequest); } protected override async Task NewIssue(NotificationOptions model, SlackNotificationSettings settings) { await Run(model, settings, NotificationType.Issue); } protected override async Task IssueComment(NotificationOptions model, SlackNotificationSettings settings) { await Run(model, settings, NotificationType.IssueComment); } protected override async Task IssueResolved(NotificationOptions model, SlackNotificationSettings settings) { await Run(model, settings, NotificationType.IssueResolved); } protected override async Task AddedToRequestQueue(NotificationOptions model, SlackNotificationSettings settings) { await Run(model, settings, NotificationType.ItemAddedToFaultQueue); } protected override async Task RequestDeclined(NotificationOptions model, SlackNotificationSettings settings) { await Run(model, settings, NotificationType.RequestAvailable); } protected override async Task RequestApproved(NotificationOptions model, SlackNotificationSettings settings) { await Run(model, settings, NotificationType.RequestApproved); } protected override async Task AvailableRequest(NotificationOptions model, SlackNotificationSettings settings) { await Run(model, settings, NotificationType.RequestAvailable); } protected override async Task Send(NotificationMessage model, SlackNotificationSettings settings) { try { var body = new SlackNotificationBody { channel = settings.Channel, icon_emoji = settings.IconEmoji, icon_url = settings.IconUrl, text = model.Message, username = settings.Username }; await Api.PushAsync(settings.Team, settings.Token, settings.Service, body); } catch (Exception e) { Logger.LogError(LoggingEvents.SlackNotification, e, "Failed to send Slack Notification"); } } protected override async Task Test(NotificationOptions model, SlackNotificationSettings settings) { var message = $"This is a test from Ombi, if you can see this then we have successfully pushed a notification!"; var notification = new NotificationMessage { Message = message, }; await Send(notification, settings); } private async Task Run(NotificationOptions model, SlackNotificationSettings settings, NotificationType type) { var parsed = await LoadTemplate(NotificationAgent.Slack, type, model); if (parsed.Disabled) { Logger.LogInformation($"Template {type} is disabled for {NotificationAgent.Slack}"); return; } var notification = new NotificationMessage { Message = parsed.Message, }; notification.Other.Add("image", parsed.Image); await Send(notification, settings); } } }