using System; using System.Collections.Generic; using System.Globalization; using System.IO; using System.Linq; using System.Text.Json.Serialization; using System.Threading; using System.Threading.Tasks; using MediaBrowser.Common.Progress; using MediaBrowser.Controller.Channels; using MediaBrowser.Controller.Collections; using MediaBrowser.Controller.Configuration; using MediaBrowser.Controller.Dto; using MediaBrowser.Controller.Entities.Audio; using MediaBrowser.Controller.Entities.Movies; using MediaBrowser.Controller.Entities.TV; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Providers; using MediaBrowser.Model.Dto; using MediaBrowser.Model.IO; using MediaBrowser.Model.Querying; using Microsoft.Extensions.Logging; namespace MediaBrowser.Controller.Entities { /// /// Class Folder /// public class Folder : BaseItem { public static IUserManager UserManager { get; set; } public static IUserViewManager UserViewManager { get; set; } /// /// Gets or sets a value indicating whether this instance is root. /// /// true if this instance is root; otherwise, false. public bool IsRoot { get; set; } public LinkedChild[] LinkedChildren { get; set; } [JsonIgnore] public DateTime? DateLastMediaAdded { get; set; } public Folder() { LinkedChildren = Array.Empty(); } [JsonIgnore] public override bool SupportsThemeMedia => true; [JsonIgnore] public virtual bool IsPreSorted => false; [JsonIgnore] public virtual bool IsPhysicalRoot => false; [JsonIgnore] public override bool SupportsInheritedParentImages => true; [JsonIgnore] public override bool SupportsPlayedStatus => true; /// /// Gets a value indicating whether this instance is folder. /// /// true if this instance is folder; otherwise, false. [JsonIgnore] public override bool IsFolder => true; [JsonIgnore] public override bool IsDisplayedAsFolder => true; [JsonIgnore] public virtual bool SupportsCumulativeRunTimeTicks => false; [JsonIgnore] public virtual bool SupportsDateLastMediaAdded => false; public override bool CanDelete() { if (IsRoot) { return false; } return base.CanDelete(); } public override bool RequiresRefresh() { var baseResult = base.RequiresRefresh(); if (SupportsCumulativeRunTimeTicks && !RunTimeTicks.HasValue) { baseResult = true; } return baseResult; } [JsonIgnore] public override string FileNameWithoutExtension { get { if (IsFileProtocol) { return System.IO.Path.GetFileName(Path); } return null; } } protected override bool IsAllowTagFilterEnforced() { if (this is ICollectionFolder) { return false; } if (this is UserView) { return false; } return true; } [JsonIgnore] protected virtual bool SupportsShortcutChildren => false; /// /// Adds the child. /// /// The item. /// The cancellation token. /// Task. /// Unable to add + item.Name public void AddChild(BaseItem item, CancellationToken cancellationToken) { item.SetParent(this); if (item.Id.Equals(Guid.Empty)) { item.Id = LibraryManager.GetNewItemId(item.Path, item.GetType()); } if (item.DateCreated == DateTime.MinValue) { item.DateCreated = DateTime.UtcNow; } if (item.DateModified == DateTime.MinValue) { item.DateModified = DateTime.UtcNow; } LibraryManager.CreateItem(item, this); } /// /// Gets the actual children. /// /// The actual children. [JsonIgnore] public virtual IEnumerable Children => LoadChildren(); /// /// thread-safe access to all recursive children of this folder - without regard to user /// /// The recursive children. [JsonIgnore] public IEnumerable RecursiveChildren => GetRecursiveChildren(); public override bool IsVisible(User user) { if (this is ICollectionFolder && !(this is BasePluginFolder)) { if (user.Policy.BlockedMediaFolders != null) { if (user.Policy.BlockedMediaFolders.Contains(Id.ToString("N", CultureInfo.InvariantCulture), StringComparer.OrdinalIgnoreCase) || // Backwards compatibility user.Policy.BlockedMediaFolders.Contains(Name, StringComparer.OrdinalIgnoreCase)) { return false; } } else { if (!user.Policy.EnableAllFolders && !user.Policy.EnabledFolders.Contains(Id.ToString("N", CultureInfo.InvariantCulture), StringComparer.OrdinalIgnoreCase)) { return false; } } } return base.IsVisible(user); } /// /// Loads our children. Validation will occur externally. /// We want this sychronous. /// protected virtual List LoadChildren() { //logger.LogDebug("Loading children from {0} {1} {2}", GetType().Name, Id, Path); //just load our children from the repo - the library will be validated and maintained in other processes return GetCachedChildren(); } public override double? GetRefreshProgress() { return ProviderManager.GetRefreshProgress(Id); } public Task ValidateChildren(IProgress progress, CancellationToken cancellationToken) { return ValidateChildren(progress, cancellationToken, new MetadataRefreshOptions(new DirectoryService(FileSystem))); } /// /// Validates that the children of the folder still exist /// /// The progress. /// The cancellation token. /// The metadata refresh options. /// if set to true [recursive]. /// Task. public Task ValidateChildren(IProgress progress, CancellationToken cancellationToken, MetadataRefreshOptions metadataRefreshOptions, bool recursive = true) { return ValidateChildrenInternal(progress, cancellationToken, recursive, true, metadataRefreshOptions, metadataRefreshOptions.DirectoryService); } private Dictionary GetActualChildrenDictionary() { var dictionary = new Dictionary(); var childrenList = Children.ToList(); foreach (var child in childrenList) { var id = child.Id; if (dictionary.ContainsKey(id)) { Logger.LogError("Found folder containing items with duplicate id. Path: {path}, Child Name: {ChildName}", Path ?? Name, child.Path ?? child.Name); } else { dictionary[id] = child; } } return dictionary; } protected override void TriggerOnRefreshStart() { } protected override void TriggerOnRefreshComplete() { } /// /// Validates the children internal. /// /// The progress. /// The cancellation token. /// if set to true [recursive]. /// if set to true [refresh child metadata]. /// The refresh options. /// The directory service. /// Task. protected virtual async Task ValidateChildrenInternal(IProgress progress, CancellationToken cancellationToken, bool recursive, bool refreshChildMetadata, MetadataRefreshOptions refreshOptions, IDirectoryService directoryService) { if (recursive) { ProviderManager.OnRefreshStart(this); } try { await ValidateChildrenInternal2(progress, cancellationToken, recursive, refreshChildMetadata, refreshOptions, directoryService).ConfigureAwait(false); } finally { if (recursive) { ProviderManager.OnRefreshComplete(this); } } } private async Task ValidateChildrenInternal2(IProgress progress, CancellationToken cancellationToken, bool recursive, bool refreshChildMetadata, MetadataRefreshOptions refreshOptions, IDirectoryService directoryService) { cancellationToken.ThrowIfCancellationRequested(); var validChildren = new List(); var validChildrenNeedGeneration = false; if (IsFileProtocol) { IEnumerable nonCachedChildren; try { nonCachedChildren = GetNonCachedChildren(directoryService); } catch (Exception ex) { Logger.LogError(ex, "Error retrieving children folder"); return; } progress.Report(5); if (recursive) { ProviderManager.OnRefreshProgress(this, 5); } // Build a dictionary of the current children we have now by Id so we can compare quickly and easily var currentChildren = GetActualChildrenDictionary(); // Create a list for our validated children var newItems = new List(); cancellationToken.ThrowIfCancellationRequested(); foreach (var child in nonCachedChildren) { if (currentChildren.TryGetValue(child.Id, out BaseItem currentChild)) { validChildren.Add(currentChild); if (currentChild.UpdateFromResolvedItem(child) > ItemUpdateType.None) { currentChild.UpdateToRepository(ItemUpdateType.MetadataImport, cancellationToken); } continue; } // Brand new item - needs to be added child.SetParent(this); newItems.Add(child); validChildren.Add(child); } // If any items were added or removed.... if (newItems.Count > 0 || currentChildren.Count != validChildren.Count) { // That's all the new and changed ones - now see if there are any that are missing var itemsRemoved = currentChildren.Values.Except(validChildren).ToList(); foreach (var item in itemsRemoved) { if (item.IsFileProtocol) { Logger.LogDebug("Removed item: " + item.Path); item.SetParent(null); LibraryManager.DeleteItem(item, new DeleteOptions { DeleteFileLocation = false }, this, false); } } LibraryManager.CreateItems(newItems, this, cancellationToken); } } else { validChildrenNeedGeneration = true; } progress.Report(10); if (recursive) { ProviderManager.OnRefreshProgress(this, 10); } cancellationToken.ThrowIfCancellationRequested(); if (recursive) { var innerProgress = new ActionableProgress(); var folder = this; innerProgress.RegisterAction(p => { double newPct = 0.80 * p + 10; progress.Report(newPct); ProviderManager.OnRefreshProgress(folder, newPct); }); if (validChildrenNeedGeneration) { validChildren = Children.ToList(); validChildrenNeedGeneration = false; } await ValidateSubFolders(validChildren.OfType().ToList(), directoryService, innerProgress, cancellationToken).ConfigureAwait(false); } if (refreshChildMetadata) { progress.Report(90); if (recursive) { ProviderManager.OnRefreshProgress(this, 90); } var container = this as IMetadataContainer; var innerProgress = new ActionableProgress(); var folder = this; innerProgress.RegisterAction(p => { double newPct = 0.10 * p + 90; progress.Report(newPct); if (recursive) { ProviderManager.OnRefreshProgress(folder, newPct); } }); if (container != null) { await RefreshAllMetadataForContainer(container, refreshOptions, innerProgress, cancellationToken).ConfigureAwait(false); } else { if (validChildrenNeedGeneration) { validChildren = Children.ToList(); } await RefreshMetadataRecursive(validChildren, refreshOptions, recursive, innerProgress, cancellationToken); } } } private async Task RefreshMetadataRecursive(List children, MetadataRefreshOptions refreshOptions, bool recursive, IProgress progress, CancellationToken cancellationToken) { var numComplete = 0; var count = children.Count; double currentPercent = 0; foreach (var child in children) { cancellationToken.ThrowIfCancellationRequested(); var innerProgress = new ActionableProgress(); // Avoid implicitly captured closure var currentInnerPercent = currentPercent; innerProgress.RegisterAction(p => { double innerPercent = currentInnerPercent; innerPercent += p / (count); progress.Report(innerPercent); }); await RefreshChildMetadata(child, refreshOptions, recursive && child.IsFolder, innerProgress, cancellationToken) .ConfigureAwait(false); numComplete++; double percent = numComplete; percent /= count; percent *= 100; currentPercent = percent; progress.Report(percent); } } private async Task RefreshAllMetadataForContainer(IMetadataContainer container, MetadataRefreshOptions refreshOptions, IProgress progress, CancellationToken cancellationToken) { var series = container as Series; if (series != null) { await series.RefreshMetadata(refreshOptions, cancellationToken).ConfigureAwait(false); } await container.RefreshAllMetadata(refreshOptions, progress, cancellationToken).ConfigureAwait(false); } private async Task RefreshChildMetadata(BaseItem child, MetadataRefreshOptions refreshOptions, bool recursive, IProgress progress, CancellationToken cancellationToken) { var container = child as IMetadataContainer; if (container != null) { await RefreshAllMetadataForContainer(container, refreshOptions, progress, cancellationToken).ConfigureAwait(false); } else { if (refreshOptions.RefreshItem(child)) { await child.RefreshMetadata(refreshOptions, cancellationToken).ConfigureAwait(false); } if (recursive && child is Folder folder) { await folder.RefreshMetadataRecursive(folder.Children.ToList(), refreshOptions, true, progress, cancellationToken); } } } /// /// Refreshes the children. /// /// The children. /// The directory service. /// The progress. /// The cancellation token. /// Task. private async Task ValidateSubFolders(IList children, IDirectoryService directoryService, IProgress progress, CancellationToken cancellationToken) { var numComplete = 0; var count = children.Count; double currentPercent = 0; foreach (var child in children) { cancellationToken.ThrowIfCancellationRequested(); var innerProgress = new ActionableProgress(); // Avoid implicitly captured closure var currentInnerPercent = currentPercent; innerProgress.RegisterAction(p => { double innerPercent = currentInnerPercent; innerPercent += p / (count); progress.Report(innerPercent); }); await child.ValidateChildrenInternal(innerProgress, cancellationToken, true, false, null, directoryService) .ConfigureAwait(false); numComplete++; double percent = numComplete; percent /= count; percent *= 100; currentPercent = percent; progress.Report(percent); } } /// /// Get the children of this folder from the actual file system /// /// IEnumerable{BaseItem}. protected virtual IEnumerable GetNonCachedChildren(IDirectoryService directoryService) { var collectionType = LibraryManager.GetContentType(this); var libraryOptions = LibraryManager.GetLibraryOptions(this); return LibraryManager.ResolvePaths(GetFileSystemChildren(directoryService), directoryService, this, libraryOptions, collectionType); } /// /// Get our children from the repo - stubbed for now /// /// IEnumerable{BaseItem}. protected List GetCachedChildren() { return ItemRepository.GetItemList(new InternalItemsQuery { Parent = this, GroupByPresentationUniqueKey = false, DtoOptions = new DtoOptions(true) }); } public virtual int GetChildCount(User user) { if (LinkedChildren.Length > 0) { if (!(this is ICollectionFolder)) { return GetChildren(user, true).Count; } } var result = GetItems(new InternalItemsQuery(user) { Recursive = false, Limit = 0, Parent = this, DtoOptions = new DtoOptions(false) { EnableImages = false } }); return result.TotalRecordCount; } public virtual int GetRecursiveChildCount(User user) { return GetItems(new InternalItemsQuery(user) { Recursive = true, IsFolder = false, IsVirtualItem = false, EnableTotalRecordCount = true, Limit = 0, DtoOptions = new DtoOptions(false) { EnableImages = false } }).TotalRecordCount; } public QueryResult QueryRecursive(InternalItemsQuery query) { var user = query.User; if (!query.ForceDirect && RequiresPostFiltering(query)) { IEnumerable items; Func filter = i => UserViewBuilder.Filter(i, user, query, UserDataManager, LibraryManager); if (query.User == null) { items = GetRecursiveChildren(filter); } else { items = GetRecursiveChildren(user, query); } return PostFilterAndSort(items, query, true); } if (!(this is UserRootFolder) && !(this is AggregateFolder) && query.ParentId == Guid.Empty) { query.Parent = this; } if (RequiresPostFiltering2(query)) { return QueryWithPostFiltering2(query); } return LibraryManager.GetItemsResult(query); } private QueryResult QueryWithPostFiltering2(InternalItemsQuery query) { var startIndex = query.StartIndex; var limit = query.Limit; query.StartIndex = null; query.Limit = null; IEnumerable itemsList = LibraryManager.GetItemList(query); var user = query.User; if (user != null) { // needed for boxsets itemsList = itemsList.Where(i => i.IsVisibleStandalone(query.User)); } IEnumerable returnItems; int totalCount = 0; if (query.EnableTotalRecordCount) { var itemArray = itemsList.ToArray(); totalCount = itemArray.Length; returnItems = itemArray; } else { returnItems = itemsList; } if (limit.HasValue) { returnItems = returnItems.Skip(startIndex ?? 0).Take(limit.Value); } else if (startIndex.HasValue) { returnItems = returnItems.Skip(startIndex.Value); } return new QueryResult { TotalRecordCount = totalCount, Items = returnItems.ToArray() }; } private bool RequiresPostFiltering2(InternalItemsQuery query) { if (query.IncludeItemTypes.Length == 1 && string.Equals(query.IncludeItemTypes[0], typeof(BoxSet).Name, StringComparison.OrdinalIgnoreCase)) { Logger.LogDebug("Query requires post-filtering due to BoxSet query"); return true; } return false; } private bool RequiresPostFiltering(InternalItemsQuery query) { if (LinkedChildren.Length > 0) { if (!(this is ICollectionFolder)) { Logger.LogDebug("Query requires post-filtering due to LinkedChildren. Type: " + GetType().Name); return true; } } // Filter by Video3DFormat if (query.Is3D.HasValue) { Logger.LogDebug("Query requires post-filtering due to Is3D"); return true; } if (query.HasOfficialRating.HasValue) { Logger.LogDebug("Query requires post-filtering due to HasOfficialRating"); return true; } if (query.IsPlaceHolder.HasValue) { Logger.LogDebug("Query requires post-filtering due to IsPlaceHolder"); return true; } if (query.HasSpecialFeature.HasValue) { Logger.LogDebug("Query requires post-filtering due to HasSpecialFeature"); return true; } if (query.HasSubtitles.HasValue) { Logger.LogDebug("Query requires post-filtering due to HasSubtitles"); return true; } if (query.HasTrailer.HasValue) { Logger.LogDebug("Query requires post-filtering due to HasTrailer"); return true; } // Filter by VideoType if (query.VideoTypes.Length > 0) { Logger.LogDebug("Query requires post-filtering due to VideoTypes"); return true; } if (CollapseBoxSetItems(query, this, query.User, ConfigurationManager)) { Logger.LogDebug("Query requires post-filtering due to CollapseBoxSetItems"); return true; } if (!string.IsNullOrEmpty(query.AdjacentTo)) { Logger.LogDebug("Query requires post-filtering due to AdjacentTo"); return true; } if (query.SeriesStatuses.Length > 0) { Logger.LogDebug("Query requires post-filtering due to SeriesStatuses"); return true; } if (query.AiredDuringSeason.HasValue) { Logger.LogDebug("Query requires post-filtering due to AiredDuringSeason"); return true; } if (query.IsPlayed.HasValue) { if (query.IncludeItemTypes.Length == 1 && query.IncludeItemTypes.Contains(typeof(Series).Name)) { Logger.LogDebug("Query requires post-filtering due to IsPlayed"); return true; } } return false; } public QueryResult GetItems(InternalItemsQuery query) { if (query.ItemIds.Length > 0) { return LibraryManager.GetItemsResult(query); } return GetItemsInternal(query); } public IReadOnlyList GetItemList(InternalItemsQuery query) { query.EnableTotalRecordCount = false; if (query.ItemIds.Length > 0) { return LibraryManager.GetItemList(query); } return GetItemsInternal(query).Items; } protected virtual QueryResult GetItemsInternal(InternalItemsQuery query) { if (SourceType == SourceType.Channel) { try { query.Parent = this; query.ChannelIds = new Guid[] { ChannelId }; // Don't blow up here because it could cause parent screens with other content to fail return ChannelManager.GetChannelItemsInternal(query, new SimpleProgress(), CancellationToken.None).Result; } catch { // Already logged at lower levels return new QueryResult(); } } if (query.Recursive) { return QueryRecursive(query); } var user = query.User; Func filter = i => UserViewBuilder.Filter(i, user, query, UserDataManager, LibraryManager); IEnumerable items; if (query.User == null) { items = Children.Where(filter); } else { items = GetChildren(user, true).Where(filter); } return PostFilterAndSort(items, query, true); } public static ICollectionManager CollectionManager { get; set; } protected QueryResult PostFilterAndSort(IEnumerable items, InternalItemsQuery query, bool enableSorting) { var user = query.User; // Check recursive - don't substitute in plain folder views if (user != null) { items = CollapseBoxSetItemsIfNeeded(items, query, this, user, ConfigurationManager, CollectionManager); } if (!string.IsNullOrEmpty(query.NameStartsWithOrGreater)) { items = items.Where(i => string.Compare(query.NameStartsWithOrGreater, i.SortName, StringComparison.CurrentCultureIgnoreCase) < 1); } if (!string.IsNullOrEmpty(query.NameStartsWith)) { items = items.Where(i => i.SortName.StartsWith(query.NameStartsWith, StringComparison.OrdinalIgnoreCase)); } if (!string.IsNullOrEmpty(query.NameLessThan)) { items = items.Where(i => string.Compare(query.NameLessThan, i.SortName, StringComparison.CurrentCultureIgnoreCase) == 1); } // This must be the last filter if (!string.IsNullOrEmpty(query.AdjacentTo)) { items = UserViewBuilder.FilterForAdjacency(items.ToList(), query.AdjacentTo); } return UserViewBuilder.SortAndPage(items, null, query, LibraryManager, enableSorting); } private static IEnumerable CollapseBoxSetItemsIfNeeded(IEnumerable items, InternalItemsQuery query, BaseItem queryParent, User user, IServerConfigurationManager configurationManager, ICollectionManager collectionManager) { if (items == null) { throw new ArgumentNullException(nameof(items)); } if (CollapseBoxSetItems(query, queryParent, user, configurationManager)) { items = collectionManager.CollapseItemsWithinBoxSets(items, user); } return items; } private static bool CollapseBoxSetItems(InternalItemsQuery query, BaseItem queryParent, User user, IServerConfigurationManager configurationManager) { // Could end up stuck in a loop like this if (queryParent is BoxSet) { return false; } if (queryParent is Series) { return false; } if (queryParent is Season) { return false; } if (queryParent is MusicAlbum) { return false; } if (queryParent is MusicArtist) { return false; } var param = query.CollapseBoxSetItems; if (!param.HasValue) { if (user != null && !configurationManager.Configuration.EnableGroupingIntoCollections) { return false; } if (query.IncludeItemTypes.Length == 0 || query.IncludeItemTypes.Contains("Movie", StringComparer.OrdinalIgnoreCase)) { param = true; } } return param.HasValue && param.Value && AllowBoxSetCollapsing(query); } private static bool AllowBoxSetCollapsing(InternalItemsQuery request) { if (request.IsFavorite.HasValue) { return false; } if (request.IsFavoriteOrLiked.HasValue) { return false; } if (request.IsLiked.HasValue) { return false; } if (request.IsPlayed.HasValue) { return false; } if (request.IsResumable.HasValue) { return false; } if (request.IsFolder.HasValue) { return false; } if (request.Genres.Length > 0) { return false; } if (request.GenreIds.Length > 0) { return false; } if (request.HasImdbId.HasValue) { return false; } if (request.HasOfficialRating.HasValue) { return false; } if (request.HasOverview.HasValue) { return false; } if (request.HasParentalRating.HasValue) { return false; } if (request.HasSpecialFeature.HasValue) { return false; } if (request.HasSubtitles.HasValue) { return false; } if (request.HasThemeSong.HasValue) { return false; } if (request.HasThemeVideo.HasValue) { return false; } if (request.HasTmdbId.HasValue) { return false; } if (request.HasTrailer.HasValue) { return false; } if (request.ImageTypes.Length > 0) { return false; } if (request.Is3D.HasValue) { return false; } if (request.IsHD.HasValue) { return false; } if (request.IsLocked.HasValue) { return false; } if (request.IsPlaceHolder.HasValue) { return false; } if (request.IsPlayed.HasValue) { return false; } if (!string.IsNullOrWhiteSpace(request.Person)) { return false; } if (request.PersonIds.Length > 0) { return false; } if (request.ItemIds.Length > 0) { return false; } if (request.StudioIds.Length > 0) { return false; } if (request.GenreIds.Length > 0) { return false; } if (request.VideoTypes.Length > 0) { return false; } if (request.Years.Length > 0) { return false; } if (request.Tags.Length > 0) { return false; } if (request.OfficialRatings.Length > 0) { return false; } if (request.MinCommunityRating.HasValue) { return false; } if (request.MinCriticRating.HasValue) { return false; } if (request.MinIndexNumber.HasValue) { return false; } return true; } public List GetChildren(User user, bool includeLinkedChildren) { if (user == null) { throw new ArgumentNullException(nameof(user)); } return GetChildren(user, includeLinkedChildren, null); } public virtual List GetChildren(User user, bool includeLinkedChildren, InternalItemsQuery query) { if (user == null) { throw new ArgumentNullException(nameof(user)); } //the true root should return our users root folder children if (IsPhysicalRoot) { return LibraryManager.GetUserRootFolder().GetChildren(user, includeLinkedChildren); } var result = new Dictionary(); AddChildren(user, includeLinkedChildren, result, false, query); return result.Values.ToList(); } protected virtual IEnumerable GetEligibleChildrenForRecursiveChildren(User user) { return Children; } /// /// Adds the children to list. /// /// true if XXXX, false otherwise private void AddChildren(User user, bool includeLinkedChildren, Dictionary result, bool recursive, InternalItemsQuery query) { foreach (var child in GetEligibleChildrenForRecursiveChildren(user)) { bool? isVisibleToUser = null; if (query == null || UserViewBuilder.FilterItem(child, query)) { isVisibleToUser = child.IsVisible(user); if (isVisibleToUser.Value) { result[child.Id] = child; } } if (isVisibleToUser ?? child.IsVisible(user)) { if (recursive && child.IsFolder) { var folder = (Folder)child; folder.AddChildren(user, includeLinkedChildren, result, true, query); } } } if (includeLinkedChildren) { foreach (var child in GetLinkedChildren(user)) { if (query == null || UserViewBuilder.FilterItem(child, query)) { if (child.IsVisible(user)) { result[child.Id] = child; } } } } } /// /// Gets allowed recursive children of an item /// /// The user. /// if set to true [include linked children]. /// IEnumerable{BaseItem}. /// public IEnumerable GetRecursiveChildren(User user, bool includeLinkedChildren = true) { return GetRecursiveChildren(user, null); } public virtual IEnumerable GetRecursiveChildren(User user, InternalItemsQuery query) { if (user == null) { throw new ArgumentNullException(nameof(user)); } var result = new Dictionary(); AddChildren(user, true, result, true, query); return result.Values; } /// /// Gets the recursive children. /// /// IList{BaseItem}. public IList GetRecursiveChildren() { return GetRecursiveChildren(true); } public IList GetRecursiveChildren(bool includeLinkedChildren) { return GetRecursiveChildren(i => true, includeLinkedChildren); } public IList GetRecursiveChildren(Func filter) { return GetRecursiveChildren(filter, true); } public IList GetRecursiveChildren(Func filter, bool includeLinkedChildren) { var result = new Dictionary(); AddChildrenToList(result, includeLinkedChildren, true, filter); return result.Values.ToList(); } /// /// Adds the children to list. /// private void AddChildrenToList(Dictionary result, bool includeLinkedChildren, bool recursive, Func filter) { foreach (var child in Children) { if (filter == null || filter(child)) { result[child.Id] = child; } if (recursive && child.IsFolder) { var folder = (Folder)child; // We can only support includeLinkedChildren for the first folder, or we might end up stuck in a loop of linked items folder.AddChildrenToList(result, false, true, filter); } } if (includeLinkedChildren) { foreach (var child in GetLinkedChildren()) { if (filter == null || filter(child)) { result[child.Id] = child; } } } } /// /// Gets the linked children. /// /// IEnumerable{BaseItem}. public List GetLinkedChildren() { var linkedChildren = LinkedChildren; var list = new List(linkedChildren.Length); foreach (var i in linkedChildren) { var child = GetLinkedChild(i); if (child != null) { list.Add(child); } } return list; } protected virtual bool FilterLinkedChildrenPerUser => false; public bool ContainsLinkedChildByItemId(Guid itemId) { var linkedChildren = LinkedChildren; foreach (var i in linkedChildren) { if (i.ItemId.HasValue && i.ItemId.Value == itemId) { return true; } var child = GetLinkedChild(i); if (child != null && child.Id == itemId) { return true; } } return false; } public List GetLinkedChildren(User user) { if (!FilterLinkedChildrenPerUser || user == null) { return GetLinkedChildren(); } var linkedChildren = LinkedChildren; var list = new List(linkedChildren.Length); if (linkedChildren.Length == 0) { return list; } var allUserRootChildren = LibraryManager.GetUserRootFolder() .GetChildren(user, true) .OfType() .ToList(); var collectionFolderIds = allUserRootChildren .Select(i => i.Id) .ToList(); foreach (var i in linkedChildren) { var child = GetLinkedChild(i); if (child == null) { continue; } var childOwner = child.GetOwner() ?? child; if (childOwner != null && !(child is IItemByName)) { var childProtocol = childOwner.PathProtocol; if (!childProtocol.HasValue || childProtocol.Value != Model.MediaInfo.MediaProtocol.File) { if (!childOwner.IsVisibleStandalone(user)) { continue; } } else { var itemCollectionFolderIds = LibraryManager.GetCollectionFolders(childOwner, allUserRootChildren).Select(f => f.Id); if (!itemCollectionFolderIds.Any(collectionFolderIds.Contains)) { continue; } } } list.Add(child); } return list; } /// /// Gets the linked children. /// /// IEnumerable{BaseItem}. public IEnumerable> GetLinkedChildrenInfos() { return LinkedChildren .Select(i => new Tuple(i, GetLinkedChild(i))) .Where(i => i.Item2 != null); } [JsonIgnore] protected override bool SupportsOwnedItems => base.SupportsOwnedItems || SupportsShortcutChildren; protected override async Task RefreshedOwnedItems(MetadataRefreshOptions options, List fileSystemChildren, CancellationToken cancellationToken) { var changesFound = false; if (IsFileProtocol) { if (RefreshLinkedChildren(fileSystemChildren)) { changesFound = true; } } var baseHasChanges = await base.RefreshedOwnedItems(options, fileSystemChildren, cancellationToken).ConfigureAwait(false); return baseHasChanges || changesFound; } /// /// Refreshes the linked children. /// /// true if XXXX, false otherwise protected virtual bool RefreshLinkedChildren(IEnumerable fileSystemChildren) { if (SupportsShortcutChildren) { var newShortcutLinks = fileSystemChildren .Where(i => !i.IsDirectory && FileSystem.IsShortcut(i.FullName)) .Select(i => { try { Logger.LogDebug("Found shortcut at {0}", i.FullName); var resolvedPath = CollectionFolder.ApplicationHost.ExpandVirtualPath(FileSystem.ResolveShortcut(i.FullName)); if (!string.IsNullOrEmpty(resolvedPath)) { return new LinkedChild { Path = resolvedPath, Type = LinkedChildType.Shortcut }; } Logger.LogError("Error resolving shortcut {0}", i.FullName); return null; } catch (IOException ex) { Logger.LogError(ex, "Error resolving shortcut {0}", i.FullName); return null; } }) .Where(i => i != null) .ToList(); var currentShortcutLinks = LinkedChildren.Where(i => i.Type == LinkedChildType.Shortcut).ToList(); if (!newShortcutLinks.SequenceEqual(currentShortcutLinks, new LinkedChildComparer(FileSystem))) { Logger.LogInformation("Shortcut links have changed for {0}", Path); newShortcutLinks.AddRange(LinkedChildren.Where(i => i.Type == LinkedChildType.Manual)); LinkedChildren = newShortcutLinks.ToArray(); return true; } } foreach (var child in LinkedChildren) { // Reset the cached value child.ItemId = null; } return false; } /// /// Marks the played. /// /// The user. /// The date played. /// if set to true [reset position]. /// Task. public override void MarkPlayed(User user, DateTime? datePlayed, bool resetPosition) { var query = new InternalItemsQuery { User = user, Recursive = true, IsFolder = false, EnableTotalRecordCount = false }; if (!user.Configuration.DisplayMissingEpisodes) { query.IsVirtualItem = false; } var itemsResult = GetItemList(query); // Sweep through recursively and update status foreach (var item in itemsResult) { if (item.IsVirtualItem) { // The querying doesn't support virtual unaired var episode = item as Episode; if (episode != null && episode.IsUnaired) { continue; } } item.MarkPlayed(user, datePlayed, resetPosition); } } /// /// Marks the unplayed. /// /// The user. /// Task. public override void MarkUnplayed(User user) { var itemsResult = GetItemList(new InternalItemsQuery { User = user, Recursive = true, IsFolder = false, EnableTotalRecordCount = false }); // Sweep through recursively and update status foreach (var item in itemsResult) { item.MarkUnplayed(user); } } public override bool IsPlayed(User user) { var itemsResult = GetItemList(new InternalItemsQuery(user) { Recursive = true, IsFolder = false, IsVirtualItem = false, EnableTotalRecordCount = false }); return itemsResult .All(i => i.IsPlayed(user)); } public override bool IsUnplayed(User user) { return !IsPlayed(user); } [JsonIgnore] public virtual bool SupportsUserDataFromChildren { get { // These are just far too slow. if (this is ICollectionFolder) { return false; } if (this is UserView) { return false; } if (this is UserRootFolder) { return false; } if (this is Channel) { return false; } if (SourceType != SourceType.Library) { return false; } var iItemByName = this as IItemByName; if (iItemByName != null) { var hasDualAccess = this as IHasDualAccess; if (hasDualAccess == null || hasDualAccess.IsAccessedByName) { return false; } } return true; } } public override void FillUserDataDtoValues(UserItemDataDto dto, UserItemData userData, BaseItemDto itemDto, User user, DtoOptions fields) { if (!SupportsUserDataFromChildren) { return; } if (itemDto != null) { if (fields.ContainsField(ItemFields.RecursiveItemCount)) { itemDto.RecursiveItemCount = GetRecursiveChildCount(user); } } if (SupportsPlayedStatus) { var unplayedQueryResult = GetItems(new InternalItemsQuery(user) { Recursive = true, IsFolder = false, IsVirtualItem = false, EnableTotalRecordCount = true, Limit = 0, IsPlayed = false, DtoOptions = new DtoOptions(false) { EnableImages = false } }); double unplayedCount = unplayedQueryResult.TotalRecordCount; dto.UnplayedItemCount = unplayedQueryResult.TotalRecordCount; if (itemDto != null && itemDto.RecursiveItemCount.HasValue) { if (itemDto.RecursiveItemCount.Value > 0) { var unplayedPercentage = (unplayedCount / itemDto.RecursiveItemCount.Value) * 100; dto.PlayedPercentage = 100 - unplayedPercentage; dto.Played = dto.PlayedPercentage.Value >= 100; } } else { dto.Played = (dto.UnplayedItemCount ?? 0) == 0; } } } } }