using System;
using System.Collections.Generic;
using System.Net;
using System.ServiceModel.Syndication;
using NLog;
using NzbDrone.Core.Model;
using NzbDrone.Core.Providers.Core;
using NzbDrone.Core.Repository;
namespace NzbDrone.Core.Providers.Indexer
{
public abstract class IndexerBase
{
protected readonly Logger _logger;
private readonly HttpProvider _httpProvider;
protected readonly ConfigProvider _configProvider;
private readonly IndexerProvider _indexerProvider;
protected IndexerBase(HttpProvider httpProvider, ConfigProvider configProvider, IndexerProvider indexerProvider)
{
_httpProvider = httpProvider;
_configProvider = configProvider;
_indexerProvider = indexerProvider;
_logger = LogManager.GetLogger(GetType().ToString());
}
///
/// Gets the name for the feed
///
public abstract string Name { get; }
///
/// Gets the source URL for the feed
///
protected abstract string[] Urls { get; }
///
/// Gets the credential.
///
protected virtual NetworkCredential Credentials
{
get { return null; }
}
public IndexerSetting Settings
{
get
{
return _indexerProvider.GetSettings(GetType());
}
}
///
/// Fetches RSS feed and process each news item.
///
public List Fetch()
{
_logger.Debug("Fetching feeds from " + Settings.Name);
var result = new List();
foreach (var url in Urls)
{
try
{
_logger.Trace("Downloading RSS " + url);
var reader = new SyndicationFeedXmlReader(_httpProvider.DownloadStream(url, Credentials));
var feed = SyndicationFeed.Load(reader).Items;
foreach (var item in feed)
{
try
{
var parsedEpisode = ParseFeed(item);
if (parsedEpisode != null)
{
parsedEpisode.NzbUrl = NzbDownloadUrl(item);
parsedEpisode.Indexer = Name;
parsedEpisode.NzbTitle = item.Title.Text;
result.Add(parsedEpisode);
}
}
catch (Exception itemEx)
{
_logger.ErrorException("An error occurred while processing feed item", itemEx);
}
}
}
catch (Exception feedEx)
{
_logger.ErrorException("An error occurred while processing feed", feedEx);
}
}
_logger.Info("Finished processing feeds from " + Settings.Name);
return result;
}
///
/// Parses the RSS feed item
///
/// RSS feed item to parse
/// Detailed episode info
public EpisodeParseResult ParseFeed(SyndicationItem item)
{
var episodeParseResult = Parser.ParseEpisodeInfo(item.Title.Text);
return CustomParser(item, episodeParseResult);
}
///
/// This method can be overwritten to provide indexer specific info parsing
///
/// RSS item that needs to be parsed
/// Result of the built in parse function.
///
protected virtual EpisodeParseResult CustomParser(SyndicationItem item, EpisodeParseResult currentResult)
{
return currentResult;
}
///
/// Generates direct link to download an NZB
///
/// RSS Feed item to generate the link for
/// Download link URL
protected abstract string NzbDownloadUrl(SyndicationItem item);
}
}