| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425 | using System;using System.Collections.Generic;using System.Globalization;using System.IO;using System.Linq;using System.Threading.Tasks;using MediaBrowser.Common;using MediaBrowser.Controller.Channels;using MediaBrowser.Controller.Drawing;using MediaBrowser.Controller.Dto;using MediaBrowser.Controller.Entities;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.Persistence;using MediaBrowser.Controller.Playlists;using MediaBrowser.Controller.Providers;using MediaBrowser.Model.Drawing;using MediaBrowser.Model.Dto;using MediaBrowser.Model.Entities;using MediaBrowser.Model.Querying;using Microsoft.Extensions.Logging;namespace Emby.Server.Implementations.Dto{    public class DtoService : IDtoService    {        private readonly ILogger _logger;        private readonly ILibraryManager _libraryManager;        private readonly IUserDataManager _userDataRepository;        private readonly IItemRepository _itemRepo;        private readonly IImageProcessor _imageProcessor;        private readonly IProviderManager _providerManager;        private readonly IApplicationHost _appHost;        private readonly Func<IMediaSourceManager> _mediaSourceManager;        private readonly Func<ILiveTvManager> _livetvManager;        public DtoService(            ILoggerFactory loggerFactory,            ILibraryManager libraryManager,            IUserDataManager userDataRepository,            IItemRepository itemRepo,            IImageProcessor imageProcessor,            IProviderManager providerManager,            IApplicationHost appHost,            Func<IMediaSourceManager> mediaSourceManager,            Func<ILiveTvManager> livetvManager)        {            _logger = loggerFactory.CreateLogger(nameof(DtoService));            _libraryManager = libraryManager;            _userDataRepository = userDataRepository;            _itemRepo = itemRepo;            _imageProcessor = imageProcessor;            _providerManager = providerManager;            _appHost = appHost;            _mediaSourceManager = mediaSourceManager;            _livetvManager = livetvManager;        }        /// <summary>        /// Converts a BaseItem to a DTOBaseItem        /// </summary>        /// <param name="item">The item.</param>        /// <param name="fields">The fields.</param>        /// <param name="user">The user.</param>        /// <param name="owner">The owner.</param>        /// <returns>Task{DtoBaseItem}.</returns>        /// <exception cref="ArgumentNullException">item</exception>        public BaseItemDto GetBaseItemDto(BaseItem item, ItemFields[] fields, User user = null, BaseItem owner = null)        {            var options = new DtoOptions            {                Fields = fields            };            return GetBaseItemDto(item, options, user, owner);        }        /// <inheritdoc />        public IReadOnlyList<BaseItemDto> GetBaseItemDtos(IReadOnlyList<BaseItem> items, DtoOptions options, User user = null, BaseItem owner = null)        {            var returnItems = new BaseItemDto[items.Count];            var programTuples = new List<(BaseItem, BaseItemDto)>();            var channelTuples = new List<(BaseItemDto, LiveTvChannel)>();            for (int index = 0; index < items.Count; index++)            {                var item = items[index];                var dto = GetBaseItemDtoInternal(item, options, user, owner);                if (item is LiveTvChannel tvChannel)                {                    channelTuples.Add((dto, tvChannel));                }                else if (item is LiveTvProgram)                {                    programTuples.Add((item, dto));                }                if (item is IItemByName byName)                {                    if (options.ContainsField(ItemFields.ItemCounts))                    {                        var libraryItems = byName.GetTaggedItems(new InternalItemsQuery(user)                        {                            Recursive = true,                            DtoOptions = new DtoOptions(false)                            {                                EnableImages = false                            }                        });                        SetItemByNameInfo(item, dto, libraryItems, user);                    }                }                returnItems[index] = dto;            }            if (programTuples.Count > 0)            {                _livetvManager().AddInfoToProgramDto(programTuples, options.Fields, user).GetAwaiter().GetResult();            }            if (channelTuples.Count > 0)            {                _livetvManager().AddChannelInfo(channelTuples, options, user);            }            return returnItems;        }        public BaseItemDto GetBaseItemDto(BaseItem item, DtoOptions options, User user = null, BaseItem owner = null)        {            var dto = GetBaseItemDtoInternal(item, options, user, owner);            if (item is LiveTvChannel tvChannel)            {                var list = new List<(BaseItemDto, LiveTvChannel)>(1) { (dto, tvChannel) };                _livetvManager().AddChannelInfo(list, options, user);            }            else if (item is LiveTvProgram)            {                var list = new List<(BaseItem, BaseItemDto)>(1) { (item, dto) };                var task = _livetvManager().AddInfoToProgramDto(list, options.Fields, user);                Task.WaitAll(task);            }            if (item is IItemByName itemByName                && options.ContainsField(ItemFields.ItemCounts))            {                SetItemByNameInfo(                    item,                    dto,                    GetTaggedItems(                        itemByName,                        user,                        new DtoOptions(false)                        {                            EnableImages = false                        }),                    user);            }            return dto;        }        private static IList<BaseItem> GetTaggedItems(IItemByName byName, User user, DtoOptions options)        {            return byName.GetTaggedItems(                new InternalItemsQuery(user)                {                    Recursive = true,                    DtoOptions = options                });        }        private BaseItemDto GetBaseItemDtoInternal(BaseItem item, DtoOptions options, User user = null, BaseItem owner = null)        {            var dto = new BaseItemDto            {                ServerId = _appHost.SystemId            };            if (item.SourceType == SourceType.Channel)            {                dto.SourceType = item.SourceType.ToString();            }            if (options.ContainsField(ItemFields.People))            {                AttachPeople(dto, item);            }            if (options.ContainsField(ItemFields.PrimaryImageAspectRatio))            {                try                {                    AttachPrimaryImageAspectRatio(dto, item);                }                catch (Exception ex)                {                    // Have to use a catch-all unfortunately because some .net image methods throw plain Exceptions                    _logger.LogError(ex, "Error generating PrimaryImageAspectRatio for {itemName}", item.Name);                }            }            if (options.ContainsField(ItemFields.DisplayPreferencesId))            {                dto.DisplayPreferencesId = item.DisplayPreferencesId.ToString("N", CultureInfo.InvariantCulture);            }            if (user != null)            {                AttachUserSpecificInfo(dto, item, user, options);            }            if (item is IHasMediaSources                && options.ContainsField(ItemFields.MediaSources))            {                dto.MediaSources = _mediaSourceManager().GetStaticMediaSources(item, true, user).ToArray();                NormalizeMediaSourceContainers(dto);            }            if (options.ContainsField(ItemFields.Studios))            {                AttachStudios(dto, item);            }            AttachBasicFields(dto, item, owner, options);            if (options.ContainsField(ItemFields.CanDelete))            {                dto.CanDelete = user == null                    ? item.CanDelete()                    : item.CanDelete(user);            }            if (options.ContainsField(ItemFields.CanDownload))            {                dto.CanDownload = user == null                    ? item.CanDownload()                    : item.CanDownload(user);            }            if (options.ContainsField(ItemFields.Etag))            {                dto.Etag = item.GetEtag(user);            }            var liveTvManager = _livetvManager();            var activeRecording = liveTvManager.GetActiveRecordingInfo(item.Path);            if (activeRecording != null)            {                dto.Type = "Recording";                dto.CanDownload = false;                dto.RunTimeTicks = null;                if (!string.IsNullOrEmpty(dto.SeriesName))                {                    dto.EpisodeTitle = dto.Name;                    dto.Name = dto.SeriesName;                }                liveTvManager.AddInfoToRecordingDto(item, dto, activeRecording, user);            }            return dto;        }        private static void NormalizeMediaSourceContainers(BaseItemDto dto)        {            foreach (var mediaSource in dto.MediaSources)            {                var container = mediaSource.Container;                if (string.IsNullOrEmpty(container))                {                    continue;                }                var containers = container.Split(new[] { ',' });                if (containers.Length < 2)                {                    continue;                }                var path = mediaSource.Path;                string fileExtensionContainer = null;                if (!string.IsNullOrEmpty(path))                {                    path = Path.GetExtension(path);                    if (!string.IsNullOrEmpty(path))                    {                        path = Path.GetExtension(path);                        if (!string.IsNullOrEmpty(path))                        {                            path = path.TrimStart('.');                        }                        if (!string.IsNullOrEmpty(path) && containers.Contains(path, StringComparer.OrdinalIgnoreCase))                        {                            fileExtensionContainer = path;                        }                    }                }                mediaSource.Container = fileExtensionContainer ?? containers[0];            }        }        public BaseItemDto GetItemByNameDto(BaseItem item, DtoOptions options, List<BaseItem> taggedItems, User user = null)        {            var dto = GetBaseItemDtoInternal(item, options, user);            if (taggedItems != null && options.ContainsField(ItemFields.ItemCounts))            {                SetItemByNameInfo(item, dto, taggedItems, user);            }            return dto;        }        private static void SetItemByNameInfo(BaseItem item, BaseItemDto dto, IList<BaseItem> taggedItems, User user = null)        {            if (item is MusicArtist)            {                dto.AlbumCount = taggedItems.Count(i => i is MusicAlbum);                dto.MusicVideoCount = taggedItems.Count(i => i is MusicVideo);                dto.SongCount = taggedItems.Count(i => i is Audio);            }            else if (item is MusicGenre)            {                dto.ArtistCount = taggedItems.Count(i => i is MusicArtist);                dto.AlbumCount = taggedItems.Count(i => i is MusicAlbum);                dto.MusicVideoCount = taggedItems.Count(i => i is MusicVideo);                dto.SongCount = taggedItems.Count(i => i is Audio);            }            else            {                // This populates them all and covers Genre, Person, Studio, Year                dto.ArtistCount = taggedItems.Count(i => i is MusicArtist);                dto.AlbumCount = taggedItems.Count(i => i is MusicAlbum);                dto.EpisodeCount = taggedItems.Count(i => i is Episode);                dto.MovieCount = taggedItems.Count(i => i is Movie);                dto.TrailerCount = taggedItems.Count(i => i is Trailer);                dto.MusicVideoCount = taggedItems.Count(i => i is MusicVideo);                dto.SeriesCount = taggedItems.Count(i => i is Series);                dto.ProgramCount = taggedItems.Count(i => i is LiveTvProgram);                dto.SongCount = taggedItems.Count(i => i is Audio);            }            dto.ChildCount = taggedItems.Count;        }        /// <summary>        /// Attaches the user specific info.        /// </summary>        private void AttachUserSpecificInfo(BaseItemDto dto, BaseItem item, User user, DtoOptions options)        {            if (item.IsFolder)            {                var folder = (Folder)item;                if (options.EnableUserData)                {                    dto.UserData = _userDataRepository.GetUserDataDto(item, dto, user, options);                }                if (!dto.ChildCount.HasValue && item.SourceType == SourceType.Library)                {                    // For these types we can try to optimize and assume these values will be equal                    if (item is MusicAlbum || item is Season || item is Playlist)                    {                        dto.ChildCount = dto.RecursiveItemCount;                    }                    if (options.ContainsField(ItemFields.ChildCount))                    {                        dto.ChildCount = dto.ChildCount ?? GetChildCount(folder, user);                    }                }                if (options.ContainsField(ItemFields.CumulativeRunTimeTicks))                {                    dto.CumulativeRunTimeTicks = item.RunTimeTicks;                }                if (options.ContainsField(ItemFields.DateLastMediaAdded))                {                    dto.DateLastMediaAdded = folder.DateLastMediaAdded;                }            }            else            {                if (options.EnableUserData)                {                    dto.UserData = _userDataRepository.GetUserDataDto(item, user);                }            }            if (options.ContainsField(ItemFields.PlayAccess))            {                dto.PlayAccess = item.GetPlayAccess(user);            }            if (options.ContainsField(ItemFields.BasicSyncInfo))            {                var userCanSync = user != null && user.Policy.EnableContentDownloading;                if (userCanSync && item.SupportsExternalTransfer)                {                    dto.SupportsSync = true;                }            }        }        private static int GetChildCount(Folder folder, User user)        {            // Right now this is too slow to calculate for top level folders on a per-user basis            // Just return something so that apps that are expecting a value won't think the folders are empty            if (folder is ICollectionFolder || folder is UserView)            {                return new Random().Next(1, 10);            }            return folder.GetChildCount(user);        }        /// <summary>        /// Gets client-side Id of a server-side BaseItem        /// </summary>        /// <param name="item">The item.</param>        /// <returns>System.String.</returns>        /// <exception cref="ArgumentNullException">item</exception>        public string GetDtoId(BaseItem item)        {            return item.Id.ToString("N", CultureInfo.InvariantCulture);        }        private static void SetBookProperties(BaseItemDto dto, Book item)        {            dto.SeriesName = item.SeriesName;        }        private static void SetPhotoProperties(BaseItemDto dto, Photo item)        {            dto.CameraMake = item.CameraMake;            dto.CameraModel = item.CameraModel;            dto.Software = item.Software;            dto.ExposureTime = item.ExposureTime;            dto.FocalLength = item.FocalLength;            dto.ImageOrientation = item.Orientation;            dto.Aperture = item.Aperture;            dto.ShutterSpeed = item.ShutterSpeed;            dto.Latitude = item.Latitude;            dto.Longitude = item.Longitude;            dto.Altitude = item.Altitude;            dto.IsoSpeedRating = item.IsoSpeedRating;            var album = item.AlbumEntity;            if (album != null)            {                dto.Album = album.Name;                dto.AlbumId = album.Id;            }        }        private void SetMusicVideoProperties(BaseItemDto dto, MusicVideo item)        {            if (!string.IsNullOrEmpty(item.Album))            {                var parentAlbumIds = _libraryManager.GetItemIds(new InternalItemsQuery                {                    IncludeItemTypes = new[] { typeof(MusicAlbum).Name },                    Name = item.Album,                    Limit = 1                });                if (parentAlbumIds.Count > 0)                {                    dto.AlbumId = parentAlbumIds[0];                }            }            dto.Album = item.Album;        }        private string[] GetImageTags(BaseItem item, List<ItemImageInfo> images)        {            return images                .Select(p => GetImageCacheTag(item, p))                .Where(i => i != null)                .ToArray();        }        private string GetImageCacheTag(BaseItem item, ImageType type)        {            try            {                return _imageProcessor.GetImageCacheTag(item, type);            }            catch (Exception ex)            {                _logger.LogError(ex, "Error getting {type} image info", type);                return null;            }        }        private string GetImageCacheTag(BaseItem item, ItemImageInfo image)        {            try            {                return _imageProcessor.GetImageCacheTag(item, image);            }            catch (Exception ex)            {                _logger.LogError(ex, "Error getting {imageType} image info for {path}", image.Type, image.Path);                return null;            }        }        /// <summary>        /// Attaches People DTO's to a DTOBaseItem        /// </summary>        /// <param name="dto">The dto.</param>        /// <param name="item">The item.</param>        /// <returns>Task.</returns>        private void AttachPeople(BaseItemDto dto, BaseItem item)        {            // Ordering by person type to ensure actors and artists are at the front.            // This is taking advantage of the fact that they both begin with A            // This should be improved in the future            var people = _libraryManager.GetPeople(item).OrderBy(i => i.SortOrder ?? int.MaxValue)                .ThenBy(i =>                {                    if (i.IsType(PersonType.Actor))                    {                        return 0;                    }                    if (i.IsType(PersonType.GuestStar))                    {                        return 1;                    }                    if (i.IsType(PersonType.Director))                    {                        return 2;                    }                    if (i.IsType(PersonType.Writer))                    {                        return 3;                    }                    if (i.IsType(PersonType.Producer))                    {                        return 4;                    }                    if (i.IsType(PersonType.Composer))                    {                        return 4;                    }                    return 10;                })                .ToList();            var list = new List<BaseItemPerson>();            var dictionary = people.Select(p => p.Name)                .Distinct(StringComparer.OrdinalIgnoreCase).Select(c =>                {                    try                    {                        return _libraryManager.GetPerson(c);                    }                    catch (Exception ex)                    {                        _logger.LogError(ex, "Error getting person {Name}", c);                        return null;                    }                }).Where(i => i != null)                .GroupBy(i => i.Name, StringComparer.OrdinalIgnoreCase)                .Select(x => x.First())                .ToDictionary(i => i.Name, StringComparer.OrdinalIgnoreCase);            for (var i = 0; i < people.Count; i++)            {                var person = people[i];                var baseItemPerson = new BaseItemPerson                {                    Name = person.Name,                    Role = person.Role,                    Type = person.Type                };                if (dictionary.TryGetValue(person.Name, out Person entity))                {                    baseItemPerson.PrimaryImageTag = GetImageCacheTag(entity, ImageType.Primary);                    baseItemPerson.Id = entity.Id.ToString("N", CultureInfo.InvariantCulture);                    list.Add(baseItemPerson);                }            }            dto.People = list.ToArray();        }        /// <summary>        /// Attaches the studios.        /// </summary>        /// <param name="dto">The dto.</param>        /// <param name="item">The item.</param>        /// <returns>Task.</returns>        private void AttachStudios(BaseItemDto dto, BaseItem item)        {            dto.Studios = item.Studios                .Where(i => !string.IsNullOrEmpty(i))                .Select(i => new NameGuidPair                {                    Name = i,                    Id = _libraryManager.GetStudioId(i)                })                .ToArray();        }        private void AttachGenreItems(BaseItemDto dto, BaseItem item)        {            dto.GenreItems = item.Genres                .Where(i => !string.IsNullOrEmpty(i))                .Select(i => new NameGuidPair                {                    Name = i,                    Id = GetGenreId(i, item)                })                .ToArray();        }        private Guid GetGenreId(string name, BaseItem owner)        {            if (owner is IHasMusicGenres)            {                return _libraryManager.GetMusicGenreId(name);            }            return _libraryManager.GetGenreId(name);        }        /// <summary>        /// Sets simple property values on a DTOBaseItem        /// </summary>        /// <param name="dto">The dto.</param>        /// <param name="item">The item.</param>        /// <param name="owner">The owner.</param>        /// <param name="options">The options.</param>        private void AttachBasicFields(BaseItemDto dto, BaseItem item, BaseItem owner, DtoOptions options)        {            if (options.ContainsField(ItemFields.DateCreated))            {                dto.DateCreated = item.DateCreated;            }            if (options.ContainsField(ItemFields.Settings))            {                dto.LockedFields = item.LockedFields;                dto.LockData = item.IsLocked;                dto.ForcedSortName = item.ForcedSortName;            }            dto.Container = item.Container;            dto.EndDate = item.EndDate;            if (options.ContainsField(ItemFields.ExternalUrls))            {                dto.ExternalUrls = _providerManager.GetExternalUrls(item).ToArray();            }            if (options.ContainsField(ItemFields.Tags))            {                dto.Tags = item.Tags;            }            var hasAspectRatio = item as IHasAspectRatio;            if (hasAspectRatio != null)            {                dto.AspectRatio = hasAspectRatio.AspectRatio;            }            var backdropLimit = options.GetImageLimit(ImageType.Backdrop);            if (backdropLimit > 0)            {                dto.BackdropImageTags = GetImageTags(item, item.GetImages(ImageType.Backdrop).Take(backdropLimit).ToList());            }            if (options.ContainsField(ItemFields.ScreenshotImageTags))            {                var screenshotLimit = options.GetImageLimit(ImageType.Screenshot);                if (screenshotLimit > 0)                {                    dto.ScreenshotImageTags = GetImageTags(item, item.GetImages(ImageType.Screenshot).Take(screenshotLimit).ToList());                }            }            if (options.ContainsField(ItemFields.Genres))            {                dto.Genres = item.Genres;                AttachGenreItems(dto, item);            }            if (options.EnableImages)            {                dto.ImageTags = new Dictionary<ImageType, string>();                // Prevent implicitly captured closure                var currentItem = item;                foreach (var image in currentItem.ImageInfos.Where(i => !currentItem.AllowsMultipleImages(i.Type))                    .ToList())                {                    if (options.GetImageLimit(image.Type) > 0)                    {                        var tag = GetImageCacheTag(item, image);                        if (tag != null)                        {                            dto.ImageTags[image.Type] = tag;                        }                    }                }            }            dto.Id = item.Id;            dto.IndexNumber = item.IndexNumber;            dto.ParentIndexNumber = item.ParentIndexNumber;            if (item.IsFolder)            {                dto.IsFolder = true;            }            else if (item is IHasMediaSources)            {                dto.IsFolder = false;            }            dto.MediaType = item.MediaType;            if (!(item is LiveTvProgram))            {                dto.LocationType = item.LocationType;            }            dto.Audio = item.Audio;            if (options.ContainsField(ItemFields.Settings))            {                dto.PreferredMetadataCountryCode = item.PreferredMetadataCountryCode;                dto.PreferredMetadataLanguage = item.PreferredMetadataLanguage;            }            dto.CriticRating = item.CriticRating;            if (item is IHasDisplayOrder hasDisplayOrder)            {                dto.DisplayOrder = hasDisplayOrder.DisplayOrder;            }            if (item is IHasCollectionType hasCollectionType)            {                dto.CollectionType = hasCollectionType.CollectionType;            }            if (options.ContainsField(ItemFields.RemoteTrailers))            {                dto.RemoteTrailers = item.RemoteTrailers;            }            dto.Name = item.Name;            dto.OfficialRating = item.OfficialRating;            if (options.ContainsField(ItemFields.Overview))            {                dto.Overview = item.Overview;            }            if (options.ContainsField(ItemFields.OriginalTitle))            {                dto.OriginalTitle = item.OriginalTitle;            }            if (options.ContainsField(ItemFields.ParentId))            {                dto.ParentId = item.DisplayParentId;            }            AddInheritedImages(dto, item, options, owner);            if (options.ContainsField(ItemFields.Path))            {                dto.Path = GetMappedPath(item, owner);            }            if (options.ContainsField(ItemFields.EnableMediaSourceDisplay))            {                dto.EnableMediaSourceDisplay = item.EnableMediaSourceDisplay;            }            dto.PremiereDate = item.PremiereDate;            dto.ProductionYear = item.ProductionYear;            if (options.ContainsField(ItemFields.ProviderIds))            {                dto.ProviderIds = item.ProviderIds;            }            dto.RunTimeTicks = item.RunTimeTicks;            if (options.ContainsField(ItemFields.SortName))            {                dto.SortName = item.SortName;            }            if (options.ContainsField(ItemFields.CustomRating))            {                dto.CustomRating = item.CustomRating;            }            if (options.ContainsField(ItemFields.Taglines))            {                if (!string.IsNullOrEmpty(item.Tagline))                {                    dto.Taglines = new string[] { item.Tagline };                }                if (dto.Taglines == null)                {                    dto.Taglines = Array.Empty<string>();                }            }            dto.Type = item.GetClientTypeName();            if ((item.CommunityRating ?? 0) > 0)            {                dto.CommunityRating = item.CommunityRating;            }            var supportsPlaceHolders = item as ISupportsPlaceHolders;            if (supportsPlaceHolders != null && supportsPlaceHolders.IsPlaceHolder)            {                dto.IsPlaceHolder = supportsPlaceHolders.IsPlaceHolder;            }            // Add audio info            var audio = item as Audio;            if (audio != null)            {                dto.Album = audio.Album;                if (audio.ExtraType.HasValue)                {                    dto.ExtraType = audio.ExtraType.Value.ToString();                }                var albumParent = audio.AlbumEntity;                if (albumParent != null)                {                    dto.AlbumId = albumParent.Id;                    dto.AlbumPrimaryImageTag = GetImageCacheTag(albumParent, ImageType.Primary);                }                //if (options.ContainsField(ItemFields.MediaSourceCount))                //{                // Songs always have one                //}            }            if (item is IHasArtist hasArtist)            {                dto.Artists = hasArtist.Artists;                //var artistItems = _libraryManager.GetArtists(new InternalItemsQuery                //{                //    EnableTotalRecordCount = false,                //    ItemIds = new[] { item.Id.ToString("N", CultureInfo.InvariantCulture) }                //});                //dto.ArtistItems = artistItems.Items                //    .Select(i =>                //    {                //        var artist = i.Item1;                //        return new NameIdPair                //        {                //            Name = artist.Name,                //            Id = artist.Id.ToString("N", CultureInfo.InvariantCulture)                //        };                //    })                //    .ToList();                // Include artists that are not in the database yet, e.g., just added via metadata editor                //var foundArtists = artistItems.Items.Select(i => i.Item1.Name).ToList();                dto.ArtistItems = hasArtist.Artists                    //.Except(foundArtists, new DistinctNameComparer())                    .Select(i =>                    {                        // This should not be necessary but we're seeing some cases of it                        if (string.IsNullOrEmpty(i))                        {                            return null;                        }                        var artist = _libraryManager.GetArtist(i, new DtoOptions(false)                        {                            EnableImages = false                        });                        if (artist != null)                        {                            return new NameGuidPair                            {                                Name = artist.Name,                                Id = artist.Id                            };                        }                        return null;                    }).Where(i => i != null).ToArray();            }            var hasAlbumArtist = item as IHasAlbumArtist;            if (hasAlbumArtist != null)            {                dto.AlbumArtist = hasAlbumArtist.AlbumArtists.FirstOrDefault();                //var artistItems = _libraryManager.GetAlbumArtists(new InternalItemsQuery                //{                //    EnableTotalRecordCount = false,                //    ItemIds = new[] { item.Id.ToString("N", CultureInfo.InvariantCulture) }                //});                //dto.AlbumArtists = artistItems.Items                //    .Select(i =>                //    {                //        var artist = i.Item1;                //        return new NameIdPair                //        {                //            Name = artist.Name,                //            Id = artist.Id.ToString("N", CultureInfo.InvariantCulture)                //        };                //    })                //    .ToList();                dto.AlbumArtists = hasAlbumArtist.AlbumArtists                    //.Except(foundArtists, new DistinctNameComparer())                    .Select(i =>                    {                        // This should not be necessary but we're seeing some cases of it                        if (string.IsNullOrEmpty(i))                        {                            return null;                        }                        var artist = _libraryManager.GetArtist(i, new DtoOptions(false)                        {                            EnableImages = false                        });                        if (artist != null)                        {                            return new NameGuidPair                            {                                Name = artist.Name,                                Id = artist.Id                            };                        }                        return null;                    }).Where(i => i != null).ToArray();            }            // Add video info            var video = item as Video;            if (video != null)            {                dto.VideoType = video.VideoType;                dto.Video3DFormat = video.Video3DFormat;                dto.IsoType = video.IsoType;                if (video.HasSubtitles)                {                    dto.HasSubtitles = video.HasSubtitles;                }                if (video.AdditionalParts.Length != 0)                {                    dto.PartCount = video.AdditionalParts.Length + 1;                }                if (options.ContainsField(ItemFields.MediaSourceCount))                {                    var mediaSourceCount = video.MediaSourceCount;                    if (mediaSourceCount != 1)                    {                        dto.MediaSourceCount = mediaSourceCount;                    }                }                if (options.ContainsField(ItemFields.Chapters))                {                    dto.Chapters = _itemRepo.GetChapters(item);                }                if (video.ExtraType.HasValue)                {                    dto.ExtraType = video.ExtraType.Value.ToString();                }            }            if (options.ContainsField(ItemFields.MediaStreams))            {                // Add VideoInfo                var iHasMediaSources = item as IHasMediaSources;                if (iHasMediaSources != null)                {                    MediaStream[] mediaStreams;                    if (dto.MediaSources != null && dto.MediaSources.Length > 0)                    {                        if (item.SourceType == SourceType.Channel)                        {                            mediaStreams = dto.MediaSources[0].MediaStreams.ToArray();                        }                        else                        {                            string id = item.Id.ToString("N", CultureInfo.InvariantCulture);                            mediaStreams = dto.MediaSources.Where(i => string.Equals(i.Id, id, StringComparison.OrdinalIgnoreCase))                                .SelectMany(i => i.MediaStreams)                                .ToArray();                        }                    }                    else                    {                        mediaStreams = _mediaSourceManager().GetStaticMediaSources(item, true)[0].MediaStreams.ToArray();                    }                    dto.MediaStreams = mediaStreams;                }            }            BaseItem[] allExtras = null;            if (options.ContainsField(ItemFields.SpecialFeatureCount))            {                if (allExtras == null)                {                    allExtras = item.GetExtras().ToArray();                }                dto.SpecialFeatureCount = allExtras.Count(i => i.ExtraType.HasValue && BaseItem.DisplayExtraTypes.Contains(i.ExtraType.Value));            }            if (options.ContainsField(ItemFields.LocalTrailerCount))            {                int trailerCount = 0;                if (allExtras == null)                {                    allExtras = item.GetExtras().ToArray();                }                trailerCount += allExtras.Count(i => i.ExtraType.HasValue && i.ExtraType.Value == ExtraType.Trailer);                if (item is IHasTrailers hasTrailers)                {                    trailerCount += hasTrailers.GetTrailerCount();                }                dto.LocalTrailerCount = trailerCount;            }            // Add EpisodeInfo            if (item is Episode episode)            {                dto.IndexNumberEnd = episode.IndexNumberEnd;                dto.SeriesName = episode.SeriesName;                if (options.ContainsField(ItemFields.SpecialEpisodeNumbers))                {                    dto.AirsAfterSeasonNumber = episode.AirsAfterSeasonNumber;                    dto.AirsBeforeEpisodeNumber = episode.AirsBeforeEpisodeNumber;                    dto.AirsBeforeSeasonNumber = episode.AirsBeforeSeasonNumber;                }                dto.SeasonName = episode.SeasonName;                dto.SeasonId = episode.SeasonId;                dto.SeriesId = episode.SeriesId;                Series episodeSeries = null;                // this block will add the series poster for episodes without a poster                // TODO maybe remove the if statement entirely                //if (options.ContainsField(ItemFields.SeriesPrimaryImage))                {                    episodeSeries = episodeSeries ?? episode.Series;                    if (episodeSeries != null)                    {                        dto.SeriesPrimaryImageTag = GetImageCacheTag(episodeSeries, ImageType.Primary);                    }                }                if (options.ContainsField(ItemFields.SeriesStudio))                {                    episodeSeries = episodeSeries ?? episode.Series;                    if (episodeSeries != null)                    {                        dto.SeriesStudio = episodeSeries.Studios.FirstOrDefault();                    }                }            }            // Add SeriesInfo            if (item is Series series)            {                dto.AirDays = series.AirDays;                dto.AirTime = series.AirTime;                dto.Status = series.Status.HasValue ? series.Status.Value.ToString() : null;            }            // Add SeasonInfo            if (item is Season season)            {                dto.SeriesName = season.SeriesName;                dto.SeriesId = season.SeriesId;                series = null;                if (options.ContainsField(ItemFields.SeriesStudio))                {                    series = series ?? season.Series;                    if (series != null)                    {                        dto.SeriesStudio = series.Studios.FirstOrDefault();                    }                }                // this block will add the series poster for seasons without a poster                // TODO maybe remove the if statement entirely                //if (options.ContainsField(ItemFields.SeriesPrimaryImage))                {                    series = series ?? season.Series;                    if (series != null)                    {                        dto.SeriesPrimaryImageTag = GetImageCacheTag(series, ImageType.Primary);                    }                }            }            if (item is MusicVideo musicVideo)            {                SetMusicVideoProperties(dto, musicVideo);            }            if (item is Book book)            {                SetBookProperties(dto, book);            }            if (options.ContainsField(ItemFields.ProductionLocations))            {                if (item.ProductionLocations.Length > 0 || item is Movie)                {                    dto.ProductionLocations = item.ProductionLocations;                }            }            if (options.ContainsField(ItemFields.Width))            {                var width = item.Width;                if (width > 0)                {                    dto.Width = width;                }            }            if (options.ContainsField(ItemFields.Height))            {                var height = item.Height;                if (height > 0)                {                    dto.Height = height;                }            }            if (options.ContainsField(ItemFields.IsHD))            {                // Compatibility                if (item.IsHD)                {                    dto.IsHD = true;                }            }            if (item is Photo photo)            {                SetPhotoProperties(dto, photo);            }            dto.ChannelId = item.ChannelId;            if (item.SourceType == SourceType.Channel)            {                var channel = _libraryManager.GetItemById(item.ChannelId);                if (channel != null)                {                    dto.ChannelName = channel.Name;                }            }        }        private BaseItem GetImageDisplayParent(BaseItem currentItem, BaseItem originalItem)        {            if (currentItem is MusicAlbum musicAlbum)            {                var artist = musicAlbum.GetMusicArtist(new DtoOptions(false));                if (artist != null)                {                    return artist;                }            }            var parent = currentItem.DisplayParent ?? currentItem.GetOwner() ?? currentItem.GetParent();            if (parent == null && !(originalItem is UserRootFolder) && !(originalItem is UserView) && !(originalItem is AggregateFolder) && !(originalItem is ICollectionFolder) && !(originalItem is Channel))            {                parent = _libraryManager.GetCollectionFolders(originalItem).FirstOrDefault();            }            return parent;        }        private void AddInheritedImages(BaseItemDto dto, BaseItem item, DtoOptions options, BaseItem owner)        {            if (!item.SupportsInheritedParentImages)            {                return;            }            var logoLimit = options.GetImageLimit(ImageType.Logo);            var artLimit = options.GetImageLimit(ImageType.Art);            var thumbLimit = options.GetImageLimit(ImageType.Thumb);            var backdropLimit = options.GetImageLimit(ImageType.Backdrop);            // For now. Emby apps are not using this            artLimit = 0;            if (logoLimit == 0 && artLimit == 0 && thumbLimit == 0 && backdropLimit == 0)            {                return;            }            BaseItem parent = null;            var isFirst = true;            var imageTags = dto.ImageTags;            while (((!(imageTags != null && imageTags.ContainsKey(ImageType.Logo)) && logoLimit > 0) || (!(imageTags != null && imageTags.ContainsKey(ImageType.Art)) && artLimit > 0) || (!(imageTags != null && imageTags.ContainsKey(ImageType.Thumb)) && thumbLimit > 0) || parent is Series) &&                (parent = parent ?? (isFirst ? GetImageDisplayParent(item, item) ?? owner : parent)) != null)            {                if (parent == null)                {                    break;                }                var allImages = parent.ImageInfos;                if (logoLimit > 0 && !(imageTags != null && imageTags.ContainsKey(ImageType.Logo)) && dto.ParentLogoItemId == null)                {                    var image = allImages.FirstOrDefault(i => i.Type == ImageType.Logo);                    if (image != null)                    {                        dto.ParentLogoItemId = GetDtoId(parent);                        dto.ParentLogoImageTag = GetImageCacheTag(parent, image);                    }                }                if (artLimit > 0 && !(imageTags != null && imageTags.ContainsKey(ImageType.Art)) && dto.ParentArtItemId == null)                {                    var image = allImages.FirstOrDefault(i => i.Type == ImageType.Art);                    if (image != null)                    {                        dto.ParentArtItemId = GetDtoId(parent);                        dto.ParentArtImageTag = GetImageCacheTag(parent, image);                    }                }                if (thumbLimit > 0 && !(imageTags != null && imageTags.ContainsKey(ImageType.Thumb)) && (dto.ParentThumbItemId == null || parent is Series) && !(parent is ICollectionFolder) && !(parent is UserView))                {                    var image = allImages.FirstOrDefault(i => i.Type == ImageType.Thumb);                    if (image != null)                    {                        dto.ParentThumbItemId = GetDtoId(parent);                        dto.ParentThumbImageTag = GetImageCacheTag(parent, image);                    }                }                if (backdropLimit > 0 && !((dto.BackdropImageTags != null && dto.BackdropImageTags.Length > 0) || (dto.ParentBackdropImageTags != null && dto.ParentBackdropImageTags.Length > 0)))                {                    var images = allImages.Where(i => i.Type == ImageType.Backdrop).Take(backdropLimit).ToList();                    if (images.Count > 0)                    {                        dto.ParentBackdropItemId = GetDtoId(parent);                        dto.ParentBackdropImageTags = GetImageTags(parent, images);                    }                }                isFirst = false;                if (!parent.SupportsInheritedParentImages)                {                    break;                }                parent = GetImageDisplayParent(parent, item);            }        }        private string GetMappedPath(BaseItem item, BaseItem ownerItem)        {            var path = item.Path;            if (item.IsFileProtocol)            {                path = _libraryManager.GetPathAfterNetworkSubstitution(path, ownerItem ?? item);            }            return path;        }        /// <summary>        /// Attaches the primary image aspect ratio.        /// </summary>        /// <param name="dto">The dto.</param>        /// <param name="item">The item.</param>        /// <returns>Task.</returns>        public void AttachPrimaryImageAspectRatio(IItemDto dto, BaseItem item)        {            dto.PrimaryImageAspectRatio = GetPrimaryImageAspectRatio(item);        }        public double? GetPrimaryImageAspectRatio(BaseItem item)        {            var imageInfo = item.GetImageInfo(ImageType.Primary, 0);            if (imageInfo == null)            {                return null;            }            var supportedEnhancers = _imageProcessor.GetSupportedEnhancers(item, ImageType.Primary).ToArray();            ImageDimensions size;            var defaultAspectRatio = item.GetDefaultPrimaryImageAspectRatio();            if (defaultAspectRatio > 0)            {                if (supportedEnhancers.Length == 0)                {                    return defaultAspectRatio;                }                int dummyWidth = 200;                int dummyHeight = Convert.ToInt32(dummyWidth / defaultAspectRatio);                size = new ImageDimensions(dummyWidth, dummyHeight);            }            else            {                if (!imageInfo.IsLocalFile)                {                    return null;                }                try                {                    size = _imageProcessor.GetImageDimensions(item, imageInfo);                    if (size.Width <= 0 || size.Height <= 0)                    {                        return null;                    }                }                catch (Exception ex)                {                    _logger.LogError(ex, "Failed to determine primary image aspect ratio for {0}", imageInfo.Path);                    return null;                }            }            foreach (var enhancer in supportedEnhancers)            {                try                {                    size = enhancer.GetEnhancedImageSize(item, ImageType.Primary, 0, size);                }                catch (Exception ex)                {                    _logger.LogError(ex, "Error in image enhancer: {0}", enhancer.GetType().Name);                }            }            var width = size.Width;            var height = size.Height;            if (width <= 0 || height <= 0)            {                return null;            }            return width / height;        }    }}
 |