2019-01-13 14:54:44 -05:00
|
|
|
using System;
|
2019-01-13 14:18:07 -05:00
|
|
|
using System.Collections.Generic;
|
2019-02-28 17:22:57 -05:00
|
|
|
using System.Globalization;
|
2019-01-13 14:18:07 -05:00
|
|
|
using System.Linq;
|
|
|
|
using System.Threading;
|
|
|
|
using System.Threading.Tasks;
|
2020-05-20 13:07:53 -04:00
|
|
|
using Jellyfin.Data.Entities;
|
2020-05-12 22:10:35 -04:00
|
|
|
using Jellyfin.Data.Enums;
|
2019-01-13 14:18:07 -05:00
|
|
|
using MediaBrowser.Common.Configuration;
|
2014-07-02 14:34:08 -04:00
|
|
|
using MediaBrowser.Common.Extensions;
|
2014-04-26 23:42:05 -04:00
|
|
|
using MediaBrowser.Controller.Configuration;
|
|
|
|
using MediaBrowser.Controller.Entities;
|
2014-04-25 16:15:50 -04:00
|
|
|
using MediaBrowser.Controller.Library;
|
|
|
|
using MediaBrowser.Controller.Notifications;
|
2018-09-12 13:26:21 -04:00
|
|
|
using MediaBrowser.Model.Dto;
|
2019-01-13 14:18:07 -05:00
|
|
|
using MediaBrowser.Model.Notifications;
|
|
|
|
using Microsoft.Extensions.Logging;
|
2014-04-25 16:15:50 -04:00
|
|
|
|
2018-09-12 13:26:21 -04:00
|
|
|
namespace Emby.Notifications
|
2014-04-25 16:15:50 -04:00
|
|
|
{
|
2019-12-20 15:30:51 -05:00
|
|
|
/// <summary>
|
|
|
|
/// NotificationManager class.
|
|
|
|
/// </summary>
|
2014-04-25 16:15:50 -04:00
|
|
|
public class NotificationManager : INotificationManager
|
|
|
|
{
|
2020-06-05 20:15:56 -04:00
|
|
|
private readonly ILogger<NotificationManager> _logger;
|
2014-04-25 16:15:50 -04:00
|
|
|
private readonly IUserManager _userManager;
|
2014-04-26 23:42:05 -04:00
|
|
|
private readonly IServerConfigurationManager _config;
|
|
|
|
|
2019-12-20 15:30:51 -05:00
|
|
|
private INotificationService[] _services = Array.Empty<INotificationService>();
|
|
|
|
private INotificationTypeFactory[] _typeFactories = Array.Empty<INotificationTypeFactory>();
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
/// Initializes a new instance of the <see cref="NotificationManager" /> class.
|
|
|
|
/// </summary>
|
|
|
|
/// <param name="logger">The logger.</param>
|
2020-02-08 16:25:44 -05:00
|
|
|
/// <param name="userManager">The user manager.</param>
|
2019-12-20 15:30:51 -05:00
|
|
|
/// <param name="config">The server configuration manager.</param>
|
|
|
|
public NotificationManager(
|
|
|
|
ILogger<NotificationManager> logger,
|
|
|
|
IUserManager userManager,
|
|
|
|
IServerConfigurationManager config)
|
2014-04-25 16:15:50 -04:00
|
|
|
{
|
2019-12-20 15:30:51 -05:00
|
|
|
_logger = logger;
|
2014-04-25 16:15:50 -04:00
|
|
|
_userManager = userManager;
|
2014-04-26 23:42:05 -04:00
|
|
|
_config = config;
|
2014-04-25 16:15:50 -04:00
|
|
|
}
|
|
|
|
|
2014-07-02 14:34:08 -04:00
|
|
|
private NotificationOptions GetConfiguration()
|
|
|
|
{
|
|
|
|
return _config.GetConfiguration<NotificationOptions>("notifications");
|
|
|
|
}
|
|
|
|
|
2019-12-20 15:30:51 -05:00
|
|
|
/// <inheritdoc />
|
2014-04-25 16:15:50 -04:00
|
|
|
public Task SendNotification(NotificationRequest request, CancellationToken cancellationToken)
|
2017-10-16 02:11:09 -04:00
|
|
|
{
|
|
|
|
return SendNotification(request, null, cancellationToken);
|
|
|
|
}
|
|
|
|
|
2019-12-20 15:30:51 -05:00
|
|
|
/// <inheritdoc />
|
|
|
|
public Task SendNotification(NotificationRequest request, BaseItem? relatedItem, CancellationToken cancellationToken)
|
2014-04-25 16:15:50 -04:00
|
|
|
{
|
2014-04-26 23:42:05 -04:00
|
|
|
var notificationType = request.NotificationType;
|
|
|
|
|
2018-09-12 13:26:21 -04:00
|
|
|
var options = string.IsNullOrEmpty(notificationType) ?
|
2014-04-27 13:54:43 -04:00
|
|
|
null :
|
2014-07-02 14:34:08 -04:00
|
|
|
GetConfiguration().GetOptions(notificationType);
|
2014-04-27 13:54:43 -04:00
|
|
|
|
2014-04-27 15:30:12 -04:00
|
|
|
var users = GetUserIds(request, options)
|
2017-10-16 02:11:09 -04:00
|
|
|
.Select(i => _userManager.GetUserById(i))
|
|
|
|
.Where(i => relatedItem == null || relatedItem.IsVisibleStandalone(i))
|
|
|
|
.ToArray();
|
2014-04-27 13:54:43 -04:00
|
|
|
|
2018-09-12 13:26:21 -04:00
|
|
|
var title = request.Name;
|
|
|
|
var description = request.Description;
|
2014-04-26 23:42:05 -04:00
|
|
|
|
|
|
|
var tasks = _services.Where(i => IsEnabled(i, notificationType))
|
2014-04-30 23:24:55 -04:00
|
|
|
.Select(i => SendNotification(request, i, users, title, description, cancellationToken));
|
2014-04-25 16:15:50 -04:00
|
|
|
|
|
|
|
return Task.WhenAll(tasks);
|
|
|
|
}
|
|
|
|
|
2019-12-20 15:30:51 -05:00
|
|
|
private Task SendNotification(
|
|
|
|
NotificationRequest request,
|
2014-04-25 16:15:50 -04:00
|
|
|
INotificationService service,
|
2020-05-20 13:07:53 -04:00
|
|
|
IEnumerable<User> users,
|
2014-04-26 23:42:05 -04:00
|
|
|
string title,
|
2014-04-30 11:07:02 -04:00
|
|
|
string description,
|
2014-04-25 16:15:50 -04:00
|
|
|
CancellationToken cancellationToken)
|
|
|
|
{
|
2022-01-04 11:05:36 -05:00
|
|
|
users = users.Where(i => IsEnabledForUser(service, i));
|
2014-04-25 16:15:50 -04:00
|
|
|
|
2014-04-30 11:07:02 -04:00
|
|
|
var tasks = users.Select(i => SendNotification(request, service, title, description, i, cancellationToken));
|
2014-04-25 16:15:50 -04:00
|
|
|
|
|
|
|
return Task.WhenAll(tasks);
|
|
|
|
}
|
|
|
|
|
2019-12-20 15:30:51 -05:00
|
|
|
private IEnumerable<Guid> GetUserIds(NotificationRequest request, NotificationOption? options)
|
2014-04-27 13:54:43 -04:00
|
|
|
{
|
|
|
|
if (request.SendToUserMode.HasValue)
|
|
|
|
{
|
|
|
|
switch (request.SendToUserMode.Value)
|
|
|
|
{
|
|
|
|
case SendToUserType.Admins:
|
2020-05-12 22:10:35 -04:00
|
|
|
return _userManager.Users.Where(i => i.HasPermission(PermissionKind.IsAdministrator))
|
2018-09-12 13:26:21 -04:00
|
|
|
.Select(i => i.Id);
|
2014-04-27 13:54:43 -04:00
|
|
|
case SendToUserType.All:
|
2019-06-09 16:08:01 -04:00
|
|
|
return _userManager.UsersIds;
|
2014-04-27 13:54:43 -04:00
|
|
|
case SendToUserType.Custom:
|
|
|
|
return request.UserIds;
|
|
|
|
default:
|
|
|
|
throw new ArgumentException("Unrecognized SendToUserMode: " + request.SendToUserMode.Value);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-09-12 13:26:21 -04:00
|
|
|
if (options != null && !string.IsNullOrEmpty(request.NotificationType))
|
2014-04-27 13:54:43 -04:00
|
|
|
{
|
2014-07-03 22:22:57 -04:00
|
|
|
var config = GetConfiguration();
|
|
|
|
|
2014-07-17 20:39:07 -04:00
|
|
|
return _userManager.Users
|
2020-05-12 22:10:35 -04:00
|
|
|
.Where(i => config.IsEnabledToSendToUser(request.NotificationType, i.Id.ToString("N", CultureInfo.InvariantCulture), i))
|
2018-09-12 13:26:21 -04:00
|
|
|
.Select(i => i.Id);
|
2014-04-27 13:54:43 -04:00
|
|
|
}
|
|
|
|
|
2014-04-30 11:07:02 -04:00
|
|
|
return request.UserIds;
|
2014-04-27 13:54:43 -04:00
|
|
|
}
|
|
|
|
|
2019-12-20 15:30:51 -05:00
|
|
|
private async Task SendNotification(
|
|
|
|
NotificationRequest request,
|
2014-04-25 16:15:50 -04:00
|
|
|
INotificationService service,
|
2014-04-26 23:42:05 -04:00
|
|
|
string title,
|
2014-04-30 11:07:02 -04:00
|
|
|
string description,
|
2020-05-20 13:07:53 -04:00
|
|
|
User user,
|
2014-04-25 16:15:50 -04:00
|
|
|
CancellationToken cancellationToken)
|
|
|
|
{
|
|
|
|
var notification = new UserNotification
|
|
|
|
{
|
|
|
|
Date = request.Date,
|
2014-04-30 11:07:02 -04:00
|
|
|
Description = description,
|
2014-04-25 16:15:50 -04:00
|
|
|
Level = request.Level,
|
2014-04-26 23:42:05 -04:00
|
|
|
Name = title,
|
2014-04-25 16:15:50 -04:00
|
|
|
Url = request.Url,
|
|
|
|
User = user
|
|
|
|
};
|
|
|
|
|
2020-05-12 22:10:35 -04:00
|
|
|
_logger.LogDebug("Sending notification via {0} to user {1}", service.Name, user.Username);
|
2014-04-26 23:42:05 -04:00
|
|
|
|
2014-04-25 16:15:50 -04:00
|
|
|
try
|
|
|
|
{
|
|
|
|
await service.SendNotification(notification, cancellationToken).ConfigureAwait(false);
|
|
|
|
}
|
|
|
|
catch (Exception ex)
|
|
|
|
{
|
2018-12-20 07:11:26 -05:00
|
|
|
_logger.LogError(ex, "Error sending notification to {0}", service.Name);
|
2014-04-25 16:15:50 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-05-20 13:07:53 -04:00
|
|
|
private bool IsEnabledForUser(INotificationService service, User user)
|
2014-04-25 16:15:50 -04:00
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
return service.IsEnabledForUser(user);
|
|
|
|
}
|
|
|
|
catch (Exception ex)
|
|
|
|
{
|
2018-12-20 07:11:26 -05:00
|
|
|
_logger.LogError(ex, "Error in IsEnabledForUser");
|
2014-04-25 16:15:50 -04:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-26 23:42:05 -04:00
|
|
|
private bool IsEnabled(INotificationService service, string notificationType)
|
|
|
|
{
|
2015-09-19 22:06:56 -04:00
|
|
|
if (string.IsNullOrEmpty(notificationType))
|
|
|
|
{
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
return GetConfiguration().IsServiceEnabled(service.Name, notificationType);
|
2014-04-26 23:42:05 -04:00
|
|
|
}
|
|
|
|
|
2019-12-20 15:30:51 -05:00
|
|
|
/// <inheritdoc />
|
2014-04-26 23:42:05 -04:00
|
|
|
public void AddParts(IEnumerable<INotificationService> services, IEnumerable<INotificationTypeFactory> notificationTypeFactories)
|
2014-04-25 16:15:50 -04:00
|
|
|
{
|
|
|
|
_services = services.ToArray();
|
2014-04-26 23:42:05 -04:00
|
|
|
_typeFactories = notificationTypeFactories.ToArray();
|
|
|
|
}
|
|
|
|
|
2019-12-20 15:30:51 -05:00
|
|
|
/// <inheritdoc />
|
2017-08-19 15:43:35 -04:00
|
|
|
public List<NotificationTypeInfo> GetNotificationTypes()
|
2014-04-26 23:42:05 -04:00
|
|
|
{
|
|
|
|
var list = _typeFactories.Select(i =>
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
return i.GetNotificationTypes().ToList();
|
|
|
|
}
|
|
|
|
catch (Exception ex)
|
|
|
|
{
|
2018-12-20 07:11:26 -05:00
|
|
|
_logger.LogError(ex, "Error in GetNotificationTypes");
|
2014-04-26 23:42:05 -04:00
|
|
|
return new List<NotificationTypeInfo>();
|
|
|
|
}
|
|
|
|
}).SelectMany(i => i).ToList();
|
|
|
|
|
2014-07-02 14:34:08 -04:00
|
|
|
var config = GetConfiguration();
|
|
|
|
|
2014-04-26 23:42:05 -04:00
|
|
|
foreach (var i in list)
|
|
|
|
{
|
2014-07-02 14:34:08 -04:00
|
|
|
i.Enabled = config.IsEnabled(i.Type);
|
2014-04-26 23:42:05 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
return list;
|
|
|
|
}
|
|
|
|
|
2019-12-20 15:30:51 -05:00
|
|
|
/// <inheritdoc />
|
2018-09-12 13:26:21 -04:00
|
|
|
public IEnumerable<NameIdPair> GetNotificationServices()
|
2014-04-26 23:42:05 -04:00
|
|
|
{
|
2018-09-12 13:26:21 -04:00
|
|
|
return _services.Select(i => new NameIdPair
|
2014-04-26 23:42:05 -04:00
|
|
|
{
|
|
|
|
Name = i.Name,
|
2019-02-28 17:22:57 -05:00
|
|
|
Id = i.Name.GetMD5().ToString("N", CultureInfo.InvariantCulture)
|
2014-04-26 23:42:05 -04:00
|
|
|
}).OrderBy(i => i.Name);
|
2014-04-25 16:15:50 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|