UserViewManager.cs 7.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165
  1. using MediaBrowser.Common.IO;
  2. using MediaBrowser.Controller;
  3. using MediaBrowser.Controller.Channels;
  4. using MediaBrowser.Controller.Entities;
  5. using MediaBrowser.Controller.Entities.Audio;
  6. using MediaBrowser.Controller.Entities.Movies;
  7. using MediaBrowser.Controller.Entities.TV;
  8. using MediaBrowser.Controller.Library;
  9. using MediaBrowser.Controller.LiveTv;
  10. using MediaBrowser.Controller.Localization;
  11. using MediaBrowser.Controller.Playlists;
  12. using MediaBrowser.Model.Channels;
  13. using MediaBrowser.Model.Entities;
  14. using MediaBrowser.Model.Library;
  15. using MediaBrowser.Model.Querying;
  16. using System;
  17. using System.Collections.Generic;
  18. using System.Linq;
  19. using System.Threading;
  20. using System.Threading.Tasks;
  21. namespace MediaBrowser.Server.Implementations.Library
  22. {
  23. public class UserViewManager : IUserViewManager
  24. {
  25. private readonly ILibraryManager _libraryManager;
  26. private readonly ILocalizationManager _localizationManager;
  27. private readonly IFileSystem _fileSystem;
  28. private readonly IUserManager _userManager;
  29. private readonly IChannelManager _channelManager;
  30. private readonly ILiveTvManager _liveTvManager;
  31. private readonly IServerApplicationPaths _appPaths;
  32. private readonly IPlaylistManager _playlists;
  33. public UserViewManager(ILibraryManager libraryManager, ILocalizationManager localizationManager, IFileSystem fileSystem, IUserManager userManager, IChannelManager channelManager, ILiveTvManager liveTvManager, IServerApplicationPaths appPaths, IPlaylistManager playlists)
  34. {
  35. _libraryManager = libraryManager;
  36. _localizationManager = localizationManager;
  37. _fileSystem = fileSystem;
  38. _userManager = userManager;
  39. _channelManager = channelManager;
  40. _liveTvManager = liveTvManager;
  41. _appPaths = appPaths;
  42. _playlists = playlists;
  43. }
  44. public async Task<IEnumerable<Folder>> GetUserViews(UserViewQuery query, CancellationToken cancellationToken)
  45. {
  46. var user = _userManager.GetUserById(query.UserId);
  47. var folders = user.RootFolder
  48. .GetChildren(user, true)
  49. .OfType<Folder>()
  50. .ToList();
  51. var list = new List<Folder>();
  52. var excludeFolderIds = user.Configuration.ExcludeFoldersFromGrouping.Select(i => new Guid(i)).ToList();
  53. var standaloneFolders = folders.Where(i => UserView.IsExcludedFromGrouping(i) || excludeFolderIds.Contains(i.Id)).ToList();
  54. var foldersWithViewTypes = folders
  55. .Except(standaloneFolders)
  56. .OfType<ICollectionFolder>()
  57. .ToList();
  58. list.AddRange(standaloneFolders);
  59. if (foldersWithViewTypes.Any(i => string.Equals(i.CollectionType, CollectionType.TvShows, StringComparison.OrdinalIgnoreCase)) ||
  60. foldersWithViewTypes.Any(i => string.IsNullOrWhiteSpace(i.CollectionType)))
  61. {
  62. list.Add(await GetUserView(CollectionType.TvShows, user, string.Empty, cancellationToken).ConfigureAwait(false));
  63. }
  64. if (foldersWithViewTypes.Any(i => string.Equals(i.CollectionType, CollectionType.Music, StringComparison.OrdinalIgnoreCase)) ||
  65. foldersWithViewTypes.Any(i => string.Equals(i.CollectionType, CollectionType.MusicVideos, StringComparison.OrdinalIgnoreCase)))
  66. {
  67. list.Add(await GetUserView(CollectionType.Music, user, string.Empty, cancellationToken).ConfigureAwait(false));
  68. }
  69. if (foldersWithViewTypes.Any(i => string.Equals(i.CollectionType, CollectionType.Movies, StringComparison.OrdinalIgnoreCase)) ||
  70. foldersWithViewTypes.Any(i => string.IsNullOrWhiteSpace(i.CollectionType)))
  71. {
  72. list.Add(await GetUserView(CollectionType.Movies, user, string.Empty, cancellationToken).ConfigureAwait(false));
  73. }
  74. if (foldersWithViewTypes.Any(i => string.Equals(i.CollectionType, CollectionType.Games, StringComparison.OrdinalIgnoreCase)))
  75. {
  76. list.Add(await GetUserView(CollectionType.Games, user, string.Empty, cancellationToken).ConfigureAwait(false));
  77. }
  78. if (user.Configuration.DisplayCollectionsView &&
  79. folders
  80. .Except(standaloneFolders)
  81. .SelectMany(i => i.GetRecursiveChildren(user, false)).OfType<BoxSet>().Any())
  82. {
  83. list.Add(await GetUserView(CollectionType.BoxSets, user, string.Empty, cancellationToken).ConfigureAwait(false));
  84. }
  85. if (foldersWithViewTypes.Any(i => string.Equals(i.CollectionType, CollectionType.Playlists, StringComparison.OrdinalIgnoreCase)))
  86. {
  87. list.Add(_playlists.GetPlaylistsFolder(user.Id.ToString("N")));
  88. }
  89. if (user.Configuration.DisplayFoldersView)
  90. {
  91. list.Add(await GetUserView(CollectionType.Folders, user, "zz_" + CollectionType.Folders, cancellationToken).ConfigureAwait(false));
  92. }
  93. if (query.IncludeExternalContent)
  94. {
  95. var channelResult = await _channelManager.GetChannels(new ChannelQuery
  96. {
  97. UserId = query.UserId
  98. }, cancellationToken).ConfigureAwait(false);
  99. var channels = channelResult.Items;
  100. var embeddedChannels = channels
  101. .Where(i => user.Configuration.DisplayChannelsWithinViews.Contains(i.Id))
  102. .ToList();
  103. list.AddRange(embeddedChannels.Select(i => _channelManager.GetChannel(i.Id)));
  104. if (channels.Length > embeddedChannels.Count)
  105. {
  106. list.Add(await _channelManager.GetInternalChannelFolder(query.UserId, cancellationToken).ConfigureAwait(false));
  107. }
  108. if (_liveTvManager.GetEnabledUsers().Select(i => i.Id.ToString("N")).Contains(query.UserId))
  109. {
  110. list.Add(await _liveTvManager.GetInternalLiveTvFolder(query.UserId, cancellationToken).ConfigureAwait(false));
  111. }
  112. }
  113. var sorted = _libraryManager.Sort(list, user, new[] { ItemSortBy.SortName }, SortOrder.Ascending).ToList();
  114. var orders = user.Configuration.OrderedViews.ToList();
  115. return list
  116. .OrderBy(i =>
  117. {
  118. var index = orders.IndexOf(i.Id.ToString("N"));
  119. return index == -1 ? int.MaxValue : index;
  120. })
  121. .ThenBy(sorted.IndexOf)
  122. .ThenBy(i => i.SortName);
  123. }
  124. public Task<UserView> GetUserView(string category, string type, User user, string sortName, CancellationToken cancellationToken)
  125. {
  126. var name = _localizationManager.GetLocalizedString("ViewType" + type);
  127. return _libraryManager.GetNamedView(name, category, type, sortName, cancellationToken);
  128. }
  129. public Task<UserView> GetUserView(string type, User user, string sortName, CancellationToken cancellationToken)
  130. {
  131. return GetUserView(null, type, user, sortName, cancellationToken);
  132. }
  133. }
  134. }