SeasonMetadataService.cs 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105
  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 (updateType <= ItemUpdateType.None)
  35. {
  36. if (!string.Equals(item.SeriesName, item.FindSeriesName(), StringComparison.Ordinal))
  37. {
  38. updateType |= ItemUpdateType.MetadataImport;
  39. }
  40. }
  41. if (updateType <= ItemUpdateType.None)
  42. {
  43. if (!string.Equals(item.SeriesSortName, item.FindSeriesSortName(), StringComparison.Ordinal))
  44. {
  45. updateType |= ItemUpdateType.MetadataImport;
  46. }
  47. }
  48. if (updateType <= ItemUpdateType.None)
  49. {
  50. if (item.SeriesId != item.FindSeriesId())
  51. {
  52. updateType |= ItemUpdateType.MetadataImport;
  53. }
  54. }
  55. return updateType;
  56. }
  57. protected override void MergeData(MetadataResult<Season> source, MetadataResult<Season> target, List<MetadataFields> lockedFields, bool replaceData, bool mergeMetadataSettings)
  58. {
  59. ProviderUtils.MergeBaseItemData(source, target, lockedFields, replaceData, mergeMetadataSettings);
  60. }
  61. private ItemUpdateType SavePremiereDate(Season item, List<Episode> episodes)
  62. {
  63. var dates = episodes.Where(i => i.PremiereDate.HasValue).Select(i => i.PremiereDate.Value).ToList();
  64. DateTime? premiereDate = null;
  65. if (dates.Count > 0)
  66. {
  67. premiereDate = dates.Min();
  68. }
  69. if (item.PremiereDate != premiereDate)
  70. {
  71. item.PremiereDate = premiereDate;
  72. return ItemUpdateType.MetadataEdit;
  73. }
  74. return ItemUpdateType.None;
  75. }
  76. private ItemUpdateType SaveIsVirtualItem(Season item, List<Episode> episodes)
  77. {
  78. var isVirtualItem = item.LocationType == LocationType.Virtual && (episodes.Count == 0 || episodes.All(i => i.LocationType == LocationType.Virtual));
  79. if (item.IsVirtualItem != isVirtualItem)
  80. {
  81. item.IsVirtualItem = isVirtualItem;
  82. return ItemUpdateType.MetadataEdit;
  83. }
  84. return ItemUpdateType.None;
  85. }
  86. public SeasonMetadataService(IServerConfigurationManager serverConfigurationManager, ILogger logger, IProviderManager providerManager, IFileSystem fileSystem, IUserDataManager userDataManager, ILibraryManager libraryManager) : base(serverConfigurationManager, logger, providerManager, fileSystem, userDataManager, libraryManager)
  87. {
  88. }
  89. }
  90. }