mirror of https://github.com/lidarr/Lidarr
42 lines
1.3 KiB
C#
42 lines
1.3 KiB
C#
|
using System.Collections.Generic;
|
||
|
using System.Linq;
|
||
|
using NzbDrone.Core.MediaFiles.TrackImport.Manual;
|
||
|
using NzbDrone.Core.Qualities;
|
||
|
using Lidarr.Http;
|
||
|
|
||
|
namespace Lidarr.Api.V3.ManualImport
|
||
|
{
|
||
|
public class ManualImportModule : LidarrRestModule<ManualImportResource>
|
||
|
{
|
||
|
private readonly IManualImportService _manualImportService;
|
||
|
|
||
|
public ManualImportModule(IManualImportService manualImportService)
|
||
|
: base("/manualimport")
|
||
|
{
|
||
|
_manualImportService = manualImportService;
|
||
|
|
||
|
GetResourceAll = GetMediaFiles;
|
||
|
}
|
||
|
|
||
|
private List<ManualImportResource> GetMediaFiles()
|
||
|
{
|
||
|
var folder = (string)Request.Query.folder;
|
||
|
var downloadId = (string)Request.Query.downloadId;
|
||
|
|
||
|
return _manualImportService.GetMediaFiles(folder, downloadId).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;
|
||
|
}
|
||
|
}
|
||
|
}
|