453 lines
16 KiB
C#
453 lines
16 KiB
C#
using MediaBrowser.Common.Plugins;
|
|
using MediaBrowser.Common.ScheduledTasks;
|
|
using MediaBrowser.Common.Updates;
|
|
using MediaBrowser.Controller;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Entities.Audio;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.Notifications;
|
|
using MediaBrowser.Controller.Plugins;
|
|
using MediaBrowser.Controller.Session;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.Events;
|
|
using MediaBrowser.Model.Logging;
|
|
using MediaBrowser.Model.Notifications;
|
|
using MediaBrowser.Model.Tasks;
|
|
using MediaBrowser.Model.Updates;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Globalization;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace MediaBrowser.Server.Implementations.EntryPoints.Notifications
|
|
{
|
|
/// <summary>
|
|
/// Creates notifications for various system events
|
|
/// </summary>
|
|
public class Notifications : IServerEntryPoint
|
|
{
|
|
private readonly IInstallationManager _installationManager;
|
|
private readonly IUserManager _userManager;
|
|
private readonly ILogger _logger;
|
|
|
|
private readonly ITaskManager _taskManager;
|
|
private readonly INotificationManager _notificationManager;
|
|
|
|
private readonly ILibraryManager _libraryManager;
|
|
private readonly ISessionManager _sessionManager;
|
|
private readonly IServerApplicationHost _appHost;
|
|
|
|
private Timer LibraryUpdateTimer { get; set; }
|
|
private readonly object _libraryChangedSyncLock = new object();
|
|
|
|
public Notifications(IInstallationManager installationManager, IUserManager userManager, ILogger logger, ITaskManager taskManager, INotificationManager notificationManager, ILibraryManager libraryManager, ISessionManager sessionManager, IServerApplicationHost appHost)
|
|
{
|
|
_installationManager = installationManager;
|
|
_userManager = userManager;
|
|
_logger = logger;
|
|
_taskManager = taskManager;
|
|
_notificationManager = notificationManager;
|
|
_libraryManager = libraryManager;
|
|
_sessionManager = sessionManager;
|
|
_appHost = appHost;
|
|
}
|
|
|
|
public void Run()
|
|
{
|
|
_installationManager.PluginInstalled += _installationManager_PluginInstalled;
|
|
_installationManager.PluginUpdated += _installationManager_PluginUpdated;
|
|
_installationManager.PackageInstallationFailed += _installationManager_PackageInstallationFailed;
|
|
_installationManager.PluginUninstalled += _installationManager_PluginUninstalled;
|
|
|
|
_taskManager.TaskCompleted += _taskManager_TaskCompleted;
|
|
|
|
_userManager.UserCreated += _userManager_UserCreated;
|
|
_libraryManager.ItemAdded += _libraryManager_ItemAdded;
|
|
_sessionManager.PlaybackStart += _sessionManager_PlaybackStart;
|
|
_sessionManager.PlaybackStopped += _sessionManager_PlaybackStopped;
|
|
_appHost.HasPendingRestartChanged += _appHost_HasPendingRestartChanged;
|
|
_appHost.HasUpdateAvailableChanged += _appHost_HasUpdateAvailableChanged;
|
|
_appHost.ApplicationUpdated += _appHost_ApplicationUpdated;
|
|
}
|
|
|
|
async void _appHost_ApplicationUpdated(object sender, GenericEventArgs<PackageVersionInfo> e)
|
|
{
|
|
var type = NotificationType.ApplicationUpdateInstalled.ToString();
|
|
|
|
var notification = new NotificationRequest
|
|
{
|
|
NotificationType = type
|
|
};
|
|
|
|
notification.Variables["Version"] = e.Argument.versionStr;
|
|
notification.Variables["ReleaseNotes"] = e.Argument.description;
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
|
|
async void _installationManager_PluginUpdated(object sender, GenericEventArgs<Tuple<IPlugin, PackageVersionInfo>> e)
|
|
{
|
|
var type = NotificationType.PluginUpdateInstalled.ToString();
|
|
|
|
var installationInfo = e.Argument.Item1;
|
|
|
|
var notification = new NotificationRequest
|
|
{
|
|
Description = e.Argument.Item2.description,
|
|
NotificationType = type
|
|
};
|
|
|
|
notification.Variables["Name"] = installationInfo.Name;
|
|
notification.Variables["Version"] = installationInfo.Version.ToString();
|
|
notification.Variables["ReleaseNotes"] = e.Argument.Item2.description;
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
|
|
async void _installationManager_PluginInstalled(object sender, GenericEventArgs<PackageVersionInfo> e)
|
|
{
|
|
var type = NotificationType.PluginInstalled.ToString();
|
|
|
|
var installationInfo = e.Argument;
|
|
|
|
var notification = new NotificationRequest
|
|
{
|
|
Description = installationInfo.description,
|
|
NotificationType = type
|
|
};
|
|
|
|
notification.Variables["Name"] = installationInfo.name;
|
|
notification.Variables["Version"] = installationInfo.versionStr;
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
|
|
async void _appHost_HasUpdateAvailableChanged(object sender, EventArgs e)
|
|
{
|
|
// This notification is for users who can't auto-update (aka running as service)
|
|
if (!_appHost.HasUpdateAvailable || _appHost.CanSelfUpdate)
|
|
{
|
|
return;
|
|
}
|
|
|
|
var type = NotificationType.ApplicationUpdateAvailable.ToString();
|
|
|
|
var notification = new NotificationRequest
|
|
{
|
|
Description = "Please see mediabrowser.tv for details.",
|
|
NotificationType = type
|
|
};
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
|
|
async void _appHost_HasPendingRestartChanged(object sender, EventArgs e)
|
|
{
|
|
if (!_appHost.HasPendingRestart)
|
|
{
|
|
return;
|
|
}
|
|
|
|
var type = NotificationType.ServerRestartRequired.ToString();
|
|
|
|
var notification = new NotificationRequest
|
|
{
|
|
NotificationType = type
|
|
};
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
|
|
void _sessionManager_PlaybackStart(object sender, PlaybackProgressEventArgs e)
|
|
{
|
|
var item = e.MediaInfo;
|
|
|
|
if (item == null)
|
|
{
|
|
_logger.Warn("PlaybackStart reported with null media info.");
|
|
return;
|
|
}
|
|
|
|
var type = GetPlaybackNotificationType(item.MediaType);
|
|
|
|
SendPlaybackNotification(type, e);
|
|
}
|
|
|
|
void _sessionManager_PlaybackStopped(object sender, PlaybackStopEventArgs e)
|
|
{
|
|
var item = e.MediaInfo;
|
|
|
|
if (item == null)
|
|
{
|
|
_logger.Warn("PlaybackStopped reported with null media info.");
|
|
return;
|
|
}
|
|
|
|
var type = GetPlaybackStoppedNotificationType(item.MediaType);
|
|
|
|
SendPlaybackNotification(type, e);
|
|
}
|
|
|
|
private async void SendPlaybackNotification(string type, PlaybackProgressEventArgs e)
|
|
{
|
|
var user = e.Users.FirstOrDefault();
|
|
|
|
var item = e.MediaInfo;
|
|
|
|
if (e.Item != null && e.Item.Parent == null)
|
|
{
|
|
// Don't report theme song or local trailer playback
|
|
// TODO: This will also cause movie specials to not be reported
|
|
return;
|
|
}
|
|
|
|
var notification = new NotificationRequest
|
|
{
|
|
NotificationType = type,
|
|
|
|
ExcludeUserIds = e.Users.Select(i => i.Id.ToString("N")).ToList()
|
|
};
|
|
|
|
notification.Variables["ItemName"] = item.Name;
|
|
notification.Variables["UserName"] = user == null ? "Unknown user" : user.Name;
|
|
notification.Variables["AppName"] = e.ClientName;
|
|
notification.Variables["DeviceName"] = e.DeviceName;
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
|
|
private string GetPlaybackNotificationType(string mediaType)
|
|
{
|
|
if (string.Equals(mediaType, MediaType.Audio, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return NotificationType.AudioPlayback.ToString();
|
|
}
|
|
if (string.Equals(mediaType, MediaType.Game, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return NotificationType.GamePlayback.ToString();
|
|
}
|
|
if (string.Equals(mediaType, MediaType.Video, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return NotificationType.VideoPlayback.ToString();
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
private string GetPlaybackStoppedNotificationType(string mediaType)
|
|
{
|
|
if (string.Equals(mediaType, MediaType.Audio, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return NotificationType.AudioPlaybackStopped.ToString();
|
|
}
|
|
if (string.Equals(mediaType, MediaType.Game, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return NotificationType.GamePlaybackStopped.ToString();
|
|
}
|
|
if (string.Equals(mediaType, MediaType.Video, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return NotificationType.VideoPlaybackStopped.ToString();
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
private readonly List<BaseItem> _itemsAdded = new List<BaseItem>();
|
|
void _libraryManager_ItemAdded(object sender, ItemChangeEventArgs e)
|
|
{
|
|
if (e.Item.LocationType == LocationType.FileSystem && !e.Item.IsFolder)
|
|
{
|
|
lock (_libraryChangedSyncLock)
|
|
{
|
|
if (LibraryUpdateTimer == null)
|
|
{
|
|
LibraryUpdateTimer = new Timer(LibraryUpdateTimerCallback, null, 5000,
|
|
Timeout.Infinite);
|
|
}
|
|
else
|
|
{
|
|
LibraryUpdateTimer.Change(5000, Timeout.Infinite);
|
|
}
|
|
|
|
_itemsAdded.Add(e.Item);
|
|
}
|
|
}
|
|
}
|
|
|
|
private async void LibraryUpdateTimerCallback(object state)
|
|
{
|
|
List<BaseItem> items;
|
|
|
|
lock (_libraryChangedSyncLock)
|
|
{
|
|
items = _itemsAdded.ToList();
|
|
_itemsAdded.Clear();
|
|
DisposeLibraryUpdateTimer();
|
|
}
|
|
|
|
if (items.Count == 1)
|
|
{
|
|
var item = items.First();
|
|
|
|
var notification = new NotificationRequest
|
|
{
|
|
NotificationType = NotificationType.NewLibraryContent.ToString()
|
|
};
|
|
|
|
notification.Variables["Name"] = GetItemName(item);
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
else
|
|
{
|
|
var notification = new NotificationRequest
|
|
{
|
|
NotificationType = NotificationType.NewLibraryContentMultiple.ToString()
|
|
};
|
|
|
|
notification.Variables["ItemCount"] = items.Count.ToString(CultureInfo.InvariantCulture);
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
public static string GetItemName(BaseItem item)
|
|
{
|
|
var name = item.Name;
|
|
|
|
var hasSeries = item as IHasSeries;
|
|
|
|
if (hasSeries != null)
|
|
{
|
|
name = hasSeries.SeriesName + " - " + name;
|
|
}
|
|
|
|
var hasArtist = item as IHasArtist;
|
|
if (hasArtist != null)
|
|
{
|
|
var artists = hasArtist.AllArtists;
|
|
|
|
if (artists.Count > 0)
|
|
{
|
|
name = hasArtist.AllArtists[0] + " - " + name;
|
|
}
|
|
}
|
|
|
|
return name;
|
|
}
|
|
|
|
async void _userManager_UserCreated(object sender, GenericEventArgs<User> e)
|
|
{
|
|
var notification = new NotificationRequest
|
|
{
|
|
UserIds = new List<string> { e.Argument.Id.ToString("N") },
|
|
Name = "Welcome to Media Browser!",
|
|
Description = "Check back here for more notifications."
|
|
};
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
|
|
async void _taskManager_TaskCompleted(object sender, TaskCompletionEventArgs e)
|
|
{
|
|
var result = e.Result;
|
|
|
|
if (result.Status == TaskCompletionStatus.Failed)
|
|
{
|
|
var type = NotificationType.TaskFailed.ToString();
|
|
|
|
var notification = new NotificationRequest
|
|
{
|
|
Description = result.ErrorMessage,
|
|
Level = NotificationLevel.Error,
|
|
NotificationType = type
|
|
};
|
|
|
|
notification.Variables["Name"] = result.Name;
|
|
notification.Variables["ErrorMessage"] = result.ErrorMessage;
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
}
|
|
|
|
async void _installationManager_PluginUninstalled(object sender, GenericEventArgs<IPlugin> e)
|
|
{
|
|
var type = NotificationType.PluginUninstalled.ToString();
|
|
|
|
var plugin = e.Argument;
|
|
|
|
var notification = new NotificationRequest
|
|
{
|
|
NotificationType = type
|
|
};
|
|
|
|
notification.Variables["Name"] = plugin.Name;
|
|
notification.Variables["Version"] = plugin.Version.ToString();
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
|
|
async void _installationManager_PackageInstallationFailed(object sender, InstallationFailedEventArgs e)
|
|
{
|
|
var installationInfo = e.InstallationInfo;
|
|
|
|
var type = NotificationType.InstallationFailed.ToString();
|
|
|
|
var notification = new NotificationRequest
|
|
{
|
|
Level = NotificationLevel.Error,
|
|
Description = e.Exception.Message,
|
|
NotificationType = type
|
|
};
|
|
|
|
notification.Variables["Name"] = installationInfo.Name;
|
|
notification.Variables["Version"] = installationInfo.Version;
|
|
|
|
await SendNotification(notification).ConfigureAwait(false);
|
|
}
|
|
|
|
private async Task SendNotification(NotificationRequest notification)
|
|
{
|
|
try
|
|
{
|
|
await _notificationManager.SendNotification(notification, CancellationToken.None).ConfigureAwait(false);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.ErrorException("Error sending notification", ex);
|
|
}
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
DisposeLibraryUpdateTimer();
|
|
|
|
_installationManager.PluginInstalled -= _installationManager_PluginInstalled;
|
|
_installationManager.PluginUpdated -= _installationManager_PluginUpdated;
|
|
_installationManager.PackageInstallationFailed -= _installationManager_PackageInstallationFailed;
|
|
_installationManager.PluginUninstalled -= _installationManager_PluginUninstalled;
|
|
|
|
_taskManager.TaskCompleted -= _taskManager_TaskCompleted;
|
|
|
|
_userManager.UserCreated -= _userManager_UserCreated;
|
|
_libraryManager.ItemAdded -= _libraryManager_ItemAdded;
|
|
_sessionManager.PlaybackStart -= _sessionManager_PlaybackStart;
|
|
|
|
_appHost.HasPendingRestartChanged -= _appHost_HasPendingRestartChanged;
|
|
_appHost.HasUpdateAvailableChanged -= _appHost_HasUpdateAvailableChanged;
|
|
_appHost.ApplicationUpdated -= _appHost_ApplicationUpdated;
|
|
}
|
|
|
|
private void DisposeLibraryUpdateTimer()
|
|
{
|
|
if (LibraryUpdateTimer != null)
|
|
{
|
|
LibraryUpdateTimer.Dispose();
|
|
LibraryUpdateTimer = null;
|
|
}
|
|
}
|
|
}
|
|
}
|