New: Indexers - AvistaZ, CinemaZ, ExoticaZ

2026-04-21 22:25:03 -04:00 · 2021-03-20 16:50:17 -04:00
parent acbb6e3549
commit d2e3bf3964
9 changed files with 6869 additions and 31 deletions
@@ -0,0 +1,126 @@
+using System;
+using System.Net;
+using System.Threading.Tasks;
+using FluentValidation.Results;
+using NLog;
+using NzbDrone.Common.Http;
+using NzbDrone.Core.Configuration;
+using NzbDrone.Core.Messaging.Events;
+
+namespace NzbDrone.Core.Indexers.Definitions.Avistaz
+{
+    public abstract class AvistazBase : HttpIndexerBase<AvistazSettings>
+    {
+        public override DownloadProtocol Protocol => DownloadProtocol.Torrent;
+        public override string BaseUrl => "";
+        protected virtual string LoginUrl => BaseUrl + "api/v1/jackett/auth";
+        public override bool SupportsRss => true;
+        public override bool SupportsSearch => true;
+        public override int PageSize => 50;
+        public override IndexerCapabilities Capabilities => SetCapabilities();
+        private IIndexerRepository _indexerRepository;
+
+        public AvistazBase(IIndexerRepository indexerRepository,
+                       IHttpClient httpClient,
+                       IEventAggregator eventAggregator,
+                       IIndexerStatusService indexerStatusService,
+                       IConfigService configService,
+                       Logger logger)
+            : base(httpClient, eventAggregator, indexerStatusService, configService, logger)
+        {
+            _indexerRepository = indexerRepository;
+        }
+
+        public override IIndexerRequestGenerator GetRequestGenerator()
+        {
+            return new AvistazRequestGenerator()
+            {
+                Settings = Settings,
+                HttpClient = _httpClient,
+                Logger = _logger,
+                Capabilities = Capabilities,
+                BaseUrl = BaseUrl
+            };
+        }
+
+        public override IParseIndexerResponse GetParser()
+        {
+            return new AvistazParser();
+        }
+
+        protected virtual IndexerCapabilities SetCapabilities()
+        {
+            var caps = new IndexerCapabilities();
+
+            return caps;
+        }
+
+        protected override async Task DoLogin()
+        {
+            Settings.Token = await GetToken();
+
+            if (Definition.Id > 0)
+            {
+                _indexerRepository.UpdateSettings((IndexerDefinition)Definition);
+            }
+
+            _logger.Debug("Avistaz authentication succeeded.");
+        }
+
+        protected override bool CheckIfLoginNeeded(HttpResponse response)
+        {
+            if (response.StatusCode == HttpStatusCode.Unauthorized || response.StatusCode == HttpStatusCode.PreconditionFailed)
+            {
+                return true;
+            }
+
+            return false;
+        }
+
+        protected override void ModifyRequest(IndexerRequest request)
+        {
+            request.HttpRequest.Headers.Set("Authorization", $"Bearer {Settings.Token}");
+            base.ModifyRequest(request);
+        }
+
+        protected override async Task<ValidationFailure> TestConnection()
+        {
+            try
+            {
+                await GetToken();
+            }
+            catch (Exception ex)
+            {
+                _logger.Warn(ex, "Unable to connect to indexer");
+
+                return new ValidationFailure(string.Empty, "Unable to connect to indexer, check the log for more details");
+            }
+
+            return null;
+        }
+
+        private async Task<string> GetToken()
+        {
+            var requestBuilder = new HttpRequestBuilder(LoginUrl)
+            {
+                LogResponseContent = true
+            };
+
+            requestBuilder.Method = HttpMethod.POST;
+            requestBuilder.PostProcess += r => r.RequestTimeout = TimeSpan.FromSeconds(15);
+
+            var authLoginRequest = requestBuilder
+                .AddFormParameter("username", Settings.Username)
+                .AddFormParameter("password", Settings.Password)
+                .AddFormParameter("pid", Settings.Pid.Trim())
+                .SetHeader("Content-Type", "application/json")
+                .Accept(HttpAccept.Json)
+                .Build();
+
+            var response = await _httpClient.PostAsync<AvistazAuthResponse>(authLoginRequest);
+            var token = response.Resource.Token;
+
+            return token;
+        }
+    }
+}