2018-03-14 20:41:36 +00:00
|
|
|
using System.IO;
|
2013-07-16 02:56:46 +00:00
|
|
|
using NLog;
|
2014-01-06 06:20:08 +00:00
|
|
|
using NzbDrone.Common.Disk;
|
2020-08-28 02:33:54 +00:00
|
|
|
using NzbDrone.Common.Extensions;
|
2020-01-16 02:52:45 +00:00
|
|
|
using NzbDrone.Core.MediaFiles.MovieImport;
|
2013-07-16 02:56:46 +00:00
|
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.MediaFiles
|
|
|
|
{
|
|
|
|
public interface IUpgradeMediaFiles
|
|
|
|
{
|
2017-01-03 19:24:55 +00:00
|
|
|
MovieFileMoveResult UpgradeMovieFile(MovieFile movieFile, LocalMovie localMovie, bool copyOnly = false);
|
2013-07-16 02:56:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public class UpgradeMediaFileService : IUpgradeMediaFiles
|
|
|
|
{
|
|
|
|
private readonly IRecycleBinProvider _recycleBinProvider;
|
|
|
|
private readonly IMediaFileService _mediaFileService;
|
2017-01-03 19:24:55 +00:00
|
|
|
private readonly IMoveMovieFiles _movieFileMover;
|
2019-12-22 22:08:53 +00:00
|
|
|
private readonly IRenameMovieFileService _movieFileRenamer;
|
2013-07-31 05:11:54 +00:00
|
|
|
private readonly IDiskProvider _diskProvider;
|
2013-07-16 02:56:46 +00:00
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
|
|
public UpgradeMediaFileService(IRecycleBinProvider recycleBinProvider,
|
|
|
|
IMediaFileService mediaFileService,
|
2017-01-03 19:24:55 +00:00
|
|
|
IMoveMovieFiles movieFileMover,
|
2013-07-31 05:11:54 +00:00
|
|
|
IDiskProvider diskProvider,
|
2019-12-22 21:24:10 +00:00
|
|
|
IRenameMovieFileService movieFileRenamer,
|
2013-07-16 02:56:46 +00:00
|
|
|
Logger logger)
|
|
|
|
{
|
|
|
|
_recycleBinProvider = recycleBinProvider;
|
|
|
|
_mediaFileService = mediaFileService;
|
2017-01-03 19:24:55 +00:00
|
|
|
_movieFileMover = movieFileMover;
|
2013-07-31 05:11:54 +00:00
|
|
|
_diskProvider = diskProvider;
|
2019-12-22 21:24:10 +00:00
|
|
|
_movieFileRenamer = movieFileRenamer;
|
2013-07-16 02:56:46 +00:00
|
|
|
_logger = logger;
|
|
|
|
}
|
|
|
|
|
2017-03-01 23:45:12 +00:00
|
|
|
public MovieFileMoveResult UpgradeMovieFile(MovieFile movieFile, LocalMovie localMovie, bool copyOnly = false)
|
2017-01-03 19:24:55 +00:00
|
|
|
{
|
2017-03-01 23:45:12 +00:00
|
|
|
_logger.Trace("Upgrading existing movie file.");
|
2017-01-03 19:24:55 +00:00
|
|
|
var moveFileResult = new MovieFileMoveResult();
|
|
|
|
|
2020-08-28 02:33:54 +00:00
|
|
|
var existingFile = localMovie.Movie.MovieFileId > 0 ? localMovie.Movie.MovieFile : null;
|
2017-01-03 19:24:55 +00:00
|
|
|
|
2020-01-16 02:52:45 +00:00
|
|
|
var rootFolder = _diskProvider.GetParentFolder(localMovie.Movie.Path);
|
|
|
|
|
|
|
|
// If there are existing movie files and the root folder is missing, throw, so the old file isn't left behind during the import process.
|
|
|
|
if (existingFile != null && !_diskProvider.FolderExists(rootFolder))
|
|
|
|
{
|
|
|
|
throw new RootFolderNotFoundException($"Root folder '{rootFolder}' was not found.");
|
|
|
|
}
|
|
|
|
|
2017-03-02 01:20:42 +00:00
|
|
|
if (existingFile != null)
|
2017-01-20 17:26:18 +00:00
|
|
|
{
|
2017-03-02 01:20:42 +00:00
|
|
|
var movieFilePath = Path.Combine(localMovie.Movie.Path, existingFile.RelativePath);
|
2020-08-28 02:33:54 +00:00
|
|
|
var subfolder = rootFolder.GetRelativePath(_diskProvider.GetParentFolder(movieFilePath));
|
2017-03-02 01:20:42 +00:00
|
|
|
|
|
|
|
if (_diskProvider.FileExists(movieFilePath))
|
|
|
|
{
|
|
|
|
_logger.Debug("Removing existing movie file: {0}", existingFile);
|
2020-08-28 02:33:54 +00:00
|
|
|
_recycleBinProvider.DeleteFile(movieFilePath, subfolder);
|
2017-03-02 01:20:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
moveFileResult.OldFiles.Add(existingFile);
|
|
|
|
_mediaFileService.Delete(existingFile, DeleteMediaFileReason.Upgrade);
|
2017-01-20 17:26:18 +00:00
|
|
|
}
|
2017-01-03 19:24:55 +00:00
|
|
|
|
|
|
|
if (copyOnly)
|
|
|
|
{
|
2017-03-01 23:45:12 +00:00
|
|
|
moveFileResult.MovieFile = _movieFileMover.CopyMovieFile(movieFile, localMovie);
|
2017-01-03 19:24:55 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2017-03-01 23:45:12 +00:00
|
|
|
moveFileResult.MovieFile = _movieFileMover.MoveMovieFile(movieFile, localMovie);
|
2017-01-03 19:24:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return moveFileResult;
|
|
|
|
}
|
2013-07-16 02:56:46 +00:00
|
|
|
}
|
|
|
|
}
|