UserView.cs 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  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. }