mirror of https://github.com/Sonarr/Sonarr
147 lines
5.5 KiB
C#
147 lines
5.5 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 readonly PneumaticProvider _pneumaticProvider;
|
|
|
|
private static readonly Logger logger = LogManager.GetCurrentClassLogger();
|
|
|
|
[Inject]
|
|
public DownloadProvider(SabProvider sabProvider, HistoryProvider historyProvider,
|
|
EpisodeProvider episodeProvider, ExternalNotificationProvider externalNotificationProvider,
|
|
ConfigProvider configProvider, BlackholeProvider blackholeProvider,
|
|
SignalRProvider signalRProvider, PneumaticProvider pneumaticProvider)
|
|
{
|
|
_sabProvider = sabProvider;
|
|
_historyProvider = historyProvider;
|
|
_episodeProvider = episodeProvider;
|
|
_externalNotificationProvider = externalNotificationProvider;
|
|
_configProvider = configProvider;
|
|
_blackholeProvider = blackholeProvider;
|
|
_signalRProvider = signalRProvider;
|
|
_pneumaticProvider = pneumaticProvider;
|
|
}
|
|
|
|
public DownloadProvider()
|
|
{
|
|
}
|
|
|
|
public virtual bool DownloadReport(EpisodeParseResult parseResult)
|
|
{
|
|
var downloadTitle = GetDownloadTitle(parseResult);
|
|
|
|
var provider = GetActiveDownloadClient();
|
|
|
|
bool success = provider.DownloadNzb(parseResult.NzbUrl, downloadTitle);
|
|
|
|
if (success)
|
|
{
|
|
logger.Trace("Download added to Queue: {0}", downloadTitle);
|
|
|
|
foreach (var episode in parseResult.Episodes)
|
|
{
|
|
var history = new History
|
|
{
|
|
Date = DateTime.Now,
|
|
Indexer = parseResult.Indexer,
|
|
IsProper = parseResult.Quality.Proper,
|
|
Quality = parseResult.Quality.QualityType,
|
|
NzbTitle = parseResult.OriginalString,
|
|
EpisodeId = episode.EpisodeId,
|
|
SeriesId = episode.SeriesId,
|
|
NzbInfoUrl = parseResult.NzbInfoUrl,
|
|
ReleaseGroup = parseResult.ReleaseGroup,
|
|
};
|
|
|
|
_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;
|
|
|
|
case DownloadClientType.Pneumatic:
|
|
return _pneumaticProvider;
|
|
|
|
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.Episodes)
|
|
{
|
|
episodeString.Add(String.Format("{0}x{1:00}", parseResult.SeasonNumber, episode.EpisodeNumber));
|
|
}
|
|
|
|
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;
|
|
}
|
|
}
|
|
} |