UserViewManager.cs 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377
  1. using MediaBrowser.Controller.Channels;
  2. using MediaBrowser.Controller.Collections;
  3. using MediaBrowser.Controller.Configuration;
  4. using MediaBrowser.Controller.Entities;
  5. using MediaBrowser.Controller.Library;
  6. using MediaBrowser.Controller.LiveTv;
  7. using MediaBrowser.Controller.Localization;
  8. using MediaBrowser.Controller.Playlists;
  9. using MediaBrowser.Model.Channels;
  10. using MediaBrowser.Model.Entities;
  11. using MediaBrowser.Model.Library;
  12. using MediaBrowser.Model.Querying;
  13. using MoreLinq;
  14. using System;
  15. using System.Collections.Generic;
  16. using System.Linq;
  17. using System.Threading;
  18. using System.Threading.Tasks;
  19. namespace MediaBrowser.Server.Implementations.Library
  20. {
  21. public class UserViewManager : IUserViewManager
  22. {
  23. private readonly ILibraryManager _libraryManager;
  24. private readonly ILocalizationManager _localizationManager;
  25. private readonly IUserManager _userManager;
  26. private readonly IChannelManager _channelManager;
  27. private readonly ILiveTvManager _liveTvManager;
  28. private readonly IPlaylistManager _playlists;
  29. private readonly ICollectionManager _collectionManager;
  30. private readonly IServerConfigurationManager _config;
  31. public UserViewManager(ILibraryManager libraryManager, ILocalizationManager localizationManager, IUserManager userManager, IChannelManager channelManager, ILiveTvManager liveTvManager, IPlaylistManager playlists, ICollectionManager collectionManager, IServerConfigurationManager config)
  32. {
  33. _libraryManager = libraryManager;
  34. _localizationManager = localizationManager;
  35. _userManager = userManager;
  36. _channelManager = channelManager;
  37. _liveTvManager = liveTvManager;
  38. _playlists = playlists;
  39. _collectionManager = collectionManager;
  40. _config = config;
  41. }
  42. public async Task<IEnumerable<Folder>> GetUserViews(UserViewQuery query, CancellationToken cancellationToken)
  43. {
  44. var user = _userManager.GetUserById(query.UserId);
  45. var folders = user.RootFolder
  46. .GetChildren(user, true)
  47. .OfType<Folder>()
  48. .ToList();
  49. var excludeFolderIds = user.Configuration.ExcludeFoldersFromGrouping.Select(i => new Guid(i)).ToList();
  50. var plainFolderIds = user.Configuration.PlainFolderViews.Select(i => new Guid(i)).ToList();
  51. var standaloneFolders = folders
  52. .Where(i => UserView.IsExcludedFromGrouping(i) || excludeFolderIds.Contains(i.Id))
  53. .ToList();
  54. var foldersWithViewTypes = folders
  55. .Except(standaloneFolders)
  56. .OfType<ICollectionFolder>()
  57. .ToList();
  58. var list = new List<Folder>();
  59. if (_config.Configuration.EnableUserSpecificUserViews)
  60. {
  61. foreach (var folder in standaloneFolders)
  62. {
  63. var collectionFolder = folder as ICollectionFolder;
  64. var folderViewType = collectionFolder == null ? null : collectionFolder.CollectionType;
  65. if (plainFolderIds.Contains(folder.Id))
  66. {
  67. list.Add(await GetUserView(folder.Id, folder.Name, folderViewType, false, string.Empty, user, cancellationToken).ConfigureAwait(false));
  68. }
  69. else if (!string.IsNullOrWhiteSpace(folderViewType))
  70. {
  71. list.Add(await GetUserView(folder.Id, folder.Name, folderViewType, true, string.Empty, user, cancellationToken).ConfigureAwait(false));
  72. }
  73. else
  74. {
  75. list.Add(folder);
  76. }
  77. }
  78. }
  79. else
  80. {
  81. list.AddRange(standaloneFolders);
  82. }
  83. var parents = foldersWithViewTypes.Where(i => string.Equals(i.CollectionType, CollectionType.TvShows, StringComparison.OrdinalIgnoreCase) || string.IsNullOrWhiteSpace(i.CollectionType))
  84. .ToList();
  85. if (parents.Count > 0)
  86. {
  87. list.Add(await GetUserView(parents, CollectionType.TvShows, string.Empty, user, cancellationToken).ConfigureAwait(false));
  88. }
  89. parents = foldersWithViewTypes.Where(i => string.Equals(i.CollectionType, CollectionType.Music, StringComparison.OrdinalIgnoreCase) || string.Equals(i.CollectionType, CollectionType.MusicVideos, StringComparison.OrdinalIgnoreCase) || string.IsNullOrWhiteSpace(i.CollectionType))
  90. .ToList();
  91. if (parents.Count > 0)
  92. {
  93. list.Add(await GetUserView(parents, CollectionType.Music, string.Empty, user, cancellationToken).ConfigureAwait(false));
  94. }
  95. parents = foldersWithViewTypes.Where(i => string.Equals(i.CollectionType, CollectionType.Movies, StringComparison.OrdinalIgnoreCase) || string.IsNullOrWhiteSpace(i.CollectionType))
  96. .ToList();
  97. if (parents.Count > 0)
  98. {
  99. list.Add(await GetUserView(parents, CollectionType.Movies, string.Empty, user, cancellationToken).ConfigureAwait(false));
  100. }
  101. parents = foldersWithViewTypes.Where(i => string.Equals(i.CollectionType, CollectionType.Games, StringComparison.OrdinalIgnoreCase))
  102. .ToList();
  103. if (parents.Count > 0)
  104. {
  105. list.Add(await GetUserView(parents, CollectionType.Games, string.Empty, user, cancellationToken).ConfigureAwait(false));
  106. }
  107. parents = foldersWithViewTypes.Where(i => string.Equals(i.CollectionType, CollectionType.BoxSets, StringComparison.OrdinalIgnoreCase))
  108. .ToList();
  109. if (parents.Count > 0)
  110. {
  111. list.Add(await GetUserView(parents, CollectionType.BoxSets, string.Empty, user, cancellationToken).ConfigureAwait(false));
  112. }
  113. parents = foldersWithViewTypes.Where(i => string.Equals(i.CollectionType, CollectionType.Playlists, StringComparison.OrdinalIgnoreCase))
  114. .ToList();
  115. if (parents.Count > 0)
  116. {
  117. list.Add(await GetUserView(parents, CollectionType.Playlists, string.Empty, user, cancellationToken).ConfigureAwait(false));
  118. }
  119. if (user.Configuration.DisplayFoldersView)
  120. {
  121. list.Add(await GetUserView(new List<ICollectionFolder>(), CollectionType.Folders, "zz_" + CollectionType.Folders, user, cancellationToken).ConfigureAwait(false));
  122. }
  123. if (query.IncludeExternalContent)
  124. {
  125. var channelResult = await _channelManager.GetChannels(new ChannelQuery
  126. {
  127. UserId = query.UserId
  128. }, cancellationToken).ConfigureAwait(false);
  129. var channels = channelResult.Items;
  130. var embeddedChannels = channels
  131. .Where(i => user.Configuration.DisplayChannelsWithinViews.Contains(i.Id))
  132. .ToList();
  133. list.AddRange(embeddedChannels.Select(i => _channelManager.GetChannel(i.Id)));
  134. if (channels.Length > embeddedChannels.Count)
  135. {
  136. list.Add(await _channelManager.GetInternalChannelFolder(query.UserId, cancellationToken).ConfigureAwait(false));
  137. }
  138. if (_liveTvManager.GetEnabledUsers().Select(i => i.Id.ToString("N")).Contains(query.UserId))
  139. {
  140. //list.Add(await _liveTvManager.GetInternalLiveTvFolder(query.UserId, cancellationToken).ConfigureAwait(false));
  141. list.Add(await GetUserView(new List<ICollectionFolder>(), CollectionType.LiveTv, string.Empty, user, cancellationToken).ConfigureAwait(false));
  142. }
  143. }
  144. var sorted = _libraryManager.Sort(list, user, new[] { ItemSortBy.SortName }, SortOrder.Ascending).ToList();
  145. var orders = user.Configuration.OrderedViews.ToList();
  146. return list
  147. .OrderBy(i =>
  148. {
  149. var index = orders.IndexOf(i.Id.ToString("N"));
  150. return index == -1 ? int.MaxValue : index;
  151. })
  152. .ThenBy(sorted.IndexOf)
  153. .ThenBy(i => i.SortName);
  154. }
  155. public Task<UserView> GetUserSubView(string name, string parentId, string type, User user, string sortName, CancellationToken cancellationToken)
  156. {
  157. return _libraryManager.GetNamedView(user, name, parentId, type, sortName, cancellationToken);
  158. }
  159. public Task<UserView> GetUserSubView(string parentId, string type, User user, string sortName, CancellationToken cancellationToken)
  160. {
  161. var name = _localizationManager.GetLocalizedString("ViewType" + type);
  162. return GetUserSubView(name, parentId, type, user, sortName, cancellationToken);
  163. }
  164. public async Task<UserView> GetUserView(List<ICollectionFolder> parents, string viewType, string sortName, User user, CancellationToken cancellationToken)
  165. {
  166. if (parents.Count == 1 && parents.All(i => string.Equals(i.CollectionType, viewType, StringComparison.OrdinalIgnoreCase)))
  167. {
  168. var name = parents[0].Name;
  169. var parentId = parents[0].Id;
  170. var enableRichView = !user.Configuration.PlainFolderViews.Contains(parentId.ToString("N"), StringComparer.OrdinalIgnoreCase);
  171. if (_config.Configuration.EnableUserSpecificUserViews)
  172. {
  173. viewType = enableRichView ? viewType : null;
  174. var view = await _libraryManager.GetNamedView(user, name, viewType, sortName, cancellationToken).ConfigureAwait(false);
  175. if (view.ParentId != parentId)
  176. {
  177. view.ParentId = parentId;
  178. await view.UpdateToRepository(ItemUpdateType.MetadataEdit, cancellationToken).ConfigureAwait(false);
  179. }
  180. return view;
  181. }
  182. viewType = enableRichView ? viewType : CollectionType.Folders;
  183. return await _libraryManager.GetNamedView(user, name, viewType, sortName, cancellationToken).ConfigureAwait(false);
  184. }
  185. else
  186. {
  187. var name = _localizationManager.GetLocalizedString("ViewType" + viewType);
  188. return await _libraryManager.GetNamedView(user, name, viewType, sortName, cancellationToken).ConfigureAwait(false);
  189. }
  190. }
  191. public Task<UserView> GetUserView(Guid parentId, string name, string viewType, bool enableRichView, string sortName, User user, CancellationToken cancellationToken)
  192. {
  193. return _libraryManager.GetNamedView(user, name, parentId.ToString("N"), viewType, sortName, cancellationToken);
  194. }
  195. public List<Tuple<BaseItem, List<BaseItem>>> GetLatestItems(LatestItemsQuery request)
  196. {
  197. var user = _userManager.GetUserById(request.UserId);
  198. var includeTypes = request.IncludeItemTypes;
  199. var currentUser = user;
  200. Func<BaseItem, bool> filter = i =>
  201. {
  202. if (includeTypes.Length > 0)
  203. {
  204. if (!includeTypes.Contains(i.GetType().Name, StringComparer.OrdinalIgnoreCase))
  205. {
  206. return false;
  207. }
  208. }
  209. if (request.IsPlayed.HasValue)
  210. {
  211. var val = request.IsPlayed.Value;
  212. if (i is Video && i.IsPlayed(currentUser) != val)
  213. {
  214. return false;
  215. }
  216. }
  217. return i.LocationType != LocationType.Virtual && !i.IsFolder;
  218. };
  219. // Avoid implicitly captured closure
  220. var libraryItems = string.IsNullOrEmpty(request.ParentId) && user != null ?
  221. GetItemsConfiguredForLatest(user, filter) :
  222. GetAllLibraryItems(request.UserId, _userManager, _libraryManager, request.ParentId, filter);
  223. libraryItems = libraryItems.OrderByDescending(i => i.DateCreated);
  224. if (request.IsPlayed.HasValue)
  225. {
  226. var takeLimit = (request.Limit ?? 20) * 20;
  227. libraryItems = libraryItems.Take(takeLimit);
  228. }
  229. // Avoid implicitly captured closure
  230. var items = libraryItems
  231. .ToList();
  232. var list = new List<Tuple<BaseItem, List<BaseItem>>>();
  233. foreach (var item in items)
  234. {
  235. // Only grab the index container for media
  236. var container = item.IsFolder || !request.GroupItems ? null : item.LatestItemsIndexContainer;
  237. if (container == null)
  238. {
  239. list.Add(new Tuple<BaseItem, List<BaseItem>>(null, new List<BaseItem> { item }));
  240. }
  241. else
  242. {
  243. var current = list.FirstOrDefault(i => i.Item1 != null && i.Item1.Id == container.Id);
  244. if (current != null)
  245. {
  246. current.Item2.Add(item);
  247. }
  248. else
  249. {
  250. list.Add(new Tuple<BaseItem, List<BaseItem>>(container, new List<BaseItem> { item }));
  251. }
  252. }
  253. if (list.Count >= request.Limit)
  254. {
  255. break;
  256. }
  257. }
  258. return list;
  259. }
  260. protected IList<BaseItem> GetAllLibraryItems(string userId, IUserManager userManager, ILibraryManager libraryManager, string parentId, Func<BaseItem, bool> filter)
  261. {
  262. if (!string.IsNullOrEmpty(parentId))
  263. {
  264. var folder = (Folder)libraryManager.GetItemById(new Guid(parentId));
  265. if (!string.IsNullOrWhiteSpace(userId))
  266. {
  267. var user = userManager.GetUserById(userId);
  268. if (user == null)
  269. {
  270. throw new ArgumentException("User not found");
  271. }
  272. return folder
  273. .GetRecursiveChildren(user, filter)
  274. .ToList();
  275. }
  276. return folder
  277. .GetRecursiveChildren(filter);
  278. }
  279. if (!string.IsNullOrWhiteSpace(userId))
  280. {
  281. var user = userManager.GetUserById(userId);
  282. if (user == null)
  283. {
  284. throw new ArgumentException("User not found");
  285. }
  286. return user
  287. .RootFolder
  288. .GetRecursiveChildren(user, filter)
  289. .ToList();
  290. }
  291. return libraryManager
  292. .RootFolder
  293. .GetRecursiveChildren(filter);
  294. }
  295. private IEnumerable<BaseItem> GetItemsConfiguredForLatest(User user, Func<BaseItem, bool> filter)
  296. {
  297. // Avoid implicitly captured closure
  298. var currentUser = user;
  299. return user.RootFolder.GetChildren(user, true)
  300. .OfType<Folder>()
  301. .Where(i => !user.Configuration.LatestItemsExcludes.Contains(i.Id.ToString("N")))
  302. .SelectMany(i => i.GetRecursiveChildren(currentUser, filter))
  303. .DistinctBy(i => i.Id);
  304. }
  305. }
  306. }