Prowlarr/src/NzbDrone.Core/Applications/LazyLibrarian/LazyLibrarianV1Proxy.cs

using System;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using System.Net.Http;
using FluentValidation.Results;
using NLog;
using NzbDrone.Common.Http;
using NzbDrone.Common.Serializer;

namespace NzbDrone.Core.Applications.LazyLibrarian
{
    public interface ILazyLibrarianV1Proxy
    {
        LazyLibrarianIndexer AddIndexer(LazyLibrarianIndexer indexer, LazyLibrarianSettings settings);
        List<LazyLibrarianIndexer> GetIndexers(LazyLibrarianSettings settings);
        LazyLibrarianIndexer GetIndexer(string indexerName, LazyLibrarianProviderType indexerType, LazyLibrarianSettings settings);
        void RemoveIndexer(string indexerName, LazyLibrarianProviderType indexerType, LazyLibrarianSettings settings);
        LazyLibrarianIndexer UpdateIndexer(LazyLibrarianIndexer indexer, LazyLibrarianSettings settings);
        ValidationFailure TestConnection(LazyLibrarianSettings settings);
    }

    public class LazyLibrarianV1Proxy : ILazyLibrarianV1Proxy
    {
        private const int ProwlarrHighestPriority = 50;

        private readonly IHttpClient _httpClient;
        private readonly Logger _logger;

        public LazyLibrarianV1Proxy(IHttpClient httpClient, Logger logger)
        {
            _httpClient = httpClient;
            _logger = logger;
        }

        public LazyLibrarianStatus GetStatus(LazyLibrarianSettings settings)
        {
            var request = BuildRequest(settings, "/api", "getVersion", HttpMethod.Get);
            return Execute<LazyLibrarianStatus>(request);
        }

        public List<LazyLibrarianIndexer> GetIndexers(LazyLibrarianSettings settings)
        {
            var request = BuildRequest(settings, "/api", "listNabProviders", HttpMethod.Get);

            var response = Execute<LazyLibrarianIndexerResponse>(request);

            if (!response.Success)
            {
                throw new LazyLibrarianException(string.Format("LazyLibrarian Error - Code {0}: {1}", response.Error.Code, response.Error.Message));
            }

            var indexers = new List<LazyLibrarianIndexer>();

            var torIndexers = response.Data.Torznabs;
            torIndexers.ForEach(i => i.Type = LazyLibrarianProviderType.Torznab);

            var nzbIndexers = response.Data.Newznabs;
            nzbIndexers.ForEach(i => i.Type = LazyLibrarianProviderType.Newznab);

            indexers.AddRange(torIndexers);
            indexers.AddRange(nzbIndexers);
            indexers.ForEach(i => i.Altername = i.Name);

            return indexers;
        }

        public LazyLibrarianIndexer GetIndexer(string indexerName, LazyLibrarianProviderType indexerType, LazyLibrarianSettings settings)
        {
            var indexers = GetIndexers(settings);

            return indexers.SingleOrDefault(i => i.Name == indexerName && i.Type == indexerType);
        }

        public void RemoveIndexer(string indexerName, LazyLibrarianProviderType indexerType, LazyLibrarianSettings settings)
        {
            var parameters = new Dictionary<string, string>
            {
                { "name", indexerName },
                { "providertype", indexerType.ToString().ToLower() }
            };

            var request = BuildRequest(settings, "/api", "delProvider", HttpMethod.Get, parameters);
            CheckForError(Execute<LazyLibrarianStatus>(request));
        }

