mirror of https://github.com/Ombi-app/Ombi
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
143 lines
5.2 KiB
143 lines
5.2 KiB
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<SlackNotificationSettings>, ISlackNotification
|
|
{
|
|
public SlackNotification(ISlackApi api, ISettingsService<SlackNotificationSettings> sn, ILogger<SlackNotification> log, INotificationTemplatesRepository r, IMovieRequestRepository m, ITvRequestRepository t,
|
|
ISettingsService<CustomizationSettings> s, IRepository<RequestSubscription> sub, IMusicRequestRepository music,
|
|
IRepository<UserNotificationPreferences> 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<SlackNotification> 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);
|
|
}
|
|
}
|
|
}
|