TmdbSeriesImageProvider.cs 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188
  1. #pragma warning disable CS1591
  2. using System;
  3. using System.Collections.Generic;
  4. using System.Linq;
  5. using System.Net.Http;
  6. using System.Threading;
  7. using System.Threading.Tasks;
  8. using MediaBrowser.Controller.Entities;
  9. using MediaBrowser.Controller.Entities.TV;
  10. using MediaBrowser.Controller.Providers;
  11. using MediaBrowser.Model.Dto;
  12. using MediaBrowser.Model.Entities;
  13. using MediaBrowser.Model.IO;
  14. using MediaBrowser.Model.Providers;
  15. using MediaBrowser.Model.Serialization;
  16. using MediaBrowser.Providers.Plugins.Tmdb.Models.General;
  17. using MediaBrowser.Providers.Plugins.Tmdb.Models.TV;
  18. using MediaBrowser.Providers.Plugins.Tmdb.Movies;
  19. namespace MediaBrowser.Providers.Plugins.Tmdb.TV
  20. {
  21. public class TmdbSeriesImageProvider : IRemoteImageProvider, IHasOrder
  22. {
  23. private readonly IJsonSerializer _jsonSerializer;
  24. private readonly IHttpClientFactory _httpClientFactory;
  25. private readonly IFileSystem _fileSystem;
  26. public TmdbSeriesImageProvider(IJsonSerializer jsonSerializer, IHttpClientFactory httpClientFactory, IFileSystem fileSystem)
  27. {
  28. _jsonSerializer = jsonSerializer;
  29. _httpClientFactory = httpClientFactory;
  30. _fileSystem = fileSystem;
  31. }
  32. public string Name => ProviderName;
  33. public static string ProviderName => TmdbUtils.ProviderName;
  34. public bool Supports(BaseItem item)
  35. {
  36. return item is Series;
  37. }
  38. public IEnumerable<ImageType> GetSupportedImages(BaseItem item)
  39. {
  40. return new List<ImageType>
  41. {
  42. ImageType.Primary,
  43. ImageType.Backdrop
  44. };
  45. }
  46. public async Task<IEnumerable<RemoteImageInfo>> GetImages(BaseItem item, CancellationToken cancellationToken)
  47. {
  48. var list = new List<RemoteImageInfo>();
  49. var results = await FetchImages(item, null, _jsonSerializer, cancellationToken).ConfigureAwait(false);
  50. if (results == null)
  51. {
  52. return list;
  53. }
  54. var tmdbSettings = await TmdbMovieProvider.Current.GetTmdbSettings(cancellationToken).ConfigureAwait(false);
  55. var tmdbImageUrl = tmdbSettings.images.GetImageUrl("original");
  56. var language = item.GetPreferredMetadataLanguage();
  57. list.AddRange(GetPosters(results).Select(i => new RemoteImageInfo
  58. {
  59. Url = tmdbImageUrl + i.File_Path,
  60. CommunityRating = i.Vote_Average,
  61. VoteCount = i.Vote_Count,
  62. Width = i.Width,
  63. Height = i.Height,
  64. Language = TmdbMovieProvider.AdjustImageLanguage(i.Iso_639_1, language),
  65. ProviderName = Name,
  66. Type = ImageType.Primary,
  67. RatingType = RatingType.Score
  68. }));
  69. list.AddRange(GetBackdrops(results).Select(i => new RemoteImageInfo
  70. {
  71. Url = tmdbImageUrl + i.File_Path,
  72. CommunityRating = i.Vote_Average,
  73. VoteCount = i.Vote_Count,
  74. Width = i.Width,
  75. Height = i.Height,
  76. ProviderName = Name,
  77. Type = ImageType.Backdrop,
  78. RatingType = RatingType.Score
  79. }));
  80. var isLanguageEn = string.Equals(language, "en", StringComparison.OrdinalIgnoreCase);
  81. return list.OrderByDescending(i =>
  82. {
  83. if (string.Equals(language, i.Language, StringComparison.OrdinalIgnoreCase))
  84. {
  85. return 3;
  86. }
  87. if (!isLanguageEn)
  88. {
  89. if (string.Equals("en", i.Language, StringComparison.OrdinalIgnoreCase))
  90. {
  91. return 2;
  92. }
  93. }
  94. if (string.IsNullOrEmpty(i.Language))
  95. {
  96. return isLanguageEn ? 3 : 2;
  97. }
  98. return 0;
  99. })
  100. .ThenByDescending(i => i.CommunityRating ?? 0)
  101. .ThenByDescending(i => i.VoteCount ?? 0);
  102. }
  103. /// <summary>
  104. /// Gets the posters.
  105. /// </summary>
  106. /// <param name="images">The images.</param>
  107. private IEnumerable<Poster> GetPosters(Images images)
  108. {
  109. return images.Posters ?? new List<Poster>();
  110. }
  111. /// <summary>
  112. /// Gets the backdrops.
  113. /// </summary>
  114. /// <param name="images">The images.</param>
  115. private IEnumerable<Backdrop> GetBackdrops(Images images)
  116. {
  117. var eligibleBackdrops = images.Backdrops ?? new List<Backdrop>();
  118. return eligibleBackdrops.OrderByDescending(i => i.Vote_Average)
  119. .ThenByDescending(i => i.Vote_Count);
  120. }
  121. /// <summary>
  122. /// Fetches the images.
  123. /// </summary>
  124. /// <param name="item">The item.</param>
  125. /// <param name="language">The language.</param>
  126. /// <param name="jsonSerializer">The json serializer.</param>
  127. /// <param name="cancellationToken">The cancellation token.</param>
  128. /// <returns>Task{MovieImages}.</returns>
  129. private async Task<Images> FetchImages(BaseItem item, string language, IJsonSerializer jsonSerializer,
  130. CancellationToken cancellationToken)
  131. {
  132. var tmdbId = item.GetProviderId(MetadataProvider.Tmdb);
  133. if (string.IsNullOrEmpty(tmdbId))
  134. {
  135. return null;
  136. }
  137. await TmdbSeriesProvider.Current.EnsureSeriesInfo(tmdbId, language, cancellationToken).ConfigureAwait(false);
  138. var path = TmdbSeriesProvider.Current.GetDataFilePath(tmdbId, language);
  139. if (!string.IsNullOrEmpty(path))
  140. {
  141. var fileInfo = _fileSystem.GetFileInfo(path);
  142. if (fileInfo.Exists)
  143. {
  144. return jsonSerializer.DeserializeFromFile<SeriesResult>(path).Images;
  145. }
  146. }
  147. return null;
  148. }
  149. // After tvdb and fanart
  150. public int Order => 2;
  151. public Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
  152. {
  153. return _httpClientFactory.CreateClient().GetAsync(url, cancellationToken);
  154. }
  155. }
  156. }