UserView.cs 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Text.Json.Serialization;
  5. using System.Threading.Tasks;
  6. using Jellyfin.Data.Entities;
  7. using MediaBrowser.Controller.TV;
  8. using MediaBrowser.Model.Querying;
  9. using Microsoft.Extensions.Logging;
  10. namespace MediaBrowser.Controller.Entities
  11. {
  12. public class UserView : Folder, IHasCollectionType
  13. {
  14. /// <inheritdoc />
  15. public string ViewType { get; set; }
  16. /// <inheritdoc />
  17. public new Guid DisplayParentId { get; set; }
  18. /// <inheritdoc />
  19. public Guid? UserId { get; set; }
  20. public static ITVSeriesManager TVSeriesManager;
  21. /// <inheritdoc />
  22. [JsonIgnore]
  23. public string CollectionType => ViewType;
  24. /// <inheritdoc />
  25. public override IEnumerable<Guid> GetIdsForAncestorQuery()
  26. {
  27. if (!DisplayParentId.Equals(Guid.Empty))
  28. {
  29. yield return DisplayParentId;
  30. }
  31. else if (!ParentId.Equals(Guid.Empty))
  32. {
  33. yield return ParentId;
  34. }
  35. else
  36. {
  37. yield return Id;
  38. }
  39. }
  40. [JsonIgnore]
  41. public override bool SupportsInheritedParentImages => false;
  42. [JsonIgnore]
  43. public override bool SupportsPlayedStatus => false;
  44. public override int GetChildCount(User user)
  45. {
  46. return GetChildren(user, true).Count;
  47. }
  48. protected override QueryResult<BaseItem> GetItemsInternal(InternalItemsQuery query)
  49. {
  50. var parent = this as Folder;
  51. if (!DisplayParentId.Equals(Guid.Empty))
  52. {
  53. parent = LibraryManager.GetItemById(DisplayParentId) as Folder ?? parent;
  54. }
  55. else if (!ParentId.Equals(Guid.Empty))
  56. {
  57. parent = LibraryManager.GetItemById(ParentId) as Folder ?? parent;
  58. }
  59. return new UserViewBuilder(UserViewManager, LibraryManager, LoggerFactory.CreateLogger<UserViewBuilder>(), UserDataManager, TVSeriesManager, ConfigurationManager)
  60. .GetUserItems(parent, this, CollectionType, query);
  61. }
  62. public override List<BaseItem> GetChildren(User user, bool includeLinkedChildren, InternalItemsQuery query)
  63. {
  64. if (query == null)
  65. {
  66. query = new InternalItemsQuery(user);
  67. }
  68. query.EnableTotalRecordCount = false;
  69. var result = GetItemList(query);
  70. return result.ToList();
  71. }
  72. public override bool CanDelete()
  73. {
  74. return false;
  75. }
  76. public override bool IsSaveLocalMetadataEnabled()
  77. {
  78. return true;
  79. }
  80. public override IEnumerable<BaseItem> GetRecursiveChildren(User user, InternalItemsQuery query)
  81. {
  82. query.SetUser(user);
  83. query.Recursive = true;
  84. query.EnableTotalRecordCount = false;
  85. query.ForceDirect = true;
  86. return GetItemList(query);
  87. }
  88. protected override IEnumerable<BaseItem> GetEligibleChildrenForRecursiveChildren(User user)
  89. {
  90. return GetChildren(user, false);
  91. }
  92. private static string[] UserSpecificViewTypes = new string[]
  93. {
  94. Model.Entities.CollectionType.Playlists
  95. };
  96. public static bool IsUserSpecific(Folder folder)
  97. {
  98. var collectionFolder = folder as ICollectionFolder;
  99. if (collectionFolder == null)
  100. {
  101. return false;
  102. }
  103. var supportsUserSpecific = folder as ISupportsUserSpecificView;
  104. if (supportsUserSpecific != null && supportsUserSpecific.EnableUserSpecificView)
  105. {
  106. return true;
  107. }
  108. return UserSpecificViewTypes.Contains(collectionFolder.CollectionType ?? string.Empty, StringComparer.OrdinalIgnoreCase);
  109. }
  110. public static bool IsEligibleForGrouping(Folder folder)
  111. {
  112. return folder is ICollectionFolder collectionFolder
  113. && IsEligibleForGrouping(collectionFolder.CollectionType);
  114. }
  115. private static string[] ViewTypesEligibleForGrouping = new string[]
  116. {
  117. Model.Entities.CollectionType.Movies,
  118. Model.Entities.CollectionType.TvShows,
  119. string.Empty
  120. };
  121. public static bool IsEligibleForGrouping(string viewType)
  122. {
  123. return ViewTypesEligibleForGrouping.Contains(viewType ?? string.Empty, StringComparer.OrdinalIgnoreCase);
  124. }
  125. private static string[] OriginalFolderViewTypes = new string[]
  126. {
  127. Model.Entities.CollectionType.Books,
  128. Model.Entities.CollectionType.MusicVideos,
  129. Model.Entities.CollectionType.HomeVideos,
  130. Model.Entities.CollectionType.Photos,
  131. Model.Entities.CollectionType.Music,
  132. Model.Entities.CollectionType.BoxSets
  133. };
  134. public static bool EnableOriginalFolder(string viewType)
  135. {
  136. return OriginalFolderViewTypes.Contains(viewType ?? string.Empty, StringComparer.OrdinalIgnoreCase);
  137. }
  138. protected override Task ValidateChildrenInternal(IProgress<double> progress, System.Threading.CancellationToken cancellationToken, bool recursive, bool refreshChildMetadata, Providers.MetadataRefreshOptions refreshOptions, Providers.IDirectoryService directoryService)
  139. {
  140. return Task.CompletedTask;
  141. }
  142. [JsonIgnore]
  143. public override bool SupportsPeople => false;
  144. }
  145. }