mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-10-31 16:02:29 +01:00
101 lines
3.0 KiB
C#
101 lines
3.0 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using NLog;
|
|
using NzbDrone.Core.DecisionEngine;
|
|
using NzbDrone.Core.Download;
|
|
using NzbDrone.Core.Model;
|
|
|
|
namespace NzbDrone.Core.Indexers
|
|
{
|
|
|
|
public interface ISyncRss
|
|
{
|
|
void Sync();
|
|
}
|
|
|
|
public class RssSyncService : ISyncRss
|
|
{
|
|
private readonly IFetchAndParseRss _rssFetcherAndParser;
|
|
private readonly IMakeDownloadDecision _downloadDecisionMaker;
|
|
private readonly IDownloadService _downloadService;
|
|
private readonly Logger _logger;
|
|
|
|
public RssSyncService(IFetchAndParseRss rssFetcherAndParser, IMakeDownloadDecision downloadDecisionMaker, IDownloadService downloadService, Logger logger)
|
|
{
|
|
_rssFetcherAndParser = rssFetcherAndParser;
|
|
_downloadDecisionMaker = downloadDecisionMaker;
|
|
_downloadService = downloadService;
|
|
_logger = logger;
|
|
}
|
|
|
|
|
|
public void Sync()
|
|
{
|
|
_logger.Info("Starting RSS Sync");
|
|
|
|
var parseResults = _rssFetcherAndParser.Fetch();
|
|
var decisions = _downloadDecisionMaker.GetRssDecision(parseResults);
|
|
|
|
var qualifiedReports = decisions
|
|
.Where(c => c.Approved)
|
|
.Select(c => c.ParseResult)
|
|
.OrderByDescending(c => c.Quality)
|
|
.ThenBy(c => c.EpisodeNumbers.MinOrDefault())
|
|
.ThenBy(c => c.Age);
|
|
|
|
|
|
foreach (var episodeParseResult in qualifiedReports)
|
|
{
|
|
try
|
|
{
|
|
_downloadService.DownloadReport(episodeParseResult);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
_logger.WarnException("Couldn't add report to download queue. " + episodeParseResult, e);
|
|
}
|
|
}
|
|
|
|
_logger.Info("RSS Sync Completed. Reports found {0}, Fetches attempted {1}", parseResults.Count, qualifiedReports);
|
|
}
|
|
}
|
|
|
|
|
|
public interface IFetchAndParseRss
|
|
{
|
|
List<IndexerParseResult> Fetch();
|
|
}
|
|
|
|
public class FetchAndParseRssService : IFetchAndParseRss
|
|
{
|
|
private readonly IIndexerService _indexerService;
|
|
private readonly IFetchFeedFromIndexers _feedFetcher;
|
|
|
|
public FetchAndParseRssService(IIndexerService indexerService, IFetchFeedFromIndexers feedFetcher)
|
|
{
|
|
_indexerService = indexerService;
|
|
_feedFetcher = feedFetcher;
|
|
}
|
|
|
|
public List<IndexerParseResult> Fetch()
|
|
{
|
|
var result = new List<IndexerParseResult>();
|
|
|
|
var indexers = _indexerService.GetAvailableIndexers();
|
|
|
|
Parallel.ForEach(indexers, indexer =>
|
|
{
|
|
var indexerFeed = _feedFetcher.FetchRss(indexer);
|
|
|
|
lock (result)
|
|
{
|
|
result.AddRange(indexerFeed);
|
|
}
|
|
});
|
|
|
|
return result;
|
|
}
|
|
}
|
|
} |