using System; using System.Collections.Generic; using System.Linq; using System.Net; using NLog; using NzbDrone.Core.Providers.Indexer; using NzbDrone.Core.Repository; using PetaPoco; namespace NzbDrone.Core.Providers { public class NewznabProvider { private static readonly Logger Logger = LogManager.GetCurrentClassLogger(); private readonly IDatabase _database; public NewznabProvider(IDatabase database) { _database = database; var newznabIndexers = new List { new NewznabDefinition { Enable = false, Name = "Nzbs.org", Url = "http://nzbs.org", BuiltIn = true }, new NewznabDefinition { Enable = false, Name = "Nzb.su", Url = "https://nzb.su", BuiltIn = true }, new NewznabDefinition { Enable = false, Name = "Dognzb.cr", Url = "https://dognzb.cr", BuiltIn = true } }; InitializeNewznabIndexers(newznabIndexers); } public NewznabProvider() { } public virtual List Enabled() { return _database.Fetch("WHERE Enable = 1"); } public virtual List All() { return _database.Fetch(); } public virtual int Save(NewznabDefinition definition) { //Cleanup the URL if it is not null or whitespace if (!String.IsNullOrWhiteSpace(definition.Url)) definition.Url = (new Uri(definition.Url).ParentUriString()); if (definition.Id == 0) { Logger.Debug("Adding Newznab definitions for {0}", definition.Name); return Convert.ToInt32(_database.Insert(definition)); } Logger.Debug("Updating Newznab definitions for {0}", definition.Name); return _database.Update(definition); } public virtual void SaveAll(IEnumerable definitions) { var definitionsList = definitions.ToList(); //Cleanup the URL for each definition foreach (var newznabDefinition in definitionsList) { CheckHostname(newznabDefinition.Url); //newznabDefinition.Url = new Uri(newznabDefinition.Url).ParentUriString(); } _database.UpdateMany(definitionsList); } private void InitializeNewznabIndexers(IList indexers) { Logger.Debug("Initializing Newznab indexers. Count {0}", indexers.Count); try { var currentIndexers = All(); Logger.Debug("Deleting broken Newznab indexer"); var brokenIndexers = currentIndexers.Where(i => String.IsNullOrEmpty(i.Name) || String.IsNullOrWhiteSpace(i.Url)).ToList(); brokenIndexers.ForEach(e => _database.Delete(e.Id)); currentIndexers = All(); foreach (var feedProvider in indexers) { try { NewznabDefinition indexerLocal = feedProvider; var currentIndexer = currentIndexers .FirstOrDefault( c => new Uri(c.Url.ToLower()).Host == new Uri(indexerLocal.Url.ToLower()).Host); if (currentIndexer == null) { var settings = new NewznabDefinition { Enable = false, Name = indexerLocal.Name, Url = indexerLocal.Url, ApiKey = indexerLocal.ApiKey, BuiltIn = true }; Save(settings); } else { currentIndexer.Url = indexerLocal.Url; currentIndexer.BuiltIn = true; Save(currentIndexer); } } catch (Exception ex) { Logger.ErrorException("An error occurred while setting up indexer: " + feedProvider.Name, ex); } } } catch (Exception ex) { Logger.ErrorException("An Error occurred while initializing Newznab Indexers", ex); } } public virtual void Delete(int id) { _database.Delete(id); } public virtual void CheckHostname(string url) { try { var uri = new Uri(url); var hostname = uri.DnsSafeHost; Dns.GetHostEntry(hostname); } catch (Exception ex) { Logger.Error("Invalid address {0}, please correct the site URL.", url); Logger.TraceException(ex.Message, ex); throw; } } } }