jellyfin/Emby.Server.Implementations/Collections/CollectionManager.cs

302 lines
10 KiB
C#
Raw Normal View History

2014-07-08 01:41:03 +00:00
using MediaBrowser.Common.Events;
2014-03-06 05:17:13 +00:00
using MediaBrowser.Controller.Collections;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Movies;
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Providers;
2014-07-08 01:41:03 +00:00
using MediaBrowser.Model.Logging;
2014-03-06 05:17:13 +00:00
using System;
using System.Collections.Generic;
2014-03-06 05:17:13 +00:00
using System.IO;
using System.Linq;
2014-03-06 05:17:13 +00:00
using System.Threading;
using System.Threading.Tasks;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Model.IO;
2014-03-06 05:17:13 +00:00
namespace Emby.Server.Implementations.Collections
2014-03-06 05:17:13 +00:00
{
public class CollectionManager : ICollectionManager
{
private readonly ILibraryManager _libraryManager;
private readonly IFileSystem _fileSystem;
private readonly ILibraryMonitor _iLibraryMonitor;
2014-07-08 01:41:03 +00:00
private readonly ILogger _logger;
2015-09-29 16:29:06 +00:00
private readonly IProviderManager _providerManager;
2014-03-06 05:17:13 +00:00
2014-07-08 01:41:03 +00:00
public event EventHandler<CollectionCreatedEventArgs> CollectionCreated;
public event EventHandler<CollectionModifiedEventArgs> ItemsAddedToCollection;
public event EventHandler<CollectionModifiedEventArgs> ItemsRemovedFromCollection;
2015-09-29 16:29:06 +00:00
public CollectionManager(ILibraryManager libraryManager, IFileSystem fileSystem, ILibraryMonitor iLibraryMonitor, ILogger logger, IProviderManager providerManager)
2014-03-06 05:17:13 +00:00
{
_libraryManager = libraryManager;
_fileSystem = fileSystem;
_iLibraryMonitor = iLibraryMonitor;
2014-07-08 01:41:03 +00:00
_logger = logger;
2015-09-29 16:29:06 +00:00
_providerManager = providerManager;
2014-03-06 05:17:13 +00:00
}
2014-07-01 04:06:28 +00:00
public Folder GetCollectionsFolder(string userId)
{
2014-08-02 02:34:45 +00:00
return _libraryManager.RootFolder.Children.OfType<ManualCollectionsFolder>()
2015-10-29 13:28:05 +00:00
.FirstOrDefault() ?? _libraryManager.GetUserRootFolder().Children.OfType<ManualCollectionsFolder>()
2014-07-01 04:06:28 +00:00
.FirstOrDefault();
}
2015-01-24 19:03:55 +00:00
public IEnumerable<BoxSet> GetCollections(User user)
{
var folder = GetCollectionsFolder(user.Id.ToString("N"));
return folder == null ?
new List<BoxSet>() :
folder.GetChildren(user, true).OfType<BoxSet>();
}
2014-03-15 15:17:46 +00:00
public async Task<BoxSet> CreateCollection(CollectionCreationOptions options)
2014-03-06 05:17:13 +00:00
{
var name = options.Name;
2014-03-08 04:20:31 +00:00
// Need to use the [boxset] suffix
// If internet metadata is not found, or if xml saving is off there will be no collection.xml
// This could cause it to get re-resolved as a plain folder
var folderName = _fileSystem.GetValidFilename(name) + " [boxset]";
2014-03-06 05:17:13 +00:00
var parentFolder = GetParentFolder(options.ParentId);
2014-03-06 05:17:13 +00:00
if (parentFolder == null)
{
throw new ArgumentException();
}
var path = Path.Combine(parentFolder.Path, folderName);
_iLibraryMonitor.ReportFileSystemChangeBeginning(path);
try
{
2015-10-14 05:02:30 +00:00
_fileSystem.CreateDirectory(path);
2014-03-06 05:17:13 +00:00
var collection = new BoxSet
{
Name = name,
Path = path,
2014-04-27 03:42:05 +00:00
IsLocked = options.IsLocked,
2014-12-13 03:56:30 +00:00
ProviderIds = options.ProviderIds,
Shares = options.UserIds.Select(i => new Share
{
2015-04-26 04:39:40 +00:00
UserId = i.ToString("N"),
CanEdit = true
2014-12-13 03:56:30 +00:00
}).ToList()
2014-03-06 05:17:13 +00:00
};
await parentFolder.AddChild(collection, CancellationToken.None).ConfigureAwait(false);
2014-03-15 15:17:46 +00:00
if (options.ItemIdList.Count > 0)
{
2015-10-14 05:02:30 +00:00
await AddToCollection(collection.Id, options.ItemIdList, false, new MetadataRefreshOptions(_fileSystem)
{
// The initial adding of items is going to create a local metadata file
// This will cause internet metadata to be skipped as a result
MetadataRefreshMode = MetadataRefreshMode.FullRefresh
});
2014-03-15 15:17:46 +00:00
}
2015-09-29 16:29:06 +00:00
else
{
2017-04-30 02:37:51 +00:00
_providerManager.QueueRefresh(collection.Id, new MetadataRefreshOptions(_fileSystem), RefreshPriority.High);
2015-09-29 16:29:06 +00:00
}
2014-03-15 15:17:46 +00:00
2014-07-08 01:41:03 +00:00
EventHelper.FireEventIfNotNull(CollectionCreated, this, new CollectionCreatedEventArgs
{
Collection = collection,
Options = options
}, _logger);
2014-03-15 15:17:46 +00:00
return collection;
2014-03-06 05:17:13 +00:00
}
finally
{
// Refresh handled internally
_iLibraryMonitor.ReportFileSystemChangeComplete(path, false);
}
}
private Folder GetParentFolder(Guid? parentId)
2014-03-06 05:17:13 +00:00
{
if (parentId.HasValue)
{
if (parentId.Value == Guid.Empty)
{
throw new ArgumentNullException("parentId");
}
2014-03-10 17:38:53 +00:00
var folder = _libraryManager.GetItemById(parentId.Value) as Folder;
// Find an actual physical folder
if (folder is CollectionFolder)
{
2014-05-07 02:28:19 +00:00
var child = _libraryManager.RootFolder.Children.OfType<Folder>()
.FirstOrDefault(i => folder.PhysicalLocations.Contains(i.Path, StringComparer.OrdinalIgnoreCase));
if (child != null)
{
return child;
}
2014-03-10 17:38:53 +00:00
}
}
2014-07-01 04:06:28 +00:00
return GetCollectionsFolder(string.Empty);
2014-03-06 05:17:13 +00:00
}
2014-07-08 01:41:03 +00:00
public Task AddToCollection(Guid collectionId, IEnumerable<Guid> ids)
{
2015-10-14 05:02:30 +00:00
return AddToCollection(collectionId, ids, true, new MetadataRefreshOptions(_fileSystem));
2014-07-08 01:41:03 +00:00
}
2015-10-14 05:02:30 +00:00
private async Task AddToCollection(Guid collectionId, IEnumerable<Guid> ids, bool fireEvent, MetadataRefreshOptions refreshOptions)
2014-03-06 05:17:13 +00:00
{
var collection = _libraryManager.GetItemById(collectionId) as BoxSet;
if (collection == null)
{
throw new ArgumentException("No collection exists with the supplied Id");
}
var list = new List<LinkedChild>();
2014-07-08 01:41:03 +00:00
var itemList = new List<BaseItem>();
var currentLinkedChildren = collection.GetLinkedChildren().ToList();
foreach (var itemId in ids)
{
var item = _libraryManager.GetItemById(itemId);
2017-05-30 18:24:50 +00:00
if (string.IsNullOrWhiteSpace(item.Path))
{
continue;
}
if (item == null)
{
throw new ArgumentException("No item exists with the supplied Id");
}
2014-07-08 01:41:03 +00:00
itemList.Add(item);
2014-12-13 03:56:30 +00:00
2015-10-07 21:42:29 +00:00
if (currentLinkedChildren.All(i => i.Id != itemId))
{
2015-10-07 21:42:29 +00:00
list.Add(LinkedChild.Create(item));
}
}
2015-10-07 21:42:29 +00:00
if (list.Count > 0)
{
collection.LinkedChildren.AddRange(list);
2015-10-07 21:42:29 +00:00
collection.UpdateRatingToContent();
2015-10-07 21:42:29 +00:00
await collection.UpdateToRepository(ItemUpdateType.MetadataEdit, CancellationToken.None).ConfigureAwait(false);
2015-09-29 16:29:06 +00:00
2017-04-30 02:37:51 +00:00
_providerManager.QueueRefresh(collection.Id, refreshOptions, RefreshPriority.High);
2014-07-08 01:41:03 +00:00
2015-10-07 21:42:29 +00:00
if (fireEvent)
2014-07-08 01:41:03 +00:00
{
2015-10-07 21:42:29 +00:00
EventHelper.FireEventIfNotNull(ItemsAddedToCollection, this, new CollectionModifiedEventArgs
{
Collection = collection,
ItemsChanged = itemList
2014-07-08 01:41:03 +00:00
2015-10-07 21:42:29 +00:00
}, _logger);
}
2014-07-08 01:41:03 +00:00
}
}
public async Task RemoveFromCollection(Guid collectionId, IEnumerable<Guid> itemIds)
{
var collection = _libraryManager.GetItemById(collectionId) as BoxSet;
if (collection == null)
{
throw new ArgumentException("No collection exists with the supplied Id");
}
var list = new List<LinkedChild>();
2014-07-08 01:41:03 +00:00
var itemList = new List<BaseItem>();
foreach (var itemId in itemIds)
{
2016-10-10 18:18:28 +00:00
var childItem = _libraryManager.GetItemById(itemId);
var child = collection.LinkedChildren.FirstOrDefault(i => (i.ItemId.HasValue && i.ItemId.Value == itemId) || (childItem != null && string.Equals(childItem.Path, i.Path, StringComparison.OrdinalIgnoreCase)));
if (child == null)
{
throw new ArgumentException("No collection title exists with the supplied Id");
}
list.Add(child);
2014-03-15 04:14:07 +00:00
2014-07-08 01:41:03 +00:00
if (childItem != null)
{
itemList.Add(childItem);
}
}
2016-10-10 18:18:28 +00:00
foreach (var child in list)
{
2016-10-10 18:18:28 +00:00
collection.LinkedChildren.Remove(child);
}
2015-01-19 04:29:57 +00:00
collection.UpdateRatingToContent();
2015-01-19 04:29:57 +00:00
await collection.UpdateToRepository(ItemUpdateType.MetadataEdit, CancellationToken.None).ConfigureAwait(false);
2017-04-30 02:37:51 +00:00
_providerManager.QueueRefresh(collection.Id, new MetadataRefreshOptions(_fileSystem), RefreshPriority.High);
2014-07-08 01:41:03 +00:00
EventHelper.FireEventIfNotNull(ItemsRemovedFromCollection, this, new CollectionModifiedEventArgs
{
Collection = collection,
ItemsChanged = itemList
}, _logger);
2014-03-06 05:17:13 +00:00
}
2014-04-22 17:25:54 +00:00
public IEnumerable<BaseItem> CollapseItemsWithinBoxSets(IEnumerable<BaseItem> items, User user)
{
2015-01-22 16:41:34 +00:00
var results = new Dictionary<Guid, BaseItem>();
2015-01-24 19:03:55 +00:00
var allBoxsets = GetCollections(user).ToList();
2014-04-22 17:25:54 +00:00
2015-01-22 16:41:34 +00:00
foreach (var item in items)
2014-04-22 17:25:54 +00:00
{
2015-01-22 16:41:34 +00:00
var grouping = item as ISupportsBoxSetGrouping;
2014-04-22 17:25:54 +00:00
2015-01-22 16:41:34 +00:00
if (grouping == null)
{
results[item.Id] = item;
}
else
2014-04-22 17:25:54 +00:00
{
2015-01-22 16:41:34 +00:00
var itemId = item.Id;
var currentBoxSets = allBoxsets
.Where(i => i.GetLinkedChildren().Any(j => j.Id == itemId))
.ToList();
if (currentBoxSets.Count > 0)
{
foreach (var boxset in currentBoxSets)
{
results[boxset.Id] = boxset;
}
}
else
{
results[item.Id] = item;
}
2014-04-22 17:25:54 +00:00
}
}
2015-01-22 16:41:34 +00:00
return results.Values;
2014-04-22 17:25:54 +00:00
}
2014-03-06 05:17:13 +00:00
}
}