2
0

UserView.cs 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  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)).OfType<GameSystem>();
  19. case CollectionType.BoxSets:
  20. return mediaFolders.SelectMany(i => i.GetRecursiveChildren(user, includeLinkedChildren)).OfType<BoxSet>();
  21. case CollectionType.TvShows:
  22. return mediaFolders.SelectMany(i => i.GetRecursiveChildren(user, includeLinkedChildren)).OfType<Series>();
  23. default:
  24. return mediaFolders.SelectMany(i => i.GetChildren(user, includeLinkedChildren));
  25. }
  26. }
  27. protected override IEnumerable<BaseItem> GetEligibleChildrenForRecursiveChildren(User user)
  28. {
  29. return GetChildren(user, false);
  30. }
  31. private IEnumerable<Folder> GetMediaFolders(User user)
  32. {
  33. var excludeFolderIds = user.Configuration.ExcludeFoldersFromGrouping.Select(i => new Guid(i)).ToList();
  34. return user.RootFolder
  35. .GetChildren(user, true)
  36. .OfType<Folder>()
  37. .Where(i => !excludeFolderIds.Contains(i.Id) && !IsExcludedFromGrouping(i));
  38. }
  39. public static bool IsExcludedFromGrouping(Folder folder)
  40. {
  41. var standaloneTypes = new List<string>
  42. {
  43. CollectionType.AdultVideos,
  44. CollectionType.Books,
  45. CollectionType.HomeVideos,
  46. CollectionType.Photos,
  47. CollectionType.Trailers
  48. };
  49. var collectionFolder = folder as CollectionFolder;
  50. if (collectionFolder == null)
  51. {
  52. return false;
  53. }
  54. return standaloneTypes.Contains(collectionFolder.CollectionType ?? string.Empty);
  55. }
  56. }
  57. }