convert episode providers to new system

2025-12-17 06:23:03 +03:00 · 2014-02-04 15:19:29 -05:00
parent 351cfef7a7
commit 04d62d3420
50 changed files with 1862 additions and 1113 deletions
--- a/MediaBrowser.Providers/TV/EpisodeMetadataService.cs
+++ b/MediaBrowser.Providers/TV/EpisodeMetadataService.cs
@@ -0,0 +1,137 @@
+using MediaBrowser.Common.IO;
+using MediaBrowser.Controller.Configuration;
+using MediaBrowser.Controller.Entities.TV;
+using MediaBrowser.Controller.Library;
+using MediaBrowser.Controller.Providers;
+using MediaBrowser.Model.Entities;
+using MediaBrowser.Model.Logging;
+using MediaBrowser.Providers.Manager;
+using System.Collections.Generic;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace MediaBrowser.Providers.TV
+{
+    public class EpisodeMetadataService : MetadataService<Episode, EpisodeId>
+    {
+        private readonly ILibraryManager _libraryManager;
+
+        public EpisodeMetadataService(IServerConfigurationManager serverConfigurationManager, ILogger logger, IProviderManager providerManager, IProviderRepository providerRepo, IFileSystem fileSystem, ILibraryManager libraryManager)
+            : base(serverConfigurationManager, logger, providerManager, providerRepo, fileSystem)
+        {
+            _libraryManager = libraryManager;
+        }
+
+        /// <summary>
+        /// Merges the specified source.
+        /// </summary>
+        /// <param name="source">The source.</param>
+        /// <param name="target">The target.</param>
+        /// <param name="lockedFields">The locked fields.</param>
+        /// <param name="replaceData">if set to <c>true</c> [replace data].</param>
+        /// <param name="mergeMetadataSettings">if set to <c>true</c> [merge metadata settings].</param>
+        protected override void MergeData(Episode source, Episode target, List<MetadataFields> lockedFields, bool replaceData, bool mergeMetadataSettings)
+        {
+            ProviderUtils.MergeBaseItemData(source, target, lockedFields, replaceData, mergeMetadataSettings);
+
+            if (replaceData || !target.AirsBeforeSeasonNumber.HasValue)
+            {
+                target.AirsBeforeSeasonNumber = source.AirsBeforeSeasonNumber;
+            }
+
+            if (replaceData || !target.AirsAfterSeasonNumber.HasValue)
+            {
+                target.AirsAfterSeasonNumber = source.AirsAfterSeasonNumber;
+            }
+
+            if (replaceData || !target.AirsBeforeEpisodeNumber.HasValue)
+            {
+                target.AirsBeforeEpisodeNumber = source.AirsBeforeEpisodeNumber;
+            }
+
+            if (replaceData || !target.DvdSeasonNumber.HasValue)
+            {
+                target.DvdSeasonNumber = source.DvdSeasonNumber;
+            }
+
+            if (replaceData || !target.DvdEpisodeNumber.HasValue)
+            {
+                target.DvdEpisodeNumber = source.DvdEpisodeNumber;
+            }
+
+            if (replaceData || !target.AbsoluteEpisodeNumber.HasValue)
+            {
+                target.AbsoluteEpisodeNumber = source.AbsoluteEpisodeNumber;
+            }
+
+            if (replaceData || !target.IndexNumberEnd.HasValue)
+            {
+                target.IndexNumberEnd = source.IndexNumberEnd;
+            }
+        }
+
+        protected override Task SaveItem(Episode item, ItemUpdateType reason, CancellationToken cancellationToken)
+        {
+            return _libraryManager.UpdateItem(item, reason, cancellationToken);
+        }
+
+        protected override EpisodeId GetId(Episode item)
+        {
+            var id = base.GetId(item);
+
+            var series = item.Series;
+
+            if (series != null)
+            {
+                id.SeriesProviderIds = series.ProviderIds;
+            }
+
+            id.IndexNumberEnd = item.IndexNumberEnd;
+
+            return id;
+        }
+
+        protected override ItemUpdateType BeforeMetadataRefresh(Episode item)
+        {
+            var updateType = base.BeforeMetadataRefresh(item);
+
+            var locationType = item.LocationType;
+            if (locationType == LocationType.FileSystem || locationType == LocationType.Offline)
+            {
+                var currentIndexNumber = item.IndexNumber;
+                var currentIndexNumberEnd = item.IndexNumberEnd;
+                var currentParentIndexNumber = item.ParentIndexNumber;
+
+                item.IndexNumber = item.IndexNumber ?? TVUtils.GetEpisodeNumberFromFile(item.Path, item.Parent is Season);
+                item.IndexNumberEnd = item.IndexNumberEnd ?? TVUtils.GetEndingEpisodeNumberFromFile(item.Path);
+
+                if (!item.ParentIndexNumber.HasValue)
+                {
+                    var season = item.Season;
+
+                    if (season != null)
+                    {
+                        item.ParentIndexNumber = season.IndexNumber;
+                    }
+                }
+
+                if ((currentIndexNumber ?? -1) != (item.IndexNumber ?? -1))
+                {
+                    updateType = updateType | ItemUpdateType.MetadataImport;
+                }
+
+                if ((currentIndexNumberEnd ?? -1) != (item.IndexNumberEnd ?? -1))
+                {
+                    updateType = updateType | ItemUpdateType.MetadataImport;
+                }
+
+                if ((currentParentIndexNumber ?? -1) != (item.ParentIndexNumber ?? -1))
+                {
+                    updateType = updateType | ItemUpdateType.MetadataImport;
+                }
+            }
+
+            return updateType;
+        }
+    }
+}