Save episode quality after change

2026-04-22 22:14:44 -04:00 · 2013-08-20 23:46:23 -07:00
parent df1cb91a84
commit 7c915bd522
9 changed files with 79 additions and 5 deletions
@@ -0,0 +1,34 @@
+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;
+        }
+    }
+}