using MediaBrowser.Common.Extensions; 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.Persistence; using MediaBrowser.Model.Drawing; using MediaBrowser.Model.Dto; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Logging; using MediaBrowser.Model.Querying; using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Threading.Tasks; namespace MediaBrowser.Controller.Dto { /// /// Generates DTO's from domain entities /// public class DtoBuilder { /// /// The index folder delimeter /// const string IndexFolderDelimeter = "-index-"; private readonly ILogger _logger; private readonly ILibraryManager _libraryManager; private readonly IUserDataRepository _userDataRepository; public DtoBuilder(ILogger logger, ILibraryManager libraryManager, IUserDataRepository userDataRepository) { _logger = logger; _libraryManager = libraryManager; _userDataRepository = userDataRepository; } /// /// Gets the dto base item. /// /// The item. /// The fields. /// Task{DtoBaseItem}. /// item public async Task GetBaseItemDto(BaseItem item, List fields) { if (item == null) { throw new ArgumentNullException("item"); } if (fields == null) { throw new ArgumentNullException("fields"); } var dto = new BaseItemDto(); var tasks = new List(); if (fields.Contains(ItemFields.Studios)) { dto.Studios = item.Studios; } if (fields.Contains(ItemFields.People)) { tasks.Add(AttachPeople(dto, item)); } if (fields.Contains(ItemFields.PrimaryImageAspectRatio)) { try { await AttachPrimaryImageAspectRatio(dto, item, _logger).ConfigureAwait(false); } catch (Exception ex) { // Have to use a catch-all unfortunately because some .net image methods throw plain Exceptions _logger.ErrorException("Error generating PrimaryImageAspectRatio for {0}", ex, item.Name); } } AttachBasicFields(dto, item, fields); // Make sure all the tasks we kicked off have completed. if (tasks.Count > 0) { await Task.WhenAll(tasks).ConfigureAwait(false); } return dto; } /// /// Converts a BaseItem to a DTOBaseItem /// /// The item. /// The user. /// The fields. /// Task{DtoBaseItem}. /// item public async Task GetBaseItemDto(BaseItem item, User user, List fields) { if (item == null) { throw new ArgumentNullException("item"); } if (user == null) { throw new ArgumentNullException("user"); } if (fields == null) { throw new ArgumentNullException("fields"); } var dto = new BaseItemDto(); var tasks = new List(); if (fields.Contains(ItemFields.Studios)) { dto.Studios = item.Studios; } if (fields.Contains(ItemFields.People)) { tasks.Add(AttachPeople(dto, item)); } tasks.Add(AttachUserSpecificInfo(dto, item, user, fields)); if (fields.Contains(ItemFields.PrimaryImageAspectRatio)) { try { await AttachPrimaryImageAspectRatio(dto, item, _logger).ConfigureAwait(false); } catch (Exception ex) { // Have to use a catch-all unfortunately because some .net image methods throw plain Exceptions _logger.ErrorException("Error generating PrimaryImageAspectRatio for {0}", ex, item.Name); } } AttachBasicFields(dto, item, fields); // Make sure all the tasks we kicked off have completed. if (tasks.Count > 0) { await Task.WhenAll(tasks).ConfigureAwait(false); } return dto; } /// /// Attaches the user specific info. /// /// The dto. /// The item. /// The user. /// The fields. private async Task AttachUserSpecificInfo(BaseItemDto dto, BaseItem item, User user, List fields) { if (fields.Contains(ItemFields.UserData)) { var userData = await _userDataRepository.GetUserData(user.Id, item.GetUserDataKey()).ConfigureAwait(false); dto.UserData = GetUserItemDataDto(userData); } if (item.IsFolder && fields.Contains(ItemFields.DisplayPreferencesId)) { dto.DisplayPreferencesId = ((Folder) item).GetDisplayPreferencesId(user.Id).ToString(); } if (item.IsFolder) { if (fields.Contains(ItemFields.ItemCounts)) { var folder = (Folder)item; // Skip sorting since all we want is a count dto.ChildCount = folder.GetChildren(user).Count(); await SetSpecialCounts(folder, user, dto, _userDataRepository).ConfigureAwait(false); } } } /// /// Attaches the primary image aspect ratio. /// /// The dto. /// The item. /// The _logger. /// Task. internal static async Task AttachPrimaryImageAspectRatio(IItemDto dto, BaseItem item, ILogger logger) { var path = item.PrimaryImagePath; if (string.IsNullOrEmpty(path)) { return; } var metaFileEntry = item.ResolveArgs.GetMetaFileByPath(path); // See if we can avoid a file system lookup by looking for the file in ResolveArgs var dateModified = metaFileEntry == null ? File.GetLastWriteTimeUtc(path) : metaFileEntry.Value.LastWriteTimeUtc; ImageSize size; try { size = await Kernel.Instance.ImageManager.GetImageSize(path, dateModified).ConfigureAwait(false); } catch (FileNotFoundException) { logger.Error("Image file does not exist: {0}", path); return; } catch (Exception ex) { logger.ErrorException("Failed to determine primary image aspect ratio for {0}", ex, path); return; } foreach (var enhancer in Kernel.Instance.ImageEnhancers .Where(i => i.Supports(item, ImageType.Primary))) { size = enhancer.GetEnhancedImageSize(item, ImageType.Primary, 0, size); } dto.PrimaryImageAspectRatio = size.Width / size.Height; } /// /// Sets simple property values on a DTOBaseItem /// /// The dto. /// The item. /// The fields. private void AttachBasicFields(BaseItemDto dto, BaseItem item, List fields) { if (fields.Contains(ItemFields.DateCreated)) { dto.DateCreated = item.DateCreated; } if (fields.Contains(ItemFields.DisplayMediaType)) { dto.DisplayMediaType = item.DisplayMediaType; } if (fields.Contains(ItemFields.Budget)) { dto.Budget = item.Budget; } if (fields.Contains(ItemFields.Revenue)) { dto.Revenue = item.Revenue; } if (fields.Contains(ItemFields.EndDate)) { dto.EndDate = item.EndDate; } if (fields.Contains(ItemFields.HomePageUrl)) { dto.HomePageUrl = item.HomePageUrl; } if (fields.Contains(ItemFields.ProductionLocations)) { dto.ProductionLocations = item.ProductionLocations; } dto.AspectRatio = item.AspectRatio; dto.BackdropImageTags = GetBackdropImageTags(item); if (fields.Contains(ItemFields.Genres)) { dto.Genres = item.Genres; } if (item.Images != null) { dto.ImageTags = new Dictionary(); foreach (var image in item.Images) { ImageType type; if (Enum.TryParse(image.Key, true, out type)) { dto.ImageTags[type] = Kernel.Instance.ImageManager.GetImageCacheTag(item, type, image.Value); } } } dto.Id = GetClientItemId(item); dto.IndexNumber = item.IndexNumber; dto.IsFolder = item.IsFolder; dto.Language = item.Language; dto.MediaType = item.MediaType; dto.LocationType = item.LocationType; var localTrailerCount = item.LocalTrailers == null ? 0 : item.LocalTrailers.Count; if (localTrailerCount > 0) { dto.LocalTrailerCount = localTrailerCount; } dto.Name = item.Name; dto.OfficialRating = item.OfficialRating; var hasOverview = fields.Contains(ItemFields.Overview); var hasHtmlOverview = fields.Contains(ItemFields.OverviewHtml); if (hasOverview || hasHtmlOverview) { var strippedOverview = string.IsNullOrEmpty(item.Overview) ? item.Overview : item.Overview.StripHtml(); if (hasOverview) { dto.Overview = strippedOverview; } // Only supply the html version if there was actually html content if (hasHtmlOverview) { dto.OverviewHtml = item.Overview; } } // If there are no backdrops, indicate what parent has them in case the Ui wants to allow inheritance if (dto.BackdropImageTags.Count == 0) { var parentWithBackdrop = GetParentBackdropItem(item); if (parentWithBackdrop != null) { dto.ParentBackdropItemId = GetClientItemId(parentWithBackdrop); dto.ParentBackdropImageTags = GetBackdropImageTags(parentWithBackdrop); } } if (item.Parent != null && fields.Contains(ItemFields.ParentId)) { dto.ParentId = GetClientItemId(item.Parent); } dto.ParentIndexNumber = item.ParentIndexNumber; // If there is no logo, indicate what parent has one in case the Ui wants to allow inheritance if (!dto.HasLogo) { var parentWithLogo = GetParentLogoItem(item); if (parentWithLogo != null) { dto.ParentLogoItemId = GetClientItemId(parentWithLogo); dto.ParentLogoImageTag = Kernel.Instance.ImageManager.GetImageCacheTag(parentWithLogo, ImageType.Logo, parentWithLogo.GetImage(ImageType.Logo)); } } if (fields.Contains(ItemFields.Path)) { dto.Path = item.Path; } dto.PremiereDate = item.PremiereDate; dto.ProductionYear = item.ProductionYear; if (fields.Contains(ItemFields.ProviderIds)) { dto.ProviderIds = item.ProviderIds; } dto.RunTimeTicks = item.RunTimeTicks; if (fields.Contains(ItemFields.SortName)) { dto.SortName = item.SortName; } if (fields.Contains(ItemFields.Taglines)) { dto.Taglines = item.Taglines; } if (fields.Contains(ItemFields.TrailerUrls)) { dto.TrailerUrls = item.TrailerUrls; } dto.Type = item.GetType().Name; dto.CommunityRating = item.CommunityRating; if (item.IsFolder) { var folder = (Folder)item; if (fields.Contains(ItemFields.IndexOptions)) { dto.IndexOptions = folder.IndexByOptionStrings.ToArray(); } } // Add audio info if (fields.Contains(ItemFields.AudioInfo)) { var audio = item as Audio; if (audio != null) { dto.Album = audio.Album; dto.AlbumArtist = audio.AlbumArtist; dto.Artist = audio.Artist; } var album = item as MusicAlbum; if (album != null) { var songs = album.Children.OfType