2013-05-17 11:29:22 -04:00
|
|
|
|
using MediaBrowser.Common.Net;
|
2013-05-21 23:42:25 -04:00
|
|
|
|
using MediaBrowser.Controller.Entities;
|
2013-05-17 11:29:22 -04:00
|
|
|
|
using MediaBrowser.Controller.Library;
|
|
|
|
|
using MediaBrowser.Controller.Plugins;
|
|
|
|
|
using MediaBrowser.Controller.Session;
|
|
|
|
|
using MediaBrowser.Model.Entities;
|
2013-05-21 23:42:25 -04:00
|
|
|
|
using MoreLinq;
|
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
2013-05-17 11:29:22 -04:00
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Threading;
|
|
|
|
|
|
|
|
|
|
namespace MediaBrowser.ServerApplication.EntryPoints
|
|
|
|
|
{
|
|
|
|
|
public class LibraryChangedNotifier : IServerEntryPoint
|
|
|
|
|
{
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// The _library manager
|
|
|
|
|
/// </summary>
|
|
|
|
|
private readonly ILibraryManager _libraryManager;
|
|
|
|
|
|
|
|
|
|
private readonly ISessionManager _sessionManager;
|
|
|
|
|
private readonly IServerManager _serverManager;
|
2013-05-21 23:42:25 -04:00
|
|
|
|
private readonly IUserManager _userManager;
|
|
|
|
|
|
2013-05-17 11:29:22 -04:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// The _library changed sync lock
|
|
|
|
|
/// </summary>
|
|
|
|
|
private readonly object _libraryChangedSyncLock = new object();
|
|
|
|
|
|
2013-05-21 23:42:25 -04:00
|
|
|
|
private readonly List<Folder> _foldersAddedTo = new List<Folder>();
|
|
|
|
|
private readonly List<Folder> _foldersRemovedFrom = new List<Folder>();
|
|
|
|
|
|
|
|
|
|
private readonly List<BaseItem> _itemsAdded = new List<BaseItem>();
|
|
|
|
|
private readonly List<BaseItem> _itemsRemoved = new List<BaseItem>();
|
|
|
|
|
private readonly List<BaseItem> _itemsUpdated = new List<BaseItem>();
|
2013-05-17 11:29:22 -04:00
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Gets or sets the library update timer.
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <value>The library update timer.</value>
|
|
|
|
|
private Timer LibraryUpdateTimer { get; set; }
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// The library update duration
|
|
|
|
|
/// </summary>
|
|
|
|
|
private const int LibraryUpdateDuration = 60000;
|
|
|
|
|
|
2013-05-21 23:42:25 -04:00
|
|
|
|
public LibraryChangedNotifier(ILibraryManager libraryManager, ISessionManager sessionManager, IServerManager serverManager, IUserManager userManager)
|
2013-05-17 11:29:22 -04:00
|
|
|
|
{
|
|
|
|
|
_libraryManager = libraryManager;
|
|
|
|
|
_sessionManager = sessionManager;
|
|
|
|
|
_serverManager = serverManager;
|
2013-05-21 23:42:25 -04:00
|
|
|
|
_userManager = userManager;
|
2013-05-17 11:29:22 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void Run()
|
|
|
|
|
{
|
|
|
|
|
_libraryManager.ItemAdded += libraryManager_ItemAdded;
|
|
|
|
|
_libraryManager.ItemUpdated += libraryManager_ItemUpdated;
|
|
|
|
|
_libraryManager.ItemRemoved += libraryManager_ItemRemoved;
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <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>
|
|
|
|
|
void libraryManager_ItemAdded(object sender, ItemChangeEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
lock (_libraryChangedSyncLock)
|
|
|
|
|
{
|
|
|
|
|
if (LibraryUpdateTimer == null)
|
|
|
|
|
{
|
|
|
|
|
LibraryUpdateTimer = new Timer(LibraryUpdateTimerCallback, null, LibraryUpdateDuration,
|
|
|
|
|
Timeout.Infinite);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
LibraryUpdateTimer.Change(LibraryUpdateDuration, Timeout.Infinite);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (e.Item.Parent != null)
|
|
|
|
|
{
|
2013-05-21 23:42:25 -04:00
|
|
|
|
_foldersAddedTo.Add(e.Item.Parent);
|
2013-05-17 11:29:22 -04:00
|
|
|
|
}
|
|
|
|
|
|
2013-05-21 23:42:25 -04:00
|
|
|
|
_itemsAdded.Add(e.Item);
|
2013-05-17 11:29:22 -04:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Handles the ItemUpdated 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>
|
|
|
|
|
void libraryManager_ItemUpdated(object sender, ItemChangeEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
lock (_libraryChangedSyncLock)
|
|
|
|
|
{
|
|
|
|
|
if (LibraryUpdateTimer == null)
|
|
|
|
|
{
|
|
|
|
|
LibraryUpdateTimer = new Timer(LibraryUpdateTimerCallback, null, LibraryUpdateDuration,
|
|
|
|
|
Timeout.Infinite);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
LibraryUpdateTimer.Change(LibraryUpdateDuration, Timeout.Infinite);
|
|
|
|
|
}
|
|
|
|
|
|
2013-05-21 23:42:25 -04:00
|
|
|
|
_itemsUpdated.Add(e.Item);
|
2013-05-17 11:29:22 -04:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <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>
|
|
|
|
|
void libraryManager_ItemRemoved(object sender, ItemChangeEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
lock (_libraryChangedSyncLock)
|
|
|
|
|
{
|
|
|
|
|
if (LibraryUpdateTimer == null)
|
|
|
|
|
{
|
|
|
|
|
LibraryUpdateTimer = new Timer(LibraryUpdateTimerCallback, null, LibraryUpdateDuration,
|
|
|
|
|
Timeout.Infinite);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
LibraryUpdateTimer.Change(LibraryUpdateDuration, Timeout.Infinite);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (e.Item.Parent != null)
|
|
|
|
|
{
|
2013-05-21 23:42:25 -04:00
|
|
|
|
_foldersRemovedFrom.Add(e.Item.Parent);
|
2013-05-17 11:29:22 -04:00
|
|
|
|
}
|
|
|
|
|
|
2013-05-21 23:42:25 -04:00
|
|
|
|
_itemsRemoved.Add(e.Item);
|
2013-05-17 11:29:22 -04:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Libraries the update timer callback.
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="state">The state.</param>
|
|
|
|
|
private void LibraryUpdateTimerCallback(object state)
|
|
|
|
|
{
|
|
|
|
|
lock (_libraryChangedSyncLock)
|
|
|
|
|
{
|
|
|
|
|
// Remove dupes in case some were saved multiple times
|
2013-05-21 23:42:25 -04:00
|
|
|
|
var foldersAddedTo = _foldersAddedTo.DistinctBy(i => i.Id).ToList();
|
2013-05-17 11:29:22 -04:00
|
|
|
|
|
2013-05-21 23:42:25 -04:00
|
|
|
|
var foldersRemovedFrom = _foldersRemovedFrom.DistinctBy(i => i.Id).ToList();
|
2013-05-17 11:29:22 -04:00
|
|
|
|
|
2013-05-21 23:42:25 -04:00
|
|
|
|
var itemsUpdated = _itemsUpdated
|
|
|
|
|
.Where(i => !_itemsAdded.Contains(i))
|
|
|
|
|
.DistinctBy(i => i.Id)
|
2013-05-17 11:29:22 -04:00
|
|
|
|
.ToList();
|
|
|
|
|
|
2013-05-21 23:42:25 -04:00
|
|
|
|
SendChangeNotifications(_itemsAdded.ToList(), itemsUpdated, _itemsRemoved.ToList(), foldersAddedTo, foldersRemovedFrom, CancellationToken.None);
|
2013-05-17 11:29:22 -04:00
|
|
|
|
|
|
|
|
|
if (LibraryUpdateTimer != null)
|
|
|
|
|
{
|
|
|
|
|
LibraryUpdateTimer.Dispose();
|
|
|
|
|
LibraryUpdateTimer = null;
|
|
|
|
|
}
|
|
|
|
|
|
2013-05-21 23:42:25 -04:00
|
|
|
|
_itemsAdded.Clear();
|
|
|
|
|
_itemsRemoved.Clear();
|
|
|
|
|
_itemsUpdated.Clear();
|
|
|
|
|
_foldersAddedTo.Clear();
|
|
|
|
|
_foldersRemovedFrom.Clear();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <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 void SendChangeNotifications(IEnumerable<BaseItem> itemsAdded, IEnumerable<BaseItem> itemsUpdated, IEnumerable<BaseItem> itemsRemoved, IEnumerable<Folder> foldersAddedTo, IEnumerable<Folder> foldersRemovedFrom, CancellationToken cancellationToken)
|
|
|
|
|
{
|
|
|
|
|
var currentSessions = _sessionManager.Sessions.ToList();
|
|
|
|
|
|
|
|
|
|
var users = currentSessions.Select(i => i.UserId ?? Guid.Empty).Where(i => i != Guid.Empty).Distinct().ToList();
|
|
|
|
|
|
|
|
|
|
foreach (var userId in users)
|
|
|
|
|
{
|
|
|
|
|
var id = userId;
|
|
|
|
|
var webSockets = currentSessions.Where(u => u.UserId.HasValue && u.UserId.Value == id).SelectMany(i => i.WebSockets).ToList();
|
|
|
|
|
|
|
|
|
|
await _serverManager.SendWebSocketMessageAsync("LibraryChanged", () => GetLibraryUpdateInfo(itemsAdded, itemsUpdated, itemsRemoved, foldersAddedTo, foldersRemovedFrom, id), webSockets, cancellationToken).ConfigureAwait(false);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <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(IEnumerable<BaseItem> itemsAdded, IEnumerable<BaseItem> itemsUpdated, IEnumerable<BaseItem> itemsRemoved, IEnumerable<Folder> foldersAddedTo, IEnumerable<Folder> foldersRemovedFrom, Guid userId)
|
|
|
|
|
{
|
|
|
|
|
var user = _userManager.GetUserById(userId);
|
|
|
|
|
|
|
|
|
|
var collections = user.RootFolder.GetChildren(user).ToList();
|
|
|
|
|
|
|
|
|
|
var allRecursiveChildren = user.RootFolder.GetRecursiveChildren(user).ToDictionary(i => i.Id);
|
|
|
|
|
|
|
|
|
|
return new LibraryUpdateInfo
|
|
|
|
|
{
|
|
|
|
|
ItemsAdded = itemsAdded.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user, collections, allRecursiveChildren)).Select(i => i.Id).Distinct().ToList(),
|
|
|
|
|
|
|
|
|
|
ItemsUpdated = itemsUpdated.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user, collections, allRecursiveChildren)).Select(i => i.Id).Distinct().ToList(),
|
|
|
|
|
|
|
|
|
|
ItemsRemoved = itemsRemoved.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user, collections, allRecursiveChildren)).Select(i => i.Id).Distinct().ToList(),
|
|
|
|
|
|
|
|
|
|
FoldersAddedTo = foldersAddedTo.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user, collections, allRecursiveChildren)).Select(i => i.Id).Distinct().ToList(),
|
|
|
|
|
|
|
|
|
|
FoldersRemovedFrom = foldersRemovedFrom.SelectMany(i => TranslatePhysicalItemToUserLibrary(i, user, collections, allRecursiveChildren)).Select(i => i.Id).Distinct().ToList()
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Translates the physical item to user library.
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <typeparam name="T"></typeparam>
|
|
|
|
|
/// <param name="item">The item.</param>
|
|
|
|
|
/// <param name="user">The user.</param>
|
|
|
|
|
/// <param name="collections">The collections.</param>
|
|
|
|
|
/// <param name="allRecursiveChildren">All recursive children.</param>
|
|
|
|
|
/// <returns>IEnumerable{``0}.</returns>
|
|
|
|
|
private IEnumerable<T> TranslatePhysicalItemToUserLibrary<T>(T item, User user, List<BaseItem> collections, Dictionary<Guid, BaseItem> allRecursiveChildren)
|
|
|
|
|
where T : BaseItem
|
|
|
|
|
{
|
|
|
|
|
// If the physical root changed, return the user root
|
|
|
|
|
if (item is AggregateFolder)
|
|
|
|
|
{
|
|
|
|
|
return new T[] { user.RootFolder as T };
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Need to find what user collection folder this belongs to
|
|
|
|
|
if (item.Parent is AggregateFolder)
|
|
|
|
|
{
|
|
|
|
|
return new T[] { user.RootFolder as T };
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// If it's a user root, return it only if it's the right one
|
|
|
|
|
if (item is UserRootFolder)
|
|
|
|
|
{
|
|
|
|
|
if (item.Id == user.RootFolder.Id)
|
|
|
|
|
{
|
|
|
|
|
return new T[] { item };
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return new T[] { };
|
2013-05-17 11:29:22 -04:00
|
|
|
|
}
|
2013-05-21 23:42:25 -04:00
|
|
|
|
|
|
|
|
|
// Return it only if it's in the user's library
|
|
|
|
|
if (allRecursiveChildren.ContainsKey(item.Id))
|
|
|
|
|
{
|
|
|
|
|
return new T[] { item };
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return new T[] { };
|
2013-05-17 11:29:22 -04:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Performs application-defined tasks associated with freeing, releasing, or resetting unmanaged resources.
|
|
|
|
|
/// </summary>
|
|
|
|
|
public void Dispose()
|
|
|
|
|
{
|
|
|
|
|
Dispose(true);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Releases unmanaged and - optionally - managed resources.
|
|
|
|
|
/// </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)
|
|
|
|
|
{
|
|
|
|
|
if (LibraryUpdateTimer != null)
|
|
|
|
|
{
|
|
|
|
|
LibraryUpdateTimer.Dispose();
|
|
|
|
|
LibraryUpdateTimer = null;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
_libraryManager.ItemAdded -= libraryManager_ItemAdded;
|
|
|
|
|
_libraryManager.ItemUpdated -= libraryManager_ItemUpdated;
|
|
|
|
|
_libraryManager.ItemRemoved -= libraryManager_ItemRemoved;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|