using MediaBrowser.Common.Extensions; using MediaBrowser.Common.IO; using MediaBrowser.Controller; using MediaBrowser.Controller.Channels; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Entities.Audio; using MediaBrowser.Controller.Entities.Movies; using MediaBrowser.Controller.Entities.TV; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.LiveTv; using MediaBrowser.Controller.Localization; using MediaBrowser.Controller.Playlists; using MediaBrowser.Model.Channels; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Library; using MediaBrowser.Model.Querying; using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Threading; using System.Threading.Tasks; namespace MediaBrowser.Server.Implementations.Library { public class UserViewManager : IUserViewManager { private readonly ILibraryManager _libraryManager; private readonly ILocalizationManager _localizationManager; private readonly IFileSystem _fileSystem; private readonly IUserManager _userManager; private readonly IChannelManager _channelManager; private readonly ILiveTvManager _liveTvManager; private readonly IServerApplicationPaths _appPaths; private readonly IPlaylistManager _playlists; public UserViewManager(ILibraryManager libraryManager, ILocalizationManager localizationManager, IFileSystem fileSystem, IUserManager userManager, IChannelManager channelManager, ILiveTvManager liveTvManager, IServerApplicationPaths appPaths, IPlaylistManager playlists) { _libraryManager = libraryManager; _localizationManager = localizationManager; _fileSystem = fileSystem; _userManager = userManager; _channelManager = channelManager; _liveTvManager = liveTvManager; _appPaths = appPaths; _playlists = playlists; } public async Task> GetUserViews(UserViewQuery query, CancellationToken cancellationToken) { var user = _userManager.GetUserById(new Guid(query.UserId)); var folders = user.RootFolder .GetChildren(user, true) .OfType() .ToList(); var list = new List(); var excludeFolderIds = user.Configuration.ExcludeFoldersFromGrouping.Select(i => new Guid(i)).ToList(); var standaloneFolders = folders.Where(i => UserView.IsExcludedFromGrouping(i) || excludeFolderIds.Contains(i.Id)).ToList(); list.AddRange(standaloneFolders); var recursiveChildren = folders .Except(standaloneFolders) .SelectMany(i => i.GetRecursiveChildren(user, false)) .ToList(); if (recursiveChildren.OfType().Any()) { list.Add(await GetUserView(CollectionType.TvShows, user, string.Empty, cancellationToken).ConfigureAwait(false)); } if (recursiveChildren.OfType().Any() || recursiveChildren.OfType().Any()) { list.Add(await GetUserView(CollectionType.Music, user, string.Empty, cancellationToken).ConfigureAwait(false)); } if (recursiveChildren.OfType().Any()) { list.Add(await GetUserView(CollectionType.Movies, user, string.Empty, cancellationToken).ConfigureAwait(false)); } if (recursiveChildren.OfType().Any()) { list.Add(await GetUserView(CollectionType.Games, user, string.Empty, cancellationToken).ConfigureAwait(false)); } if (user.Configuration.DisplayCollectionsView && recursiveChildren.OfType().Any()) { list.Add(await GetUserView(CollectionType.BoxSets, user, string.Empty, cancellationToken).ConfigureAwait(false)); } if (recursiveChildren.OfType().Any()) { list.Add(_playlists.GetPlaylistsFolder(user.Id.ToString("N"))); } if (user.Configuration.DisplayFoldersView) { list.Add(await GetUserView(CollectionType.Folders, user, "zz_" + CollectionType.Folders, cancellationToken).ConfigureAwait(false)); } if (query.IncludeExternalContent) { var channelResult = await _channelManager.GetChannels(new ChannelQuery { UserId = query.UserId }, cancellationToken).ConfigureAwait(false); var channels = channelResult.Items; var embeddedChannels = channels .Where(i => user.Configuration.DisplayChannelsWithinViews.Contains(i.Id)) .ToList(); list.AddRange(embeddedChannels.Select(i => _channelManager.GetChannel(i.Id))); if (channels.Length > embeddedChannels.Count) { list.Add(await _channelManager.GetInternalChannelFolder(query.UserId, cancellationToken).ConfigureAwait(false)); } if (_liveTvManager.GetEnabledUsers().Select(i => i.Id.ToString("N")).Contains(query.UserId)) { list.Add(await _liveTvManager.GetInternalLiveTvFolder(query.UserId, cancellationToken).ConfigureAwait(false)); } } return _libraryManager.Sort(list, user, new[] { ItemSortBy.SortName }, SortOrder.Ascending).Cast(); } public Task GetUserView(string type, User user, string sortName, CancellationToken cancellationToken) { var name = _localizationManager.GetLocalizedString("ViewType" + type); return _libraryManager.GetNamedView(name, type, sortName, cancellationToken); } public async Task GetSpecialFolder(string name, SpecialFolderType type, string itemType, CancellationToken cancellationToken) { var path = Path.Combine(_appPaths.ItemsByNamePath, "specialfolders", _fileSystem.GetValidFilename(name)); var id = (path + "_specialfolder_" + name).GetMBId(typeof(SpecialFolder)); var item = _libraryManager.GetItemById(id) as SpecialFolder; if (item == null) { Directory.CreateDirectory(Path.GetDirectoryName(path)); item = new SpecialFolder { Path = path, Id = id, DateCreated = DateTime.UtcNow, Name = name, SpecialFolderType = type, ItemTypeName = itemType }; await _libraryManager.CreateItem(item, cancellationToken).ConfigureAwait(false); await item.RefreshMetadata(cancellationToken).ConfigureAwait(false); } return item; } } }