Fix some Build Issues, Renaming

2026-04-21 22:04:31 -04:00 · 2017-09-05 23:38:05 -04:00
parent 95051cbd63
commit a747c5f135
191 changed files with 16 additions and 16 deletions
@@ -0,0 +1,120 @@
+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.V3.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 remoteEpisode = _remoteAlbumCache.Find(release.Guid);
+
+            if (remoteEpisode == 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(remoteEpisode);
+            }
+            catch (ReleaseDownloadException ex)
+            {
+                _logger.ErrorException(ex.Message, ex);
+                throw new NzbDroneClientException(HttpStatusCode.Conflict, "Getting release from indexer failed");
+            }
+
+            return release.AsResponse();
+        }
+
+        private List<ReleaseResource> GetReleases()
+        {
+            if (Request.Query.episodeId.HasValue)
+            {
+                return GetEpisodeReleases(Request.Query.episodeId);
+            }
+
+            return GetRss();
+        }
+
+        private List<ReleaseResource> GetEpisodeReleases(int albumId)
+        {
+            try
+            {
+                var decisions = _nzbSearchService.AlbumSearch(albumId, true, true);
+                var prioritizedDecisions = _prioritizeDownloadDecision.PrioritizeDecisions(decisions);
+
+                return MapDecisions(prioritizedDecisions);
+            }
+            catch (Exception ex)
+            {
+                _logger.ErrorException("Episode search failed: " + ex.Message, ex);
+            }
+
+            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)
+        {
+            _remoteAlbumCache.Set(decision.RemoteAlbum.Release.Guid, decision.RemoteAlbum, TimeSpan.FromMinutes(30));
+           return base.MapDecision(decision, initialWeight);
+        }
+    }
+}