Implement NetImportSearchService

check it out

src/NzbDrone.Core/NetImport/CouchPotato/CouchPotatoImport.cs

@@ -17,6 +17,7 @@ namespace NzbDrone.Core.NetImport.CouchPotato
     {
         public override string Name => "CouchPotato";
         public override bool Enabled => true;
+        public override bool EnableAuto => false;
 
         public CouchPotatoImport(IHttpClient httpClient, IConfigService configService, IParsingService parsingService, Logger logger)
             : base(httpClient, configService, parsingService, logger)

src/NzbDrone.Core/NetImport/INetImport.cs

@@ -9,7 +9,8 @@ namespace NzbDrone.Core.NetImport
     public interface INetImport : IProvider
     {
         bool Enabled { get; }
-        
+        bool EnableAuto { get; }
+
         IList<Movie> Fetch();
     }
 }

src/NzbDrone.Core/NetImport/NetImportBase.cs

@@ -23,6 +23,7 @@ namespace NzbDrone.Core.NetImport
         public abstract string Name { get; }
 
         public abstract bool Enabled { get; }
+        public abstract bool EnableAuto { get; }
 
         public NetImportBase(IConfigService configService, IParsingService parsingService, Logger logger)
         {

src/NzbDrone.Core/NetImport/NetImportFactory.cs

@@ -14,19 +14,16 @@ namespace NzbDrone.Core.NetImport
 
     public class NetImportFactory : ProviderFactory<INetImport, NetImportDefinition>, INetImportFactory
     {
-        //private readonly IIndexerStatusService _indexerStatusService;
         private readonly INetImportRepository _providerRepository;
         private readonly Logger _logger;
 
-        public NetImportFactory(//IIndexerStatusService indexerStatusService,
-                              INetImportRepository providerRepository,
+        public NetImportFactory(INetImportRepository providerRepository,
                               IEnumerable<INetImport> providers,
                               IContainer container, 
                               IEventAggregator eventAggregator,
                               Logger logger)
             : base(providerRepository, providers, container, eventAggregator, logger)
         {
-            //_indexerStatusService = indexerStatusService;
             _providerRepository = providerRepository;
             _logger = logger;
         }
@@ -43,27 +40,16 @@ namespace NzbDrone.Core.NetImport
 
         public List<INetImport> Enabled()
         {
-            var enabledIndexers = GetAvailableProviders().Where(n => ((NetImportDefinition)n.Definition).Enabled);
-
-            var indexers = FilterBlockedIndexers(enabledIndexers);
-
+            var enabledImporters = GetAvailableProviders().Where(n => ((NetImportDefinition)n.Definition).Enabled);
+            var indexers = FilterBlockedIndexers(enabledImporters);
             return indexers.ToList();
         }
 
-        private IEnumerable<INetImport> FilterBlockedIndexers(IEnumerable<INetImport> indexers)
+        private IEnumerable<INetImport> FilterBlockedIndexers(IEnumerable<INetImport> importers)
         {
-            //var blockedIndexers = _indexerStatusService.GetBlockedIndexers().ToDictionary(v => v.IndexerId, v => v);
-
-            foreach (var indexer in indexers)
+            foreach (var importer in importers)
             {
-                /*IndexerStatus blockedIndexerStatus;
-                if (blockedIndexers.TryGetValue(indexer.Definition.Id, out blockedIndexerStatus))
-                {
-                    _logger.Debug("Temporarily ignoring indexer {0} till {1} due to recent failures.", indexer.Definition.Name, blockedIndexerStatus.DisabledTill.Value.ToLocalTime());
-                    continue;
-                }*/
-
-                yield return indexer;
+                yield return importer;
             }
         }
     }

src/NzbDrone.Core/NetImport/NetImportSearchService.cs

@@ -0,0 +1,86 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using NLog;
+using NzbDrone.Core.Tv;
+
+namespace NzbDrone.Core.NetImport
+{
+    public interface IFetchNetImport
+    {
+        List<Movie> Fetch(int listId, bool onlyEnableAuto);
+        List<Movie> FetchAndFilter(int listId, bool onlyEnableAuto);
+    }
+
+    public class NetImportSearchService : IFetchNetImport
+    {
+        private readonly Logger _logger;
+        private readonly INetImportFactory _netImportFactory;
+        private readonly IMovieService _movieService;
+
+        public NetImportSearchService(INetImportFactory netImportFactory, IMovieService movieService, Logger logger)
+        {
+            _netImportFactory = netImportFactory;
+            _movieService = movieService;
+            _logger = logger;
+        }
+
+        public List<Movie> Fetch(int listId, bool onlyEnableAuto)
+        {
+            var movies = new List<Movie>();
+
+            // Get all the lists
+            var importLists = _netImportFactory.GetAvailableProviders();
+
+            // No listId is set return all movies in all lists
+            var lists = listId == 0 ? importLists.Where(n => ((NetImportDefinition) n.Definition).Enabled == true) : importLists.Where(n => ((NetImportDefinition) n.Definition).Id == listId);
+
+            // Only return lists where enabledAuto is truthy
+            if (onlyEnableAuto)
+            {
+                lists = importLists.Where(a => a.EnableAuto == true);
+            }
+
+            foreach (var list in lists)
+            {
+                movies.AddRange(list.Fetch());
+            }
+
+            return movies;
+        }
+
+        public List<Movie> FetchAndFilter(int listId, bool onlyEnableAuto)
+        {
+            var movies = new List<Movie>();
+
+            // Get all the lists
+            var importLists = _netImportFactory.GetAvailableProviders();
+
+            // No listId is set return all movies in all lists
+            var lists = listId == 0 ? importLists.Where(n => ((NetImportDefinition)n.Definition).Enabled == true) : importLists.Where(n => ((NetImportDefinition)n.Definition).Id == listId);
+
+            // Only return lists where enabledAuto is truthy
+            if (onlyEnableAuto)
+            {
+                lists = importLists.Where(a => a.EnableAuto == true);
+            }
+
+            // Get all existing movies
+            var existingMovies = _movieService.GetAllMovies();
+
+            foreach (var list in lists)
+            {
+                movies = (List<Movie>)list.Fetch();
+            }
+
+            // remove from movies list where existMovies (choose one)
+            // movies.RemoveAll(x => existingMovies.Contains(x));
+            // return movies;
+            // movies.RemoveAll(a => existingMovies.Exists(w => w.TmdbId == a.TmdbId));
+            // return movies;
+
+            return movies.Where(x => !existingMovies.Contains(x)).ToList();
+        }
+    }
+}

src/NzbDrone.Core/NetImport/RSSImport/RSSImport.cs

@@ -17,6 +17,7 @@ namespace NzbDrone.Core.NetImport.RSSImport
     {
         public override string Name => "RSSList";
         public override bool Enabled => true;
+        public override bool EnableAuto => true;
 
         public RSSImport(IHttpClient httpClient, IConfigService configService, IParsingService parsingService, Logger logger)
             : base(httpClient, configService, parsingService, logger)

src/NzbDrone.Core/NzbDrone.Core.csproj

@@ -130,6 +130,7 @@
     <Compile Include="NetImport\CouchPotato\CouchPotatoImport.cs" />
     <Compile Include="NetImport\Exceptions\NetImportException.cs" />
     <Compile Include="NetImport\HttpNetImportBase.cs" />
+    <Compile Include="NetImport\NetImportSearchService.cs" />
     <Compile Include="NetImport\NetImportFactory.cs" />
     <Compile Include="NetImport\IProcessNetImportResponse.cs" />
     <Compile Include="NetImport\NetImportBaseSettings.cs" />