Lidarr/NzbDrone.Core/Download/Clients/BlackholeProvider.cs

68 lines
2.3 KiB
C#
Raw Normal View History

using System;
using System.IO;
using NLog;
using NzbDrone.Common;
2013-02-24 06:48:52 +00:00
using NzbDrone.Core.Configuration;
using NzbDrone.Core.DecisionEngine.Specifications;
using NzbDrone.Core.Model;
using NzbDrone.Core.Organizer;
2013-03-05 05:33:34 +00:00
namespace NzbDrone.Core.Download.Clients
{
public class BlackholeProvider : IDownloadClient
{
2013-02-24 06:48:52 +00:00
private readonly IConfigService _configService;
2013-04-10 23:41:45 +00:00
private readonly IHttpProvider _httpProvider;
private readonly DiskProvider _diskProvider;
private readonly UpgradeHistorySpecification _upgradeHistorySpecification;
private static readonly Logger logger = LogManager.GetCurrentClassLogger();
2013-04-10 23:41:45 +00:00
public BlackholeProvider(IConfigService configService, IHttpProvider httpProvider,
DiskProvider diskProvider, UpgradeHistorySpecification upgradeHistorySpecification)
{
2013-02-24 06:48:52 +00:00
_configService = configService;
_httpProvider = httpProvider;
_diskProvider = diskProvider;
_upgradeHistorySpecification = upgradeHistorySpecification;
}
public BlackholeProvider()
{
}
public virtual bool DownloadNzb(string url, string title, bool recentlyAired)
{
try
{
title = FileNameBuilder.CleanFilename(title);
2013-02-24 06:48:52 +00:00
var filename = Path.Combine(_configService.BlackholeDirectory, title + ".nzb");
if (_diskProvider.FileExists(filename))
{
//Return true so a lesser quality is not returned.
2012-02-11 08:16:26 +00:00
logger.Info("NZB already exists on disk: {0}", filename);
return true;
}
logger.Trace("Downloading NZB from: {0} to: {1}", url, filename);
_httpProvider.DownloadFile(url, filename);
logger.Trace("NZB Download succeeded, saved to: {0}", filename);
return true;
}
catch (Exception ex)
{
logger.WarnException("Failed to download NZB: " + url, ex);
return false;
}
}
public virtual bool IsInQueue(IndexerParseResult newParseResult)
{
return !_upgradeHistorySpecification.IsSatisfiedBy(newParseResult);
}
}
}