OmdbEpisodeProvider.cs 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. #pragma warning disable CS1591
  2. using System.Collections.Generic;
  3. using System.Net.Http;
  4. using System.Threading;
  5. using System.Threading.Tasks;
  6. using MediaBrowser.Controller.Configuration;
  7. using MediaBrowser.Controller.Entities.TV;
  8. using MediaBrowser.Controller.Library;
  9. using MediaBrowser.Controller.Providers;
  10. using MediaBrowser.Model.Entities;
  11. using MediaBrowser.Model.IO;
  12. using MediaBrowser.Model.Providers;
  13. namespace MediaBrowser.Providers.Plugins.Omdb
  14. {
  15. public class OmdbEpisodeProvider : IRemoteMetadataProvider<Episode, EpisodeInfo>, IHasOrder
  16. {
  17. private readonly OmdbItemProvider _itemProvider;
  18. private readonly OmdbProvider _omdbProvider;
  19. public OmdbEpisodeProvider(
  20. IHttpClientFactory httpClientFactory,
  21. ILibraryManager libraryManager,
  22. IFileSystem fileSystem,
  23. IServerConfigurationManager configurationManager)
  24. {
  25. _itemProvider = new OmdbItemProvider(httpClientFactory, libraryManager, fileSystem, configurationManager);
  26. _omdbProvider = new OmdbProvider(httpClientFactory, fileSystem, configurationManager);
  27. }
  28. // After TheTvDb
  29. public int Order => 1;
  30. public string Name => "The Open Movie Database";
  31. public Task<IEnumerable<RemoteSearchResult>> GetSearchResults(EpisodeInfo searchInfo, CancellationToken cancellationToken)
  32. {
  33. return _itemProvider.GetSearchResults(searchInfo, cancellationToken);
  34. }
  35. public async Task<MetadataResult<Episode>> GetMetadata(EpisodeInfo info, CancellationToken cancellationToken)
  36. {
  37. var result = new MetadataResult<Episode>
  38. {
  39. Item = new Episode(),
  40. QueriedById = true
  41. };
  42. // Allowing this will dramatically increase scan times
  43. if (info.IsMissingEpisode)
  44. {
  45. return result;
  46. }
  47. if (info.SeriesProviderIds.TryGetValue(MetadataProvider.Imdb.ToString(), out string? seriesImdbId)
  48. && !string.IsNullOrEmpty(seriesImdbId)
  49. && info.IndexNumber.HasValue)
  50. {
  51. result.HasMetadata = await _omdbProvider.FetchEpisodeData(
  52. result,
  53. info.IndexNumber.Value,
  54. info.ParentIndexNumber ?? 1,
  55. info.GetProviderId(MetadataProvider.Imdb),
  56. seriesImdbId,
  57. info.MetadataLanguage,
  58. info.MetadataCountryCode,
  59. cancellationToken).ConfigureAwait(false);
  60. }
  61. return result;
  62. }
  63. public Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
  64. {
  65. return _itemProvider.GetImageResponse(url, cancellationToken);
  66. }
  67. }
  68. }