SeriesMetadataService.cs 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105
  1. #pragma warning disable CS1591
  2. using System;
  3. using System.Threading;
  4. using System.Threading.Tasks;
  5. using MediaBrowser.Controller.Configuration;
  6. using MediaBrowser.Controller.Entities.TV;
  7. using MediaBrowser.Controller.Library;
  8. using MediaBrowser.Controller.Providers;
  9. using MediaBrowser.Model.Entities;
  10. using MediaBrowser.Model.Globalization;
  11. using MediaBrowser.Model.IO;
  12. using MediaBrowser.Providers.Manager;
  13. using MediaBrowser.Providers.Plugins.TheTvdb;
  14. using Microsoft.Extensions.Logging;
  15. namespace MediaBrowser.Providers.TV
  16. {
  17. public class SeriesMetadataService : MetadataService<Series, SeriesInfo>
  18. {
  19. private readonly ILocalizationManager _localization;
  20. private readonly TvdbClientManager _tvdbClientManager;
  21. public SeriesMetadataService(
  22. IServerConfigurationManager serverConfigurationManager,
  23. ILogger<SeriesMetadataService> logger,
  24. IProviderManager providerManager,
  25. IFileSystem fileSystem,
  26. ILibraryManager libraryManager,
  27. ILocalizationManager localization,
  28. TvdbClientManager tvdbClientManager)
  29. : base(serverConfigurationManager, logger, providerManager, fileSystem, libraryManager)
  30. {
  31. _localization = localization;
  32. _tvdbClientManager = tvdbClientManager;
  33. }
  34. /// <inheritdoc />
  35. protected override async Task AfterMetadataRefresh(Series item, MetadataRefreshOptions refreshOptions, CancellationToken cancellationToken)
  36. {
  37. await base.AfterMetadataRefresh(item, refreshOptions, cancellationToken).ConfigureAwait(false);
  38. var seasonProvider = new DummySeasonProvider(Logger, _localization, LibraryManager, FileSystem);
  39. await seasonProvider.Run(item, cancellationToken).ConfigureAwait(false);
  40. // TODO why does it not register this itself omg
  41. var provider = new MissingEpisodeProvider(
  42. Logger,
  43. ServerConfigurationManager,
  44. LibraryManager,
  45. _localization,
  46. FileSystem,
  47. _tvdbClientManager);
  48. try
  49. {
  50. await provider.Run(item, true, CancellationToken.None).ConfigureAwait(false);
  51. }
  52. catch (Exception ex)
  53. {
  54. Logger.LogError(ex, "Error in DummySeasonProvider for {ItemPath}", item.Path);
  55. }
  56. }
  57. /// <inheritdoc />
  58. protected override bool IsFullLocalMetadata(Series item)
  59. {
  60. if (string.IsNullOrWhiteSpace(item.Overview))
  61. {
  62. return false;
  63. }
  64. if (!item.ProductionYear.HasValue)
  65. {
  66. return false;
  67. }
  68. return base.IsFullLocalMetadata(item);
  69. }
  70. /// <inheritdoc />
  71. protected override void MergeData(MetadataResult<Series> source, MetadataResult<Series> target, MetadataField[] lockedFields, bool replaceData, bool mergeMetadataSettings)
  72. {
  73. ProviderUtils.MergeBaseItemData(source, target, lockedFields, replaceData, mergeMetadataSettings);
  74. var sourceItem = source.Item;
  75. var targetItem = target.Item;
  76. if (replaceData || string.IsNullOrEmpty(targetItem.AirTime))
  77. {
  78. targetItem.AirTime = sourceItem.AirTime;
  79. }
  80. if (replaceData || !targetItem.Status.HasValue)
  81. {
  82. targetItem.Status = sourceItem.Status;
  83. }
  84. if (replaceData || targetItem.AirDays == null || targetItem.AirDays.Length == 0)
  85. {
  86. targetItem.AirDays = sourceItem.AirDays;
  87. }
  88. }
  89. }
  90. }