New: Parse existing subtitles and extra files

Towards #459
2026-04-19 21:44:30 -04:00 · 2015-12-25 01:22:00 -08:00
parent 816cf608fc
commit 2e96c4e798
78 changed files with 2013 additions and 678 deletions
@@ -0,0 +1,70 @@
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using NzbDrone.Common;
+using NzbDrone.Common.Disk;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Core.Configuration;
+using NzbDrone.Core.MediaFiles;
+using NzbDrone.Core.Tv;
+
+namespace NzbDrone.Core.Extras.Files
+{
+    public interface IManageExtraFiles
+    {
+        int Order { get; }
+        IEnumerable<ExtraFile> CreateAfterSeriesScan(Series series, List<EpisodeFile> episodeFiles);
+        IEnumerable<ExtraFile> CreateAfterEpisodeImport(Series series, EpisodeFile episodeFile);
+        IEnumerable<ExtraFile> CreateAfterEpisodeImport(Series series, string seriesFolder, string seasonFolder);
+        IEnumerable<ExtraFile> MoveFilesAfterRename(Series series, List<EpisodeFile> episodeFiles);
+        ExtraFile Import(Series series, EpisodeFile episodeFile, string path, string extension, bool readOnly);
+    }
+
+    public abstract class ExtraFileManager<TExtraFile> : IManageExtraFiles
+        where TExtraFile : ExtraFile, new()
+
+    {
+        private readonly IConfigService _configService;
+        private readonly IDiskTransferService _diskTransferService;
+        private readonly IExtraFileService<TExtraFile> _extraFileService;
+
+        public ExtraFileManager(IConfigService configService,
+                                IDiskTransferService diskTransferService,
+                                IExtraFileService<TExtraFile> extraFileService)
+        {
+            _configService = configService;
+            _diskTransferService = diskTransferService;
+            _extraFileService = extraFileService;
+        }
+
+        public abstract int Order { get; }
+        public abstract IEnumerable<ExtraFile> CreateAfterSeriesScan(Series series, List<EpisodeFile> episodeFiles);
+        public abstract IEnumerable<ExtraFile> CreateAfterEpisodeImport(Series series, EpisodeFile episodeFile);
+        public abstract IEnumerable<ExtraFile> CreateAfterEpisodeImport(Series series, string seriesFolder, string seasonFolder);
+        public abstract IEnumerable<ExtraFile> MoveFilesAfterRename(Series series, List<EpisodeFile> episodeFiles);
+        public abstract ExtraFile Import(Series series, EpisodeFile episodeFile, string path, string extension, bool readOnly);
+
+        protected TExtraFile ImportFile(Series series, EpisodeFile episodeFile, string path, string extension, bool readOnly)
+        {
+            var newFileName = Path.Combine(series.Path, Path.ChangeExtension(episodeFile.RelativePath, extension));
+
+            var transferMode = TransferMode.Move;
+
+            if (readOnly)
+            {
+                transferMode = _configService.CopyUsingHardlinks ? TransferMode.HardLinkOrCopy : TransferMode.Copy;
+            }
+
+            _diskTransferService.TransferFile(path, newFileName, transferMode, true, false);
+
+            return new TExtraFile
+            {
+                SeriesId = series.Id,
+                SeasonNumber = episodeFile.SeasonNumber,
+                EpisodeFileId = episodeFile.Id,
+                RelativePath = series.Path.GetRelativePath(newFileName),
+                Extension = Path.GetExtension(path)
+            };
+        }
+    }
+}