OmdbEpisodeProvider.cs 2.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980
  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. && info.ParentIndexNumber.HasValue)
  51. {
  52. result.HasMetadata = await _omdbProvider.FetchEpisodeData(
  53. result,
  54. info.IndexNumber.Value,
  55. info.ParentIndexNumber.Value,
  56. info.GetProviderId(MetadataProvider.Imdb),
  57. seriesImdbId,
  58. info.MetadataLanguage,
  59. info.MetadataCountryCode,
  60. cancellationToken).ConfigureAwait(false);
  61. }
  62. return result;
  63. }
  64. public Task<HttpResponseMessage> GetImageResponse(string url, CancellationToken cancellationToken)
  65. {
  66. return _itemProvider.GetImageResponse(url, cancellationToken);
  67. }
  68. }
  69. }