OmdbImageProvider.cs 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. using MediaBrowser.Common.Net;
  2. using MediaBrowser.Controller.Entities;
  3. using MediaBrowser.Controller.Entities.Movies;
  4. using MediaBrowser.Controller.Entities.TV;
  5. using MediaBrowser.Controller.LiveTv;
  6. using MediaBrowser.Controller.Providers;
  7. using MediaBrowser.Model.Entities;
  8. using MediaBrowser.Model.Providers;
  9. using System.Collections.Generic;
  10. using System.IO;
  11. using System.Text;
  12. using System.Threading;
  13. using System.Threading.Tasks;
  14. namespace MediaBrowser.Providers.Omdb
  15. {
  16. public class OmdbImageProvider : IRemoteImageProvider, IHasOrder
  17. {
  18. private readonly IHttpClient _httpClient;
  19. public OmdbImageProvider(IHttpClient httpClient)
  20. {
  21. _httpClient = httpClient;
  22. }
  23. public IEnumerable<ImageType> GetSupportedImages(IHasImages item)
  24. {
  25. return new List<ImageType>
  26. {
  27. ImageType.Primary
  28. };
  29. }
  30. public async Task<IEnumerable<RemoteImageInfo>> GetImages(IHasImages item, CancellationToken cancellationToken)
  31. {
  32. var imdbId = item.GetProviderId(MetadataProviders.Imdb);
  33. var list = new List<RemoteImageInfo>();
  34. if (!string.IsNullOrWhiteSpace(imdbId) && OmdbProvider.Current != null)
  35. {
  36. OmdbProvider.RootObject rootObject = await OmdbProvider.Current.GetRootObject(imdbId, cancellationToken);
  37. if (!string.IsNullOrEmpty(rootObject.Poster))
  38. {
  39. list.Add(new RemoteImageInfo
  40. {
  41. ProviderName = Name,
  42. Url = string.Format("https://img.omdbapi.com/?i={0}&apikey=82e83907", imdbId)
  43. });
  44. }
  45. }
  46. return list;
  47. }
  48. public Task<HttpResponseInfo> GetImageResponse(string url, CancellationToken cancellationToken)
  49. {
  50. return _httpClient.GetResponse(new HttpRequestOptions
  51. {
  52. CancellationToken = cancellationToken,
  53. Url = url,
  54. ResourcePool = OmdbProvider.ResourcePool
  55. });
  56. }
  57. public string Name
  58. {
  59. get { return "The Open Movie Database"; }
  60. }
  61. public bool Supports(IHasImages item)
  62. {
  63. // Supports images for tv movies
  64. var tvProgram = item as LiveTvProgram;
  65. if (tvProgram != null && tvProgram.IsMovie)
  66. {
  67. return true;
  68. }
  69. return item is Movie || item is Trailer || item is Episode;
  70. }
  71. public int Order
  72. {
  73. get
  74. {
  75. // After other internet providers, because they're better
  76. // But before fallback providers like screengrab
  77. return 90;
  78. }
  79. }
  80. }
  81. }