mirror of
https://github.com/lidarr/lidarr.git
synced 2025-08-20 13:33:34 -07:00
parent
e0993fb52e
commit
dd875a0c49
2 changed files with 46 additions and 14 deletions
|
@ -1,6 +1,7 @@
|
||||||
using System;
|
using System;
|
||||||
using System.Collections.Generic;
|
using System.Collections.Generic;
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
|
using System.Linq.Expressions;
|
||||||
using NLog;
|
using NLog;
|
||||||
using NzbDrone.Common.Extensions;
|
using NzbDrone.Common.Extensions;
|
||||||
using NzbDrone.Common.Instrumentation.Extensions;
|
using NzbDrone.Common.Instrumentation.Extensions;
|
||||||
|
@ -14,23 +15,26 @@ using NzbDrone.Core.Music;
|
||||||
namespace NzbDrone.Core.IndexerSearch
|
namespace NzbDrone.Core.IndexerSearch
|
||||||
{
|
{
|
||||||
class AlbumSearchService : IExecute<AlbumSearchCommand>,
|
class AlbumSearchService : IExecute<AlbumSearchCommand>,
|
||||||
IExecute<MissingAlbumSearchCommand>
|
IExecute<MissingAlbumSearchCommand>,
|
||||||
//IExecute<CutoffUnmetAlbumSearchCommand>
|
IExecute<CutoffUnmetAlbumSearchCommand>
|
||||||
{
|
{
|
||||||
private readonly ISearchForNzb _nzbSearchService;
|
private readonly ISearchForNzb _nzbSearchService;
|
||||||
private readonly IAlbumService _albumService;
|
private readonly IAlbumService _albumService;
|
||||||
|
private readonly IAlbumCutoffService _albumCutoffService;
|
||||||
private readonly IQueueService _queueService;
|
private readonly IQueueService _queueService;
|
||||||
private readonly IProcessDownloadDecisions _processDownloadDecisions;
|
private readonly IProcessDownloadDecisions _processDownloadDecisions;
|
||||||
private readonly Logger _logger;
|
private readonly Logger _logger;
|
||||||
|
|
||||||
public AlbumSearchService(ISearchForNzb nzbSearchService,
|
public AlbumSearchService(ISearchForNzb nzbSearchService,
|
||||||
IAlbumService albumService,
|
IAlbumService albumService,
|
||||||
|
IAlbumCutoffService albumCutoffService,
|
||||||
IQueueService queueService,
|
IQueueService queueService,
|
||||||
IProcessDownloadDecisions processDownloadDecisions,
|
IProcessDownloadDecisions processDownloadDecisions,
|
||||||
Logger logger)
|
Logger logger)
|
||||||
{
|
{
|
||||||
_nzbSearchService = nzbSearchService;
|
_nzbSearchService = nzbSearchService;
|
||||||
_albumService = albumService;
|
_albumService = albumService;
|
||||||
|
_albumCutoffService = albumCutoffService;
|
||||||
_queueService = queueService;
|
_queueService = queueService;
|
||||||
_processDownloadDecisions = processDownloadDecisions;
|
_processDownloadDecisions = processDownloadDecisions;
|
||||||
_logger = logger;
|
_logger = logger;
|
||||||
|
@ -107,5 +111,39 @@ namespace NzbDrone.Core.IndexerSearch
|
||||||
|
|
||||||
SearchForMissingAlbums(missing, message.Trigger == CommandTrigger.Manual);
|
SearchForMissingAlbums(missing, message.Trigger == CommandTrigger.Manual);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public void Execute(CutoffUnmetAlbumSearchCommand message)
|
||||||
|
{
|
||||||
|
Expression<Func<Album, bool>> filterExpression;
|
||||||
|
|
||||||
|
if (message.ArtistId.HasValue)
|
||||||
|
{
|
||||||
|
filterExpression = v =>
|
||||||
|
v.ArtistId == message.ArtistId.Value &&
|
||||||
|
v.Monitored == true &&
|
||||||
|
v.Artist.Monitored == true;
|
||||||
|
}
|
||||||
|
|
||||||
|
else
|
||||||
|
{
|
||||||
|
filterExpression = v =>
|
||||||
|
v.Monitored == true &&
|
||||||
|
v.Artist.Monitored == true;
|
||||||
|
}
|
||||||
|
|
||||||
|
var albums = _albumCutoffService.AlbumsWhereCutoffUnmet(new PagingSpec<Album>
|
||||||
|
{
|
||||||
|
Page = 1,
|
||||||
|
PageSize = 100000,
|
||||||
|
SortDirection = SortDirection.Ascending,
|
||||||
|
SortKey = "Id",
|
||||||
|
FilterExpression = filterExpression
|
||||||
|
}).Records.ToList();
|
||||||
|
|
||||||
|
var queue = _queueService.GetQueue().Select(q => q.Album.Id);
|
||||||
|
var missing = albums.Where(e => !queue.Contains(e.Id)).ToList();
|
||||||
|
|
||||||
|
SearchForMissingAlbums(missing, message.Trigger == CommandTrigger.Manual);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,26 +1,20 @@
|
||||||
using NzbDrone.Core.Messaging.Commands;
|
using NzbDrone.Core.Messaging.Commands;
|
||||||
|
|
||||||
namespace NzbDrone.Core.IndexerSearch
|
namespace NzbDrone.Core.IndexerSearch
|
||||||
{
|
{
|
||||||
public class CutoffUnmetAlbumSearchCommand : Command
|
public class CutoffUnmetAlbumSearchCommand : Command
|
||||||
{
|
{
|
||||||
public int? SeriesId { get; set; }
|
public int? ArtistId { get; set; }
|
||||||
|
|
||||||
public override bool SendUpdatesToClient
|
public override bool SendUpdatesToClient => true;
|
||||||
{
|
|
||||||
get
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public CutoffUnmetAlbumSearchCommand()
|
public CutoffUnmetAlbumSearchCommand()
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
|
|
||||||
public CutoffUnmetAlbumSearchCommand(int seriesId)
|
public CutoffUnmetAlbumSearchCommand(int artistId)
|
||||||
{
|
{
|
||||||
SeriesId = seriesId;
|
ArtistId = artistId;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue