using System; using MediaBrowser.Controller.Providers; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Querying; using MediaBrowser.Model.Users; using MoreLinq; using System.Collections.Generic; using System.Linq; using System.Runtime.Serialization; using System.Threading.Tasks; using MediaBrowser.Model.Configuration; namespace MediaBrowser.Controller.Entities.TV { /// /// Class Season /// public class Season : Folder, IHasSeries, IHasLookupInfo { [IgnoreDataMember] public override bool SupportsAddingToPlaylist { get { return true; } } [IgnoreDataMember] public override bool IsPreSorted { get { return true; } } [IgnoreDataMember] public override bool SupportsDateLastMediaAdded { get { return true; } } [IgnoreDataMember] public override Guid? DisplayParentId { get { var series = Series; return series == null ? ParentId : series.Id; } } // Genre, Rating and Stuido will all be the same protected override IEnumerable GetIndexByOptions() { return new List { {"None"}, {"Performer"}, {"Director"}, {"Year"}, }; } public override List GetUserDataKeys() { var list = base.GetUserDataKeys(); var series = Series; if (series != null) { list.InsertRange(0, series.GetUserDataKeys().Select(i => i + (IndexNumber ?? 0).ToString("000"))); } return list; } /// /// This Episode's Series Instance /// /// The series. [IgnoreDataMember] public Series Series { get { return FindParent(); } } [IgnoreDataMember] public string SeriesPath { get { var series = Series; if (series != null) { return series.Path; } return System.IO.Path.GetDirectoryName(Path); } } [IgnoreDataMember] public override string PresentationUniqueKey { get { if (IndexNumber.HasValue) { var series = Series; if (series != null) { return series.PresentationUniqueKey + "-" + (IndexNumber ?? 0).ToString("000"); } } return base.PresentationUniqueKey; } } /// /// Creates the name of the sort. /// /// System.String. protected override string CreateSortName() { return IndexNumber != null ? IndexNumber.Value.ToString("0000") : Name; } public override bool RequiresRefresh() { var result = base.RequiresRefresh(); if (!result) { if (!IsMissingSeason.HasValue) { return true; } } return result; } [IgnoreDataMember] public bool? IsMissingSeason { get; set; } [IgnoreDataMember] public bool IsVirtualUnaired { get { return LocationType == LocationType.Virtual && IsUnaired; } } [IgnoreDataMember] public bool IsMissingOrVirtualUnaired { get { return (IsMissingSeason ?? false) || (LocationType == LocationType.Virtual && IsUnaired); } } [IgnoreDataMember] public bool IsSpecialSeason { get { return (IndexNumber ?? -1) == 0; } } protected override Task> GetItemsInternal(InternalItemsQuery query) { if (query.User == null) { return base.GetItemsInternal(query); } var user = query.User; Func filter = i => UserViewBuilder.Filter(i, user, query, UserDataManager, LibraryManager); var items = GetEpisodes(user).Where(filter); var result = PostFilterAndSort(items, query); return Task.FromResult(result); } /// /// Gets the episodes. /// /// The user. /// IEnumerable{Episode}. public IEnumerable GetEpisodes(User user) { var config = user.Configuration; return GetEpisodes(user, config.DisplayMissingEpisodes, config.DisplayUnairedEpisodes); } public IEnumerable GetEpisodes(User user, bool includeMissingEpisodes, bool includeVirtualUnairedEpisodes) { var series = Series; if (IndexNumber.HasValue && series != null) { return series.GetEpisodes(user, IndexNumber.Value, includeMissingEpisodes, includeVirtualUnairedEpisodes); } var episodes = GetRecursiveChildren(user) .OfType(); if (series != null && series.ContainsEpisodesWithoutSeasonFolders) { var seasonNumber = IndexNumber; var list = episodes.ToList(); if (seasonNumber.HasValue) { list.AddRange(series.GetRecursiveChildren(user).OfType() .Where(i => i.ParentIndexNumber.HasValue && i.ParentIndexNumber.Value == seasonNumber.Value)); } else { list.AddRange(series.GetRecursiveChildren(user).OfType() .Where(i => !i.ParentIndexNumber.HasValue)); } episodes = list.DistinctBy(i => i.Id); } if (!includeMissingEpisodes) { episodes = episodes.Where(i => !i.IsMissingEpisode); } if (!includeVirtualUnairedEpisodes) { episodes = episodes.Where(i => !i.IsVirtualUnaired); } return LibraryManager .Sort(episodes, user, new[] { ItemSortBy.SortName }, SortOrder.Ascending) .Cast(); } public IEnumerable GetEpisodes() { var episodes = GetRecursiveChildren().OfType(); var series = Series; if (series != null && series.ContainsEpisodesWithoutSeasonFolders) { var seasonNumber = IndexNumber; var list = episodes.ToList(); if (seasonNumber.HasValue) { list.AddRange(series.GetRecursiveChildren().OfType() .Where(i => i.ParentIndexNumber.HasValue && i.ParentIndexNumber.Value == seasonNumber.Value)); } else { list.AddRange(series.GetRecursiveChildren().OfType() .Where(i => !i.ParentIndexNumber.HasValue)); } episodes = list.DistinctBy(i => i.Id); } return episodes; } public override IEnumerable GetChildren(User user, bool includeLinkedChildren) { return GetEpisodes(user); } protected override bool GetBlockUnratedValue(UserPolicy config) { // Don't block. Let either the entire series rating or episode rating determine it return false; } public override UnratedItem GetBlockUnratedType() { return UnratedItem.Series; } [IgnoreDataMember] public string SeriesName { get { var series = Series; return series == null ? null : series.Name; } } /// /// Gets the lookup information. /// /// SeasonInfo. public SeasonInfo GetLookupInfo() { var id = GetItemLookupInfo(); var series = Series; if (series != null) { id.SeriesProviderIds = series.ProviderIds; id.AnimeSeriesIndex = series.AnimeSeriesIndex; } return id; } /// /// This is called before any metadata refresh and returns true or false indicating if changes were made /// /// true if XXXX, false otherwise. public override bool BeforeMetadataRefresh() { var hasChanges = base.BeforeMetadataRefresh(); var locationType = LocationType; if (locationType == LocationType.FileSystem || locationType == LocationType.Offline) { if (!IndexNumber.HasValue && !string.IsNullOrEmpty(Path)) { IndexNumber = IndexNumber ?? LibraryManager.GetSeasonNumberFromPath(Path); // If a change was made record it if (IndexNumber.HasValue) { hasChanges = true; } } } return hasChanges; } } }