Lidarr/src/Lidarr.Api.V1/ManualImport/ManualImportModule.cs

using System.Collections.Generic;
using System.Linq;
using NzbDrone.Core.MediaFiles.TrackImport.Manual;
using NzbDrone.Core.Qualities;
using Lidarr.Http;
using Lidarr.Http.Extensions;
using NzbDrone.SignalR;
using NzbDrone.Core.Datastore.Events;
using NzbDrone.Core.Music;
using NLog;

namespace Lidarr.Api.V1.ManualImport
{
    public class ManualImportModule : ManualImportModuleWithSignalR
    {
        private readonly IArtistService _artistService;
        private readonly IAlbumService _albumService;
        private readonly IReleaseService _releaseService;

        public ManualImportModule(IManualImportService manualImportService,
                                  IArtistService artistService,
                                  IAlbumService albumService,
                                  IReleaseService releaseService,
                                  IBroadcastSignalRMessage signalRBroadcaster,
                                  Logger logger)
        : base(manualImportService, signalRBroadcaster, logger)
        {
            _artistService = artistService;
            _albumService = albumService;
            _releaseService = releaseService;

            GetResourceAll = GetMediaFiles;
            UpdateResource = UpdateImportItem;
        }

        private List<ManualImportResource> GetMediaFiles()
        {
            var folder = (string)Request.Query.folder;
            var downloadId = (string)Request.Query.downloadId;
            var filterExistingFiles = Request.GetBooleanQueryParameter("filterExistingFiles", true);

            return _manualImportService.GetMediaFiles(folder, downloadId, filterExistingFiles).ToResource().Select(AddQualityWeight).ToList();
        }

        private ManualImportResource AddQualityWeight(ManualImportResource item)
        {
            if (item.Quality != null)
            {
                item.QualityWeight = Quality.DefaultQualityDefinitions.Single(q => q.Quality == item.Quality.Quality).Weight;
                item.QualityWeight += item.Quality.Revision.Real * 10;
                item.QualityWeight += item.Quality.Revision.Version;
            }
           
            return item;
        }

        private void UpdateImportItem(ManualImportResource resource)
        {
            var item = new ManualImportItem{
                Id = resource.Id,
                Path = resource.Path,
                RelativePath = resource.RelativePath,
                FolderName = resource.FolderName,
                Name = resource.Name,
                Size = resource.Size,
                Artist = resource.Artist == null ? null : _artistService.GetArtist(resource.Artist.Id),
                Album = resource.Album == null ? null : _albumService.GetAlbum(resource.Album.Id),
                Release = resource.AlbumReleaseId == 0 ? null : _releaseService.GetRelease(resource.AlbumReleaseId),
                Quality = resource.Quality,
                Language = resource.Language,
                DownloadId = resource.DownloadId
            };
            
            //recalculate import and broadcast
            _manualImportService.UpdateItem(item);
            BroadcastResourceChange(ModelAction.Updated, item.Id);
        }
    }
}