SeasonMetadataService.cs 3.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  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. return updateType;
  49. }
  50. protected override void MergeData(MetadataResult<Season> source, MetadataResult<Season> target, List<MetadataFields> lockedFields, bool replaceData, bool mergeMetadataSettings)
  51. {
  52. ProviderUtils.MergeBaseItemData(source, target, lockedFields, replaceData, mergeMetadataSettings);
  53. }
  54. private ItemUpdateType SavePremiereDate(Season item, List<Episode> episodes)
  55. {
  56. var dates = episodes.Where(i => i.PremiereDate.HasValue).Select(i => i.PremiereDate.Value).ToList();
  57. DateTime? premiereDate = null;
  58. if (dates.Count > 0)
  59. {
  60. premiereDate = dates.Min();
  61. }
  62. if (item.PremiereDate != premiereDate)
  63. {
  64. item.PremiereDate = premiereDate;
  65. return ItemUpdateType.MetadataEdit;
  66. }
  67. return ItemUpdateType.None;
  68. }
  69. private ItemUpdateType SaveIsVirtualItem(Season item, List<Episode> episodes)
  70. {
  71. var isVirtualItem = item.LocationType == LocationType.Virtual && (episodes.Count == 0 || episodes.All(i => i.LocationType == LocationType.Virtual));
  72. if (item.IsVirtualItem != isVirtualItem)
  73. {
  74. item.IsVirtualItem = isVirtualItem;
  75. return ItemUpdateType.MetadataEdit;
  76. }
  77. return ItemUpdateType.None;
  78. }
  79. public SeasonMetadataService(IServerConfigurationManager serverConfigurationManager, ILogger logger, IProviderManager providerManager, IFileSystem fileSystem, IUserDataManager userDataManager, ILibraryManager libraryManager) : base(serverConfigurationManager, logger, providerManager, fileSystem, userDataManager, libraryManager)
  80. {
  81. }
  82. }
  83. }