SeasonMetadataService.cs 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596
  1. using MediaBrowser.Controller.Configuration;
  2. using MediaBrowser.Controller.Entities.TV;
  3. using MediaBrowser.Controller.Library;
  4. using MediaBrowser.Controller.Providers;
  5. using MediaBrowser.Model.Entities;
  6. using MediaBrowser.Model.Logging;
  7. using MediaBrowser.Providers.Manager;
  8. using System;
  9. using System.Collections.Generic;
  10. using System.Linq;
  11. using System.Threading.Tasks;
  12. using CommonIO;
  13. namespace MediaBrowser.Providers.TV
  14. {
  15. public class SeasonMetadataService : MetadataService<Season, SeasonInfo>
  16. {
  17. protected override async Task<ItemUpdateType> BeforeSave(Season item, bool isFullRefresh, ItemUpdateType currentUpdateType)
  18. {
  19. var updateType = await base.BeforeSave(item, isFullRefresh, currentUpdateType).ConfigureAwait(false);
  20. if (item.IndexNumber.HasValue && item.IndexNumber.Value == 0)
  21. {
  22. if (!string.Equals(item.Name, ServerConfigurationManager.Configuration.SeasonZeroDisplayName, StringComparison.OrdinalIgnoreCase))
  23. {
  24. item.Name = ServerConfigurationManager.Configuration.SeasonZeroDisplayName;
  25. updateType = updateType | ItemUpdateType.MetadataEdit;
  26. }
  27. }
  28. if (isFullRefresh || currentUpdateType > ItemUpdateType.None)
  29. {
  30. var episodes = item.GetEpisodes().ToList();
  31. updateType |= SavePremiereDate(item, episodes);
  32. updateType |= SaveIsVirtualItem(item, episodes);
  33. }
  34. if (!string.Equals(item.SeriesName, item.FindSeriesName(), StringComparison.Ordinal))
  35. {
  36. updateType |= ItemUpdateType.MetadataImport;
  37. }
  38. if (!string.Equals(item.SeriesSortName, item.FindSeriesSortName(), StringComparison.Ordinal))
  39. {
  40. updateType |= ItemUpdateType.MetadataImport;
  41. }
  42. if (item.SeriesId != item.FindSeriesId())
  43. {
  44. updateType |= ItemUpdateType.MetadataImport;
  45. }
  46. return updateType;
  47. }
  48. protected override void MergeData(MetadataResult<Season> source, MetadataResult<Season> target, List<MetadataFields> lockedFields, bool replaceData, bool mergeMetadataSettings)
  49. {
  50. ProviderUtils.MergeBaseItemData(source, target, lockedFields, replaceData, mergeMetadataSettings);
  51. }
  52. private ItemUpdateType SavePremiereDate(Season item, List<Episode> episodes)
  53. {
  54. var dates = episodes.Where(i => i.PremiereDate.HasValue).Select(i => i.PremiereDate.Value).ToList();
  55. DateTime? premiereDate = null;
  56. if (dates.Count > 0)
  57. {
  58. premiereDate = dates.Min();
  59. }
  60. if (item.PremiereDate != premiereDate)
  61. {
  62. item.PremiereDate = premiereDate;
  63. return ItemUpdateType.MetadataEdit;
  64. }
  65. return ItemUpdateType.None;
  66. }
  67. private ItemUpdateType SaveIsVirtualItem(Season item, List<Episode> episodes)
  68. {
  69. var isVirtualItem = item.LocationType == LocationType.Virtual && (episodes.Count == 0 || episodes.All(i => i.LocationType == LocationType.Virtual));
  70. if (item.IsVirtualItem != isVirtualItem)
  71. {
  72. item.IsVirtualItem = isVirtualItem;
  73. return ItemUpdateType.MetadataEdit;
  74. }
  75. return ItemUpdateType.None;
  76. }
  77. public SeasonMetadataService(IServerConfigurationManager serverConfigurationManager, ILogger logger, IProviderManager providerManager, IFileSystem fileSystem, IUserDataManager userDataManager, ILibraryManager libraryManager) : base(serverConfigurationManager, logger, providerManager, fileSystem, userDataManager, libraryManager)
  78. {
  79. }
  80. }
  81. }