TmdbSeasonImageProvider.cs 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  1. #pragma warning disable CS1591
  2. using System;
  3. using System.Collections.Generic;
  4. using System.Globalization;
  5. using System.Linq;
  6. using System.Net.Http;
  7. using System.Threading;
  8. using System.Threading.Tasks;
  9. using MediaBrowser.Common.Net;
  10. using MediaBrowser.Controller.Entities;
  11. using MediaBrowser.Controller.Entities.TV;
  12. using MediaBrowser.Controller.Providers;
  13. using MediaBrowser.Model.Dto;
  14. using MediaBrowser.Model.Entities;
  15. using MediaBrowser.Model.Providers;
  16. namespace MediaBrowser.Providers.Plugins.Tmdb.TV
  17. {
  18. public class TmdbSeasonImageProvider : IRemoteImageProvider, IHasOrder
  19. {
  20. private readonly IHttpClientFactory _httpClientFactory;
  21. private readonly TmdbClientManager _tmdbClientManager;
  22. public TmdbSeasonImageProvider(IHttpClientFactory httpClientFactory, TmdbClientManager tmdbClientManager)
  23. {
  24. _httpClientFactory = httpClientFactory;
  25. _tmdbClientManager = tmdbClientManager;
  26. }
  27. public int Order => 1;
  28. public string Name => TmdbUtils.ProviderName;
  29. public Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
  30. {
  31. return _httpClientFactory.CreateClient(NamedClient.Default).GetAsync(url, cancellationToken);
  32. }
  33. public async Task<IEnumerable<RemoteImageInfo>> GetImages(BaseItem item, CancellationToken cancellationToken)
  34. {
  35. var season = (Season)item;
  36. var series = season?.Series;
  37. var seriesTmdbId = Convert.ToInt32(series?.GetProviderId(MetadataProvider.Tmdb), CultureInfo.InvariantCulture);
  38. if (seriesTmdbId <= 0 || season?.IndexNumber == null)
  39. {
  40. return Enumerable.Empty<RemoteImageInfo>();
  41. }
  42. var language = item.GetPreferredMetadataLanguage();
  43. // TODO use image languages if All Languages isn't toggled, but there's currently no way to get that value in here
  44. var seasonResult = await _tmdbClientManager
  45. .GetSeasonAsync(seriesTmdbId, season.IndexNumber.Value, null, null, cancellationToken)
  46. .ConfigureAwait(false);
  47. var posters = seasonResult?.Images?.Posters;
  48. if (posters == null)
  49. {
  50. return Enumerable.Empty<RemoteImageInfo>();
  51. }
  52. var remoteImages = new RemoteImageInfo[posters.Count];
  53. for (var i = 0; i < posters.Count; i++)
  54. {
  55. var image = posters[i];
  56. remoteImages[i] = new RemoteImageInfo
  57. {
  58. Url = _tmdbClientManager.GetPosterUrl(image.FilePath),
  59. CommunityRating = image.VoteAverage,
  60. VoteCount = image.VoteCount,
  61. Width = image.Width,
  62. Height = image.Height,
  63. Language = TmdbUtils.AdjustImageLanguage(image.Iso_639_1, language),
  64. ProviderName = Name,
  65. Type = ImageType.Primary,
  66. RatingType = RatingType.Score
  67. };
  68. }
  69. return remoteImages;
  70. }
  71. public IEnumerable<ImageType> GetSupportedImages(BaseItem item)
  72. {
  73. return new List<ImageType>
  74. {
  75. ImageType.Primary
  76. };
  77. }
  78. public bool Supports(BaseItem item)
  79. {
  80. return item is Season;
  81. }
  82. }
  83. }