sonarr-repo-only/NzbDrone.Api/Qualities/QualitySizeModule.cs

using System.Collections.Generic;
using System.Linq;
using AutoMapper;
using Nancy;
using NzbDrone.Api.Extensions;
using NzbDrone.Core.Qualities;

namespace NzbDrone.Api.Qualities
{
    public class QualitySizeModule : NzbDroneApiModule
    {
        private readonly QualitySizeService _qualityTypeProvider;

        public QualitySizeModule(QualitySizeService qualityTypeProvider)
            : base("/qualitysizes")
        {
            _qualityTypeProvider = qualityTypeProvider;

            Get["/"] = x => GetQualityType();
            Get["/{id}"] = x => GetQualityType(x.Id);
            Put["/"] = x => PutQualityType();
        }

        private Response PutQualityType()
        {
            var model = Request.Body.FromJson<QualitySizeResource>();

            var type = Mapper.Map<QualitySizeResource, QualitySize>(model);
            _qualityTypeProvider.Update(type);

            return model.AsResponse();

        }

        private Response GetQualityType(int id)
        {
            var type = _qualityTypeProvider.Get(id);
            return Mapper.Map<QualitySize, QualitySizeResource>(type).AsResponse();
        }

        private Response GetQualityType()
        {
            var types = _qualityTypeProvider.All().Where(qualityType => qualityType.QualityId != 0 && qualityType.QualityId != 10).ToList();
            var responseModel = Mapper.Map<List<QualitySize>, List<QualitySizeResource>>(types);

            return responseModel.AsResponse();
        }
    }
}