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.
112 lines
3.2 KiB
112 lines
3.2 KiB
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NzbDrone.Common.Extensions;
|
|
using Sonarr.Http.REST;
|
|
using NzbDrone.Core.Profiles.Qualities;
|
|
using NzbDrone.Core.Qualities;
|
|
|
|
namespace NzbDrone.Api.Profiles
|
|
{
|
|
public class ProfileResource : RestResource
|
|
{
|
|
public string Name { get; set; }
|
|
public bool UpgradeAllowed { get; set; }
|
|
public Quality Cutoff { get; set; }
|
|
public List<ProfileQualityItemResource> Items { get; set; }
|
|
}
|
|
|
|
public class ProfileQualityItemResource : RestResource
|
|
{
|
|
public Quality Quality { get; set; }
|
|
public bool Allowed { get; set; }
|
|
}
|
|
|
|
public static class ProfileResourceMapper
|
|
{
|
|
public static ProfileResource ToResource(this QualityProfile model)
|
|
{
|
|
if (model == null) return null;
|
|
|
|
var cutoffItem = model.Items.First(q =>
|
|
{
|
|
if (q.Id == model.Cutoff) return true;
|
|
|
|
if (q.Quality == null) return false;
|
|
|
|
return q.Quality.Id == model.Cutoff;
|
|
});
|
|
|
|
var cutoff = cutoffItem.Items == null || cutoffItem.Items.Empty()
|
|
? cutoffItem.Quality
|
|
: cutoffItem.Items.First().Quality;
|
|
|
|
return new ProfileResource
|
|
{
|
|
Id = model.Id,
|
|
|
|
Name = model.Name,
|
|
UpgradeAllowed = model.UpgradeAllowed,
|
|
Cutoff = cutoff,
|
|
|
|
// Flatten groups so things don't explode
|
|
Items = model.Items.SelectMany(i =>
|
|
{
|
|
if (i == null)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
if (i.Items.Any())
|
|
{
|
|
return i.Items.ConvertAll(ToResource);
|
|
}
|
|
|
|
return new List<ProfileQualityItemResource> {ToResource(i)};
|
|
}).ToList()
|
|
};
|
|
}
|
|
|
|
public static ProfileQualityItemResource ToResource(this QualityProfileQualityItem model)
|
|
{
|
|
if (model == null) return null;
|
|
|
|
return new ProfileQualityItemResource
|
|
{
|
|
Quality = model.Quality,
|
|
Allowed = model.Allowed
|
|
};
|
|
}
|
|
|
|
public static QualityProfile ToModel(this ProfileResource resource)
|
|
{
|
|
if (resource == null) return null;
|
|
|
|
return new QualityProfile
|
|
{
|
|
Id = resource.Id,
|
|
|
|
Name = resource.Name,
|
|
UpgradeAllowed = resource.UpgradeAllowed,
|
|
Cutoff = resource.Cutoff.Id,
|
|
Items = resource.Items.ConvertAll(ToModel)
|
|
};
|
|
}
|
|
|
|
public static QualityProfileQualityItem ToModel(this ProfileQualityItemResource resource)
|
|
{
|
|
if (resource == null) return null;
|
|
|
|
return new QualityProfileQualityItem
|
|
{
|
|
Quality = (Quality)resource.Quality.Id,
|
|
Allowed = resource.Allowed
|
|
};
|
|
}
|
|
|
|
public static List<ProfileResource> ToResource(this IEnumerable<QualityProfile> models)
|
|
{
|
|
return models.Select(ToResource).ToList();
|
|
}
|
|
}
|
|
}
|