2015-12-25 09:22:00 +00:00
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using NLog;
|
|
|
|
|
using NzbDrone.Common.Disk;
|
|
|
|
|
using NzbDrone.Core.Extras.Files;
|
|
|
|
|
using NzbDrone.Core.MediaFiles;
|
|
|
|
|
using NzbDrone.Core.MediaFiles.Events;
|
|
|
|
|
using NzbDrone.Core.Messaging.Events;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Extras
|
|
|
|
|
{
|
|
|
|
|
public class ExistingExtraFileService : IHandle<SeriesScannedEvent>
|
|
|
|
|
{
|
|
|
|
|
private readonly IDiskProvider _diskProvider;
|
2016-09-18 18:04:56 +00:00
|
|
|
|
private readonly IDiskScanService _diskScanService;
|
2015-12-25 09:22:00 +00:00
|
|
|
|
private readonly List<IImportExistingExtraFiles> _existingExtraFileImporters;
|
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
|
|
|
|
public ExistingExtraFileService(IDiskProvider diskProvider,
|
2016-09-18 18:04:56 +00:00
|
|
|
|
IDiskScanService diskScanService,
|
2015-12-25 09:22:00 +00:00
|
|
|
|
List<IImportExistingExtraFiles> existingExtraFileImporters,
|
|
|
|
|
Logger logger)
|
|
|
|
|
{
|
|
|
|
|
_diskProvider = diskProvider;
|
2016-09-18 18:04:56 +00:00
|
|
|
|
_diskScanService = diskScanService;
|
2015-12-25 09:22:00 +00:00
|
|
|
|
_existingExtraFileImporters = existingExtraFileImporters.OrderBy(e => e.Order).ToList();
|
|
|
|
|
_logger = logger;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void Handle(SeriesScannedEvent message)
|
|
|
|
|
{
|
|
|
|
|
var series = message.Series;
|
|
|
|
|
var extraFiles = new List<ExtraFile>();
|
|
|
|
|
|
|
|
|
|
if (!_diskProvider.FolderExists(series.Path))
|
|
|
|
|
{
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
_logger.Debug("Looking for existing extra files in {0}", series.Path);
|
|
|
|
|
|
2016-09-18 18:04:56 +00:00
|
|
|
|
var filesOnDisk = _diskScanService.GetNonVideoFiles(series.Path);
|
|
|
|
|
var possibleExtraFiles = _diskScanService.FilterFiles(series, filesOnDisk);
|
2015-12-25 09:22:00 +00:00
|
|
|
|
|
|
|
|
|
var filteredFiles = possibleExtraFiles;
|
|
|
|
|
var importedFiles = new List<string>();
|
|
|
|
|
|
|
|
|
|
foreach (var existingExtraFileImporter in _existingExtraFileImporters)
|
|
|
|
|
{
|
|
|
|
|
var imported = existingExtraFileImporter.ProcessFiles(series, filteredFiles, importedFiles);
|
|
|
|
|
|
|
|
|
|
importedFiles.AddRange(imported.Select(f => Path.Combine(series.Path, f.RelativePath)));
|
|
|
|
|
}
|
|
|
|
|
|
2016-09-18 18:04:56 +00:00
|
|
|
|
_logger.Info("Found {0} extra files", extraFiles.Count);
|
2015-12-25 09:22:00 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|