2017-09-04 02:20:56 +00:00
|
|
|
using System.Collections.Generic;
|
2015-12-25 09:22:00 +00:00
|
|
|
using System.IO;
|
|
|
|
using System.Linq;
|
|
|
|
using System.Text;
|
|
|
|
using NLog;
|
|
|
|
using NzbDrone.Common.Disk;
|
|
|
|
using NzbDrone.Common.Extensions;
|
|
|
|
using NzbDrone.Core.Configuration;
|
|
|
|
using NzbDrone.Core.Extras.Files;
|
|
|
|
using NzbDrone.Core.MediaFiles;
|
2017-10-17 01:40:31 +00:00
|
|
|
using NzbDrone.Core.Music;
|
2015-12-25 09:22:00 +00:00
|
|
|
|
2017-10-25 01:25:29 +00:00
|
|
|
namespace NzbDrone.Core.Extras.Lyrics
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2017-10-25 01:25:29 +00:00
|
|
|
public class LyricService : ExtraFileManager<LyricFile>
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2017-10-25 01:25:29 +00:00
|
|
|
private readonly ILyricFileService _lyricFileService;
|
2019-10-07 16:40:48 +00:00
|
|
|
private readonly IMediaFileAttributeService _mediaFileAttributeService;
|
2015-12-25 09:22:00 +00:00
|
|
|
private readonly Logger _logger;
|
|
|
|
|
2017-10-25 01:25:29 +00:00
|
|
|
public LyricService(IConfigService configService,
|
2017-02-24 06:42:04 +00:00
|
|
|
IDiskProvider diskProvider,
|
2015-12-25 09:22:00 +00:00
|
|
|
IDiskTransferService diskTransferService,
|
2017-10-25 01:25:29 +00:00
|
|
|
ILyricFileService lyricFileService,
|
2019-10-07 16:40:48 +00:00
|
|
|
IMediaFileAttributeService mediaFileAttributeService,
|
2015-12-25 09:22:00 +00:00
|
|
|
Logger logger)
|
2017-02-24 06:42:04 +00:00
|
|
|
: base(configService, diskProvider, diskTransferService, logger)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2017-10-25 01:25:29 +00:00
|
|
|
_lyricFileService = lyricFileService;
|
2019-10-07 16:40:48 +00:00
|
|
|
_mediaFileAttributeService = mediaFileAttributeService;
|
2015-12-25 09:22:00 +00:00
|
|
|
_logger = logger;
|
|
|
|
}
|
|
|
|
|
2016-12-09 06:54:15 +00:00
|
|
|
public override int Order => 1;
|
2015-12-25 09:22:00 +00:00
|
|
|
|
2020-09-30 21:37:45 +00:00
|
|
|
public override IEnumerable<ExtraFile> CreateAfterMediaCoverUpdate(Artist artist)
|
|
|
|
{
|
|
|
|
return Enumerable.Empty<ExtraFile>();
|
|
|
|
}
|
|
|
|
|
2017-12-13 02:25:30 +00:00
|
|
|
public override IEnumerable<ExtraFile> CreateAfterArtistScan(Artist artist, List<TrackFile> trackFiles)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2017-10-25 01:25:29 +00:00
|
|
|
return Enumerable.Empty<LyricFile>();
|
2015-12-25 09:22:00 +00:00
|
|
|
}
|
|
|
|
|
2017-10-17 01:40:31 +00:00
|
|
|
public override IEnumerable<ExtraFile> CreateAfterTrackImport(Artist artist, TrackFile trackFile)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2017-10-25 01:25:29 +00:00
|
|
|
return Enumerable.Empty<LyricFile>();
|
2015-12-25 09:22:00 +00:00
|
|
|
}
|
|
|
|
|
2020-09-30 21:37:45 +00:00
|
|
|
public override IEnumerable<ExtraFile> CreateAfterTrackFolder(Artist artist, Album album, string artistFolder, string albumFolder)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2017-10-25 01:25:29 +00:00
|
|
|
return Enumerable.Empty<LyricFile>();
|
2015-12-25 09:22:00 +00:00
|
|
|
}
|
|
|
|
|
2017-10-17 01:40:31 +00:00
|
|
|
public override IEnumerable<ExtraFile> MoveFilesAfterRename(Artist artist, List<TrackFile> trackFiles)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2017-10-25 01:25:29 +00:00
|
|
|
var subtitleFiles = _lyricFileService.GetFilesByArtist(artist.Id);
|
2015-12-25 09:22:00 +00:00
|
|
|
|
2017-10-25 01:25:29 +00:00
|
|
|
var movedFiles = new List<LyricFile>();
|
2015-12-25 09:22:00 +00:00
|
|
|
|
2017-10-17 01:40:31 +00:00
|
|
|
foreach (var trackFile in trackFiles)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2017-10-17 01:40:31 +00:00
|
|
|
var groupedExtraFilesForTrackFile = subtitleFiles.Where(m => m.TrackFileId == trackFile.Id)
|
2019-08-02 11:50:09 +00:00
|
|
|
.GroupBy(s => s.Extension).ToList();
|
2015-12-25 09:22:00 +00:00
|
|
|
|
2017-10-17 01:40:31 +00:00
|
|
|
foreach (var group in groupedExtraFilesForTrackFile)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
|
|
|
var groupCount = group.Count();
|
|
|
|
var copy = 1;
|
|
|
|
|
|
|
|
if (groupCount > 1)
|
|
|
|
{
|
2019-08-02 11:50:09 +00:00
|
|
|
_logger.Warn("Multiple lyric files found with the same extension for {0}", trackFile.Path);
|
2015-12-25 09:22:00 +00:00
|
|
|
}
|
|
|
|
|
2017-02-24 06:42:04 +00:00
|
|
|
foreach (var subtitleFile in group)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2019-08-02 11:50:09 +00:00
|
|
|
var suffix = GetSuffix(copy, groupCount > 1);
|
2017-10-17 01:40:31 +00:00
|
|
|
movedFiles.AddIfNotNull(MoveFile(artist, trackFile, subtitleFile, suffix));
|
2015-12-25 09:22:00 +00:00
|
|
|
|
|
|
|
copy++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-10-25 01:25:29 +00:00
|
|
|
_lyricFileService.Upsert(movedFiles);
|
2015-12-25 09:22:00 +00:00
|
|
|
|
|
|
|
return movedFiles;
|
|
|
|
}
|
|
|
|
|
2017-10-17 01:40:31 +00:00
|
|
|
public override ExtraFile Import(Artist artist, TrackFile trackFile, string path, string extension, bool readOnly)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2017-10-25 01:25:29 +00:00
|
|
|
if (LyricFileExtensions.Extensions.Contains(Path.GetExtension(path)))
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2019-08-02 11:50:09 +00:00
|
|
|
var suffix = GetSuffix(1, false);
|
2017-10-17 01:40:31 +00:00
|
|
|
var subtitleFile = ImportFile(artist, trackFile, path, readOnly, extension, suffix);
|
2015-12-25 09:22:00 +00:00
|
|
|
|
2019-10-07 16:40:48 +00:00
|
|
|
_mediaFileAttributeService.SetFilePermissions(path);
|
2017-10-25 01:25:29 +00:00
|
|
|
_lyricFileService.Upsert(subtitleFile);
|
2015-12-25 09:22:00 +00:00
|
|
|
|
|
|
|
return subtitleFile;
|
|
|
|
}
|
|
|
|
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
2019-08-02 11:50:09 +00:00
|
|
|
private string GetSuffix(int copy, bool multipleCopies = false)
|
2015-12-25 09:22:00 +00:00
|
|
|
{
|
2017-03-06 01:45:35 +00:00
|
|
|
var suffixBuilder = new StringBuilder();
|
2015-12-25 09:22:00 +00:00
|
|
|
|
|
|
|
if (multipleCopies)
|
|
|
|
{
|
2023-04-12 02:31:27 +00:00
|
|
|
suffixBuilder.Append('.');
|
2017-03-06 01:45:35 +00:00
|
|
|
suffixBuilder.Append(copy);
|
2015-12-25 09:22:00 +00:00
|
|
|
}
|
|
|
|
|
2017-03-06 01:45:35 +00:00
|
|
|
return suffixBuilder.ToString();
|
2015-12-25 09:22:00 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|