BoxSet.cs 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237
  1. #nullable disable
  2. #pragma warning disable CA1721, CA1819, CS1591
  3. using System;
  4. using System.Collections.Generic;
  5. using System.Linq;
  6. using System.Text.Json.Serialization;
  7. using Jellyfin.Data.Entities;
  8. using Jellyfin.Data.Enums;
  9. using MediaBrowser.Controller.Providers;
  10. using MediaBrowser.Model.Entities;
  11. using MediaBrowser.Model.Querying;
  12. namespace MediaBrowser.Controller.Entities.Movies
  13. {
  14. /// <summary>
  15. /// Class BoxSet.
  16. /// </summary>
  17. public class BoxSet : Folder, IHasTrailers, IHasDisplayOrder, IHasLookupInfo<BoxSetInfo>
  18. {
  19. public BoxSet()
  20. {
  21. RemoteTrailers = Array.Empty<MediaUrl>();
  22. LocalTrailerIds = Array.Empty<Guid>();
  23. RemoteTrailerIds = Array.Empty<Guid>();
  24. DisplayOrder = ItemSortBy.PremiereDate;
  25. }
  26. [JsonIgnore]
  27. protected override bool FilterLinkedChildrenPerUser => true;
  28. [JsonIgnore]
  29. public override bool SupportsInheritedParentImages => false;
  30. [JsonIgnore]
  31. public override bool SupportsPeople => true;
  32. /// <inheritdoc />
  33. public IReadOnlyList<Guid> LocalTrailerIds { get; set; }
  34. /// <inheritdoc />
  35. public IReadOnlyList<Guid> RemoteTrailerIds { get; set; }
  36. /// <summary>
  37. /// Gets or sets the display order.
  38. /// </summary>
  39. /// <value>The display order.</value>
  40. public string DisplayOrder { get; set; }
  41. [JsonIgnore]
  42. private bool IsLegacyBoxSet
  43. {
  44. get
  45. {
  46. if (string.IsNullOrEmpty(Path))
  47. {
  48. return false;
  49. }
  50. if (LinkedChildren.Length > 0)
  51. {
  52. return false;
  53. }
  54. return !FileSystem.ContainsSubPath(ConfigurationManager.ApplicationPaths.DataPath, Path);
  55. }
  56. }
  57. [JsonIgnore]
  58. public override bool IsPreSorted => true;
  59. public Guid[] LibraryFolderIds { get; set; }
  60. protected override bool GetBlockUnratedValue(User user)
  61. {
  62. return user.GetPreferenceValues<UnratedItem>(PreferenceKind.BlockUnratedItems).Contains(UnratedItem.Movie);
  63. }
  64. public override double GetDefaultPrimaryImageAspectRatio()
  65. => 2.0 / 3;
  66. public override UnratedItem GetBlockUnratedType()
  67. {
  68. return UnratedItem.Movie;
  69. }
  70. protected override IEnumerable<BaseItem> GetNonCachedChildren(IDirectoryService directoryService)
  71. {
  72. if (IsLegacyBoxSet)
  73. {
  74. return base.GetNonCachedChildren(directoryService);
  75. }
  76. return Enumerable.Empty<BaseItem>();
  77. }
  78. protected override List<BaseItem> LoadChildren()
  79. {
  80. if (IsLegacyBoxSet)
  81. {
  82. return base.LoadChildren();
  83. }
  84. // Save a trip to the database
  85. return new List<BaseItem>();
  86. }
  87. public override bool IsAuthorizedToDelete(User user, List<Folder> allCollectionFolders)
  88. {
  89. return true;
  90. }
  91. public override bool IsSaveLocalMetadataEnabled()
  92. {
  93. return true;
  94. }
  95. public override List<BaseItem> GetChildren(User user, bool includeLinkedChildren, InternalItemsQuery query)
  96. {
  97. var children = base.GetChildren(user, includeLinkedChildren, query);
  98. if (string.Equals(DisplayOrder, ItemSortBy.SortName, StringComparison.OrdinalIgnoreCase))
  99. {
  100. // Sort by name
  101. return LibraryManager.Sort(children, user, new[] { ItemSortBy.SortName }, SortOrder.Ascending).ToList();
  102. }
  103. if (string.Equals(DisplayOrder, ItemSortBy.PremiereDate, StringComparison.OrdinalIgnoreCase))
  104. {
  105. // Sort by release date
  106. return LibraryManager.Sort(children, user, new[] { ItemSortBy.ProductionYear, ItemSortBy.PremiereDate, ItemSortBy.SortName }, SortOrder.Ascending).ToList();
  107. }
  108. // Default sorting
  109. return LibraryManager.Sort(children, user, new[] { ItemSortBy.ProductionYear, ItemSortBy.PremiereDate, ItemSortBy.SortName }, SortOrder.Ascending).ToList();
  110. }
  111. public override IEnumerable<BaseItem> GetRecursiveChildren(User user, InternalItemsQuery query)
  112. {
  113. var children = base.GetRecursiveChildren(user, query);
  114. if (string.Equals(DisplayOrder, ItemSortBy.PremiereDate, StringComparison.OrdinalIgnoreCase))
  115. {
  116. // Sort by release date
  117. return LibraryManager.Sort(children, user, new[] { ItemSortBy.ProductionYear, ItemSortBy.PremiereDate, ItemSortBy.SortName }, SortOrder.Ascending).ToList();
  118. }
  119. return children;
  120. }
  121. public BoxSetInfo GetLookupInfo()
  122. {
  123. return GetItemLookupInfo<BoxSetInfo>();
  124. }
  125. public override bool IsVisible(User user)
  126. {
  127. if (IsLegacyBoxSet)
  128. {
  129. return base.IsVisible(user);
  130. }
  131. if (base.IsVisible(user))
  132. {
  133. if (LinkedChildren.Length == 0)
  134. {
  135. return true;
  136. }
  137. var userLibraryFolderIds = GetLibraryFolderIds(user);
  138. var libraryFolderIds = LibraryFolderIds ?? GetLibraryFolderIds();
  139. if (libraryFolderIds.Length == 0)
  140. {
  141. return true;
  142. }
  143. return userLibraryFolderIds.Any(i => libraryFolderIds.Contains(i));
  144. }
  145. return false;
  146. }
  147. public override bool IsVisibleStandalone(User user)
  148. {
  149. if (IsLegacyBoxSet)
  150. {
  151. return base.IsVisibleStandalone(user);
  152. }
  153. return IsVisible(user);
  154. }
  155. private Guid[] GetLibraryFolderIds(User user)
  156. {
  157. return LibraryManager.GetUserRootFolder().GetChildren(user, true)
  158. .Select(i => i.Id)
  159. .ToArray();
  160. }
  161. public Guid[] GetLibraryFolderIds()
  162. {
  163. var expandedFolders = new List<Guid>();
  164. return FlattenItems(this, expandedFolders)
  165. .SelectMany(i => LibraryManager.GetCollectionFolders(i))
  166. .Select(i => i.Id)
  167. .Distinct()
  168. .ToArray();
  169. }
  170. private IEnumerable<BaseItem> FlattenItems(IEnumerable<BaseItem> items, List<Guid> expandedFolders)
  171. {
  172. return items
  173. .SelectMany(i => FlattenItems(i, expandedFolders));
  174. }
  175. private IEnumerable<BaseItem> FlattenItems(BaseItem item, List<Guid> expandedFolders)
  176. {
  177. if (item is BoxSet boxset)
  178. {
  179. if (!expandedFolders.Contains(item.Id))
  180. {
  181. expandedFolders.Add(item.Id);
  182. return FlattenItems(boxset.GetLinkedChildren(), expandedFolders);
  183. }
  184. return Array.Empty<BaseItem>();
  185. }
  186. return new[] { item };
  187. }
  188. }
  189. }