mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-10-31 07:52:37 +01:00
142 lines
4.9 KiB
C#
142 lines
4.9 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using Ninject;
|
|
using NLog;
|
|
using NzbDrone.Core.Model;
|
|
using NzbDrone.Core.Providers.Core;
|
|
using NzbDrone.Core.Providers.DownloadClients;
|
|
using NzbDrone.Core.Repository;
|
|
|
|
namespace NzbDrone.Core.Providers
|
|
{
|
|
public class DownloadProvider
|
|
{
|
|
private readonly SabProvider _sabProvider;
|
|
private readonly HistoryProvider _historyProvider;
|
|
private readonly EpisodeProvider _episodeProvider;
|
|
private readonly ExternalNotificationProvider _externalNotificationProvider;
|
|
private readonly ConfigProvider _configProvider;
|
|
private readonly BlackholeProvider _blackholeProvider;
|
|
private readonly SignalRProvider _signalRProvider;
|
|
|
|
private static readonly Logger logger = LogManager.GetCurrentClassLogger();
|
|
|
|
[Inject]
|
|
public DownloadProvider(SabProvider sabProvider, HistoryProvider historyProvider,
|
|
EpisodeProvider episodeProvider, ExternalNotificationProvider externalNotificationProvider,
|
|
ConfigProvider configProvider, BlackholeProvider blackholeProvider,
|
|
SignalRProvider signalRProvider)
|
|
{
|
|
_sabProvider = sabProvider;
|
|
_historyProvider = historyProvider;
|
|
_episodeProvider = episodeProvider;
|
|
_externalNotificationProvider = externalNotificationProvider;
|
|
_configProvider = configProvider;
|
|
_blackholeProvider = blackholeProvider;
|
|
_signalRProvider = signalRProvider;
|
|
}
|
|
|
|
public DownloadProvider()
|
|
{
|
|
}
|
|
|
|
public virtual bool DownloadReport(EpisodeParseResult parseResult)
|
|
{
|
|
var downloadTitle = GetDownloadTitle(parseResult);
|
|
|
|
var provider = GetActiveDownloadClient();
|
|
|
|
bool success = provider.DownloadNzb(parseResult.NzbUrl, GetDownloadTitle(parseResult));
|
|
|
|
if (success)
|
|
{
|
|
logger.Trace("Download added to Queue: {0}", downloadTitle);
|
|
|
|
|
|
foreach (var episode in _episodeProvider.GetEpisodesByParseResult(parseResult))
|
|
{
|
|
var history = new History();
|
|
history.Date = DateTime.Now;
|
|
history.Indexer = parseResult.Indexer;
|
|
history.IsProper = parseResult.Quality.Proper;
|
|
history.Quality = parseResult.Quality.QualityType;
|
|
history.NzbTitle = parseResult.OriginalString;
|
|
history.EpisodeId = episode.EpisodeId;
|
|
history.SeriesId = episode.SeriesId;
|
|
|
|
_historyProvider.Add(history);
|
|
_episodeProvider.MarkEpisodeAsFetched(episode.EpisodeId);
|
|
|
|
_signalRProvider.UpdateEpisodeStatus(episode.EpisodeId, EpisodeStatusType.Downloading, null);
|
|
}
|
|
|
|
_externalNotificationProvider.OnGrab(downloadTitle);
|
|
}
|
|
|
|
return success;
|
|
}
|
|
|
|
|
|
public virtual IDownloadClient GetActiveDownloadClient()
|
|
{
|
|
switch (_configProvider.DownloadClient)
|
|
{
|
|
case DownloadClientType.Blackhole:
|
|
return _blackholeProvider;
|
|
default:
|
|
return _sabProvider;
|
|
}
|
|
}
|
|
|
|
|
|
public virtual String GetDownloadTitle(EpisodeParseResult parseResult)
|
|
{
|
|
|
|
var seriesTitle = MediaFileProvider.CleanFilename(parseResult.Series.Title);
|
|
|
|
|
|
//Handle Full Naming
|
|
if (parseResult.FullSeason)
|
|
{
|
|
var seasonResult = String.Format("{0} - Season {1} [{2}]", seriesTitle,
|
|
parseResult.SeasonNumber, parseResult.Quality.QualityType);
|
|
|
|
if (parseResult.Quality.Proper)
|
|
seasonResult += " [Proper]";
|
|
|
|
return seasonResult;
|
|
}
|
|
|
|
if (parseResult.Series.IsDaily)
|
|
{
|
|
var dailyResult = String.Format("{0} - {1:yyyy-MM-dd} - {2} [{3}]", seriesTitle,
|
|
parseResult.AirDate, parseResult.EpisodeTitle, parseResult.Quality.QualityType);
|
|
|
|
if (parseResult.Quality.Proper)
|
|
dailyResult += " [Proper]";
|
|
|
|
return dailyResult;
|
|
}
|
|
|
|
//Show Name - 1x01-1x02 - Episode Name
|
|
//Show Name - 1x01 - Episode Name
|
|
var episodeString = new List<String>();
|
|
|
|
foreach (var episode in parseResult.EpisodeNumbers)
|
|
{
|
|
episodeString.Add(String.Format("{0}x{1}", parseResult.SeasonNumber, episode));
|
|
}
|
|
|
|
var epNumberString = String.Join("-", episodeString);
|
|
|
|
var result = String.Format("{0} - {1} - {2} [{3}]", seriesTitle, epNumberString, parseResult.EpisodeTitle, parseResult.Quality.QualityType);
|
|
|
|
if (parseResult.Quality.Proper)
|
|
{
|
|
result += " [Proper]";
|
|
}
|
|
|
|
return result;
|
|
}
|
|
}
|
|
} |