UserRootFolder.cs 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. using System.Runtime.Serialization;
  2. using MediaBrowser.Controller.Providers;
  3. using MediaBrowser.Model.Dto;
  4. using MediaBrowser.Model.Library;
  5. using MediaBrowser.Model.Querying;
  6. using System;
  7. using System.Collections.Generic;
  8. using System.Linq;
  9. using System.Threading;
  10. using System.Threading.Tasks;
  11. namespace MediaBrowser.Controller.Entities
  12. {
  13. /// <summary>
  14. /// Special class used for User Roots. Children contain actual ones defined for this user
  15. /// PLUS the virtual folders from the physical root (added by plug-ins).
  16. /// </summary>
  17. public class UserRootFolder : Folder
  18. {
  19. public override async Task<QueryResult<BaseItem>> GetItems(InternalItemsQuery query)
  20. {
  21. if (query.Recursive)
  22. {
  23. return QueryRecursive(query);
  24. }
  25. var result = await UserViewManager.GetUserViews(new UserViewQuery
  26. {
  27. UserId = query.User.Id.ToString("N"),
  28. PresetViews = query.PresetViews
  29. }, CancellationToken.None).ConfigureAwait(false);
  30. var user = query.User;
  31. Func<BaseItem, bool> filter = i => UserViewBuilder.Filter(i, user, query, UserDataManager, LibraryManager);
  32. return PostFilterAndSort(result.Where(filter), query);
  33. }
  34. [IgnoreDataMember]
  35. protected override bool SupportsShortcutChildren
  36. {
  37. get
  38. {
  39. return true;
  40. }
  41. }
  42. [IgnoreDataMember]
  43. public override bool IsPreSorted
  44. {
  45. get
  46. {
  47. return true;
  48. }
  49. }
  50. protected override IEnumerable<BaseItem> GetEligibleChildrenForRecursiveChildren(User user)
  51. {
  52. var list = base.GetEligibleChildrenForRecursiveChildren(user).ToList();
  53. list.AddRange(LibraryManager.RootFolder.VirtualChildren);
  54. return list;
  55. }
  56. /// <summary>
  57. /// Get the children of this folder from the actual file system
  58. /// </summary>
  59. /// <returns>IEnumerable{BaseItem}.</returns>
  60. protected override IEnumerable<BaseItem> GetNonCachedChildren(IDirectoryService directoryService)
  61. {
  62. return base.GetNonCachedChildren(directoryService);
  63. }
  64. public override bool BeforeMetadataRefresh()
  65. {
  66. var hasChanges = base.BeforeMetadataRefresh();
  67. if (string.Equals("default", Name, StringComparison.OrdinalIgnoreCase))
  68. {
  69. Name = "Media Folders";
  70. hasChanges = true;
  71. }
  72. return hasChanges;
  73. }
  74. protected override async Task ValidateChildrenInternal(IProgress<double> progress, CancellationToken cancellationToken, bool recursive, bool refreshChildMetadata, MetadataRefreshOptions refreshOptions, IDirectoryService directoryService)
  75. {
  76. await base.ValidateChildrenInternal(progress, cancellationToken, recursive, refreshChildMetadata, refreshOptions, directoryService)
  77. .ConfigureAwait(false);
  78. // Not the best way to handle this, but it solves an issue
  79. // CollectionFolders aren't always getting saved after changes
  80. // This means that grabbing the item by Id may end up returning the old one
  81. // Fix is in two places - make sure the folder gets saved
  82. // And here to remedy it for affected users.
  83. // In theory this can be removed eventually.
  84. foreach (var item in Children)
  85. {
  86. LibraryManager.RegisterItem(item);
  87. }
  88. }
  89. public override void FillUserDataDtoValues(UserItemDataDto dto, UserItemData userData, User user)
  90. {
  91. // Nothing meaninful here and will only waste resources
  92. }
  93. }
  94. }