mirror of https://github.com/Radarr/Radarr
77 lines
2.5 KiB
C#
77 lines
2.5 KiB
C#
|
using NLog;
|
||
|
using NzbDrone.Common.Disk;
|
||
|
using NzbDrone.Common.Extensions;
|
||
|
using NzbDrone.Core.MediaFiles;
|
||
|
using NzbDrone.Core.Movies;
|
||
|
|
||
|
namespace NzbDrone.Core.Extras.Others
|
||
|
{
|
||
|
public interface IOtherExtraFileRenamer
|
||
|
{
|
||
|
void RenameOtherExtraFile(Movie movie, string path);
|
||
|
}
|
||
|
|
||
|
public class OtherExtraFileRenamer : IOtherExtraFileRenamer
|
||
|
{
|
||
|
private readonly Logger _logger;
|
||
|
private readonly IDiskProvider _diskProvider;
|
||
|
private readonly IRecycleBinProvider _recycleBinProvider;
|
||
|
private readonly IMovieService _movieService;
|
||
|
private readonly IOtherExtraFileService _otherExtraFileService;
|
||
|
|
||
|
public OtherExtraFileRenamer(IOtherExtraFileService otherExtraFileService,
|
||
|
IMovieService movieService,
|
||
|
IRecycleBinProvider recycleBinProvider,
|
||
|
IDiskProvider diskProvider,
|
||
|
Logger logger)
|
||
|
{
|
||
|
_logger = logger;
|
||
|
_diskProvider = diskProvider;
|
||
|
_recycleBinProvider = recycleBinProvider;
|
||
|
_movieService = movieService;
|
||
|
_otherExtraFileService = otherExtraFileService;
|
||
|
}
|
||
|
|
||
|
public void RenameOtherExtraFile(Movie movie, string path)
|
||
|
{
|
||
|
if (!_diskProvider.FileExists(path))
|
||
|
{
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
var relativePath = movie.Path.GetRelativePath(path);
|
||
|
|
||
|
var otherExtraFile = _otherExtraFileService.FindByPath(relativePath);
|
||
|
if (otherExtraFile != null)
|
||
|
{
|
||
|
var newPath = path + "-orig";
|
||
|
|
||
|
// Recycle an existing -orig file.
|
||
|
RemoveOtherExtraFile(movie, newPath);
|
||
|
|
||
|
// Rename the file to .*-orig
|
||
|
_diskProvider.MoveFile(path, newPath);
|
||
|
otherExtraFile.RelativePath = relativePath + "-orig";
|
||
|
otherExtraFile.Extension += "-orig";
|
||
|
_otherExtraFileService.Upsert(otherExtraFile);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
private void RemoveOtherExtraFile(Movie movie, string path)
|
||
|
{
|
||
|
if (!_diskProvider.FileExists(path))
|
||
|
{
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
var relativePath = movie.Path.GetRelativePath(path);
|
||
|
|
||
|
var otherExtraFile = _otherExtraFileService.FindByPath(relativePath);
|
||
|
if (otherExtraFile != null)
|
||
|
{
|
||
|
_recycleBinProvider.DeleteFile(path);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|