You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
3.8 KiB
98 lines
3.8 KiB
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.Music;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.MediaFiles.TrackImport.Aggregation;
|
|
|
|
namespace NzbDrone.Core.Extras.Lyrics
|
|
{
|
|
public class ExistingLyricImporter : ImportExistingExtraFilesBase<LyricFile>
|
|
{
|
|
private readonly IExtraFileService<LyricFile> _lyricFileService;
|
|
private readonly IAugmentingService _augmentingService;
|
|
private readonly Logger _logger;
|
|
|
|
public ExistingLyricImporter(IExtraFileService<LyricFile> lyricFileService,
|
|
IAugmentingService augmentingService,
|
|
Logger logger)
|
|
: base (lyricFileService)
|
|
{
|
|
_lyricFileService = lyricFileService;
|
|
_augmentingService = augmentingService;
|
|
_logger = logger;
|
|
}
|
|
|
|
public override int Order => 1;
|
|
|
|
public override IEnumerable<ExtraFile> ProcessFiles(Artist artist, List<string> filesOnDisk, List<string> importedFiles)
|
|
{
|
|
_logger.Debug("Looking for existing lyrics files in {0}", artist.Path);
|
|
|
|
var subtitleFiles = new List<LyricFile>();
|
|
var filterResult = FilterAndClean(artist, filesOnDisk, importedFiles);
|
|
|
|
foreach (var possibleLyricFile in filterResult.FilesOnDisk)
|
|
{
|
|
var extension = Path.GetExtension(possibleLyricFile);
|
|
|
|
if (LyricFileExtensions.Extensions.Contains(extension))
|
|
{
|
|
var localTrack = new LocalTrack
|
|
{
|
|
FileTrackInfo = Parser.Parser.ParseMusicPath(possibleLyricFile),
|
|
Artist = artist,
|
|
Path = possibleLyricFile
|
|
};
|
|
|
|
try
|
|
{
|
|
_augmentingService.Augment(localTrack, false);
|
|
}
|
|
catch (AugmentingFailedException)
|
|
{
|
|
_logger.Debug("Unable to parse lyric file: {0}", possibleLyricFile);
|
|
continue;
|
|
}
|
|
|
|
if (localTrack.Tracks.Empty())
|
|
{
|
|
_logger.Debug("Cannot find related tracks for: {0}", possibleLyricFile);
|
|
continue;
|
|
}
|
|
|
|
if (localTrack.Tracks.DistinctBy(e => e.TrackFileId).Count() > 1)
|
|
{
|
|
_logger.Debug("Lyric file: {0} does not match existing files.", possibleLyricFile);
|
|
continue;
|
|
}
|
|
|
|
var subtitleFile = new LyricFile
|
|
{
|
|
ArtistId = artist.Id,
|
|
AlbumId = localTrack.Album.Id,
|
|
TrackFileId = localTrack.Tracks.First().TrackFileId,
|
|
RelativePath = artist.Path.GetRelativePath(possibleLyricFile),
|
|
Language = LanguageParser.ParseSubtitleLanguage(possibleLyricFile),
|
|
Extension = extension
|
|
};
|
|
|
|
subtitleFiles.Add(subtitleFile);
|
|
}
|
|
}
|
|
|
|
_logger.Info("Found {0} existing lyric files", subtitleFiles.Count);
|
|
_lyricFileService.Upsert(subtitleFiles);
|
|
|
|
// Return files that were just imported along with files that were
|
|
// previously imported so previously imported files aren't imported twice
|
|
|
|
return subtitleFiles.Concat(filterResult.PreviouslyImported);
|
|
}
|
|
}
|
|
}
|