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