mirror of https://github.com/Radarr/Radarr
75 lines
2.8 KiB
C#
75 lines
2.8 KiB
C#
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.Extras.Files;
|
|
using NzbDrone.Core.Movies;
|
|
using NzbDrone.Core.Parser;
|
|
|
|
namespace NzbDrone.Core.Extras.Others
|
|
{
|
|
public class ExistingOtherExtraImporter : ImportExistingExtraFilesBase<OtherExtraFile>
|
|
{
|
|
private readonly IExtraFileService<OtherExtraFile> _otherExtraFileService;
|
|
private readonly IParsingService _parsingService;
|
|
private readonly Logger _logger;
|
|
|
|
public ExistingOtherExtraImporter(IExtraFileService<OtherExtraFile> otherExtraFileService,
|
|
IParsingService parsingService,
|
|
Logger logger)
|
|
: base(otherExtraFileService)
|
|
{
|
|
_otherExtraFileService = otherExtraFileService;
|
|
_parsingService = parsingService;
|
|
_logger = logger;
|
|
}
|
|
|
|
public override int Order => 2;
|
|
|
|
public override IEnumerable<ExtraFile> ProcessFiles(Movie movie, List<string> filesOnDisk, List<string> importedFiles, string fileNameBeforeRename)
|
|
{
|
|
_logger.Debug("Looking for existing extra files in {0}", movie.Path);
|
|
|
|
var extraFiles = new List<OtherExtraFile>();
|
|
var filterResult = FilterAndClean(movie, filesOnDisk, importedFiles, fileNameBeforeRename is not null);
|
|
|
|
foreach (var possibleExtraFile in filterResult.FilesOnDisk)
|
|
{
|
|
var extension = Path.GetExtension(possibleExtraFile);
|
|
|
|
if (extension.IsNullOrWhiteSpace())
|
|
{
|
|
_logger.Debug("No extension for file: {0}", possibleExtraFile);
|
|
continue;
|
|
}
|
|
|
|
var minimalInfo = _parsingService.ParseMinimalPathMovieInfo(possibleExtraFile);
|
|
|
|
if (minimalInfo == null)
|
|
{
|
|
_logger.Debug("Unable to parse extra file: {0}", possibleExtraFile);
|
|
continue;
|
|
}
|
|
|
|
var extraFile = new OtherExtraFile
|
|
{
|
|
MovieId = movie.Id,
|
|
MovieFileId = movie.MovieFileId,
|
|
RelativePath = movie.Path.GetRelativePath(possibleExtraFile),
|
|
Extension = extension
|
|
};
|
|
|
|
extraFiles.Add(extraFile);
|
|
}
|
|
|
|
_logger.Info("Found {0} existing other extra files", extraFiles.Count);
|
|
_otherExtraFileService.Upsert(extraFiles);
|
|
|
|
// Return files that were just imported along with files that were
|
|
// previously imported so previously imported files aren't imported twice
|
|
return extraFiles.Concat(filterResult.PreviouslyImported);
|
|
}
|
|
}
|
|
}
|