Prowlarr/NzbDrone.Core/Providers/IndexerProvider.cs

using System;
using System.Collections.Generic;
using System.Linq;
using NLog;
using NzbDrone.Core.Providers.Core;
using NzbDrone.Core.Providers.Indexer;
using NzbDrone.Core.Repository;
using SubSonic.Repository;

namespace NzbDrone.Core.Providers
{
    public class IndexerProvider
    {
        private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
        private readonly IRepository _repository;

        public IndexerProvider(IRepository repository)
        {
            _repository = repository;
        }

        public IndexerProvider()
        {

        }

        public virtual List<IndexerSetting> AllIndexers()
        {
            return _repository.All<IndexerSetting>().ToList();
        }

        public virtual void SaveSettings(IndexerSetting settings)
        {
            if (settings.Id == 0)
            {
                Logger.Debug("Adding Indexer settings for {0}", settings.Name);
                _repository.Add(settings);
            }
            else
            {
                Logger.Debug("Updating Indexer settings for {0}", settings.Name);
                _repository.Update(settings);
            }
        }

        public virtual IndexerSetting GetSettings(Type type)
        {
            return _repository.Single<IndexerSetting>(s => s.IndexProviderType == type.ToString());
        }

        public virtual IndexerSetting GetSettings(int id)
        {
            return _repository.Single<IndexerSetting>(s => s.Id == id);
        }

        public virtual void InitializeIndexers(IList<IndexerProviderBase> indexers)
        {
            Logger.Info("Initializing indexers. Count {0}", indexers.Count);

            var currentIndexers = AllIndexers();

            foreach (var feedProvider in indexers)
            {
                IndexerProviderBase indexerProviderLocal = feedProvider;
                if (!currentIndexers.Exists(c => c.IndexProviderType == indexerProviderLocal.GetType().ToString()))
                {
                    var settings = new IndexerSetting()
                                       {
                                           Enable = false,
                                           IndexProviderType = indexerProviderLocal.GetType().ToString(),
                                           Name = indexerProviderLocal.Name
                                       };

                    SaveSettings(settings);
                }
            }
        }
    }
}