UserView.cs 3.4 KB

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