From 68df4394986e66a26413fcfff350239b01cf8a5d Mon Sep 17 00:00:00 2001
From: Qstick <qstick@gmail.com>
Date: Mon, 12 Sep 2022 21:16:05 -0500
Subject: [PATCH] New: (Avistaz) Freeleech Only Setting

#1108
---
 .../Indexers/Definitions/Avistaz/AvistazRequestGenerator.cs  | 5 +++++
 .../Indexers/Definitions/Avistaz/AvistazSettings.cs          | 4 ++++
 2 files changed, 9 insertions(+)

diff --git a/src/NzbDrone.Core/Indexers/Definitions/Avistaz/AvistazRequestGenerator.cs b/src/NzbDrone.Core/Indexers/Definitions/Avistaz/AvistazRequestGenerator.cs
index 424b5def3..16a099870 100644
--- a/src/NzbDrone.Core/Indexers/Definitions/Avistaz/AvistazRequestGenerator.cs
+++ b/src/NzbDrone.Core/Indexers/Definitions/Avistaz/AvistazRequestGenerator.cs
@@ -34,6 +34,11 @@ protected virtual List<KeyValuePair<string, string>> GetBasicSearchParameters(in
                 { "type", categoryMapping.Any() ? categoryMapping.First() : "0" }
             };
 
+            if (Settings.FreeleechOnly)
+            {
+                qc.Add("discount[]", "1");
+            }
+
             // resolution filter to improve the search
             if (!categories.Contains(NewznabStandardCategory.Movies.Id) && !categories.Contains(NewznabStandardCategory.TV.Id) &&
                 !categories.Contains(NewznabStandardCategory.Audio.Id))
diff --git a/src/NzbDrone.Core/Indexers/Definitions/Avistaz/AvistazSettings.cs b/src/NzbDrone.Core/Indexers/Definitions/Avistaz/AvistazSettings.cs
index 2f81274f6..88b8993ca 100644
--- a/src/NzbDrone.Core/Indexers/Definitions/Avistaz/AvistazSettings.cs
+++ b/src/NzbDrone.Core/Indexers/Definitions/Avistaz/AvistazSettings.cs
@@ -22,6 +22,7 @@ public class AvistazSettings : NoAuthTorrentBaseSettings
         public AvistazSettings()
         {
             Token = "";
+            FreeleechOnly = false;
         }
 
         public string Token { get; set; }
@@ -35,6 +36,9 @@ public AvistazSettings()
         [FieldDefinition(4, Label = "PID", HelpText = "PID from My Account or My Profile page")]
         public string Pid { get; set; }
 
+        [FieldDefinition(5, Label = "Freeleech Only", HelpText = "Search freeleech only")]
+        public bool FreeleechOnly { get; set; }
+
         public override NzbDroneValidationResult Validate()
         {
             return new NzbDroneValidationResult(Validator.Validate(this));