        public LazyLibrarianIndexer AddIndexer(LazyLibrarianIndexer indexer, LazyLibrarianSettings settings)
        {
            var parameters = new Dictionary<string, string>
            {
                { "name", indexer.Name },
                { "providertype", indexer.Type.ToString().ToLower() },
                { "host", indexer.Host },
                { "prov_apikey", indexer.Apikey },
                { "enabled", indexer.Enabled.ToString().ToLower() },
                { "categories", indexer.Categories },
                { "dlpriority", CalculatePriority(indexer.Priority).ToString() }
            };

            if (indexer.Type == LazyLibrarianProviderType.Torznab)
            {
                parameters.Add("seeders", indexer.MinimumSeeders.ToString());
                parameters.Add("seed_ratio", indexer.SeedRatio.ToString(CultureInfo.InvariantCulture));
                parameters.Add("seed_duration", indexer.SeedTime.ToString());
            }

            var request = BuildRequest(settings, "/api", "addProvider", HttpMethod.Get, parameters);
            CheckForError(Execute<LazyLibrarianStatus>(request));
            return indexer;
        }

        public LazyLibrarianIndexer UpdateIndexer(LazyLibrarianIndexer indexer, LazyLibrarianSettings settings)
        {
            var parameters = new Dictionary<string, string>
            {
                { "name", indexer.Name },
                { "providertype", indexer.Type.ToString().ToLower() },
                { "host", indexer.Host },
                { "prov_apikey", indexer.Apikey },
                { "enabled", indexer.Enabled.ToString().ToLower() },
                { "categories", indexer.Categories },
                { "altername", indexer.Altername },
                { "dlpriority", CalculatePriority(indexer.Priority).ToString() }
            };

            if (indexer.Type == LazyLibrarianProviderType.Torznab)
            {
                parameters.Add("seeders", indexer.MinimumSeeders.ToString());
                parameters.Add("seed_ratio", indexer.SeedRatio.ToString(CultureInfo.InvariantCulture));
                parameters.Add("seed_duration", indexer.SeedTime.ToString());
            }

            var request = BuildRequest(settings, "/api", "changeProvider", HttpMethod.Get, parameters);
            CheckForError(Execute<LazyLibrarianStatus>(request));
            return indexer;
        }

        private void CheckForError(LazyLibrarianStatus response)
        {
            if (!response.Success)
            {
                throw new LazyLibrarianException(string.Format("LazyLibrarian Error - Code {0}: {1}", response.Error.Code, response.Error.Message));
            }
        }

        public ValidationFailure TestConnection(LazyLibrarianSettings settings)
        {
            try
            {
                var status = GetStatus(settings);

                if (!status.Success)
                {
                    return new ValidationFailure("ApiKey", status.Error.Message);
                }

                GetIndexers(settings);
            }
            catch (HttpException ex)
            {
                _logger.Error(ex, "Unable to complete application test");
                return new ValidationFailure("BaseUrl", "Unable to complete application test");
            }
            catch (LazyLibrarianException ex)
            {
                _logger.Error(ex, "Connection test failed");
                return new ValidationFailure("", ex.Message);
            }
            catch (Exception ex)
            {
                _logger.Error(ex, "Unable to complete application test");
                return new ValidationFailure("", $"Unable to send test message. {ex.Message}");
            }

            return null;
        }

        private HttpRequest BuildRequest(LazyLibrarianSettings settings, string resource, string command, HttpMethod method, Dictionary<string, string> parameters = null)
        {
            var baseUrl = settings.BaseUrl.TrimEnd('/');

            var requestBuilder = new HttpRequestBuilder(baseUrl)
                .Resource(resource)
                .Accept(HttpAccept.Json)
                .AddQueryParam("cmd", command)
                .AddQueryParam("apikey", settings.ApiKey);

            if (parameters != null)
            {
                foreach (var param in parameters)
                {
                    requestBuilder.AddQueryParam(param.Key, param.Value);
                }
            }

            var request = requestBuilder.Build();

            request.Headers.ContentType = "application/json";

            request.Method = method;
            request.AllowAutoRedirect = true;

            return request;
        }

        private TResource Execute<TResource>(HttpRequest request)
             where TResource : new()
        {
            var response = _httpClient.Execute(request);

            if ((int)response.StatusCode >= 300)
            {
                throw new HttpException(response);
            }

            return Json.Deserialize<TResource>(response.Content);
        }

        private int CalculatePriority(int indexerPriority) => ProwlarrHighestPriority - indexerPriority + 1;
    }
}