New: Parse existing subtitles and extra files

Towards #459
2026-04-24 22:35:49 -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,89 @@
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using NLog;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Core.Extras.Files;
+using NzbDrone.Core.Parser;
+using NzbDrone.Core.Tv;
+
+namespace NzbDrone.Core.Extras.Subtitles
+{
+    public class ExistingSubtitleImporter : ImportExistingExtraFilesBase<SubtitleFile>
+    {
+        private readonly IExtraFileService<SubtitleFile> _subtitleFileService;
+        private readonly IParsingService _parsingService;
+        private readonly Logger _logger;
+
+        public ExistingSubtitleImporter(IExtraFileService<SubtitleFile> subtitleFileService,
+                                        IParsingService parsingService,
+                                        Logger logger)
+            : base (subtitleFileService)
+        {
+            _subtitleFileService = subtitleFileService;
+            _parsingService = parsingService;
+            _logger = logger;
+        }
+
+        public override int Order
+        {
+            get
+            {
+                return 1;
+            }
+        }
+
+        public override IEnumerable<ExtraFile> ProcessFiles(Series series, List<string> filesOnDisk, List<string> importedFiles)
+        {
+            _logger.Debug("Looking for existing subtitle files in {0}", series.Path);
+
+            var subtitleFiles = new List<SubtitleFile>();
+            var filteredFiles = FilterAndClean(series, filesOnDisk, importedFiles);
+
+            foreach (var possibleSubtitleFile in filteredFiles)
+            {
+                var extension = Path.GetExtension(possibleSubtitleFile);
+
+                if (SubtitleFileExtensions.Extensions.Contains(extension))
+                {
+                    var localEpisode = _parsingService.GetLocalEpisode(possibleSubtitleFile, series);
+
+                    if (localEpisode == null)
+                    {
+                        _logger.Debug("Unable to parse subtitle file: {0}", possibleSubtitleFile);
+                        continue;
+                    }
+
+                    if (localEpisode.Episodes.Empty())
+                    {
+                        _logger.Debug("Cannot find related episodes for: {0}", possibleSubtitleFile);
+                        continue;
+                    }
+
+                    if (localEpisode.Episodes.DistinctBy(e => e.EpisodeFileId).Count() > 1)
+                    {
+                        _logger.Debug("Subtitle file: {0} does not match existing files.", possibleSubtitleFile);
+                        continue;
+                    }
+
+                    var subtitleFile = new SubtitleFile
+                                       {
+                                           SeriesId = series.Id,
+                                           SeasonNumber = localEpisode.SeasonNumber,
+                                           EpisodeFileId = localEpisode.Episodes.First().EpisodeFileId,
+                                           RelativePath = series.Path.GetRelativePath(possibleSubtitleFile),
+                                           Language = LanguageParser.ParseSubtitleLanguage(possibleSubtitleFile),
+                                           Extension = extension
+                                       };
+
+                    subtitleFiles.Add(subtitleFile);
+                }
+            }
+
+            _logger.Info("Found {0} existing subtitle files", subtitleFiles.Count);
+            _subtitleFileService.Upsert(subtitleFiles);
+
+            return subtitleFiles;
+        }
+    }
+}