You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.6 KiB
66 lines
2.6 KiB
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NzbDrone.Core.DecisionEngine.Specifications.Search;
|
|
using NzbDrone.Core.IndexerSearch.Definitions;
|
|
using NzbDrone.Core.Parser;
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
namespace NzbDrone.Core.DecisionEngine
|
|
{
|
|
public interface IMakeDownloadDecision
|
|
{
|
|
IEnumerable<DownloadDecision> GetRssDecision(IEnumerable<ReportInfo> reports);
|
|
IEnumerable<DownloadDecision> GetSearchDecision(IEnumerable<ReportInfo> reports, SearchDefinitionBase searchDefinitionBase);
|
|
}
|
|
|
|
public class DownloadDecisionMaker : IMakeDownloadDecision
|
|
{
|
|
private readonly IEnumerable<IRejectWithReason> _specifications;
|
|
private readonly IParsingService _parsingService;
|
|
|
|
public DownloadDecisionMaker(IEnumerable<IRejectWithReason> specifications, IParsingService parsingService)
|
|
{
|
|
_specifications = specifications;
|
|
_parsingService = parsingService;
|
|
}
|
|
|
|
public IEnumerable<DownloadDecision> GetRssDecision(IEnumerable<ReportInfo> reports)
|
|
{
|
|
foreach (var report in reports)
|
|
{
|
|
var parseResult = _parsingService.Map(report);
|
|
yield return new DownloadDecision(parseResult, GetGeneralRejectionReasons(parseResult).ToArray());
|
|
}
|
|
|
|
}
|
|
|
|
public IEnumerable<DownloadDecision> GetSearchDecision(IEnumerable<ReportInfo> reports, SearchDefinitionBase searchDefinitionBase)
|
|
{
|
|
foreach (var report in reports)
|
|
{
|
|
var parseResult = _parsingService.Map(report);
|
|
var generalReasons = GetGeneralRejectionReasons(parseResult);
|
|
var searchReasons = GetSearchRejectionReasons(parseResult, searchDefinitionBase);
|
|
|
|
yield return new DownloadDecision(parseResult, generalReasons.Union(searchReasons).ToArray());
|
|
}
|
|
}
|
|
|
|
|
|
private IEnumerable<string> GetGeneralRejectionReasons(RemoteEpisode report)
|
|
{
|
|
return _specifications
|
|
.OfType<IDecisionEngineSpecification>()
|
|
.Where(spec => !spec.IsSatisfiedBy(report))
|
|
.Select(spec => spec.RejectionReason);
|
|
}
|
|
|
|
private IEnumerable<string> GetSearchRejectionReasons(RemoteEpisode report, SearchDefinitionBase searchDefinitionBase)
|
|
{
|
|
return _specifications
|
|
.OfType<IDecisionEngineSearchSpecification>()
|
|
.Where(spec => !spec.IsSatisfiedBy(report, searchDefinitionBase))
|
|
.Select(spec => spec.RejectionReason);
|
|
}
|
|
}
|
|
} |