New: Parse existing subtitles and extra files

Towards #459
2026-04-20 21:55:03 -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,53 @@
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using NzbDrone.Common;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Core.Extras.Files;
+using NzbDrone.Core.Tv;
+
+namespace NzbDrone.Core.Extras
+{
+    public abstract class ImportExistingExtraFilesBase<TExtraFile> : IImportExistingExtraFiles
+        where TExtraFile : ExtraFile, new()
+    {
+        private readonly IExtraFileService<TExtraFile> _extraFileService;
+
+        public ImportExistingExtraFilesBase(IExtraFileService<TExtraFile> extraFileService)
+        {
+            _extraFileService = extraFileService;
+        }
+
+        public abstract int Order { get; }
+        public abstract IEnumerable<ExtraFile> ProcessFiles(Series series, List<string> filesOnDisk, List<string> importedFiles);
+
+        public virtual List<string> FilterAndClean(Series series, List<string> filesOnDisk, List<string> importedFiles)
+        {
+            var seriesFiles = _extraFileService.GetFilesBySeries(series.Id);
+
+            Clean(series, filesOnDisk, importedFiles, seriesFiles);
+
+            return Filter(series, filesOnDisk, importedFiles, seriesFiles);
+        }
+
+        private List<string> Filter(Series series, List<string> filesOnDisk, List<string> importedFiles, List<TExtraFile> seriesFiles)
+        {
+            var filteredFiles = filesOnDisk;
+
+            filteredFiles = filteredFiles.Except(seriesFiles.Select(f => Path.Combine(series.Path, f.RelativePath)).ToList(), PathEqualityComparer.Instance).ToList();
+            return filteredFiles.Except(importedFiles, PathEqualityComparer.Instance).ToList();
+        }
+
+        private void Clean(Series series, List<string> filesOnDisk, List<string> importedFiles, List<TExtraFile> seriesFiles)
+        {
+            var alreadyImportedFileIds = seriesFiles.IntersectBy(f => Path.Combine(series.Path, f.RelativePath), importedFiles, i => i, PathEqualityComparer.Instance)
+                .Select(f => f.Id);
+
+            var deletedFiles = seriesFiles.ExceptBy(f => Path.Combine(series.Path, f.RelativePath), filesOnDisk, i => i, PathEqualityComparer.Instance)
+                .Select(f => f.Id);
+
+            _extraFileService.DeleteMany(alreadyImportedFileIds);
+            _extraFileService.DeleteMany(deletedFiles);
+        }
+    }
+}