Change API Version from V3 to V1

2026-04-19 21:44:30 -04:00 · 2017-10-30 21:28:29 -04:00
parent db10057f2c
commit aae9f3cfc3
155 changed files with 248 additions and 248 deletions
@@ -0,0 +1,126 @@
+using System;
+using System.Collections.Generic;
+using FluentValidation;
+using Nancy;
+using Nancy.ModelBinding;
+using NLog;
+using NzbDrone.Common.Cache;
+using NzbDrone.Core.DecisionEngine;
+using NzbDrone.Core.Download;
+using NzbDrone.Core.Exceptions;
+using NzbDrone.Core.Indexers;
+using NzbDrone.Core.IndexerSearch;
+using NzbDrone.Core.Parser.Model;
+using Lidarr.Http.Extensions;
+using HttpStatusCode = System.Net.HttpStatusCode;
+
+namespace Lidarr.Api.V1.Indexers
+{
+    public class ReleaseModule : ReleaseModuleBase
+    {
+        private readonly IFetchAndParseRss _rssFetcherAndParser;
+        private readonly ISearchForNzb _nzbSearchService;
+        private readonly IMakeDownloadDecision _downloadDecisionMaker;
+        private readonly IPrioritizeDownloadDecision _prioritizeDownloadDecision;
+        private readonly IDownloadService _downloadService;
+        private readonly Logger _logger;
+
+        private readonly ICached<RemoteAlbum> _remoteAlbumCache;
+
+        public ReleaseModule(IFetchAndParseRss rssFetcherAndParser,
+                             ISearchForNzb nzbSearchService,
+                             IMakeDownloadDecision downloadDecisionMaker,
+                             IPrioritizeDownloadDecision prioritizeDownloadDecision,
+                             IDownloadService downloadService,
+                             ICacheManager cacheManager,
+                             Logger logger)
+        {
+            _rssFetcherAndParser = rssFetcherAndParser;
+            _nzbSearchService = nzbSearchService;
+            _downloadDecisionMaker = downloadDecisionMaker;
+            _prioritizeDownloadDecision = prioritizeDownloadDecision;
+            _downloadService = downloadService;
+            _logger = logger;
+
+            GetResourceAll = GetReleases;
+            Post["/"] = x => DownloadRelease(this.Bind<ReleaseResource>());
+
+            PostValidator.RuleFor(s => s.DownloadAllowed).Equal(true);
+            PostValidator.RuleFor(s => s.Guid).NotEmpty();
+
+            _remoteAlbumCache = cacheManager.GetCache<RemoteAlbum>(GetType(), "remoteAlbums");
+        }
+
+        private Response DownloadRelease(ReleaseResource release)
+        {
+            var remoteAlbum = _remoteAlbumCache.Find(GetCacheKey(release));
+
+            if (remoteAlbum == null)
+            {
+                _logger.Debug("Couldn't find requested release in cache, cache timeout probably expired.");
+
+                throw new NzbDroneClientException(HttpStatusCode.NotFound, "Couldn't find requested release in cache, try searching again");
+            }
+
+            try
+            {
+                _downloadService.DownloadReport(remoteAlbum);
+            }
+            catch (ReleaseDownloadException ex)
+            {
+                _logger.Error(ex, ex.Message);
+                throw new NzbDroneClientException(HttpStatusCode.Conflict, "Getting release from indexer failed");
+            }
+
+            return release.AsResponse();
+        }
+
+        private List<ReleaseResource> GetReleases()
+        {
+            if (Request.Query.albumId.HasValue)
+            {
+                return GetAlbumReleases(Request.Query.albumId);
+            }
+
+            return GetRss();
+        }
+
+        private List<ReleaseResource> GetAlbumReleases(int albumId)
+        {
+            try
+            {
+                var decisions = _nzbSearchService.AlbumSearch(albumId, true, true);
+                var prioritizedDecisions = _prioritizeDownloadDecision.PrioritizeDecisions(decisions);
+
+                return MapDecisions(prioritizedDecisions);
+            }
+            catch (Exception ex)
+            {
+                _logger.Error(ex, "Album search failed: " + ex.Message);
+            }
+
+            return new List<ReleaseResource>();
+        }
+
+        private List<ReleaseResource> GetRss()
+        {
+            var reports = _rssFetcherAndParser.Fetch();
+            var decisions = _downloadDecisionMaker.GetRssDecision(reports);
+            var prioritizedDecisions = _prioritizeDownloadDecision.PrioritizeDecisions(decisions);
+
+            return MapDecisions(prioritizedDecisions);
+        }
+
+        protected override ReleaseResource MapDecision(DownloadDecision decision, int initialWeight)
+        {
+            var resource = base.MapDecision(decision, initialWeight);
+            _remoteAlbumCache.Set(GetCacheKey(resource), decision.RemoteAlbum, TimeSpan.FromMinutes(30));
+            return resource;
+        }
+
+        private string GetCacheKey(ReleaseResource resource)
+        {
+            return string.Concat(resource.IndexerId, "_", resource.Guid);
+        }
+    }
+}