Added: Custom Class for Radarr API requests. Also implements new error handling present on staging server.

2026-04-20 21:55:03 -04:00 · 2017-06-06 19:50:53 +02:00
parent f1873a9684
commit 7bbebe53a3
4 changed files with 176 additions and 45 deletions
@@ -0,0 +1,98 @@
+using NzbDrone.Common.Http;
+using NzbDrone.Core.Configuration;
+using System;
+using Newtonsoft.Json;
+using System.Collections.Generic;
+using NzbDrone.Core.MetadataSource.SkyHook.Resource;
+
+namespace NzbDrone.Core.MetadataSource.RadarrAPI
+{
+    public interface IRadarrAPIClient
+    {
+        IHttpRequestBuilderFactory RadarrAPI { get; }
+        List<MovieResult> DiscoverMovies(string action, Func<HttpRequest, HttpRequest> enhanceRequest);
+    }
+
+    public class RadarrAPIClient : IRadarrAPIClient
+    {
+        private readonly IHttpClient _httpClient;
+
+        public RadarrAPIClient(IConfigFileProvider configFile, IHttpClient httpClient)
+        {
+            _httpClient = httpClient;
+
+            if (configFile.Branch == "nightly")
+            {
+                RadarrAPI = new HttpRequestBuilder("https://staging.api.radarr.video/{route}/{action}")
+                .CreateFactory();
+            }
+            else
+            {
+                RadarrAPI = new HttpRequestBuilder("https://api.radarr.video/v2/{route}/{action}")
+                .CreateFactory();
+            }
+
+        }
+
+        private HttpResponse Execute(HttpRequest request)
+        {
+            if (request.Method == HttpMethod.GET)
+            {
+                return _httpClient.Get(request);
+            }
+            else if (request.Method == HttpMethod.POST)
+            {
+                return _httpClient.Post(request);
+            }
+            else
+            {
+                throw new NotImplementedException($"Method {request.Method} not implemented");
+            }
+        }
+
+        private T Execute<T>(HttpRequest request)
+        {
+            request.AllowAutoRedirect = true;
+            request.Headers.Accept = HttpAccept.Json.Value;
+            request.SuppressHttpError = true;
+
+            var response = Execute(request);
+
+            try
+            {
+                var error = JsonConvert.DeserializeObject<RadarrError>(response.Content);
+
+                if (error != null && error.Errors.Count != 0)
+                {
+                    throw new RadarrAPIException(error);
+                }
+            }
+            catch (JsonSerializationException)
+            {
+                //No error!
+            }
+
+
+            if (response.StatusCode != System.Net.HttpStatusCode.OK)
+            {
+                throw new HttpException(request, response);
+            }
+
+            return JsonConvert.DeserializeObject<T>(response.Content);
+        }
+
+        public List<MovieResult> DiscoverMovies(string action, Func<HttpRequest, HttpRequest> enhanceRequest = null )
+        {
+            var request = RadarrAPI.Create().SetSegment("route", "discovery").SetSegment("action", action).Build();
+
+            if (enhanceRequest != null)
+            {
+                request = enhanceRequest(request);
+            }
+
+            return Execute<List<MovieResult>>(request);
+        }
+
+        public IHttpRequestBuilderFactory RadarrAPI { get; private set; }
+    }
+}