mirror of https://github.com/Sonarr/Sonarr
52 lines
1.5 KiB
C#
52 lines
1.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using NzbDrone.Core.DecisionEngine;
|
|
using NzbDrone.Core.Profiles.Qualities;
|
|
using Sonarr.Http.REST;
|
|
|
|
namespace Sonarr.Api.V3.Indexers
|
|
{
|
|
public abstract class ReleaseControllerBase : RestController<ReleaseResource>
|
|
{
|
|
private readonly QualityProfile _qualityProfile;
|
|
|
|
public ReleaseControllerBase(IQualityProfileService qualityProfileService)
|
|
{
|
|
_qualityProfile = qualityProfileService.GetDefaultProfile(string.Empty);
|
|
}
|
|
|
|
protected override ReleaseResource GetResourceById(int id)
|
|
{
|
|
throw new NotImplementedException();
|
|
}
|
|
|
|
protected virtual List<ReleaseResource> MapDecisions(IEnumerable<DownloadDecision> decisions)
|
|
{
|
|
var result = new List<ReleaseResource>();
|
|
|
|
foreach (var downloadDecision in decisions)
|
|
{
|
|
var release = MapDecision(downloadDecision, result.Count);
|
|
|
|
result.Add(release);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
protected virtual ReleaseResource MapDecision(DownloadDecision decision, int initialWeight)
|
|
{
|
|
var release = decision.ToResource();
|
|
|
|
release.ReleaseWeight = initialWeight;
|
|
|
|
release.QualityWeight = _qualityProfile.GetIndex(release.Quality.Quality).Index * 100;
|
|
|
|
release.QualityWeight += release.Quality.Revision.Real * 10;
|
|
release.QualityWeight += release.Quality.Revision.Version;
|
|
|
|
return release;
|
|
}
|
|
}
|
|
}
|