| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569 | using System.IO;using MediaBrowser.Controller.Channels;using MediaBrowser.Controller.Collections;using MediaBrowser.Controller.Entities.Audio;using MediaBrowser.Controller.Entities.Movies;using MediaBrowser.Controller.Entities.TV;using MediaBrowser.Controller.Library;using MediaBrowser.Controller.LiveTv;using MediaBrowser.Controller.Providers;using MediaBrowser.Controller.TV;using MediaBrowser.Model.Channels;using MediaBrowser.Model.Entities;using MediaBrowser.Model.LiveTv;using MediaBrowser.Model.Logging;using MediaBrowser.Model.Querying;using System;using System.Collections.Generic;using System.Linq;using System.Threading;using System.Threading.Tasks;namespace MediaBrowser.Controller.Entities{    public class UserViewBuilder    {        private readonly IChannelManager _channelManager;        private readonly ILiveTvManager _liveTvManager;        private readonly IUserViewManager _userViewManager;        private readonly ILibraryManager _libraryManager;        private readonly ILogger _logger;        private readonly IUserDataManager _userDataManager;        private readonly ITVSeriesManager _tvSeriesManager;        private readonly ICollectionManager _collectionManager;        public UserViewBuilder(IUserViewManager userViewManager, ILiveTvManager liveTvManager, IChannelManager channelManager, ILibraryManager libraryManager, ILogger logger, IUserDataManager userDataManager, ITVSeriesManager tvSeriesManager, ICollectionManager collectionManager)        {            _userViewManager = userViewManager;            _liveTvManager = liveTvManager;            _channelManager = channelManager;            _libraryManager = libraryManager;            _logger = logger;            _userDataManager = userDataManager;            _tvSeriesManager = tvSeriesManager;            _collectionManager = collectionManager;        }        public async Task<QueryResult<BaseItem>> GetUserItems(Folder parent, string viewType, InternalItemsQuery query)        {            var user = query.User;            switch (viewType)            {                case CollectionType.Channels:                    {                        var result = await _channelManager.GetChannelsInternal(new ChannelQuery                        {                            UserId = user.Id.ToString("N"),                            Limit = query.Limit,                            StartIndex = query.StartIndex                        }, CancellationToken.None).ConfigureAwait(false);                        return GetResult(result);                    }                case CollectionType.LiveTvChannels:                    {                        var result = await _liveTvManager.GetInternalChannels(new LiveTvChannelQuery                        {                            UserId = query.User.Id.ToString("N"),                            Limit = query.Limit,                            StartIndex = query.StartIndex                        }, CancellationToken.None).ConfigureAwait(false);                        return GetResult(result);                    }                case CollectionType.LiveTvNowPlaying:                    {                        var result = await _liveTvManager.GetRecommendedProgramsInternal(new RecommendedProgramQuery                        {                            UserId = query.User.Id.ToString("N"),                            Limit = query.Limit,                            IsAiring = true                        }, CancellationToken.None).ConfigureAwait(false);                        return GetResult(result);                    }                case CollectionType.LiveTvRecordingGroups:                    {                        var result = await _liveTvManager.GetInternalRecordings(new RecordingQuery                        {                            UserId = query.User.Id.ToString("N"),                            Status = RecordingStatus.Completed,                            Limit = query.Limit,                            StartIndex = query.StartIndex                        }, CancellationToken.None).ConfigureAwait(false);                        return GetResult(result);                    }                case CollectionType.LiveTv:                    {                        var result = await GetLiveTvFolders(user).ConfigureAwait(false);                        return GetResult(result, parent, query);                    }                case CollectionType.Folders:                    return GetResult(user.RootFolder.GetChildren(user, true), parent, query);                case CollectionType.Games:                    return await GetGameView(user, parent, query).ConfigureAwait(false);                case CollectionType.BoxSets:                    return GetResult(GetMediaFolders(user).SelectMany(i => i.GetRecursiveChildren(user)).OfType<BoxSet>(), parent, query);                case CollectionType.TvShows:                    return await GetTvView(parent, user, query).ConfigureAwait(false);                case CollectionType.Music:                    return await GetMusicFolders(parent, user, query).ConfigureAwait(false);                case CollectionType.Movies:                    return await GetMovieFolders(parent, user, query).ConfigureAwait(false);                case CollectionType.GameGenres:                    return GetGameGenres(parent, user, query);                case CollectionType.GameSystems:                    return GetGameSystems(parent, user, query);                case CollectionType.LatestGames:                    return GetLatestGames(parent, user, query);                case CollectionType.RecentlyPlayedGames:                    return GetRecentlyPlayedGames(parent, user, query);                case CollectionType.GameFavorites:                    return GetFavoriteGames(parent, user, query);                case CollectionType.TvShowSeries:                    return GetTvSeries(parent, user, query);                case CollectionType.TvGenres:                    return GetTvGenres(parent, user, query);                case CollectionType.TvResume:                    return GetTvResume(parent, user, query);                case CollectionType.TvNextUp:                    return GetTvNextUp(parent, query);                case CollectionType.TvLatest:                    return GetTvLatest(parent, user, query);                case CollectionType.MovieFavorites:                    return GetFavoriteMovies(parent, user, query);                case CollectionType.MovieLatest:                    return GetMovieLatest(parent, user, query);                case CollectionType.MovieGenres:                    return GetMovieGenres(parent, user, query);                case CollectionType.MovieResume:                    return GetMovieResume(parent, user, query);                case CollectionType.MovieMovies:                    return GetMovieMovies(parent, user, query);                case CollectionType.MovieCollections:                    return GetMovieCollections(parent, user, query);                case CollectionType.MusicLatest:                    return GetMusicLatest(parent, user, query);                case CollectionType.MusicAlbums:                    return GetMusicAlbums(parent, user, query);                case CollectionType.MusicAlbumArtists:                    return GetMusicAlbumArtists(parent, user, query);                case CollectionType.MusicArtists:                    return GetMusicArtists(parent, user, query);                case CollectionType.MusicSongs:                    return GetMusicSongs(parent, user, query);                case CollectionType.TvFavoriteEpisodes:                    return GetFavoriteEpisodes(parent, user, query);                case CollectionType.TvFavoriteSeries:                    return GetFavoriteSeries(parent, user, query);                case CollectionType.MusicFavorites:                    return await GetMusicFavorites(parent, user, query).ConfigureAwait(false);                case CollectionType.MusicFavoriteAlbums:                    return GetFavoriteAlbums(parent, user, query);                case CollectionType.MusicFavoriteArtists:                    return GetFavoriteArtists(parent, user, query);                case CollectionType.MusicFavoriteSongs:                    return GetFavoriteSongs(parent, user, query);                default:                    return GetResult(GetMediaFolders(user).SelectMany(i => i.GetChildren(user, true)), parent, query);            }        }        private int GetSpecialItemsLimit()        {            return 50;        }        private async Task<QueryResult<BaseItem>> GetMusicFolders(Folder parent, User user, InternalItemsQuery query)        {            if (query.Recursive)            {                return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Music, CollectionType.MusicVideos }), parent, query);            }            var list = new List<BaseItem>();            var category = "music";            list.Add(await GetUserView(category, CollectionType.MusicLatest, user, "0", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.MusicAlbums, user, "1", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.MusicAlbumArtists, user, "2", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.MusicSongs, user, "3", parent).ConfigureAwait(false));            //list.Add(await GetUserView(CollectionType.MusicArtists, user, "3", parent).ConfigureAwait(false));            //list.Add(await GetUserView(CollectionType.MusicGenres, user, "5", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.MusicFavorites, user, "6", parent).ConfigureAwait(false));            return GetResult(list, parent, query);        }        private async Task<QueryResult<BaseItem>> GetMusicFavorites(Folder parent, User user, InternalItemsQuery query)        {            var list = new List<BaseItem>();            var category = "music";            list.Add(await GetUserView(category, CollectionType.MusicFavoriteAlbums, user, "0", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.MusicFavoriteArtists, user, "1", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.MusicFavoriteSongs, user, "2", parent).ConfigureAwait(false));            return GetResult(list, parent, query);        }        private QueryResult<BaseItem> GetMusicAlbumArtists(Folder parent, User user, InternalItemsQuery query)        {            var artists = GetRecursiveChildren(parent, user, new[] { CollectionType.Music, CollectionType.MusicVideos })                .Where(i => !i.IsFolder)                .OfType<IHasAlbumArtist>()                .SelectMany(i => i.AlbumArtists)                .Distinct(StringComparer.OrdinalIgnoreCase)                .Select(i =>                {                    try                    {                        return _libraryManager.GetArtist(i);                    }                    catch                    {                        // Already logged at lower levels                        return null;                    }                })                .Where(i => i != null);            return GetResult(artists, parent, query);        }        private QueryResult<BaseItem> GetMusicArtists(Folder parent, User user, InternalItemsQuery query)        {            var artists = GetRecursiveChildren(parent, user, new[] { CollectionType.Music, CollectionType.MusicVideos })                .Where(i => !i.IsFolder)                .OfType<IHasArtist>()                .SelectMany(i => i.Artists)                .Distinct(StringComparer.OrdinalIgnoreCase)                .Select(i =>                {                    try                    {                        return _libraryManager.GetArtist(i);                    }                    catch                    {                        // Already logged at lower levels                        return null;                    }                })                .Where(i => i != null);            return GetResult(artists, parent, query);        }        private QueryResult<BaseItem> GetFavoriteArtists(Folder parent, User user, InternalItemsQuery query)        {            var artists = GetRecursiveChildren(parent, user, new[] { CollectionType.Music, CollectionType.MusicVideos })                .Where(i => !i.IsFolder)                .OfType<IHasAlbumArtist>()                .SelectMany(i => i.AlbumArtists)                .Distinct(StringComparer.OrdinalIgnoreCase)                .Select(i =>                {                    try                    {                        return _libraryManager.GetArtist(i);                    }                    catch                    {                        // Already logged at lower levels                        return null;                    }                })                .Where(i => i != null && _userDataManager.GetUserData(user.Id, i.GetUserDataKey()).IsFavorite);            return GetResult(artists, parent, query);        }        private QueryResult<BaseItem> GetMusicAlbums(Folder parent, User user, InternalItemsQuery query)        {            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Music, CollectionType.MusicVideos }).Where(i => i is MusicAlbum), parent, query);        }        private QueryResult<BaseItem> GetMusicSongs(Folder parent, User user, InternalItemsQuery query)        {            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Music, CollectionType.MusicVideos }).Where(i => i is Audio.Audio), parent, query);        }        private QueryResult<BaseItem> GetMusicLatest(Folder parent, User user, InternalItemsQuery query)        {            query.SortBy = new[] { ItemSortBy.DateCreated, ItemSortBy.SortName };            query.SortOrder = SortOrder.Descending;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Music, CollectionType.MusicVideos }).Where(i => i is MusicVideo || i is Audio.Audio), parent, GetSpecialItemsLimit(), query);        }        private async Task<QueryResult<BaseItem>> GetMovieFolders(Folder parent, User user, InternalItemsQuery query)        {            if (query.Recursive)            {                return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Movies, CollectionType.BoxSets, string.Empty }).Where(i => i is Movie || i is BoxSet), parent, query);            }            var list = new List<BaseItem>();            var category = "movies";            list.Add(await GetUserView(category, CollectionType.MovieResume, user, "0", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.MovieLatest, user, "1", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.MovieMovies, user, "2", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.MovieCollections, user, "3", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.MovieFavorites, user, "4", parent).ConfigureAwait(false));            //list.Add(await GetUserView(CollectionType.MovieGenres, user, "5", parent).ConfigureAwait(false));            return GetResult(list, parent, query);        }        private QueryResult<BaseItem> GetFavoriteMovies(Folder parent, User user, InternalItemsQuery query)        {            query.IsFavorite = true;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Movies, CollectionType.BoxSets, string.Empty }).Where(i => i is Movie), parent, query);        }        private QueryResult<BaseItem> GetFavoriteSeries(Folder parent, User user, InternalItemsQuery query)        {            query.IsFavorite = true;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.TvShows, string.Empty }).Where(i => i is Series), parent, query);        }        private QueryResult<BaseItem> GetFavoriteEpisodes(Folder parent, User user, InternalItemsQuery query)        {            query.IsFavorite = true;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.TvShows, string.Empty }).Where(i => i is Episode), parent, query);        }        private QueryResult<BaseItem> GetFavoriteSongs(Folder parent, User user, InternalItemsQuery query)        {            query.IsFavorite = true;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Music }).Where(i => i is Audio.Audio), parent, query);        }        private QueryResult<BaseItem> GetFavoriteAlbums(Folder parent, User user, InternalItemsQuery query)        {            query.IsFavorite = true;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Music }).Where(i => i is MusicAlbum), parent, query);        }        private QueryResult<BaseItem> GetMovieMovies(Folder parent, User user, InternalItemsQuery query)        {            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Movies, CollectionType.BoxSets, string.Empty }).Where(i => i is Movie), parent, query);        }        private QueryResult<BaseItem> GetMovieCollections(Folder parent, User user, InternalItemsQuery query)        {            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Movies, CollectionType.BoxSets, string.Empty }).Where(i => i is BoxSet), parent, query);        }        private QueryResult<BaseItem> GetMovieLatest(Folder parent, User user, InternalItemsQuery query)        {            query.SortBy = new[] { ItemSortBy.DateCreated, ItemSortBy.SortName };            query.SortOrder = SortOrder.Descending;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Movies, CollectionType.BoxSets, string.Empty }).Where(i => i is Movie), parent, GetSpecialItemsLimit(), query);        }        private QueryResult<BaseItem> GetMovieResume(Folder parent, User user, InternalItemsQuery query)        {            query.SortBy = new[] { ItemSortBy.DatePlayed, ItemSortBy.SortName };            query.SortOrder = SortOrder.Descending;            query.IsResumable = true;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Movies, CollectionType.BoxSets, string.Empty }).Where(i => i is Movie), parent, GetSpecialItemsLimit(), query);        }        private QueryResult<BaseItem> GetMovieGenres(Folder parent, User user, InternalItemsQuery query)        {            var genres = GetRecursiveChildren(parent, user, new[] { CollectionType.Movies, CollectionType.BoxSets, string.Empty })                .Where(i => i is Movie)                .SelectMany(i => i.Genres)                .Distinct(StringComparer.OrdinalIgnoreCase)                .Select(i =>                {                    try                    {                        return _libraryManager.GetGenre(i);                    }                    catch                    {                        // Full exception logged at lower levels                        _logger.Error("Error getting genre");                        return null;                    }                })                .Where(i => i != null);            return GetResult(genres, parent, query);        }        private async Task<QueryResult<BaseItem>> GetTvView(Folder parent, User user, InternalItemsQuery query)        {            if (query.Recursive)            {                return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.TvShows, string.Empty }).Where(i => i is Series || i is Season || i is Episode), parent, query);            }            var list = new List<BaseItem>();            var category = "tvshows";            list.Add(await GetUserView(category, CollectionType.TvResume, user, "0", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.TvNextUp, user, "1", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.TvLatest, user, "2", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.TvShowSeries, user, "3", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.TvFavoriteSeries, user, "4", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.TvFavoriteEpisodes, user, "5", parent).ConfigureAwait(false));            //list.Add(await GetUserView(CollectionType.TvGenres, user, "5", parent).ConfigureAwait(false));            return GetResult(list, parent, query);        }        private async Task<QueryResult<BaseItem>> GetGameView(User user, Folder parent, InternalItemsQuery query)        {            if (query.Recursive)            {                return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Games }), parent, query);            }            var list = new List<BaseItem>();            var category = "games";            list.Add(await GetUserView(category, CollectionType.LatestGames, user, "0", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.RecentlyPlayedGames, user, "1", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.GameFavorites, user, "2", parent).ConfigureAwait(false));            list.Add(await GetUserView(category, CollectionType.GameSystems, user, "3", parent).ConfigureAwait(false));            //list.Add(await GetUserView(CollectionType.GameGenres, user, "4", parent).ConfigureAwait(false));            return GetResult(list, parent, query);        }        private QueryResult<BaseItem> GetLatestGames(Folder parent, User user, InternalItemsQuery query)        {            query.SortBy = new[] { ItemSortBy.DateCreated, ItemSortBy.SortName };            query.SortOrder = SortOrder.Descending;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Games }).OfType<Game>(), parent, GetSpecialItemsLimit(), query);        }        private QueryResult<BaseItem> GetRecentlyPlayedGames(Folder parent, User user, InternalItemsQuery query)        {            query.IsPlayed = true;            query.SortBy = new[] { ItemSortBy.DatePlayed, ItemSortBy.SortName };            query.SortOrder = SortOrder.Descending;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Games }).OfType<Game>(), parent, GetSpecialItemsLimit(), query);        }        private QueryResult<BaseItem> GetFavoriteGames(Folder parent, User user, InternalItemsQuery query)        {            query.IsFavorite = true;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Games }).OfType<Game>(), parent, query);        }        private QueryResult<BaseItem> GetTvLatest(Folder parent, User user, InternalItemsQuery query)        {            query.SortBy = new[] { ItemSortBy.DateCreated, ItemSortBy.SortName };            query.SortOrder = SortOrder.Descending;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.TvShows, string.Empty }).OfType<Episode>(), parent, GetSpecialItemsLimit(), query);        }        private QueryResult<BaseItem> GetTvNextUp(Folder parent, InternalItemsQuery query)        {            var parentFolders = GetMediaFolders(parent, query.User, new[] { CollectionType.TvShows, string.Empty });            var result = _tvSeriesManager.GetNextUp(new NextUpQuery            {                Limit = query.Limit,                StartIndex = query.StartIndex,                UserId = query.User.Id.ToString("N")            }, parentFolders);            return result;        }        private QueryResult<BaseItem> GetTvResume(Folder parent, User user, InternalItemsQuery query)        {            query.SortBy = new[] { ItemSortBy.DatePlayed, ItemSortBy.SortName };            query.SortOrder = SortOrder.Descending;            query.IsResumable = true;            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.TvShows, string.Empty }).OfType<Episode>(), parent, GetSpecialItemsLimit(), query);        }        private QueryResult<BaseItem> GetTvSeries(Folder parent, User user, InternalItemsQuery query)        {            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.TvShows, string.Empty }).OfType<Series>(), parent, query);        }        private QueryResult<BaseItem> GetTvGenres(Folder parent, User user, InternalItemsQuery query)        {            var genres = GetRecursiveChildren(parent, user, new[] { CollectionType.TvShows, string.Empty })                .OfType<Series>()                .SelectMany(i => i.Genres)                .Distinct(StringComparer.OrdinalIgnoreCase)                .Select(i =>                {                    try                    {                        return _libraryManager.GetGenre(i);                    }                    catch                    {                        // Full exception logged at lower levels                        _logger.Error("Error getting genre");                        return null;                    }                })                .Where(i => i != null);            return GetResult(genres, parent, query);        }        private QueryResult<BaseItem> GetGameSystems(Folder parent, User user, InternalItemsQuery query)        {            return GetResult(GetRecursiveChildren(parent, user, new[] { CollectionType.Games }).OfType<GameSystem>(), parent, query);        }        private QueryResult<BaseItem> GetGameGenres(Folder parent, User user, InternalItemsQuery query)        {            var genres = GetRecursiveChildren(parent, user, new[] { CollectionType.Games })                .OfType<Game>()                .SelectMany(i => i.Genres)                .Distinct(StringComparer.OrdinalIgnoreCase)                .Select(i =>                {                    try                    {                        return _libraryManager.GetGameGenre(i);                    }                    catch                    {                        // Full exception logged at lower levels                        _logger.Error("Error getting game genre");                        return null;                    }                })                .Where(i => i != null);            return GetResult(genres, parent, query);        }        private QueryResult<BaseItem> GetResult<T>(QueryResult<T> result)            where T : BaseItem        {            return new QueryResult<BaseItem>            {                Items = result.Items,                TotalRecordCount = result.TotalRecordCount            };        }        private QueryResult<BaseItem> GetResult<T>(IEnumerable<T> items,            BaseItem parentItem,            InternalItemsQuery query)            where T : BaseItem        {            return GetResult(items, parentItem, null, query);        }        private QueryResult<BaseItem> GetResult<T>(IEnumerable<T> items,            BaseItem parentItem,            int? totalRecordLimit,            InternalItemsQuery query)            where T : BaseItem        {            return SortAndFilter(items, parentItem, totalRecordLimit, query, _libraryManager, _userDataManager);        }        public static QueryResult<BaseItem> SortAndFilter(IEnumerable<BaseItem> items,            BaseItem parentItem,            int? totalRecordLimit,            InternalItemsQuery query,            ILibraryManager libraryManager,            IUserDataManager userDataManager)        {            var user = query.User;            items = items.Where(i => Filter(i, user, query, userDataManager));            items = FilterVirtualEpisodes(items,                query.IsMissing,                query.IsVirtualUnaired,                query.IsUnaired);            items = CollapseBoxSetItemsIfNeeded(items, query, parentItem, user);            // This must be the last filter            if (!string.IsNullOrEmpty(query.AdjacentTo))            {                items = FilterForAdjacency(items, query.AdjacentTo);            }            return Sort(items, totalRecordLimit, query, libraryManager);        }        public static IEnumerable<BaseItem> CollapseBoxSetItemsIfNeeded(IEnumerable<BaseItem> items,            InternalItemsQuery query,            BaseItem parentItem,            User user)        {            if (CollapseBoxSetItems(query, parentItem, user))            {                items = BaseItem.CollectionManager.CollapseItemsWithinBoxSets(items, user);            }            items = ApplyPostCollectionCollapseFilters(query, items, user);            return items;        }        private static IEnumerable<BaseItem> ApplyPostCollectionCollapseFilters(InternalItemsQuery request,            IEnumerable<BaseItem> items,            User user)        {            if (!string.IsNullOrEmpty(request.NameStartsWithOrGreater))            {                items = items.Where(i => string.Compare(request.NameStartsWithOrGreater, i.SortName, StringComparison.CurrentCultureIgnoreCase) < 1);            }            if (!string.IsNullOrEmpty(request.NameStartsWith))            {                items = items.Where(i => string.Compare(request.NameStartsWith, i.SortName.Substring(0, 1), StringComparison.CurrentCultureIgnoreCase) == 0);            }            if (!string.IsNullOrEmpty(request.NameLessThan))            {                items = items.Where(i => string.Compare(request.NameLessThan, i.SortName, StringComparison.CurrentCultureIgnoreCase) == 1);            }            return items;        }        private static bool CollapseBoxSetItems(InternalItemsQuery query,            BaseItem parentItem,            User user)        {            // Could end up stuck in a loop like this            if (parentItem is BoxSet)            {                return false;            }            var param = query.CollapseBoxSetItems;            if (!param.HasValue)            {                if (user != null && !user.Configuration.GroupMoviesIntoBoxSets)                {                    return false;                }                if (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.AllGenres.Length > 0)            {                return false;            }            if (request.Genres.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.IsInBoxSet.HasValue)            {                return false;            }            if (request.IsLocked.HasValue)            {                return false;            }            if (request.IsPlaceHolder.HasValue)            {                return false;            }            if (request.IsPlayed.HasValue)            {                return false;            }            if (request.IsUnidentified.HasValue)            {                return false;            }            if (request.IsYearMismatched.HasValue)            {                return false;            }            if (!string.IsNullOrWhiteSpace(request.Person))            {                return false;            }            if (request.Studios.Length > 0)            {                return false;            }            if (request.VideoTypes.Length > 0)            {                return false;            }            if (request.Years.Length > 0)            {                return false;            }            return true;        }        public static IEnumerable<BaseItem> FilterVirtualEpisodes(            IEnumerable<BaseItem> items,            bool? isMissing,            bool? isVirtualUnaired,            bool? isUnaired)        {            items = FilterVirtualSeasons(items, isMissing, isVirtualUnaired, isUnaired);            if (isMissing.HasValue)            {                var val = isMissing.Value;                items = items.Where(i =>                {                    var e = i as Episode;                    if (e != null)                    {                        return e.IsMissingEpisode == val;                    }                    return true;                });            }            if (isUnaired.HasValue)            {                var val = isUnaired.Value;                items = items.Where(i =>                {                    var e = i as Episode;                    if (e != null)                    {                        return e.IsUnaired == val;                    }                    return true;                });            }            if (isVirtualUnaired.HasValue)            {                var val = isVirtualUnaired.Value;                items = items.Where(i =>                {                    var e = i as Episode;                    if (e != null)                    {                        return e.IsVirtualUnaired == val;                    }                    return true;                });            }            return items;        }        private static IEnumerable<BaseItem> FilterVirtualSeasons(            IEnumerable<BaseItem> items,            bool? isMissing,            bool? isVirtualUnaired,            bool? isUnaired)        {            if (isMissing.HasValue && isVirtualUnaired.HasValue)            {                if (!isMissing.Value && !isVirtualUnaired.Value)                {                    return items.Where(i =>                    {                        var e = i as Season;                        if (e != null)                        {                            return !e.IsMissingOrVirtualUnaired;                        }                        return true;                    });                }            }            if (isMissing.HasValue)            {                var val = isMissing.Value;                items = items.Where(i =>                {                    var e = i as Season;                    if (e != null)                    {                        return e.IsMissingSeason == val;                    }                    return true;                });            }            if (isUnaired.HasValue)            {                var val = isUnaired.Value;                items = items.Where(i =>                {                    var e = i as Season;                    if (e != null)                    {                        return e.IsUnaired == val;                    }                    return true;                });            }            if (isVirtualUnaired.HasValue)            {                var val = isVirtualUnaired.Value;                items = items.Where(i =>                {                    var e = i as Season;                    if (e != null)                    {                        return e.IsVirtualUnaired == val;                    }                    return true;                });            }            return items;        }        public static QueryResult<BaseItem> Sort(IEnumerable<BaseItem> items,            int? totalRecordLimit,            InternalItemsQuery query,            ILibraryManager libraryManager)        {            var user = query.User;            items = libraryManager.ReplaceVideosWithPrimaryVersions(items);            if (query.SortBy.Length > 0)            {                items = libraryManager.Sort(items, user, query.SortBy, query.SortOrder);            }            var itemsArray = totalRecordLimit.HasValue ? items.Take(totalRecordLimit.Value).ToArray() : items.ToArray();            var totalCount = itemsArray.Length;            if (query.Limit.HasValue)            {                itemsArray = itemsArray.Skip(query.StartIndex ?? 0).Take(query.Limit.Value).ToArray();            }            else if (query.StartIndex.HasValue)            {                itemsArray = itemsArray.Skip(query.StartIndex.Value).ToArray();            }            return new QueryResult<BaseItem>            {                TotalRecordCount = totalCount,                Items = itemsArray            };        }        private static bool Filter(BaseItem item, User user, InternalItemsQuery query, IUserDataManager userDataManager)        {            if (query.MediaTypes.Length > 0 && !query.MediaTypes.Contains(item.MediaType ?? string.Empty, StringComparer.OrdinalIgnoreCase))            {                return false;            }            if (query.IncludeItemTypes.Length > 0 && !query.IncludeItemTypes.Contains(item.GetClientTypeName(), StringComparer.OrdinalIgnoreCase))            {                return false;            }            if (query.ExcludeItemTypes.Length > 0 && query.ExcludeItemTypes.Contains(item.GetClientTypeName(), StringComparer.OrdinalIgnoreCase))            {                return false;            }            if (query.IsFolder.HasValue && query.IsFolder.Value != item.IsFolder)            {                return false;            }            if (query.Filter != null && !query.Filter(item, user))            {                return false;            }            UserItemData userData = null;            if (query.IsLiked.HasValue)            {                userData = userData ?? userDataManager.GetUserData(user.Id, item.GetUserDataKey());                if (!userData.Likes.HasValue || userData.Likes != query.IsLiked.Value)                {                    return false;                }            }            if (query.IsFavoriteOrLiked.HasValue)            {                userData = userData ?? userDataManager.GetUserData(user.Id, item.GetUserDataKey());                var isFavoriteOrLiked = userData.IsFavorite || (userData.Likes ?? false);                if (isFavoriteOrLiked != query.IsFavoriteOrLiked.Value)                {                    return false;                }            }            if (query.IsFavorite.HasValue)            {                userData = userData ?? userDataManager.GetUserData(user.Id, item.GetUserDataKey());                if (userData.IsFavorite != query.IsFavorite.Value)                {                    return false;                }            }            if (query.IsResumable.HasValue)            {                userData = userData ?? userDataManager.GetUserData(user.Id, item.GetUserDataKey());                var isResumable = userData.PlaybackPositionTicks > 0;                if (isResumable != query.IsResumable.Value)                {                    return false;                }            }            if (query.IsPlayed.HasValue)            {                if (item.IsPlayed(user) != query.IsPlayed.Value)                {                    return false;                }            }            if (query.IsInBoxSet.HasValue)            {                var val = query.IsInBoxSet.Value;                if (item.Parents.OfType<BoxSet>().Any() != val)                {                    return false;                }            }            // Filter by Video3DFormat            if (query.Is3D.HasValue)            {                var val = query.Is3D.Value;                var video = item as Video;                if (video == null || val != video.Video3DFormat.HasValue)                {                    return false;                }            }            if (query.IsHD.HasValue)            {                var val = query.IsHD.Value;                var video = item as Video;                if (video == null || val != video.IsHD)                {                    return false;                }            }            if (query.IsUnidentified.HasValue)            {                var val = query.IsUnidentified.Value;                if (item.IsUnidentified != val)                {                    return false;                }            }            if (query.IsLocked.HasValue)            {                var val = query.IsLocked.Value;                if (item.IsLocked != val)                {                    return false;                }            }            if (query.HasOverview.HasValue)            {                var filterValue = query.HasOverview.Value;                var hasValue = !string.IsNullOrEmpty(item.Overview);                if (hasValue != filterValue)                {                    return false;                }            }            if (query.HasImdbId.HasValue)            {                var filterValue = query.HasImdbId.Value;                var hasValue = !string.IsNullOrEmpty(item.GetProviderId(MetadataProviders.Imdb));                if (hasValue != filterValue)                {                    return false;                }            }            if (query.HasTmdbId.HasValue)            {                var filterValue = query.HasTmdbId.Value;                var hasValue = !string.IsNullOrEmpty(item.GetProviderId(MetadataProviders.Tmdb));                if (hasValue != filterValue)                {                    return false;                }            }            if (query.HasTvdbId.HasValue)            {                var filterValue = query.HasTvdbId.Value;                var hasValue = !string.IsNullOrEmpty(item.GetProviderId(MetadataProviders.Tvdb));                if (hasValue != filterValue)                {                    return false;                }            }            if (query.IsYearMismatched.HasValue)            {                var filterValue = query.IsYearMismatched.Value;                if (IsYearMismatched(item) != filterValue)                {                    return false;                }            }            if (query.HasOfficialRating.HasValue)            {                var filterValue = query.HasOfficialRating.Value;                var hasValue = !string.IsNullOrEmpty(item.OfficialRating);                if (hasValue != filterValue)                {                    return false;                }            }            if (query.IsPlaceHolder.HasValue)            {                var filterValue = query.IsPlaceHolder.Value;                var isPlaceHolder = false;                var hasPlaceHolder = item as ISupportsPlaceHolders;                if (hasPlaceHolder != null)                {                    isPlaceHolder = hasPlaceHolder.IsPlaceHolder;                }                if (isPlaceHolder != filterValue)                {                    return false;                }            }            if (query.HasSpecialFeature.HasValue)            {                var filterValue = query.HasSpecialFeature.Value;                var movie = item as IHasSpecialFeatures;                if (movie != null)                {                    var ok = filterValue                        ? movie.SpecialFeatureIds.Count > 0                        : movie.SpecialFeatureIds.Count == 0;                    if (!ok)                    {                        return false;                    }                }                else                {                    return false;                }            }            if (query.HasSubtitles.HasValue)            {                var val = query.HasSubtitles.Value;                var video = item as Video;                if (video == null || val != video.HasSubtitles)                {                    return false;                }            }            if (query.HasParentalRating.HasValue)            {                var val = query.HasParentalRating.Value;                var rating = item.CustomRating;                if (string.IsNullOrEmpty(rating))                {                    rating = item.OfficialRating;                }                if (val)                {                    if (string.IsNullOrEmpty(rating))                    {                        return false;                    }                }                else                {                    if (!string.IsNullOrEmpty(rating))                    {                        return false;                    }                }            }            if (query.HasTrailer.HasValue)            {                var val = query.HasTrailer.Value;                var trailerCount = 0;                var hasTrailers = item as IHasTrailers;                if (hasTrailers != null)                {                    trailerCount = hasTrailers.LocalTrailerIds.Count;                }                var ok = val ? trailerCount > 0 : trailerCount == 0;                if (!ok)                {                    return false;                }            }            if (query.HasThemeSong.HasValue)            {                var filterValue = query.HasThemeSong.Value;                var themeCount = 0;                var iHasThemeMedia = item as IHasThemeMedia;                if (iHasThemeMedia != null)                {                    themeCount = iHasThemeMedia.ThemeSongIds.Count;                }                var ok = filterValue ? themeCount > 0 : themeCount == 0;                if (!ok)                {                    return false;                }            }            if (query.HasThemeVideo.HasValue)            {                var filterValue = query.HasThemeVideo.Value;                var themeCount = 0;                var iHasThemeMedia = item as IHasThemeMedia;                if (iHasThemeMedia != null)                {                    themeCount = iHasThemeMedia.ThemeVideoIds.Count;                }                var ok = filterValue ? themeCount > 0 : themeCount == 0;                if (!ok)                {                    return false;                }            }            // Apply genre filter            if (query.Genres.Length > 0 && !(query.Genres.Any(v => item.Genres.Contains(v, StringComparer.OrdinalIgnoreCase))))            {                return false;            }            // Apply genre filter            if (query.AllGenres.Length > 0 && !query.AllGenres.All(v => item.Genres.Contains(v, StringComparer.OrdinalIgnoreCase)))            {                return false;            }            // Filter by VideoType            if (query.VideoTypes.Length > 0)            {                var video = item as Video;                if (video == null || !query.VideoTypes.Contains(video.VideoType))                {                    return false;                }            }            if (query.ImageTypes.Length > 0 && !query.ImageTypes.Any(item.HasImage))            {                return false;            }            // Apply studio filter            if (query.Studios.Length > 0 && !query.Studios.Any(v => item.Studios.Contains(v, StringComparer.OrdinalIgnoreCase)))            {                return false;            }            // Apply year filter            if (query.Years.Length > 0)            {                if (!(item.ProductionYear.HasValue && query.Years.Contains(item.ProductionYear.Value)))                {                    return false;                }            }            // Apply person filter            if (!string.IsNullOrEmpty(query.Person))            {                var personTypes = query.PersonTypes;                if (personTypes.Length == 0)                {                    if (!(item.People.Any(p => string.Equals(p.Name, query.Person, StringComparison.OrdinalIgnoreCase))))                    {                        return false;                    }                }                else                {                    var types = personTypes;                    var ok = new[] { item }.Any(i =>                            i.People != null &&                            i.People.Any(p =>                                p.Name.Equals(query.Person, StringComparison.OrdinalIgnoreCase) && (types.Contains(p.Type, StringComparer.OrdinalIgnoreCase) || types.Contains(p.Role, StringComparer.OrdinalIgnoreCase))));                    if (!ok)                    {                        return false;                    }                }            }            return true;        }        private IEnumerable<Folder> GetMediaFolders(User user)        {            var excludeFolderIds = user.Configuration.ExcludeFoldersFromGrouping.Select(i => new Guid(i)).ToList();            return user.RootFolder                .GetChildren(user, true, true)                .OfType<Folder>()                .Where(i => !excludeFolderIds.Contains(i.Id) && !UserView.IsExcludedFromGrouping(i));        }        private IEnumerable<Folder> GetMediaFolders(User user, IEnumerable<string> viewTypes)        {            return GetMediaFolders(user)                .Where(i =>                {                    var folder = i as ICollectionFolder;                    return folder != null && viewTypes.Contains(folder.CollectionType ?? string.Empty, StringComparer.OrdinalIgnoreCase);                });        }        private IEnumerable<Folder> GetMediaFolders(Folder parent, User user, string[] viewTypes)        {            if (parent == null || parent is UserView)            {                return GetMediaFolders(user, viewTypes);            }            return new[] { parent };        }        private IEnumerable<BaseItem> GetRecursiveChildren(Folder parent, User user, string[] viewTypes)        {            if (parent == null || parent is UserView)            {                return GetMediaFolders(user, viewTypes).SelectMany(i => i.GetRecursiveChildren(user));            }            return parent.GetRecursiveChildren(user);        }        private async Task<IEnumerable<BaseItem>> GetLiveTvFolders(User user)        {            var list = new List<BaseItem>();            list.Add(await _userViewManager.GetUserView("livetv", CollectionType.LiveTvNowPlaying, user, "0", CancellationToken.None).ConfigureAwait(false));            list.Add(await _userViewManager.GetUserView("livetv", CollectionType.LiveTvChannels, user, string.Empty, CancellationToken.None).ConfigureAwait(false));            list.Add(await _userViewManager.GetUserView("livetv", CollectionType.LiveTvRecordingGroups, user, string.Empty, CancellationToken.None).ConfigureAwait(false));            return list;        }        private async Task<UserView> GetUserView(string category, string type, User user, string sortName, Folder parent)        {            var view = await _userViewManager.GetUserView(category, type, user, sortName, CancellationToken.None)                        .ConfigureAwait(false);            if (parent.Id != view.ParentId)            {                view.ParentId = parent.Id;                await view.UpdateToRepository(ItemUpdateType.MetadataImport, CancellationToken.None)                        .ConfigureAwait(false);            }            return view;        }        public static bool IsYearMismatched(BaseItem item)        {            if (item.ProductionYear.HasValue)            {                var path = item.Path;                if (!string.IsNullOrEmpty(path))                {                    int? yearInName;                    string name;                    NameParser.ParseName(Path.GetFileName(path), out name, out yearInName);                    // Go up a level if we didn't get a year                    if (!yearInName.HasValue)                    {                        NameParser.ParseName(Path.GetFileName(Path.GetDirectoryName(path)), out name, out yearInName);                    }                    if (yearInName.HasValue)                    {                        return yearInName.Value != item.ProductionYear.Value;                    }                }            }            return false;        }        public static IEnumerable<BaseItem> FilterForAdjacency(IEnumerable<BaseItem> items, string adjacentToId)        {            var list = items.ToList();            var adjacentToIdGuid = new Guid(adjacentToId);            var adjacentToItem = list.FirstOrDefault(i => i.Id == adjacentToIdGuid);            var index = list.IndexOf(adjacentToItem);            var previousId = Guid.Empty;            var nextId = Guid.Empty;            if (index > 0)            {                previousId = list[index - 1].Id;            }            if (index < list.Count - 1)            {                nextId = list[index + 1].Id;            }            return list.Where(i => i.Id == previousId || i.Id == nextId || i.Id == adjacentToIdGuid);        }    }}
 |