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 { private readonly IDiskProvider _diskProvider; private readonly IDiskScanService _diskScanService; private readonly List _existingExtraFileImporters; private readonly Logger _logger; public ExistingExtraFileService(IDiskProvider diskProvider, IDiskScanService diskScanService, IEnumerable existingExtraFileImporters, Logger logger) { _diskProvider = diskProvider; _diskScanService = diskScanService; _existingExtraFileImporters = existingExtraFileImporters.OrderBy(e => e.Order).ToList(); _logger = logger; } public void Handle(AuthorScannedEvent message) { var author = message.Author; var extraFiles = new List(); if (!_diskProvider.FolderExists(author.Path)) { return; } _logger.Debug("Looking for existing extra files in {0}", author.Path); var filesOnDisk = _diskScanService.GetNonBookFiles(author.Path); var possibleExtraFiles = _diskScanService.FilterPaths(author.Path, filesOnDisk); var filteredFiles = possibleExtraFiles; var importedFiles = new List(); foreach (var existingExtraFileImporter in _existingExtraFileImporters) { var imported = existingExtraFileImporter.ProcessFiles(author, filteredFiles, importedFiles); importedFiles.AddRange(imported.Select(f => Path.Combine(author.Path, f.RelativePath))); } _logger.Info("Found {0} extra files", extraFiles.Count); } } }