|
|
|
@ -1,202 +1,19 @@
|
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using NLog;
|
|
|
|
|
using NzbDrone.Common.Serializer;
|
|
|
|
|
using NzbDrone.Core.Configuration;
|
|
|
|
|
using NzbDrone.Core.Indexers.Newznab;
|
|
|
|
|
using NzbDrone.Core.Lifecycle;
|
|
|
|
|
using NzbDrone.Core.Messaging.Events;
|
|
|
|
|
using NzbDrone.Core.ThingiProvider;
|
|
|
|
|
using Omu.ValueInjecter;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Indexers
|
|
|
|
|
{
|
|
|
|
|
public class Indexer
|
|
|
|
|
public interface IIndexerService : IProviderFactory<IIndexer, IndexerDefinition>
|
|
|
|
|
{
|
|
|
|
|
public int Id { get; set; }
|
|
|
|
|
public string Name { get; set; }
|
|
|
|
|
public bool Enable { get; set; }
|
|
|
|
|
public IProviderConfig Settings { get; set; }
|
|
|
|
|
public IIndexer Instance { get; set; }
|
|
|
|
|
public string Implementation { get; set; }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public interface IIndexerService
|
|
|
|
|
{
|
|
|
|
|
List<Indexer> All();
|
|
|
|
|
List<IIndexer> GetAvailableIndexers();
|
|
|
|
|
Indexer Get(int id);
|
|
|
|
|
Indexer Get(string name);
|
|
|
|
|
List<Indexer> Schema();
|
|
|
|
|
Indexer Create(Indexer indexer);
|
|
|
|
|
Indexer Update(Indexer indexer);
|
|
|
|
|
void Delete(int id);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public class IndexerService : IIndexerService, IHandle<ApplicationStartedEvent>
|
|
|
|
|
public class IndexerService : ProviderFactory<IIndexer, IndexerDefinition>
|
|
|
|
|
{
|
|
|
|
|
private readonly IIndexerRepository _indexerRepository;
|
|
|
|
|
private readonly IConfigFileProvider _configFileProvider;
|
|
|
|
|
private readonly INewznabTestService _newznabTestService;
|
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
|
|
|
|
private readonly List<IIndexer> _indexers;
|
|
|
|
|
|
|
|
|
|
public IndexerService(IIndexerRepository indexerRepository,
|
|
|
|
|
IEnumerable<IIndexer> indexers,
|
|
|
|
|
IConfigFileProvider configFileProvider,
|
|
|
|
|
INewznabTestService newznabTestService,
|
|
|
|
|
Logger logger)
|
|
|
|
|
{
|
|
|
|
|
_indexerRepository = indexerRepository;
|
|
|
|
|
_configFileProvider = configFileProvider;
|
|
|
|
|
_newznabTestService = newznabTestService;
|
|
|
|
|
_logger = logger;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (!configFileProvider.Torrent)
|
|
|
|
|
{
|
|
|
|
|
_indexers = indexers.Where(c => c.Kind != IndexerKind.Torrent).ToList();
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
_indexers = indexers.ToList();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public List<Indexer> All()
|
|
|
|
|
{
|
|
|
|
|
return _indexerRepository.All().Select(ToIndexer).ToList();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public List<IIndexer> GetAvailableIndexers()
|
|
|
|
|
public IndexerService(IProviderRepository<IndexerDefinition> providerRepository, IEnumerable<IIndexer> providers, Logger logger)
|
|
|
|
|
: base(providerRepository, providers, logger)
|
|
|
|
|
{
|
|
|
|
|
return All().Where(c => c.Enable && c.Settings.Validate().IsValid).Select(c => c.Instance).ToList();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public Indexer Get(int id)
|
|
|
|
|
{
|
|
|
|
|
return ToIndexer(_indexerRepository.Get(id));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public Indexer Get(string name)
|
|
|
|
|
{
|
|
|
|
|
return ToIndexer(_indexerRepository.Get(name));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public List<Indexer> Schema()
|
|
|
|
|
{
|
|
|
|
|
var indexers = new List<Indexer>();
|
|
|
|
|
|
|
|
|
|
var newznab = new Indexer();
|
|
|
|
|
newznab.Instance = new Newznab.Newznab();
|
|
|
|
|
newznab.Id = 1;
|
|
|
|
|
newznab.Name = "Newznab";
|
|
|
|
|
newznab.Settings = new NewznabSettings();
|
|
|
|
|
newznab.Implementation = "Newznab";
|
|
|
|
|
|
|
|
|
|
indexers.Add(newznab);
|
|
|
|
|
|
|
|
|
|
return indexers;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public Indexer Create(Indexer indexer)
|
|
|
|
|
{
|
|
|
|
|
var definition = new IndexerDefinition
|
|
|
|
|
{
|
|
|
|
|
Name = indexer.Name,
|
|
|
|
|
Enable = indexer.Enable,
|
|
|
|
|
Implementation = indexer.Implementation,
|
|
|
|
|
Settings = indexer.Settings
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
var instance = ToIndexer(definition).Instance;
|
|
|
|
|
_newznabTestService.Test(instance);
|
|
|
|
|
|
|
|
|
|
definition = _indexerRepository.Insert(definition);
|
|
|
|
|
indexer.Id = definition.Id;
|
|
|
|
|
|
|
|
|
|
return indexer;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public Indexer Update(Indexer indexer)
|
|
|
|
|
{
|
|
|
|
|
var definition = _indexerRepository.Get(indexer.Id);
|
|
|
|
|
definition.InjectFrom(indexer);
|
|
|
|
|
definition.Settings = indexer.Settings;
|
|
|
|
|
_indexerRepository.Update(definition);
|
|
|
|
|
|
|
|
|
|
return indexer;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void Delete(int id)
|
|
|
|
|
{
|
|
|
|
|
_indexerRepository.Delete(id);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Indexer ToIndexer(IndexerDefinition definition)
|
|
|
|
|
{
|
|
|
|
|
var indexer = new Indexer();
|
|
|
|
|
indexer.Id = definition.Id;
|
|
|
|
|
indexer.Enable = definition.Enable;
|
|
|
|
|
indexer.Instance = GetInstance(definition);
|
|
|
|
|
indexer.Name = definition.Name;
|
|
|
|
|
indexer.Implementation = definition.Implementation;
|
|
|
|
|
|
|
|
|
|
if (indexer.Instance.GetType().GetMethod("ImportSettingsFromJson") != null)
|
|
|
|
|
{
|
|
|
|
|
indexer.Settings = ((dynamic)indexer.Instance).ImportSettingsFromJson(definition.Settings);
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
indexer.Settings = NullSetting.Instance;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return indexer;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private IIndexer GetInstance(IndexerDefinition indexerDefinition)
|
|
|
|
|
{
|
|
|
|
|
var type = GetImplementation(indexerDefinition);
|
|
|
|
|
var instance = (IIndexer)Activator.CreateInstance(type);
|
|
|
|
|
instance.InstanceDefinition = indexerDefinition;
|
|
|
|
|
return instance;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Type GetImplementation(IndexerDefinition indexerDefinition)
|
|
|
|
|
{
|
|
|
|
|
return _indexers.Select(c => c.GetType()).SingleOrDefault(c => c.Name.Equals(indexerDefinition.Implementation, StringComparison.InvariantCultureIgnoreCase));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void Handle(ApplicationStartedEvent message)
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Initializing indexers. Count {0}", _indexers.Count);
|
|
|
|
|
|
|
|
|
|
RemoveMissingImplementations();
|
|
|
|
|
|
|
|
|
|
var definitions = _indexers.SelectMany(indexer => indexer.DefaultDefinitions);
|
|
|
|
|
|
|
|
|
|
var currentIndexer = All();
|
|
|
|
|
|
|
|
|
|
var newIndexers = definitions.Where(def => currentIndexer.All(c => c.Implementation != def.Implementation)).ToList();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (newIndexers.Any())
|
|
|
|
|
{
|
|
|
|
|
_indexerRepository.InsertMany(newIndexers);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void RemoveMissingImplementations()
|
|
|
|
|
{
|
|
|
|
|
var storedIndexers = _indexerRepository.All();
|
|
|
|
|
|
|
|
|
|
foreach (var indexerDefinition in storedIndexers.Where(i => GetImplementation(i) == null))
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Removing Indexer {0} ", indexerDefinition.Name);
|
|
|
|
|
_indexerRepository.Delete(indexerDefinition);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|