2017-08-27 20:24:44 +00:00
|
|
|
using System;
|
2013-04-07 07:30:37 +00:00
|
|
|
using System.Collections.Generic;
|
2020-01-03 12:49:24 +00:00
|
|
|
using System.Linq;
|
2013-04-07 07:30:37 +00:00
|
|
|
using System.Threading.Tasks;
|
|
|
|
using NLog;
|
2020-01-03 12:49:24 +00:00
|
|
|
using NzbDrone.Common.Extensions;
|
2014-07-23 23:43:54 +00:00
|
|
|
using NzbDrone.Common.Instrumentation.Extensions;
|
2020-01-03 12:49:24 +00:00
|
|
|
using NzbDrone.Common.TPL;
|
2013-04-07 07:30:37 +00:00
|
|
|
using NzbDrone.Core.DecisionEngine;
|
|
|
|
using NzbDrone.Core.Indexers;
|
2020-01-03 12:49:24 +00:00
|
|
|
using NzbDrone.Core.IndexerSearch.Definitions;
|
2017-07-07 21:04:03 +00:00
|
|
|
using NzbDrone.Core.Music;
|
2020-01-03 12:49:24 +00:00
|
|
|
using NzbDrone.Core.Parser.Model;
|
2013-04-07 07:30:37 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.IndexerSearch
|
|
|
|
{
|
|
|
|
public interface ISearchForNzb
|
|
|
|
{
|
2017-12-02 05:03:12 +00:00
|
|
|
List<DownloadDecision> AlbumSearch(int albumId, bool missingOnly, bool userInvokedSearch, bool interactiveSearch);
|
|
|
|
List<DownloadDecision> ArtistSearch(int artistId, bool missingOnly, bool userInvokedSearch, bool interactiveSearch);
|
2013-04-07 07:30:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public class NzbSearchService : ISearchForNzb
|
|
|
|
{
|
2013-09-24 23:42:55 +00:00
|
|
|
private readonly IIndexerFactory _indexerFactory;
|
2017-07-07 21:04:03 +00:00
|
|
|
private readonly IAlbumService _albumService;
|
2017-07-08 03:27:18 +00:00
|
|
|
private readonly IArtistService _artistService;
|
2013-04-07 07:30:37 +00:00
|
|
|
private readonly IMakeDownloadDecision _makeDownloadDecision;
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
2013-09-24 23:42:55 +00:00
|
|
|
public NzbSearchService(IIndexerFactory indexerFactory,
|
2017-07-07 21:04:03 +00:00
|
|
|
IAlbumService albumService,
|
2017-07-08 03:27:18 +00:00
|
|
|
IArtistService artistService,
|
2013-06-19 02:08:29 +00:00
|
|
|
IMakeDownloadDecision makeDownloadDecision,
|
|
|
|
Logger logger)
|
2013-04-07 07:30:37 +00:00
|
|
|
{
|
2013-09-24 23:42:55 +00:00
|
|
|
_indexerFactory = indexerFactory;
|
2017-07-07 21:04:03 +00:00
|
|
|
_albumService = albumService;
|
2017-07-08 03:27:18 +00:00
|
|
|
_artistService = artistService;
|
2013-04-07 07:30:37 +00:00
|
|
|
_makeDownloadDecision = makeDownloadDecision;
|
|
|
|
_logger = logger;
|
|
|
|
}
|
|
|
|
|
2017-12-02 05:03:12 +00:00
|
|
|
public List<DownloadDecision> AlbumSearch(int albumId, bool missingOnly, bool userInvokedSearch, bool interactiveSearch)
|
2017-07-07 21:04:03 +00:00
|
|
|
{
|
2017-07-08 03:28:47 +00:00
|
|
|
var album = _albumService.GetAlbum(albumId);
|
2017-12-02 05:03:12 +00:00
|
|
|
return AlbumSearch(album, missingOnly, userInvokedSearch, interactiveSearch);
|
2017-07-07 21:04:03 +00:00
|
|
|
}
|
|
|
|
|
2017-12-02 05:03:12 +00:00
|
|
|
public List<DownloadDecision> ArtistSearch(int artistId, bool missingOnly, bool userInvokedSearch, bool interactiveSearch)
|
2017-07-20 02:34:16 +00:00
|
|
|
{
|
|
|
|
var artist = _artistService.GetArtist(artistId);
|
2017-12-02 05:03:12 +00:00
|
|
|
return ArtistSearch(artist, missingOnly, userInvokedSearch, interactiveSearch);
|
2017-07-20 02:34:16 +00:00
|
|
|
}
|
|
|
|
|
2017-12-02 05:03:12 +00:00
|
|
|
public List<DownloadDecision> ArtistSearch(Artist artist, bool missingOnly, bool userInvokedSearch, bool interactiveSearch)
|
2017-07-20 02:34:16 +00:00
|
|
|
{
|
2017-12-02 05:03:12 +00:00
|
|
|
var searchSpec = Get<ArtistSearchCriteria>(artist, userInvokedSearch, interactiveSearch);
|
2017-08-14 02:58:42 +00:00
|
|
|
var albums = _albumService.GetAlbumsByArtist(artist.Id);
|
|
|
|
|
2018-01-08 01:55:03 +00:00
|
|
|
albums = albums.Where(a => a.Monitored).ToList();
|
|
|
|
|
2017-08-14 02:58:42 +00:00
|
|
|
searchSpec.Albums = albums;
|
2020-01-03 12:49:24 +00:00
|
|
|
|
2017-07-20 02:34:16 +00:00
|
|
|
return Dispatch(indexer => indexer.Fetch(searchSpec), searchSpec);
|
|
|
|
}
|
|
|
|
|
2017-12-02 05:03:12 +00:00
|
|
|
public List<DownloadDecision> AlbumSearch(Album album, bool missingOnly, bool userInvokedSearch, bool interactiveSearch)
|
2017-07-07 21:04:03 +00:00
|
|
|
{
|
2017-08-14 02:58:42 +00:00
|
|
|
var artist = _artistService.GetArtist(album.ArtistId);
|
|
|
|
|
2017-12-02 05:03:12 +00:00
|
|
|
var searchSpec = Get<AlbumSearchCriteria>(artist, new List<Album> { album }, userInvokedSearch, interactiveSearch);
|
2017-08-14 02:58:42 +00:00
|
|
|
|
|
|
|
searchSpec.AlbumTitle = album.Title;
|
|
|
|
if (album.ReleaseDate.HasValue)
|
|
|
|
{
|
|
|
|
searchSpec.AlbumYear = album.ReleaseDate.Value.Year;
|
|
|
|
}
|
2018-09-16 15:32:42 +00:00
|
|
|
|
|
|
|
if (album.Disambiguation.IsNotNullOrWhiteSpace())
|
|
|
|
{
|
|
|
|
searchSpec.Disambiguation = album.Disambiguation;
|
|
|
|
}
|
|
|
|
|
2017-07-07 21:04:03 +00:00
|
|
|
return Dispatch(indexer => indexer.Fetch(searchSpec), searchSpec);
|
|
|
|
}
|
|
|
|
|
2020-01-03 12:49:24 +00:00
|
|
|
private TSpec Get<TSpec>(Artist artist, List<Album> albums, bool userInvokedSearch, bool interactiveSearch)
|
|
|
|
where TSpec : SearchCriteriaBase, new()
|
2017-07-07 21:04:03 +00:00
|
|
|
{
|
|
|
|
var spec = new TSpec();
|
|
|
|
|
2017-08-14 02:58:42 +00:00
|
|
|
spec.Albums = albums;
|
|
|
|
spec.Artist = artist;
|
2017-07-07 21:04:03 +00:00
|
|
|
spec.UserInvokedSearch = userInvokedSearch;
|
2017-12-02 05:03:12 +00:00
|
|
|
spec.InteractiveSearch = interactiveSearch;
|
2017-07-07 21:04:03 +00:00
|
|
|
|
|
|
|
return spec;
|
|
|
|
}
|
|
|
|
|
2020-01-03 12:49:24 +00:00
|
|
|
private static TSpec Get<TSpec>(Artist artist, bool userInvokedSearch, bool interactiveSearch)
|
|
|
|
where TSpec : SearchCriteriaBase, new()
|
2017-07-20 02:34:16 +00:00
|
|
|
{
|
|
|
|
var spec = new TSpec();
|
|
|
|
spec.Artist = artist;
|
|
|
|
spec.UserInvokedSearch = userInvokedSearch;
|
2017-12-02 05:03:12 +00:00
|
|
|
spec.InteractiveSearch = interactiveSearch;
|
2017-07-20 02:34:16 +00:00
|
|
|
|
|
|
|
return spec;
|
|
|
|
}
|
|
|
|
|
2013-09-13 23:17:58 +00:00
|
|
|
private List<DownloadDecision> Dispatch(Func<IIndexer, IEnumerable<ReleaseInfo>> searchAction, SearchCriteriaBase criteriaBase)
|
2013-04-07 07:30:37 +00:00
|
|
|
{
|
2017-12-02 05:03:12 +00:00
|
|
|
var indexers = criteriaBase.InteractiveSearch ?
|
|
|
|
_indexerFactory.InteractiveSearchEnabled() :
|
|
|
|
_indexerFactory.AutomaticSearchEnabled();
|
|
|
|
|
2013-09-13 23:17:58 +00:00
|
|
|
var reports = new List<ReleaseInfo>();
|
2013-04-07 07:30:37 +00:00
|
|
|
|
2021-08-30 02:25:41 +00:00
|
|
|
_logger.ProgressInfo("Searching indexers for {0}. {1} active indexers", criteriaBase, indexers.Count);
|
2013-05-30 01:35:26 +00:00
|
|
|
|
|
|
|
var taskList = new List<Task>();
|
|
|
|
var taskFactory = new TaskFactory(TaskCreationOptions.LongRunning, TaskContinuationOptions.None);
|
|
|
|
|
|
|
|
foreach (var indexer in indexers)
|
2013-04-07 07:30:37 +00:00
|
|
|
{
|
2013-05-30 01:35:26 +00:00
|
|
|
var indexerLocal = indexer;
|
|
|
|
|
|
|
|
taskList.Add(taskFactory.StartNew(() =>
|
2013-04-07 07:30:37 +00:00
|
|
|
{
|
2013-05-30 01:35:26 +00:00
|
|
|
try
|
2013-04-07 07:30:37 +00:00
|
|
|
{
|
2013-05-30 01:35:26 +00:00
|
|
|
var indexerReports = searchAction(indexerLocal);
|
|
|
|
|
|
|
|
lock (reports)
|
|
|
|
{
|
|
|
|
reports.AddRange(indexerReports);
|
|
|
|
}
|
2013-04-07 07:30:37 +00:00
|
|
|
}
|
2013-05-30 01:35:26 +00:00
|
|
|
catch (Exception e)
|
|
|
|
{
|
2017-01-05 23:32:17 +00:00
|
|
|
_logger.Error(e, "Error while searching for {0}", criteriaBase);
|
2013-05-30 01:35:26 +00:00
|
|
|
}
|
2013-07-12 06:10:34 +00:00
|
|
|
}).LogExceptions());
|
2013-05-30 01:35:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
Task.WaitAll(taskList.ToArray());
|
2013-04-07 07:30:37 +00:00
|
|
|
|
2013-09-04 05:01:28 +00:00
|
|
|
_logger.Debug("Total of {0} reports were found for {1} from {2} indexers", reports.Count, criteriaBase, indexers.Count);
|
2013-04-07 07:30:37 +00:00
|
|
|
|
2013-06-06 14:42:23 +00:00
|
|
|
return _makeDownloadDecision.GetSearchDecision(reports, criteriaBase).ToList();
|
2013-04-07 07:30:37 +00:00
|
|
|
}
|
|
|
|
}
|
2013-09-13 23:17:58 +00:00
|
|
|
}
|