using System; using System.Collections.Generic; using System.Linq; using NLog; using NzbDrone.Common.Composition; using NzbDrone.Core.ThingiProvider; namespace NzbDrone.Core.Indexers { public interface IIndexerFactory : IProviderFactory { } public class IndexerFactory : ProviderFactory, IIndexerFactory { private readonly IIndexerRepository _providerRepository; private readonly IEnumerable _providers; private readonly INewznabTestService _newznabTestService; public IndexerFactory(IIndexerRepository providerRepository, IEnumerable providers, IContainer container, INewznabTestService newznabTestService, Logger logger) : base(providerRepository, providers, container, logger) { _providerRepository = providerRepository; _providers = providers; _newznabTestService = newznabTestService; } protected override void InitializeProviders() { var definitions = _providers.Where(c => c.Protocol == DownloadProtocol.Usenet) .SelectMany(indexer => indexer.DefaultDefinitions); var currentProviders = All(); var newProviders = definitions.Where(def => currentProviders.All(c => c.Implementation != def.Implementation)).ToList(); if (newProviders.Any()) { _providerRepository.InsertMany(newProviders.Cast().ToList()); } } protected override List Active() { return base.Active().Where(c => c.Enable).ToList(); } public override IndexerDefinition Create(IndexerDefinition definition) { if (definition.Implementation == typeof(Newznab.Newznab).Name) { var indexer = GetInstance(definition); _newznabTestService.Test(indexer); } return base.Create(definition); } } }