|
|
|
using System.IO;
|
|
|
|
using NLog;
|
|
|
|
using NzbDrone.Common.Disk;
|
|
|
|
using NzbDrone.Common.Extensions;
|
|
|
|
using NzbDrone.Core.MediaFiles.MovieImport;
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.MediaFiles
|
|
|
|
{
|
|
|
|
public interface IUpgradeMediaFiles
|
|
|
|
{
|
|
|
|
MovieFileMoveResult UpgradeMovieFile(MovieFile movieFile, LocalMovie localMovie, bool copyOnly = false);
|
|
|
|
}
|
|
|
|
|
|
|
|
public class UpgradeMediaFileService : IUpgradeMediaFiles
|
|
|
|
{
|
|
|
|
private readonly IRecycleBinProvider _recycleBinProvider;
|
|
|
|
private readonly IMediaFileService _mediaFileService;
|
|
|
|
private readonly IMoveMovieFiles _movieFileMover;
|
|
|
|
private readonly IRenameMovieFileService _movieFileRenamer;
|
|
|
|
private readonly IDiskProvider _diskProvider;
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
|
|
public UpgradeMediaFileService(IRecycleBinProvider recycleBinProvider,
|
|
|
|
IMediaFileService mediaFileService,
|
|
|
|
IMoveMovieFiles movieFileMover,
|
|
|
|
IDiskProvider diskProvider,
|
|
|
|
IRenameMovieFileService movieFileRenamer,
|
|
|
|
Logger logger)
|
|
|
|
{
|
|
|
|
_recycleBinProvider = recycleBinProvider;
|
|
|
|
_mediaFileService = mediaFileService;
|
|
|
|
_movieFileMover = movieFileMover;
|
|
|
|
_diskProvider = diskProvider;
|
|
|
|
_movieFileRenamer = movieFileRenamer;
|
|
|
|
_logger = logger;
|
|
|
|
}
|
|
|
|
|
|
|
|
public MovieFileMoveResult UpgradeMovieFile(MovieFile movieFile, LocalMovie localMovie, bool copyOnly = false)
|
|
|
|
{
|
|
|
|
_logger.Trace("Upgrading existing movie file.");
|
|
|
|
var moveFileResult = new MovieFileMoveResult();
|
|
|
|
|
|
|
|
var existingFile = localMovie.Movie.MovieFileId > 0 ? localMovie.Movie.MovieFile : null;
|
|
|
|
|
|
|
|
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.");
|
|
|
|
}
|
|
|
|
|
|
|
|
if (existingFile != null)
|
|
|
|
{
|
|
|
|
var movieFilePath = Path.Combine(localMovie.Movie.Path, existingFile.RelativePath);
|
|
|
|
var subfolder = rootFolder.GetRelativePath(_diskProvider.GetParentFolder(movieFilePath));
|
|
|
|
|
|
|
|
if (_diskProvider.FileExists(movieFilePath))
|
|
|
|
{
|
|
|
|
_logger.Debug("Removing existing movie file: {0}", existingFile);
|
|
|
|
_recycleBinProvider.DeleteFile(movieFilePath, subfolder);
|
|
|
|
}
|
|
|
|
|
|
|
|
moveFileResult.OldFiles.Add(existingFile);
|
|
|
|
_mediaFileService.Delete(existingFile, DeleteMediaFileReason.Upgrade);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (copyOnly)
|
|
|
|
{
|
|
|
|
moveFileResult.MovieFile = _movieFileMover.CopyMovieFile(movieFile, localMovie);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
moveFileResult.MovieFile = _movieFileMover.MoveMovieFile(movieFile, localMovie);
|
|
|
|
}
|
|
|
|
|
|
|
|
return moveFileResult;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|