2017-10-06 21:10:24 +00:00
|
|
|
using System;
|
2014-03-13 05:27:36 +00:00
|
|
|
using System.Collections.Generic;
|
2013-03-07 04:49:00 +00:00
|
|
|
using System.IO;
|
|
|
|
using System.Linq;
|
|
|
|
using NLog;
|
2014-01-06 06:20:08 +00:00
|
|
|
using NzbDrone.Common.Disk;
|
2013-11-27 06:52:11 +00:00
|
|
|
using NzbDrone.Common.EnsureThat;
|
2014-12-02 06:26:25 +00:00
|
|
|
using NzbDrone.Common.Extensions;
|
2014-01-14 19:21:20 +00:00
|
|
|
using NzbDrone.Core.Configuration;
|
2015-05-24 08:51:42 +00:00
|
|
|
using NzbDrone.Core.MediaFiles.Events;
|
|
|
|
using NzbDrone.Core.Messaging.Events;
|
2013-03-07 04:49:00 +00:00
|
|
|
using NzbDrone.Core.Organizer;
|
2013-07-06 21:47:49 +00:00
|
|
|
using NzbDrone.Core.Parser.Model;
|
2013-03-07 04:49:00 +00:00
|
|
|
using NzbDrone.Core.Tv;
|
2017-12-30 06:30:16 +00:00
|
|
|
using NzbDrone.Core.MediaFiles.EpisodeImport;
|
2013-03-07 04:49:00 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.MediaFiles
|
|
|
|
{
|
|
|
|
public interface IMoveEpisodeFiles
|
|
|
|
{
|
2014-02-28 00:32:08 +00:00
|
|
|
EpisodeFile MoveEpisodeFile(EpisodeFile episodeFile, Series series);
|
|
|
|
EpisodeFile MoveEpisodeFile(EpisodeFile episodeFile, LocalEpisode localEpisode);
|
2014-04-19 15:09:22 +00:00
|
|
|
EpisodeFile CopyEpisodeFile(EpisodeFile episodeFile, LocalEpisode localEpisode);
|
2013-03-07 04:49:00 +00:00
|
|
|
}
|
|
|
|
|
2013-11-28 08:03:58 +00:00
|
|
|
public class EpisodeFileMovingService : IMoveEpisodeFiles
|
2013-03-07 04:49:00 +00:00
|
|
|
{
|
|
|
|
private readonly IEpisodeService _episodeService;
|
2014-03-08 19:01:51 +00:00
|
|
|
private readonly IUpdateEpisodeFileService _updateEpisodeFileService;
|
2013-03-07 04:49:00 +00:00
|
|
|
private readonly IBuildFileNames _buildFileNames;
|
2014-11-15 00:42:22 +00:00
|
|
|
private readonly IDiskTransferService _diskTransferService;
|
2013-05-10 23:53:50 +00:00
|
|
|
private readonly IDiskProvider _diskProvider;
|
2014-11-18 15:38:31 +00:00
|
|
|
private readonly IMediaFileAttributeService _mediaFileAttributeService;
|
2015-05-24 08:51:42 +00:00
|
|
|
private readonly IEventAggregator _eventAggregator;
|
2014-08-17 18:39:08 +00:00
|
|
|
private readonly IConfigService _configService;
|
2013-03-07 04:49:00 +00:00
|
|
|
private readonly Logger _logger;
|
|
|
|
|
2013-11-28 08:03:58 +00:00
|
|
|
public EpisodeFileMovingService(IEpisodeService episodeService,
|
2014-03-08 19:01:51 +00:00
|
|
|
IUpdateEpisodeFileService updateEpisodeFileService,
|
2013-07-16 02:56:46 +00:00
|
|
|
IBuildFileNames buildFileNames,
|
2014-11-15 00:42:22 +00:00
|
|
|
IDiskTransferService diskTransferService,
|
2013-07-16 02:56:46 +00:00
|
|
|
IDiskProvider diskProvider,
|
2014-11-18 15:38:31 +00:00
|
|
|
IMediaFileAttributeService mediaFileAttributeService,
|
2015-05-24 08:51:42 +00:00
|
|
|
IEventAggregator eventAggregator,
|
2014-08-17 18:39:08 +00:00
|
|
|
IConfigService configService,
|
2013-07-16 02:56:46 +00:00
|
|
|
Logger logger)
|
2013-03-07 04:49:00 +00:00
|
|
|
{
|
|
|
|
_episodeService = episodeService;
|
2014-03-08 19:01:51 +00:00
|
|
|
_updateEpisodeFileService = updateEpisodeFileService;
|
2013-03-07 04:49:00 +00:00
|
|
|
_buildFileNames = buildFileNames;
|
2014-11-15 00:42:22 +00:00
|
|
|
_diskTransferService = diskTransferService;
|
2013-03-07 04:49:00 +00:00
|
|
|
_diskProvider = diskProvider;
|
2014-11-18 15:38:31 +00:00
|
|
|
_mediaFileAttributeService = mediaFileAttributeService;
|
2015-05-24 08:51:42 +00:00
|
|
|
_eventAggregator = eventAggregator;
|
2014-08-17 18:39:08 +00:00
|
|
|
_configService = configService;
|
2013-03-07 04:49:00 +00:00
|
|
|
_logger = logger;
|
|
|
|
}
|
|
|
|
|
2014-02-28 00:32:08 +00:00
|
|
|
public EpisodeFile MoveEpisodeFile(EpisodeFile episodeFile, Series series)
|
2013-07-19 05:23:04 +00:00
|
|
|
{
|
|
|
|
var episodes = _episodeService.GetEpisodesByFileId(episodeFile.Id);
|
2020-08-10 22:07:26 +00:00
|
|
|
var filePath = _buildFileNames.BuildFilePath(episodes, series, episodeFile, Path.GetExtension(episodeFile.RelativePath));
|
2014-01-22 05:35:01 +00:00
|
|
|
|
2015-05-24 08:51:42 +00:00
|
|
|
EnsureEpisodeFolder(episodeFile, series, episodes.Select(v => v.SeasonNumber).First(), filePath);
|
|
|
|
|
2014-03-13 20:12:42 +00:00
|
|
|
_logger.Debug("Renaming episode file: {0} to {1}", episodeFile, filePath);
|
2017-01-05 23:32:17 +00:00
|
|
|
|
2014-08-17 18:39:08 +00:00
|
|
|
return TransferFile(episodeFile, series, episodes, filePath, TransferMode.Move);
|
2013-07-19 05:23:04 +00:00
|
|
|
}
|
|
|
|
|
2014-02-28 00:32:08 +00:00
|
|
|
public EpisodeFile MoveEpisodeFile(EpisodeFile episodeFile, LocalEpisode localEpisode)
|
2013-07-06 21:47:49 +00:00
|
|
|
{
|
2020-08-10 22:07:26 +00:00
|
|
|
var filePath = _buildFileNames.BuildFilePath(localEpisode.Episodes, localEpisode.Series, episodeFile, Path.GetExtension(localEpisode.Path));
|
2014-01-22 05:35:01 +00:00
|
|
|
|
2015-05-24 08:51:42 +00:00
|
|
|
EnsureEpisodeFolder(episodeFile, localEpisode, filePath);
|
|
|
|
|
2015-03-18 20:51:21 +00:00
|
|
|
_logger.Debug("Moving episode file: {0} to {1}", episodeFile.Path, filePath);
|
2017-01-05 23:32:17 +00:00
|
|
|
|
2014-08-17 18:39:08 +00:00
|
|
|
return TransferFile(episodeFile, localEpisode.Series, localEpisode.Episodes, filePath, TransferMode.Move);
|
2013-07-06 21:47:49 +00:00
|
|
|
}
|
|
|
|
|
2014-04-19 15:09:22 +00:00
|
|
|
public EpisodeFile CopyEpisodeFile(EpisodeFile episodeFile, LocalEpisode localEpisode)
|
|
|
|
{
|
2020-08-10 22:07:26 +00:00
|
|
|
var filePath = _buildFileNames.BuildFilePath(localEpisode.Episodes, localEpisode.Series, episodeFile, Path.GetExtension(localEpisode.Path));
|
2014-04-19 15:09:22 +00:00
|
|
|
|
2015-05-24 08:51:42 +00:00
|
|
|
EnsureEpisodeFolder(episodeFile, localEpisode, filePath);
|
|
|
|
|
2014-08-17 18:39:08 +00:00
|
|
|
if (_configService.CopyUsingHardlinks)
|
|
|
|
{
|
2015-03-18 20:51:21 +00:00
|
|
|
_logger.Debug("Hardlinking episode file: {0} to {1}", episodeFile.Path, filePath);
|
2014-08-17 18:39:08 +00:00
|
|
|
return TransferFile(episodeFile, localEpisode.Series, localEpisode.Episodes, filePath, TransferMode.HardLinkOrCopy);
|
|
|
|
}
|
2015-03-18 20:18:34 +00:00
|
|
|
|
2015-03-18 20:51:21 +00:00
|
|
|
_logger.Debug("Copying episode file: {0} to {1}", episodeFile.Path, filePath);
|
2014-12-07 07:23:11 +00:00
|
|
|
return TransferFile(episodeFile, localEpisode.Series, localEpisode.Episodes, filePath, TransferMode.Copy);
|
2014-04-19 15:09:22 +00:00
|
|
|
}
|
2017-01-05 23:32:17 +00:00
|
|
|
|
2015-06-26 23:01:33 +00:00
|
|
|
private EpisodeFile TransferFile(EpisodeFile episodeFile, Series series, List<Episode> episodes, string destinationFilePath, TransferMode mode)
|
2013-07-06 21:47:49 +00:00
|
|
|
{
|
2013-11-30 23:53:07 +00:00
|
|
|
Ensure.That(episodeFile, () => episodeFile).IsNotNull();
|
2017-01-05 23:32:17 +00:00
|
|
|
Ensure.That(series, () => series).IsNotNull();
|
2015-06-26 23:01:33 +00:00
|
|
|
Ensure.That(destinationFilePath, () => destinationFilePath).IsValidPath();
|
2013-11-27 06:52:11 +00:00
|
|
|
|
2014-07-23 23:43:54 +00:00
|
|
|
var episodeFilePath = episodeFile.Path ?? Path.Combine(series.Path, episodeFile.RelativePath);
|
|
|
|
|
|
|
|
if (!_diskProvider.FileExists(episodeFilePath))
|
2013-03-07 04:49:00 +00:00
|
|
|
{
|
2014-07-23 23:43:54 +00:00
|
|
|
throw new FileNotFoundException("Episode file path does not exist", episodeFilePath);
|
2013-03-07 04:49:00 +00:00
|
|
|
}
|
|
|
|
|
2015-06-26 23:01:33 +00:00
|
|
|
if (episodeFilePath == destinationFilePath)
|
2013-03-07 04:49:00 +00:00
|
|
|
{
|
2014-07-23 23:43:54 +00:00
|
|
|
throw new SameFilenameException("File not moved, source and destination are the same", episodeFilePath);
|
2013-03-07 04:49:00 +00:00
|
|
|
}
|
|
|
|
|
2015-06-26 23:01:33 +00:00
|
|
|
_diskTransferService.TransferFile(episodeFilePath, destinationFilePath, mode);
|
2014-07-23 23:43:54 +00:00
|
|
|
|
2015-06-26 23:01:33 +00:00
|
|
|
episodeFile.RelativePath = series.Path.GetRelativePath(destinationFilePath);
|
2013-03-07 04:49:00 +00:00
|
|
|
|
2014-03-13 05:27:36 +00:00
|
|
|
_updateEpisodeFileService.ChangeFileDateForFile(episodeFile, series, episodes);
|
2014-03-08 19:01:51 +00:00
|
|
|
|
2013-10-20 18:17:56 +00:00
|
|
|
try
|
2013-09-28 22:25:22 +00:00
|
|
|
{
|
2014-11-18 15:38:31 +00:00
|
|
|
_mediaFileAttributeService.SetFolderLastWriteTime(series.Path, episodeFile.DateAdded);
|
2013-10-20 18:17:56 +00:00
|
|
|
|
|
|
|
if (series.SeasonFolder)
|
|
|
|
{
|
2015-06-26 23:01:33 +00:00
|
|
|
var seasonFolder = Path.GetDirectoryName(destinationFilePath);
|
2013-09-28 22:25:22 +00:00
|
|
|
|
2014-11-18 15:38:31 +00:00
|
|
|
_mediaFileAttributeService.SetFolderLastWriteTime(seasonFolder, episodeFile.DateAdded);
|
2013-10-20 18:17:56 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
catch (Exception ex)
|
|
|
|
{
|
2016-02-11 21:13:42 +00:00
|
|
|
_logger.Warn(ex, "Unable to set last write time");
|
2013-09-28 22:25:22 +00:00
|
|
|
}
|
|
|
|
|
2015-06-26 23:01:33 +00:00
|
|
|
_mediaFileAttributeService.SetFilePermissions(destinationFilePath);
|
2014-02-28 00:32:08 +00:00
|
|
|
|
|
|
|
return episodeFile;
|
2014-01-14 19:21:20 +00:00
|
|
|
}
|
2015-05-24 08:51:42 +00:00
|
|
|
|
|
|
|
private void EnsureEpisodeFolder(EpisodeFile episodeFile, LocalEpisode localEpisode, string filePath)
|
|
|
|
{
|
|
|
|
EnsureEpisodeFolder(episodeFile, localEpisode.Series, localEpisode.SeasonNumber, filePath);
|
|
|
|
}
|
|
|
|
|
|
|
|
private void EnsureEpisodeFolder(EpisodeFile episodeFile, Series series, int seasonNumber, string filePath)
|
|
|
|
{
|
|
|
|
var episodeFolder = Path.GetDirectoryName(filePath);
|
|
|
|
var seasonFolder = _buildFileNames.BuildSeasonPath(series, seasonNumber);
|
|
|
|
var seriesFolder = series.Path;
|
2016-08-30 04:22:25 +00:00
|
|
|
var rootFolder = new OsPath(seriesFolder).Directory.FullPath;
|
2015-05-24 08:51:42 +00:00
|
|
|
|
|
|
|
if (!_diskProvider.FolderExists(rootFolder))
|
|
|
|
{
|
2017-12-30 06:30:16 +00:00
|
|
|
throw new RootFolderNotFoundException(string.Format("Root folder '{0}' was not found.", rootFolder));
|
2015-05-24 08:51:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
var changed = false;
|
|
|
|
var newEvent = new EpisodeFolderCreatedEvent(series, episodeFile);
|
|
|
|
|
|
|
|
if (!_diskProvider.FolderExists(seriesFolder))
|
|
|
|
{
|
|
|
|
CreateFolder(seriesFolder);
|
|
|
|
newEvent.SeriesFolder = seriesFolder;
|
|
|
|
changed = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (seriesFolder != seasonFolder && !_diskProvider.FolderExists(seasonFolder))
|
|
|
|
{
|
|
|
|
CreateFolder(seasonFolder);
|
|
|
|
newEvent.SeasonFolder = seasonFolder;
|
|
|
|
changed = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (seasonFolder != episodeFolder && !_diskProvider.FolderExists(episodeFolder))
|
|
|
|
{
|
|
|
|
CreateFolder(episodeFolder);
|
|
|
|
newEvent.EpisodeFolder = episodeFolder;
|
|
|
|
changed = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (changed)
|
|
|
|
{
|
|
|
|
_eventAggregator.PublishEvent(newEvent);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private void CreateFolder(string directoryName)
|
|
|
|
{
|
|
|
|
Ensure.That(directoryName, () => directoryName).IsNotNullOrWhiteSpace();
|
|
|
|
|
2016-09-02 19:26:04 +00:00
|
|
|
var parentFolder = new OsPath(directoryName).Directory.FullPath;
|
2015-05-24 08:51:42 +00:00
|
|
|
if (!_diskProvider.FolderExists(parentFolder))
|
|
|
|
{
|
|
|
|
CreateFolder(parentFolder);
|
|
|
|
}
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
_diskProvider.CreateFolder(directoryName);
|
|
|
|
}
|
|
|
|
catch (IOException ex)
|
|
|
|
{
|
2017-01-05 23:32:17 +00:00
|
|
|
_logger.Error(ex, "Unable to create directory: {0}", directoryName);
|
2015-05-24 08:51:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
_mediaFileAttributeService.SetFolderPermissions(directoryName);
|
|
|
|
}
|
2013-03-07 04:49:00 +00:00
|
|
|
}
|
2014-11-18 15:38:31 +00:00
|
|
|
}
|