You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
3.7 KiB
87 lines
3.7 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common.Disk;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.Download;
|
|
using NzbDrone.Core.Download.TrackedDownloads;
|
|
using NzbDrone.Core.MediaFiles.Commands;
|
|
using NzbDrone.Core.MediaFiles.TrackImport;
|
|
using NzbDrone.Core.Messaging.Commands;
|
|
|
|
namespace NzbDrone.Core.MediaFiles
|
|
{
|
|
public class DownloadedAlbumsCommandService : IExecute<DownloadedAlbumsScanCommand>
|
|
{
|
|
private readonly IDownloadedTracksImportService _downloadedTracksImportService;
|
|
private readonly ITrackedDownloadService _trackedDownloadService;
|
|
private readonly IDiskProvider _diskProvider;
|
|
private readonly ICompletedDownloadService _completedDownloadService;
|
|
private readonly Logger _logger;
|
|
|
|
public DownloadedAlbumsCommandService(IDownloadedTracksImportService downloadedTracksImportService,
|
|
ITrackedDownloadService trackedDownloadService,
|
|
IDiskProvider diskProvider,
|
|
ICompletedDownloadService completedDownloadService,
|
|
Logger logger)
|
|
{
|
|
_downloadedTracksImportService = downloadedTracksImportService;
|
|
_trackedDownloadService = trackedDownloadService;
|
|
_diskProvider = diskProvider;
|
|
_completedDownloadService = completedDownloadService;
|
|
_logger = logger;
|
|
}
|
|
|
|
private List<ImportResult> ProcessPath(DownloadedAlbumsScanCommand message)
|
|
{
|
|
if (!_diskProvider.FolderExists(message.Path) && !_diskProvider.FileExists(message.Path))
|
|
{
|
|
_logger.Warn("Folder/File specified for import scan [{0}] doesn't exist.", message.Path);
|
|
return new List<ImportResult>();
|
|
}
|
|
|
|
if (message.DownloadClientId.IsNotNullOrWhiteSpace())
|
|
{
|
|
var trackedDownload = _trackedDownloadService.Find(message.DownloadClientId);
|
|
|
|
if (trackedDownload != null)
|
|
{
|
|
_logger.Debug("External directory scan request for known download {0}. [{1}]", message.DownloadClientId, message.Path);
|
|
|
|
var importResults = _downloadedTracksImportService.ProcessPath(message.Path, message.ImportMode, trackedDownload.RemoteAlbum.Artist, trackedDownload.DownloadItem);
|
|
|
|
_completedDownloadService.VerifyImport(trackedDownload, importResults);
|
|
|
|
return importResults;
|
|
}
|
|
|
|
_logger.Warn("External directory scan request for unknown download {0}, attempting normal import. [{1}]", message.DownloadClientId, message.Path);
|
|
}
|
|
|
|
return _downloadedTracksImportService.ProcessPath(message.Path, message.ImportMode);
|
|
}
|
|
|
|
public void Execute(DownloadedAlbumsScanCommand message)
|
|
{
|
|
List<ImportResult> importResults;
|
|
|
|
if (message.Path.IsNotNullOrWhiteSpace())
|
|
{
|
|
importResults = ProcessPath(message);
|
|
}
|
|
else
|
|
{
|
|
throw new ArgumentException("A path must be provided", "path");
|
|
}
|
|
|
|
if (importResults == null || importResults.All(v => v.Result != ImportResultType.Imported))
|
|
{
|
|
// Atm we don't report it as a command failure, coz that would cause the download to be failed.
|
|
// Changing the message won't do a thing either, coz it will get set to 'Completed' a msec later.
|
|
//message.SetMessage("Failed to import");
|
|
}
|
|
}
|
|
}
|
|
}
|