Sonarr/NzbDrone.Core/Providers/Jobs/PostDownloadScanJob.cs

97 lines
3.3 KiB
C#
Raw Normal View History

using System;
using System.IO;
using Ninject;
using NLog;
using NzbDrone.Core.Model.Notification;
using NzbDrone.Core.Providers.Core;
namespace NzbDrone.Core.Providers.Jobs
{
public class PostDownloadScanJob : IJob
{
private readonly ConfigProvider _configProvider;
private readonly DiskProvider _diskProvider;
2011-06-20 03:04:08 +00:00
private readonly DiskScanProvider _diskScanProvider;
private readonly SeriesProvider _seriesProvider;
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
[Inject]
public PostDownloadScanJob(ConfigProvider configProvider, DiskProvider diskProvider,
2011-06-20 03:04:08 +00:00
DiskScanProvider diskScanProvider, SeriesProvider seriesProvider)
{
_configProvider = configProvider;
_diskProvider = diskProvider;
2011-06-20 03:04:08 +00:00
_diskScanProvider = diskScanProvider;
_seriesProvider = seriesProvider;
}
public PostDownloadScanJob()
{
}
public string Name
{
get { return "Post Download Media File Scan"; }
}
public int DefaultInterval
{
get { return 1; }
}
public virtual void Start(ProgressNotification notification, int targetId)
{
Logger.Debug("Starting New Download Scan Job");
var dropFolder = _configProvider.SabDropDirectory;
2011-06-21 05:44:01 +00:00
if (String.IsNullOrWhiteSpace(dropFolder))
{
2011-06-21 05:44:01 +00:00
Logger.Debug("Skipping drop folder scan. No drop folder is defined.");
return;
}
if (!_diskProvider.FolderExists(dropFolder))
{
Logger.Warn("Unable to Scan for New Downloads - folder Doesn't exist: {0}", dropFolder);
return;
}
2011-06-21 05:44:01 +00:00
foreach (var subfolder in _diskProvider.GetDirectories(dropFolder))
{
2011-06-21 05:44:01 +00:00
var subfolderInfo = new DirectoryInfo(subfolder);
2011-06-21 05:44:01 +00:00
if (subfolderInfo.Name.StartsWith("_UNPACK_", StringComparison.CurrentCultureIgnoreCase))
{
2011-06-21 05:44:01 +00:00
Logger.Info("Folder [{0}] is still being unpacked. skipping.", subfolder);
continue;
}
2011-06-21 05:44:01 +00:00
if (subfolderInfo.Name.StartsWith("_FAILED_", StringComparison.CurrentCultureIgnoreCase))
{
2011-06-21 05:44:01 +00:00
Logger.Info("Folder [{0}] is marked as failed. skipping.", subfolder);
continue;
}
//Parse the Folder name
2011-06-21 05:44:01 +00:00
var seriesName = Parser.ParseSeriesName(subfolderInfo.Name);
var series = _seriesProvider.FindSeries(seriesName);
if (series == null)
{
Logger.Warn("Unable to Import new download, series is not being watched");
return;
}
2011-06-21 05:44:01 +00:00
var importedFiles = _diskScanProvider.Scan(series, subfolder);
2011-06-22 01:12:20 +00:00
importedFiles.ForEach(file => _diskScanProvider.MoveEpisodeFile(file));
//Delete the folder only if all files were removed
if (_diskProvider.GetFiles(subfolder, "*.*", SearchOption.AllDirectories).Length == 0)
_diskProvider.DeleteFolder(subfolder, false);
}
2011-06-21 05:44:01 +00:00
Logger.Debug("New Download Scan Job completed successfully");
}
}
}