TmdbSeasonImageProvider.cs 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  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.Entities;
  14. using MediaBrowser.Model.Providers;
  15. namespace MediaBrowser.Providers.Plugins.Tmdb.TV
  16. {
  17. public class TmdbSeasonImageProvider : IRemoteImageProvider, IHasOrder
  18. {
  19. private readonly IHttpClientFactory _httpClientFactory;
  20. private readonly TmdbClientManager _tmdbClientManager;
  21. public TmdbSeasonImageProvider(IHttpClientFactory httpClientFactory, TmdbClientManager tmdbClientManager)
  22. {
  23. _httpClientFactory = httpClientFactory;
  24. _tmdbClientManager = tmdbClientManager;
  25. }
  26. public int Order => 1;
  27. public string Name => TmdbUtils.ProviderName;
  28. public Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
  29. {
  30. return _httpClientFactory.CreateClient(NamedClient.Default).GetAsync(url, cancellationToken);
  31. }
  32. public async Task<IEnumerable<RemoteImageInfo>> GetImages(BaseItem item, CancellationToken cancellationToken)
  33. {
  34. var season = (Season)item;
  35. var series = season?.Series;
  36. var seriesTmdbId = Convert.ToInt32(series?.GetProviderId(MetadataProvider.Tmdb), CultureInfo.InvariantCulture);
  37. if (seriesTmdbId <= 0 || season?.IndexNumber == null)
  38. {
  39. return Enumerable.Empty<RemoteImageInfo>();
  40. }
  41. var language = item.GetPreferredMetadataLanguage();
  42. // TODO use image languages if All Languages isn't toggled, but there's currently no way to get that value in here
  43. var seasonResult = await _tmdbClientManager
  44. .GetSeasonAsync(seriesTmdbId, season.IndexNumber.Value, null, null, cancellationToken)
  45. .ConfigureAwait(false);
  46. var posters = seasonResult?.Images?.Posters;
  47. if (posters == null)
  48. {
  49. return Enumerable.Empty<RemoteImageInfo>();
  50. }
  51. var remoteImages = new List<RemoteImageInfo>(posters.Count);
  52. _tmdbClientManager.ConvertPostersToRemoteImageInfo(posters, language, remoteImages);
  53. return remoteImages;
  54. }
  55. public IEnumerable<ImageType> GetSupportedImages(BaseItem item)
  56. {
  57. return new List<ImageType>
  58. {
  59. ImageType.Primary
  60. };
  61. }
  62. public bool Supports(BaseItem item)
  63. {
  64. return item is Season;
  65. }
  66. }
  67. }