mirror of
https://github.com/jellyfin/jellyfin.git
synced 2025-07-09 03:04:24 -04:00
Merge pull request #10592 from barronpm/librarychangednotifier-cleanup
LibraryChangedNotifier Cleanup
This commit is contained in:
Bond-009
GitHub
commit
547d8c07b3
@ -1,7 +1,3 @@
|
|||||||
#nullable disable
|
|
||||||
|
|
||||||
#pragma warning disable CS1591
|
|
||||||
|
|
||||||
using System;
|
using System;
|
||||||
using System.Collections.Concurrent;
|
using System.Collections.Concurrent;
|
||||||
using System.Collections.Generic;
|
using System.Collections.Generic;
|
||||||
@ -23,476 +19,382 @@ using MediaBrowser.Model.Entities;
|
|||||||
using MediaBrowser.Model.Session;
|
using MediaBrowser.Model.Session;
|
||||||
using Microsoft.Extensions.Logging;
|
using Microsoft.Extensions.Logging;
|
||||||
|
|
||||||
namespace Emby.Server.Implementations.EntryPoints
|
namespace Emby.Server.Implementations.EntryPoints;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// A <see cref="IServerEntryPoint"/> that notifies users when libraries are updated.
|
||||||
|
/// </summary>
|
||||||
|
public sealed class LibraryChangedNotifier : IServerEntryPoint
|
||||||
{
|
{
|
||||||
public class LibraryChangedNotifier : IServerEntryPoint
|
private readonly ILibraryManager _libraryManager;
|
||||||
|
private readonly IServerConfigurationManager _configurationManager;
|
||||||
|
private readonly IProviderManager _providerManager;
|
||||||
|
private readonly ISessionManager _sessionManager;
|
||||||
|
private readonly IUserManager _userManager;
|
||||||
|
private readonly ILogger<LibraryChangedNotifier> _logger;
|
||||||
|
|
||||||
|
private readonly object _libraryChangedSyncLock = new();
|
||||||
|
private readonly List<Folder> _foldersAddedTo = new();
|
||||||
|
private readonly List<Folder> _foldersRemovedFrom = new();
|
||||||
|
private readonly List<BaseItem> _itemsAdded = new();
|
||||||
|
private readonly List<BaseItem> _itemsRemoved = new();
|
||||||
|
private readonly List<BaseItem> _itemsUpdated = new();
|
||||||
|
private readonly ConcurrentDictionary<Guid, DateTime> _lastProgressMessageTimes = new();
|
||||||
|
|
||||||
|
private Timer? _libraryUpdateTimer;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Initializes a new instance of the <see cref="LibraryChangedNotifier"/> class.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="libraryManager">The <see cref="ILibraryManager"/>.</param>
|
||||||
|
/// <param name="configurationManager">The <see cref="IServerConfigurationManager"/>.</param>
|
||||||
|
/// <param name="sessionManager">The <see cref="ISessionManager"/>.</param>
|
||||||
|
/// <param name="userManager">The <see cref="IUserManager"/>.</param>
|
||||||
|
/// <param name="logger">The <see cref="ILogger"/>.</param>
|
||||||
|
/// <param name="providerManager">The <see cref="IProviderManager"/>.</param>
|
||||||
|
public LibraryChangedNotifier(
|
||||||
|
ILibraryManager libraryManager,
|
||||||
|
IServerConfigurationManager configurationManager,
|
||||||
|
ISessionManager sessionManager,
|
||||||
|
IUserManager userManager,
|
||||||
|
ILogger<LibraryChangedNotifier> logger,
|
||||||
|
IProviderManager providerManager)
|
||||||
{
|
{
|
||||||
private readonly ILibraryManager _libraryManager;
|
_libraryManager = libraryManager;
|
||||||
private readonly IServerConfigurationManager _configurationManager;
|
_configurationManager = configurationManager;
|
||||||
private readonly IProviderManager _providerManager;
|
_sessionManager = sessionManager;
|
||||||
private readonly ISessionManager _sessionManager;
|
_userManager = userManager;
|
||||||
private readonly IUserManager _userManager;
|
_logger = logger;
|
||||||
private readonly ILogger<LibraryChangedNotifier> _logger;
|
_providerManager = providerManager;
|
||||||
|
}
|
||||||
|
|
||||||
/// <summary>
|
/// <inheritdoc />
|
||||||
/// The library changed sync lock.
|
public Task RunAsync()
|
||||||
/// </summary>
|
{
|
||||||
private readonly object _libraryChangedSyncLock = new object();
|
_libraryManager.ItemAdded += OnLibraryItemAdded;
|
||||||
|
_libraryManager.ItemUpdated += OnLibraryItemUpdated;
|
||||||
|
_libraryManager.ItemRemoved += OnLibraryItemRemoved;
|
||||||
|
|
||||||
private readonly List<Folder> _foldersAddedTo = new List<Folder>();
|
_providerManager.RefreshCompleted += OnProviderRefreshCompleted;
|
||||||
private readonly List<Folder> _foldersRemovedFrom = new List<Folder>();
|
_providerManager.RefreshStarted += OnProviderRefreshStarted;
|
||||||
private readonly List<BaseItem> _itemsAdded = new List<BaseItem>();
|
_providerManager.RefreshProgress += OnProviderRefreshProgress;
|
||||||
private readonly List<BaseItem> _itemsRemoved = new List<BaseItem>();
|
|
||||||
private readonly List<BaseItem> _itemsUpdated = new List<BaseItem>();
|
|
||||||
private readonly ConcurrentDictionary<Guid, DateTime> _lastProgressMessageTimes = new ConcurrentDictionary<Guid, DateTime>();
|
|
||||||
|
|
||||||
public LibraryChangedNotifier(
|
return Task.CompletedTask;
|
||||||
ILibraryManager libraryManager,
|
}
|
||||||
IServerConfigurationManager configurationManager,
|
|
||||||
ISessionManager sessionManager,
|
private void OnProviderRefreshProgress(object? sender, GenericEventArgs<Tuple<BaseItem, double>> e)
|
||||||
IUserManager userManager,
|
{
|
||||||
ILogger<LibraryChangedNotifier> logger,
|
var item = e.Argument.Item1;
|
||||||
IProviderManager providerManager)
|
|
||||||
|
if (!EnableRefreshMessage(item))
|
||||||
{
|
{
|
||||||
_libraryManager = libraryManager;
|
return;
|
||||||
_configurationManager = configurationManager;
|
|
||||||
_sessionManager = sessionManager;
|
|
||||||
_userManager = userManager;
|
|
||||||
_logger = logger;
|
|
||||||
_providerManager = providerManager;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
var progress = e.Argument.Item2;
|
||||||
/// Gets or sets the library update timer.
|
|
||||||
/// </summary>
|
|
||||||
/// <value>The library update timer.</value>
|
|
||||||
private Timer LibraryUpdateTimer { get; set; }
|
|
||||||
|
|
||||||
public Task RunAsync()
|
if (_lastProgressMessageTimes.TryGetValue(item.Id, out var lastMessageSendTime))
|
||||||
{
|
{
|
||||||
_libraryManager.ItemAdded += OnLibraryItemAdded;
|
if (progress > 0 && progress < 100 && (DateTime.UtcNow - lastMessageSendTime).TotalMilliseconds < 1000)
|
||||||
_libraryManager.ItemUpdated += OnLibraryItemUpdated;
|
|
||||||
_libraryManager.ItemRemoved += OnLibraryItemRemoved;
|
|
||||||
|
|
||||||
_providerManager.RefreshCompleted += OnProviderRefreshCompleted;
|
|
||||||
_providerManager.RefreshStarted += OnProviderRefreshStarted;
|
|
||||||
_providerManager.RefreshProgress += OnProviderRefreshProgress;
|
|
||||||
|
|
||||||
return Task.CompletedTask;
|
|
||||||
}
|
|
||||||
|
|
||||||
private void OnProviderRefreshProgress(object sender, GenericEventArgs<Tuple<BaseItem, double>> e)
|
|
||||||
{
|
|
||||||
var item = e.Argument.Item1;
|
|
||||||
|
|
||||||
if (!EnableRefreshMessage(item))
|
|
||||||
{
|
{
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
var progress = e.Argument.Item2;
|
_lastProgressMessageTimes.AddOrUpdate(item.Id, _ => DateTime.UtcNow, (_, _) => DateTime.UtcNow);
|
||||||
|
|
||||||
if (_lastProgressMessageTimes.TryGetValue(item.Id, out var lastMessageSendTime))
|
var dict = new Dictionary<string, string>();
|
||||||
|
dict["ItemId"] = item.Id.ToString("N", CultureInfo.InvariantCulture);
|
||||||
|
dict["Progress"] = progress.ToString(CultureInfo.InvariantCulture);
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
_sessionManager.SendMessageToAdminSessions(SessionMessageType.RefreshProgress, dict, CancellationToken.None);
|
||||||
|
}
|
||||||
|
catch
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
var collectionFolders = _libraryManager.GetCollectionFolders(item);
|
||||||
|
|
||||||
|
foreach (var collectionFolder in collectionFolders)
|
||||||
|
{
|
||||||
|
var collectionFolderDict = new Dictionary<string, string>
|
||||||
{
|
{
|
||||||
if (progress > 0 && progress < 100 && (DateTime.UtcNow - lastMessageSendTime).TotalMilliseconds < 1000)
|
["ItemId"] = collectionFolder.Id.ToString("N", CultureInfo.InvariantCulture),
|
||||||
{
|
["Progress"] = (collectionFolder.GetRefreshProgress() ?? 0).ToString(CultureInfo.InvariantCulture)
|
||||||
return;
|
};
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
_lastProgressMessageTimes.AddOrUpdate(item.Id, _ => DateTime.UtcNow, (_, _) => DateTime.UtcNow);
|
|
||||||
|
|
||||||
var dict = new Dictionary<string, string>();
|
|
||||||
dict["ItemId"] = item.Id.ToString("N", CultureInfo.InvariantCulture);
|
|
||||||
dict["Progress"] = progress.ToString(CultureInfo.InvariantCulture);
|
|
||||||
|
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
_sessionManager.SendMessageToAdminSessions(SessionMessageType.RefreshProgress, dict, CancellationToken.None);
|
_sessionManager.SendMessageToAdminSessions(SessionMessageType.RefreshProgress, collectionFolderDict, CancellationToken.None);
|
||||||
}
|
}
|
||||||
catch
|
catch
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
var collectionFolders = _libraryManager.GetCollectionFolders(item);
|
private void OnProviderRefreshStarted(object? sender, GenericEventArgs<BaseItem> e)
|
||||||
|
{
|
||||||
|
OnProviderRefreshProgress(sender, new GenericEventArgs<Tuple<BaseItem, double>>(new Tuple<BaseItem, double>(e.Argument, 0)));
|
||||||
|
}
|
||||||
|
|
||||||
foreach (var collectionFolder in collectionFolders)
|
private void OnProviderRefreshCompleted(object? sender, GenericEventArgs<BaseItem> e)
|
||||||
{
|
{
|
||||||
var collectionFolderDict = new Dictionary<string, string>
|
OnProviderRefreshProgress(sender, new GenericEventArgs<Tuple<BaseItem, double>>(new Tuple<BaseItem, double>(e.Argument, 100)));
|
||||||
{
|
|
||||||
["ItemId"] = collectionFolder.Id.ToString("N", CultureInfo.InvariantCulture),
|
|
||||||
["Progress"] = (collectionFolder.GetRefreshProgress() ?? 0).ToString(CultureInfo.InvariantCulture)
|
|
||||||
};
|
|
||||||
|
|
||||||
try
|
_lastProgressMessageTimes.TryRemove(e.Argument.Id, out _);
|
||||||
{
|
}
|
||||||
_sessionManager.SendMessageToAdminSessions(SessionMessageType.RefreshProgress, collectionFolderDict, CancellationToken.None);
|
|
||||||
}
|
private static bool EnableRefreshMessage(BaseItem item)
|
||||||
catch
|
=> item is Folder { IsRoot: false, IsTopParent: true }
|
||||||
{
|
and not (AggregateFolder or UserRootFolder or UserView or Channel);
|
||||||
}
|
|
||||||
}
|
private void OnLibraryItemAdded(object? sender, ItemChangeEventArgs e)
|
||||||
|
=> OnLibraryChange(e.Item, e.Parent, _itemsAdded, _foldersAddedTo);
|
||||||
|
|
||||||
|
private void OnLibraryItemUpdated(object? sender, ItemChangeEventArgs e)
|
||||||
|
=> OnLibraryChange(e.Item, e.Parent, _itemsUpdated, null);
|
||||||
|
|
||||||
|
private void OnLibraryItemRemoved(object? sender, ItemChangeEventArgs e)
|
||||||
|
=> OnLibraryChange(e.Item, e.Parent, _itemsRemoved, _foldersRemovedFrom);
|
||||||
|
|
||||||
|
private void OnLibraryChange(BaseItem item, BaseItem parent, List<BaseItem> itemsList, List<Folder>? foldersList)
|
||||||
|
{
|
||||||
|
if (!FilterItem(item))
|
||||||
|
{
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
private void OnProviderRefreshStarted(object sender, GenericEventArgs<BaseItem> e)
|
lock (_libraryChangedSyncLock)
|
||||||
{
|
{
|
||||||
OnProviderRefreshProgress(sender, new GenericEventArgs<Tuple<BaseItem, double>>(new Tuple<BaseItem, double>(e.Argument, 0)));
|
var updateDuration = TimeSpan.FromSeconds(_configurationManager.Configuration.LibraryUpdateDuration);
|
||||||
|
|
||||||
|
if (_libraryUpdateTimer is null)
|
||||||
|
{
|
||||||
|
_libraryUpdateTimer = new Timer(LibraryUpdateTimerCallback, null, updateDuration, Timeout.InfiniteTimeSpan);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
_libraryUpdateTimer.Change(updateDuration, Timeout.InfiniteTimeSpan);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (foldersList is not null && parent is Folder folder)
|
||||||
|
{
|
||||||
|
foldersList.Add(folder);
|
||||||
|
}
|
||||||
|
|
||||||
|
itemsList.Add(item);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private async void LibraryUpdateTimerCallback(object? state)
|
||||||
|
{
|
||||||
|
List<Folder> foldersAddedTo;
|
||||||
|
List<Folder> foldersRemovedFrom;
|
||||||
|
List<BaseItem> itemsUpdated;
|
||||||
|
List<BaseItem> itemsAdded;
|
||||||
|
List<BaseItem> itemsRemoved;
|
||||||
|
lock (_libraryChangedSyncLock)
|
||||||
|
{
|
||||||
|
// Remove dupes in case some were saved multiple times
|
||||||
|
foldersAddedTo = _foldersAddedTo
|
||||||
|
.DistinctBy(x => x.Id)
|
||||||
|
.ToList();
|
||||||
|
|
||||||
|
foldersRemovedFrom = _foldersRemovedFrom
|
||||||
|
.DistinctBy(x => x.Id)
|
||||||
|
.ToList();
|
||||||
|
|
||||||
|
itemsUpdated = _itemsUpdated
|
||||||
|
.Where(i => !_itemsAdded.Contains(i))
|
||||||
|
.DistinctBy(x => x.Id)
|
||||||
|
.ToList();
|
||||||
|
|
||||||
|
itemsAdded = _itemsAdded.ToList();
|
||||||
|
itemsRemoved = _itemsRemoved.ToList();
|
||||||
|
|
||||||
|
if (_libraryUpdateTimer is not null)
|
||||||
|
{
|
||||||
|
_libraryUpdateTimer.Dispose();
|
||||||
|
_libraryUpdateTimer = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
_itemsAdded.Clear();
|
||||||
|
_itemsRemoved.Clear();
|
||||||
|
_itemsUpdated.Clear();
|
||||||
|
_foldersAddedTo.Clear();
|
||||||
|
_foldersRemovedFrom.Clear();
|
||||||
}
|
}
|
||||||
|
|
||||||
private void OnProviderRefreshCompleted(object sender, GenericEventArgs<BaseItem> e)
|
await SendChangeNotifications(itemsAdded, itemsUpdated, itemsRemoved, foldersAddedTo, foldersRemovedFrom, CancellationToken.None).ConfigureAwait(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
private async Task SendChangeNotifications(
|
||||||
|
List<BaseItem> itemsAdded,
|
||||||
|
List<BaseItem> itemsUpdated,
|
||||||
|
List<BaseItem> itemsRemoved,
|
||||||
|
List<Folder> foldersAddedTo,
|
||||||
|
List<Folder> foldersRemovedFrom,
|
||||||
|
CancellationToken cancellationToken)
|
||||||
|
{
|
||||||
|
var userIds = _sessionManager.Sessions
|
||||||
|
.Select(i => i.UserId)
|
||||||
|
.Where(i => !i.Equals(default))
|
||||||
|
.Distinct()
|
||||||
|
.ToArray();
|
||||||
|
|
||||||
|
foreach (var userId in userIds)
|
||||||
{
|
{
|
||||||
OnProviderRefreshProgress(sender, new GenericEventArgs<Tuple<BaseItem, double>>(new Tuple<BaseItem, double>(e.Argument, 100)));
|
LibraryUpdateInfo info;
|
||||||
|
|
||||||
_lastProgressMessageTimes.TryRemove(e.Argument.Id, out _);
|
try
|
||||||
}
|
|
||||||
|
|
||||||
private static bool EnableRefreshMessage(BaseItem item)
|
|
||||||
{
|
|
||||||
if (item is not Folder folder)
|
|
||||||
{
|
{
|
||||||
return false;
|
info = GetLibraryUpdateInfo(itemsAdded, itemsUpdated, itemsRemoved, foldersAddedTo, foldersRemovedFrom, userId);
|
||||||
}
|
}
|
||||||
|
catch (Exception ex)
|
||||||
if (folder.IsRoot)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (folder is AggregateFolder || folder is UserRootFolder)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (folder is UserView || folder is Channel)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!folder.IsTopParent)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Handles the ItemAdded event of the libraryManager control.
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="sender">The source of the event.</param>
|
|
||||||
/// <param name="e">The <see cref="ItemChangeEventArgs"/> instance containing the event data.</param>
|
|
||||||
private void OnLibraryItemAdded(object sender, ItemChangeEventArgs e)
|
|
||||||
{
|
|
||||||
if (!FilterItem(e.Item))
|
|
||||||
{
|
{
|
||||||
|
_logger.LogError(ex, "Error in GetLibraryUpdateInfo");
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
lock (_libraryChangedSyncLock)
|
if (info.IsEmpty)
|
||||||
{
|
{
|
||||||
if (LibraryUpdateTimer is null)
|
continue;
|
||||||
{
|
}
|
||||||
LibraryUpdateTimer = new Timer(
|
|
||||||
LibraryUpdateTimerCallback,
|
|
||||||
null,
|
|
||||||
TimeSpan.FromSeconds(_configurationManager.Configuration.LibraryUpdateDuration),
|
|
||||||
Timeout.InfiniteTimeSpan);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
LibraryUpdateTimer.Change(TimeSpan.FromSeconds(_configurationManager.Configuration.LibraryUpdateDuration), Timeout.InfiniteTimeSpan);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (e.Item.GetParent() is Folder parent)
|
try
|
||||||
{
|
{
|
||||||
_foldersAddedTo.Add(parent);
|
await _sessionManager.SendMessageToUserSessions(
|
||||||
}
|
new List<Guid> { userId },
|
||||||
|
SessionMessageType.LibraryChanged,
|
||||||
_itemsAdded.Add(e.Item);
|
info,
|
||||||
|
cancellationToken)
|
||||||
|
.ConfigureAwait(false);
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
_logger.LogError(ex, "Error sending LibraryChanged message");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/// <summary>
|
private LibraryUpdateInfo GetLibraryUpdateInfo(
|
||||||
/// Handles the ItemUpdated event of the libraryManager control.
|
List<BaseItem> itemsAdded,
|
||||||
/// </summary>
|
List<BaseItem> itemsUpdated,
|
||||||
/// <param name="sender">The source of the event.</param>
|
List<BaseItem> itemsRemoved,
|
||||||
/// <param name="e">The <see cref="ItemChangeEventArgs"/> instance containing the event data.</param>
|
List<Folder> foldersAddedTo,
|
||||||
private void OnLibraryItemUpdated(object sender, ItemChangeEventArgs e)
|
List<Folder> foldersRemovedFrom,
|
||||||
|
Guid userId)
|
||||||
|
{
|
||||||
|
var user = _userManager.GetUserById(userId);
|
||||||
|
ArgumentNullException.ThrowIfNull(user);
|
||||||
|
|
||||||
|
var newAndRemoved = new List<BaseItem>();
|
||||||
|
newAndRemoved.AddRange(foldersAddedTo);
|
||||||
|
newAndRemoved.AddRange(foldersRemovedFrom);
|
||||||
|
|
||||||
|
var allUserRootChildren = _libraryManager.GetUserRootFolder()
|
||||||
|
.GetChildren(user, true)
|
||||||
|
.OfType<Folder>()
|
||||||
|
.ToList();
|
||||||
|
|
||||||
|
return new LibraryUpdateInfo
|
||||||
{
|
{
|
||||||
if (!FilterItem(e.Item))
|
ItemsAdded = itemsAdded.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user))
|
||||||
{
|
.Select(i => i.Id.ToString("N", CultureInfo.InvariantCulture))
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
lock (_libraryChangedSyncLock)
|
|
||||||
{
|
|
||||||
if (LibraryUpdateTimer is null)
|
|
||||||
{
|
|
||||||
LibraryUpdateTimer = new Timer(LibraryUpdateTimerCallback, null, TimeSpan.FromSeconds(_configurationManager.Configuration.LibraryUpdateDuration), Timeout.InfiniteTimeSpan);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
LibraryUpdateTimer.Change(TimeSpan.FromSeconds(_configurationManager.Configuration.LibraryUpdateDuration), Timeout.InfiniteTimeSpan);
|
|
||||||
}
|
|
||||||
|
|
||||||
_itemsUpdated.Add(e.Item);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Handles the ItemRemoved event of the libraryManager control.
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="sender">The source of the event.</param>
|
|
||||||
/// <param name="e">The <see cref="ItemChangeEventArgs"/> instance containing the event data.</param>
|
|
||||||
private void OnLibraryItemRemoved(object sender, ItemChangeEventArgs e)
|
|
||||||
{
|
|
||||||
if (!FilterItem(e.Item))
|
|
||||||
{
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
lock (_libraryChangedSyncLock)
|
|
||||||
{
|
|
||||||
if (LibraryUpdateTimer is null)
|
|
||||||
{
|
|
||||||
LibraryUpdateTimer = new Timer(LibraryUpdateTimerCallback, null, TimeSpan.FromSeconds(_configurationManager.Configuration.LibraryUpdateDuration), Timeout.InfiniteTimeSpan);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
LibraryUpdateTimer.Change(TimeSpan.FromSeconds(_configurationManager.Configuration.LibraryUpdateDuration), Timeout.InfiniteTimeSpan);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (e.Parent is Folder parent)
|
|
||||||
{
|
|
||||||
_foldersRemovedFrom.Add(parent);
|
|
||||||
}
|
|
||||||
|
|
||||||
_itemsRemoved.Add(e.Item);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Libraries the update timer callback.
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="state">The state.</param>
|
|
||||||
private async void LibraryUpdateTimerCallback(object state)
|
|
||||||
{
|
|
||||||
List<Folder> foldersAddedTo;
|
|
||||||
List<Folder> foldersRemovedFrom;
|
|
||||||
List<BaseItem> itemsUpdated;
|
|
||||||
List<BaseItem> itemsAdded;
|
|
||||||
List<BaseItem> itemsRemoved;
|
|
||||||
lock (_libraryChangedSyncLock)
|
|
||||||
{
|
|
||||||
// Remove dupes in case some were saved multiple times
|
|
||||||
foldersAddedTo = _foldersAddedTo
|
|
||||||
.DistinctBy(x => x.Id)
|
|
||||||
.ToList();
|
|
||||||
|
|
||||||
foldersRemovedFrom = _foldersRemovedFrom
|
|
||||||
.DistinctBy(x => x.Id)
|
|
||||||
.ToList();
|
|
||||||
|
|
||||||
itemsUpdated = _itemsUpdated
|
|
||||||
.Where(i => !_itemsAdded.Contains(i))
|
|
||||||
.DistinctBy(x => x.Id)
|
|
||||||
.ToList();
|
|
||||||
|
|
||||||
itemsAdded = _itemsAdded.ToList();
|
|
||||||
itemsRemoved = _itemsRemoved.ToList();
|
|
||||||
|
|
||||||
if (LibraryUpdateTimer is not null)
|
|
||||||
{
|
|
||||||
LibraryUpdateTimer.Dispose();
|
|
||||||
LibraryUpdateTimer = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
_itemsAdded.Clear();
|
|
||||||
_itemsRemoved.Clear();
|
|
||||||
_itemsUpdated.Clear();
|
|
||||||
_foldersAddedTo.Clear();
|
|
||||||
_foldersRemovedFrom.Clear();
|
|
||||||
}
|
|
||||||
|
|
||||||
await SendChangeNotifications(itemsAdded, itemsUpdated, itemsRemoved, foldersAddedTo, foldersRemovedFrom, CancellationToken.None).ConfigureAwait(false);
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Sends the change notifications.
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="itemsAdded">The items added.</param>
|
|
||||||
/// <param name="itemsUpdated">The items updated.</param>
|
|
||||||
/// <param name="itemsRemoved">The items removed.</param>
|
|
||||||
/// <param name="foldersAddedTo">The folders added to.</param>
|
|
||||||
/// <param name="foldersRemovedFrom">The folders removed from.</param>
|
|
||||||
/// <param name="cancellationToken">The cancellation token.</param>
|
|
||||||
private async Task SendChangeNotifications(List<BaseItem> itemsAdded, List<BaseItem> itemsUpdated, List<BaseItem> itemsRemoved, List<Folder> foldersAddedTo, List<Folder> foldersRemovedFrom, CancellationToken cancellationToken)
|
|
||||||
{
|
|
||||||
var userIds = _sessionManager.Sessions
|
|
||||||
.Select(i => i.UserId)
|
|
||||||
.Where(i => !i.Equals(default))
|
|
||||||
.Distinct()
|
.Distinct()
|
||||||
.ToArray();
|
.ToArray(),
|
||||||
|
ItemsUpdated = itemsUpdated.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user))
|
||||||
|
.Select(i => i.Id.ToString("N", CultureInfo.InvariantCulture))
|
||||||
|
.Distinct()
|
||||||
|
.ToArray(),
|
||||||
|
ItemsRemoved = itemsRemoved.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user, true))
|
||||||
|
.Select(i => i.Id.ToString("N", CultureInfo.InvariantCulture))
|
||||||
|
.Distinct()
|
||||||
|
.ToArray(),
|
||||||
|
FoldersAddedTo = foldersAddedTo.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user))
|
||||||
|
.Select(i => i.Id.ToString("N", CultureInfo.InvariantCulture))
|
||||||
|
.Distinct()
|
||||||
|
.ToArray(),
|
||||||
|
FoldersRemovedFrom = foldersRemovedFrom.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user))
|
||||||
|
.Select(i => i.Id.ToString("N", CultureInfo.InvariantCulture))
|
||||||
|
.Distinct()
|
||||||
|
.ToArray(),
|
||||||
|
CollectionFolders = GetTopParentIds(newAndRemoved, allUserRootChildren).ToArray()
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
foreach (var userId in userIds)
|
private static bool FilterItem(BaseItem item)
|
||||||
{
|
{
|
||||||
LibraryUpdateInfo info;
|
if (!item.IsFolder && !item.HasPathProtocol)
|
||||||
|
|
||||||
try
|
|
||||||
{
|
|
||||||
info = GetLibraryUpdateInfo(itemsAdded, itemsUpdated, itemsRemoved, foldersAddedTo, foldersRemovedFrom, userId);
|
|
||||||
}
|
|
||||||
catch (Exception ex)
|
|
||||||
{
|
|
||||||
_logger.LogError(ex, "Error in GetLibraryUpdateInfo");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (info.IsEmpty)
|
|
||||||
{
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
try
|
|
||||||
{
|
|
||||||
await _sessionManager.SendMessageToUserSessions(new List<Guid> { userId }, SessionMessageType.LibraryChanged, info, cancellationToken).ConfigureAwait(false);
|
|
||||||
}
|
|
||||||
catch (Exception ex)
|
|
||||||
{
|
|
||||||
_logger.LogError(ex, "Error sending LibraryChanged message");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Gets the library update info.
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="itemsAdded">The items added.</param>
|
|
||||||
/// <param name="itemsUpdated">The items updated.</param>
|
|
||||||
/// <param name="itemsRemoved">The items removed.</param>
|
|
||||||
/// <param name="foldersAddedTo">The folders added to.</param>
|
|
||||||
/// <param name="foldersRemovedFrom">The folders removed from.</param>
|
|
||||||
/// <param name="userId">The user id.</param>
|
|
||||||
/// <returns>LibraryUpdateInfo.</returns>
|
|
||||||
private LibraryUpdateInfo GetLibraryUpdateInfo(List<BaseItem> itemsAdded, List<BaseItem> itemsUpdated, List<BaseItem> itemsRemoved, List<Folder> foldersAddedTo, List<Folder> foldersRemovedFrom, Guid userId)
|
|
||||||
{
|
{
|
||||||
var user = _userManager.GetUserById(userId);
|
return false;
|
||||||
|
|
||||||
var newAndRemoved = new List<BaseItem>();
|
|
||||||
newAndRemoved.AddRange(foldersAddedTo);
|
|
||||||
newAndRemoved.AddRange(foldersRemovedFrom);
|
|
||||||
|
|
||||||
var allUserRootChildren = _libraryManager.GetUserRootFolder().GetChildren(user, true).OfType<Folder>().ToList();
|
|
||||||
|
|
||||||
return new LibraryUpdateInfo
|
|
||||||
{
|
|
||||||
ItemsAdded = itemsAdded.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user)).Select(i => i.Id.ToString("N", CultureInfo.InvariantCulture)).Distinct().ToArray(),
|
|
||||||
|
|
||||||
ItemsUpdated = itemsUpdated.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user)).Select(i => i.Id.ToString("N", CultureInfo.InvariantCulture)).Distinct().ToArray(),
|
|
||||||
|
|
||||||
ItemsRemoved = itemsRemoved.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user, true)).Select(i => i.Id.ToString("N", CultureInfo.InvariantCulture)).Distinct().ToArray(),
|
|
||||||
|
|
||||||
FoldersAddedTo = foldersAddedTo.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user)).Select(i => i.Id.ToString("N", CultureInfo.InvariantCulture)).Distinct().ToArray(),
|
|
||||||
|
|
||||||
FoldersRemovedFrom = foldersRemovedFrom.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user)).Select(i => i.Id.ToString("N", CultureInfo.InvariantCulture)).Distinct().ToArray(),
|
|
||||||
|
|
||||||
CollectionFolders = GetTopParentIds(newAndRemoved, allUserRootChildren).ToArray()
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private static bool FilterItem(BaseItem item)
|
if (item is IItemByName && item is not MusicArtist)
|
||||||
{
|
{
|
||||||
if (!item.IsFolder && !item.HasPathProtocol)
|
return false;
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (item is IItemByName && item is not MusicArtist)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return item.SourceType == SourceType.Library;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private IEnumerable<string> GetTopParentIds(List<BaseItem> items, List<Folder> allUserRootChildren)
|
return item.SourceType == SourceType.Library;
|
||||||
{
|
}
|
||||||
var list = new List<string>();
|
|
||||||
|
|
||||||
foreach (var item in items)
|
private IEnumerable<string> GetTopParentIds(List<BaseItem> items, List<Folder> allUserRootChildren)
|
||||||
{
|
{
|
||||||
// If the physical root changed, return the user root
|
var list = new List<string>();
|
||||||
if (item is AggregateFolder)
|
|
||||||
{
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
foreach (var folder in allUserRootChildren)
|
foreach (var item in items)
|
||||||
{
|
|
||||||
list.Add(folder.Id.ToString("N", CultureInfo.InvariantCulture));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return list.Distinct(StringComparer.Ordinal);
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Translates the physical item to user library.
|
|
||||||
/// </summary>
|
|
||||||
/// <typeparam name="T">The type of item.</typeparam>
|
|
||||||
/// <param name="item">The item.</param>
|
|
||||||
/// <param name="user">The user.</param>
|
|
||||||
/// <param name="includeIfNotFound">if set to <c>true</c> [include if not found].</param>
|
|
||||||
/// <returns>IEnumerable{``0}.</returns>
|
|
||||||
private IEnumerable<T> TranslatePhysicalItemToUserLibrary<T>(T item, User user, bool includeIfNotFound = false)
|
|
||||||
where T : BaseItem
|
|
||||||
{
|
{
|
||||||
// If the physical root changed, return the user root
|
// If the physical root changed, return the user root
|
||||||
if (item is AggregateFolder)
|
if (item is AggregateFolder)
|
||||||
{
|
{
|
||||||
return new[] { _libraryManager.GetUserRootFolder() as T };
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Return it only if it's in the user's library
|
foreach (var folder in allUserRootChildren)
|
||||||
if (includeIfNotFound || item.IsVisibleStandalone(user))
|
|
||||||
{
|
{
|
||||||
return new[] { item };
|
list.Add(folder.Id.ToString("N", CultureInfo.InvariantCulture));
|
||||||
}
|
}
|
||||||
|
|
||||||
return Array.Empty<T>();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
return list.Distinct(StringComparer.Ordinal);
|
||||||
/// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources.
|
}
|
||||||
/// </summary>
|
|
||||||
public void Dispose()
|
private IEnumerable<T> TranslatePhysicalItemToUserLibrary<T>(T item, User user, bool includeIfNotFound = false)
|
||||||
|
where T : BaseItem
|
||||||
|
{
|
||||||
|
// If the physical root changed, return the user root
|
||||||
|
if (item is AggregateFolder)
|
||||||
{
|
{
|
||||||
Dispose(true);
|
return _libraryManager.GetUserRootFolder() is T t ? new[] { t } : Array.Empty<T>();
|
||||||
GC.SuppressFinalize(this);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// <summary>
|
// Return it only if it's in the user's library
|
||||||
/// Releases unmanaged and - optionally - managed resources.
|
if (includeIfNotFound || item.IsVisibleStandalone(user))
|
||||||
/// </summary>
|
|
||||||
/// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param>
|
|
||||||
protected virtual void Dispose(bool dispose)
|
|
||||||
{
|
{
|
||||||
if (dispose)
|
return new[] { item };
|
||||||
{
|
}
|
||||||
if (LibraryUpdateTimer is not null)
|
|
||||||
{
|
|
||||||
LibraryUpdateTimer.Dispose();
|
|
||||||
LibraryUpdateTimer = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
_libraryManager.ItemAdded -= OnLibraryItemAdded;
|
return Array.Empty<T>();
|
||||||
_libraryManager.ItemUpdated -= OnLibraryItemUpdated;
|
}
|
||||||
_libraryManager.ItemRemoved -= OnLibraryItemRemoved;
|
|
||||||
|
|
||||||
_providerManager.RefreshCompleted -= OnProviderRefreshCompleted;
|
/// <inheritdoc />
|
||||||
_providerManager.RefreshStarted -= OnProviderRefreshStarted;
|
public void Dispose()
|
||||||
_providerManager.RefreshProgress -= OnProviderRefreshProgress;
|
{
|
||||||
}
|
_libraryManager.ItemAdded -= OnLibraryItemAdded;
|
||||||
|
_libraryManager.ItemUpdated -= OnLibraryItemUpdated;
|
||||||
|
_libraryManager.ItemRemoved -= OnLibraryItemRemoved;
|
||||||
|
|
||||||
|
_providerManager.RefreshCompleted -= OnProviderRefreshCompleted;
|
||||||
|
_providerManager.RefreshStarted -= OnProviderRefreshStarted;
|
||||||
|
_providerManager.RefreshProgress -= OnProviderRefreshProgress;
|
||||||
|
|
||||||
|
if (_libraryUpdateTimer is not null)
|
||||||
|
{
|
||||||
|
_libraryUpdateTimer.Dispose();
|
||||||
|
_libraryUpdateTimer = null;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user