using System; using System.Collections.Generic; using System.Globalization; using System.Linq; using System.Threading; using System.Threading.Tasks; using MediaBrowser.Common.Net; using MediaBrowser.Controller.Entities; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Providers; using MediaBrowser.Model.Entities; using MediaBrowser.Model.Providers; using Microsoft.Extensions.Logging; using TvDbSharper; using TvDbSharper.Dto; using RatingType = MediaBrowser.Model.Dto.RatingType; using Series = MediaBrowser.Controller.Entities.TV.Series; namespace MediaBrowser.Providers.TV.TheTVDB { public class TvdbSeriesImageProvider : IRemoteImageProvider, IHasOrder { private readonly IHttpClient _httpClient; private readonly ILogger _logger; private readonly TvDbClientManager _tvDbClientManager; public TvdbSeriesImageProvider(IHttpClient httpClient, ILogger logger) { _httpClient = httpClient; _logger = logger; _tvDbClientManager = TvDbClientManager.Instance; } public string Name => ProviderName; public static string ProviderName => "TheTVDB"; public bool Supports(BaseItem item) { return item is Series; } public IEnumerable GetSupportedImages(BaseItem item) { return new List { ImageType.Primary, ImageType.Banner, ImageType.Backdrop }; } public async Task> GetImages(BaseItem item, CancellationToken cancellationToken) { if (!TvdbSeriesProvider.IsValidSeries(item.ProviderIds)) { return new RemoteImageInfo[] { }; } var language = item.GetPreferredMetadataLanguage(); _tvDbClientManager.TvDbClient.AcceptedLanguage = language; var remoteImages = new List(); var keyTypes = new[] {KeyType.Poster, KeyType.Series, KeyType.Fanart}; var tvdbId = Convert.ToInt32(item.GetProviderId(MetadataProviders.Tvdb)); foreach (KeyType keyType in keyTypes) { var imageQuery = new ImagesQuery { KeyType = keyType }; try { var imageResults = await _tvDbClientManager.GetImagesAsync(tvdbId, imageQuery, cancellationToken); remoteImages.AddRange(GetImages(imageResults.Data, language)); } catch (TvDbServerException e) { _logger.LogError(e, "Failed to retrieve images of type {KeyType} for series {TvDbId}", keyType, tvdbId); } } return remoteImages; } private IEnumerable GetImages(Image[] images, string preferredLanguage) { var list = new List(); var languages = TvDbClientManager.Instance.GetLanguagesAsync(CancellationToken.None).Result.Data; foreach (Image image in images) { var imageInfo = new RemoteImageInfo { RatingType = RatingType.Score, CommunityRating = (double?)image.RatingsInfo.Average, VoteCount = image.RatingsInfo.Count, Url = TVUtils.BannerUrl + image.FileName, ProviderName = Name, Language = languages.FirstOrDefault(l => l.Id == image.LanguageId)?.Abbreviation, ThumbnailUrl = TVUtils.BannerUrl + image.Thumbnail }; var resolution = image.Resolution.Split('x'); if (resolution.Length == 2) { imageInfo.Width = Convert.ToInt32(resolution[0]); imageInfo.Height = Convert.ToInt32(resolution[1]); } if (string.Equals(image.KeyType, "poster", StringComparison.OrdinalIgnoreCase)) { imageInfo.Type = ImageType.Primary; } else if (string.Equals(image.KeyType, "series", StringComparison.OrdinalIgnoreCase)) { imageInfo.Type = ImageType.Banner; } else if (string.Equals(image.KeyType, "fanart", StringComparison.OrdinalIgnoreCase)) { imageInfo.Type = ImageType.Backdrop; } list.Add(imageInfo); } var isLanguageEn = string.Equals(preferredLanguage, "en", StringComparison.OrdinalIgnoreCase); return list.OrderByDescending(i => { if (string.Equals(preferredLanguage, i.Language, StringComparison.OrdinalIgnoreCase)) { return 3; } if (!isLanguageEn) { if (string.Equals("en", i.Language, StringComparison.OrdinalIgnoreCase)) { return 2; } } if (string.IsNullOrEmpty(i.Language)) { return isLanguageEn ? 3 : 2; } return 0; }) .ThenByDescending(i => i.CommunityRating ?? 0) .ThenByDescending(i => i.VoteCount ?? 0); } public int Order => 0; public Task GetImageResponse(string url, CancellationToken cancellationToken) { return _httpClient.GetResponse(new HttpRequestOptions { CancellationToken = cancellationToken, Url = url }); } } }