UserView.cs 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  1. using MediaBrowser.Controller.Entities.Movies;
  2. using MediaBrowser.Controller.Entities.TV;
  3. using MediaBrowser.Model.Entities;
  4. using MoreLinq;
  5. using System;
  6. using System.Collections.Generic;
  7. using System.Linq;
  8. namespace MediaBrowser.Controller.Entities
  9. {
  10. public class UserView : Folder
  11. {
  12. public string ViewType { get; set; }
  13. public override IEnumerable<BaseItem> GetChildren(User user, bool includeLinkedChildren)
  14. {
  15. var mediaFolders = GetMediaFolders(user);
  16. switch (ViewType)
  17. {
  18. case CollectionType.Games:
  19. return mediaFolders.SelectMany(i => i.GetRecursiveChildren(user, includeLinkedChildren))
  20. .OfType<GameSystem>();
  21. case CollectionType.BoxSets:
  22. return mediaFolders.SelectMany(i => i.GetRecursiveChildren(user, includeLinkedChildren))
  23. .OfType<BoxSet>();
  24. case CollectionType.TvShows:
  25. return mediaFolders.SelectMany(i => i.GetRecursiveChildren(user, includeLinkedChildren))
  26. .OfType<Series>();
  27. case CollectionType.Trailers:
  28. return mediaFolders.SelectMany(i => i.GetRecursiveChildren(user, includeLinkedChildren))
  29. .OfType<Trailer>();
  30. case CollectionType.Movies:
  31. return mediaFolders.SelectMany(i => i.GetRecursiveChildren(user, includeLinkedChildren))
  32. .Where(i => i is Movie || i is BoxSet)
  33. .DistinctBy(i => i.Id);
  34. default:
  35. return mediaFolders.SelectMany(i => i.GetChildren(user, includeLinkedChildren));
  36. }
  37. }
  38. protected override IEnumerable<BaseItem> GetEligibleChildrenForRecursiveChildren(User user)
  39. {
  40. return GetChildren(user, false);
  41. }
  42. private IEnumerable<Folder> GetMediaFolders(User user)
  43. {
  44. var excludeFolderIds = user.Configuration.ExcludeFoldersFromGrouping.Select(i => new Guid(i)).ToList();
  45. return user.RootFolder
  46. .GetChildren(user, true, true)
  47. .OfType<Folder>()
  48. .Where(i => !excludeFolderIds.Contains(i.Id) && !IsExcludedFromGrouping(i));
  49. }
  50. public static bool IsExcludedFromGrouping(Folder folder)
  51. {
  52. var standaloneTypes = new List<string>
  53. {
  54. CollectionType.AdultVideos,
  55. CollectionType.Books,
  56. CollectionType.HomeVideos,
  57. CollectionType.Photos,
  58. CollectionType.Trailers
  59. };
  60. var collectionFolder = folder as CollectionFolder;
  61. if (collectionFolder == null)
  62. {
  63. return false;
  64. }
  65. return standaloneTypes.Contains(collectionFolder.CollectionType ?? string.Empty);
  66. }
  67. }
  68. public class SpecialFolder : Folder
  69. {
  70. public SpecialFolderType SpecialFolderType { get; set; }
  71. public string ItemTypeName { get; set; }
  72. public string ParentId { get; set; }
  73. public override IEnumerable<BaseItem> GetChildren(User user, bool includeLinkedChildren)
  74. {
  75. var parent = (Folder)LibraryManager.GetItemById(new Guid(ParentId));
  76. if (SpecialFolderType == SpecialFolderType.ItemsByType)
  77. {
  78. var items = parent.GetRecursiveChildren(user, includeLinkedChildren);
  79. return items.Where(i => string.Equals(i.GetType().Name, ItemTypeName, StringComparison.OrdinalIgnoreCase));
  80. }
  81. return new List<BaseItem>();
  82. }
  83. }
  84. public enum SpecialFolderType
  85. {
  86. ItemsByType = 1
  87. }
  88. }