TvdbPersonImageProvider.cs 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Threading;
  5. using System.Threading.Tasks;
  6. using MediaBrowser.Common.Net;
  7. using MediaBrowser.Controller.Dto;
  8. using MediaBrowser.Controller.Entities;
  9. using MediaBrowser.Controller.Entities.TV;
  10. using MediaBrowser.Controller.Library;
  11. using MediaBrowser.Controller.Providers;
  12. using MediaBrowser.Model.Entities;
  13. using MediaBrowser.Model.Providers;
  14. using MediaBrowser.Providers.TV.TheTVDB;
  15. using Microsoft.Extensions.Logging;
  16. using TvDbSharper;
  17. namespace MediaBrowser.Providers.People
  18. {
  19. public class TvdbPersonImageProvider : IRemoteImageProvider, IHasOrder
  20. {
  21. private readonly IHttpClient _httpClient;
  22. private readonly ILogger _logger;
  23. private readonly ILibraryManager _libraryManager;
  24. private readonly TvDbClientManager _tvDbClientManager;
  25. public TvdbPersonImageProvider(ILibraryManager libraryManager, IHttpClient httpClient, ILogger logger)
  26. {
  27. _libraryManager = libraryManager;
  28. _httpClient = httpClient;
  29. _logger = logger;
  30. _tvDbClientManager = TvDbClientManager.Instance;
  31. }
  32. public string Name => ProviderName;
  33. public static string ProviderName => "TheTVDB";
  34. public bool Supports(BaseItem item)
  35. {
  36. return item is Person;
  37. }
  38. public IEnumerable<ImageType> GetSupportedImages(BaseItem item)
  39. {
  40. return new List<ImageType>
  41. {
  42. ImageType.Primary
  43. };
  44. }
  45. public async Task<IEnumerable<RemoteImageInfo>> GetImages(BaseItem item, CancellationToken cancellationToken)
  46. {
  47. var seriesWithPerson = _libraryManager.GetItemList(new InternalItemsQuery
  48. {
  49. IncludeItemTypes = new[] { typeof(Series).Name },
  50. PersonIds = new[] { item.Id },
  51. DtoOptions = new DtoOptions(false)
  52. {
  53. EnableImages = false
  54. }
  55. }).Cast<Series>()
  56. .Where(i => TvdbSeriesProvider.IsValidSeries(i.ProviderIds))
  57. .ToList();
  58. var infos = (await Task.WhenAll(seriesWithPerson.Select(async i => await GetImageFromSeriesData(i, item.Name, cancellationToken))))
  59. .Where(i => i != null)
  60. .Take(1);
  61. return infos;
  62. }
  63. private async Task<RemoteImageInfo> GetImageFromSeriesData(Series series, string personName, CancellationToken cancellationToken)
  64. {
  65. var tvdbId = Convert.ToInt32(series.GetProviderId(MetadataProviders.Tvdb));
  66. try
  67. {
  68. var actorsResult = await _tvDbClientManager.GetActorsAsync(tvdbId, series.GetPreferredMetadataLanguage(), cancellationToken);
  69. var actor = actorsResult.Data.FirstOrDefault(a =>
  70. string.Equals(a.Name, personName, StringComparison.OrdinalIgnoreCase) &&
  71. !string.IsNullOrEmpty(a.Image));
  72. if (actor == null)
  73. {
  74. return null;
  75. }
  76. return new RemoteImageInfo
  77. {
  78. Url = TvdbUtils.BannerUrl + actor.Image,
  79. Type = ImageType.Primary,
  80. ProviderName = Name
  81. };
  82. }
  83. catch (TvDbServerException e)
  84. {
  85. _logger.LogError(e, "Failed to retrieve actor {ActorName} from series {SeriesTvdbId}", personName, tvdbId);
  86. return null;
  87. }
  88. }
  89. public int Order => 1;
  90. public Task<HttpResponseInfo> GetImageResponse(string url, CancellationToken cancellationToken)
  91. {
  92. return _httpClient.GetResponse(new HttpRequestOptions
  93. {
  94. CancellationToken = cancellationToken,
  95. Url = url
  96. });
  97. }
  98. }
  99. }