Sonarr/NzbDrone.Api/EpisodeFiles/EpisodeFileModule.cs

using System.Collections.Generic;
using NzbDrone.Api.REST;
using NzbDrone.Core.MediaFiles;
using NzbDrone.Core.Tv;
using NzbDrone.Api.Extensions;
using System.Linq;
using Omu.ValueInjecter;

namespace NzbDrone.Api.EpisodeFiles
{
    public class EpisodeModule : NzbDroneRestModule<EpisodeFileResource>
    {
        private readonly IMediaFileService _mediaFileService;

        public EpisodeModule(IMediaFileService mediaFileService)
            : base("/episodefile")
        {
            _mediaFileService = mediaFileService;

            UpdateResource = SetQuality;
        }

        private EpisodeFileResource SetQuality(EpisodeFileResource episodeFileResource)
        {
            var episodeFile = _mediaFileService.Get(episodeFileResource.Id);
            episodeFile.Quality = episodeFileResource.Quality;

            _mediaFileService.Update(episodeFile);
            episodeFileResource.InjectFrom(episodeFile);

            return episodeFileResource;
        }
    }
}