New: Nested Settings and Seed Ratio Setting (#379)

* New: Nested Settings and Seed Ratio Setting * Fixed: Sonarr related variable naming
2026-04-22 22:14:44 -04:00 · 2018-06-01 21:59:54 -04:00
parent dabb9bc18a
commit 089d213816
42 changed files with 657 additions and 172 deletions
@@ -0,0 +1,66 @@
+using System;
+using System.Linq;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Core.Datastore;
+using NzbDrone.Core.Download.Clients;
+using NzbDrone.Core.Indexers;
+using NzbDrone.Core.Parser.Model;
+
+namespace NzbDrone.Core.Indexers
+{
+    public interface ISeedConfigProvider
+    {
+        TorrentSeedConfiguration GetSeedConfiguration(RemoteAlbum release);
+    }
+
+    public class SeedConfigProvider : ISeedConfigProvider
+    {
+        private readonly IIndexerFactory _indexerFactory;
+
+        public SeedConfigProvider(IIndexerFactory indexerFactory)
+        {
+            _indexerFactory = indexerFactory;
+        }
+
+        public TorrentSeedConfiguration GetSeedConfiguration(RemoteAlbum remoteAlbum)
+        {
+            if (remoteAlbum.Release.DownloadProtocol != DownloadProtocol.Torrent)
+            {
+                return null;
+            }
+
+            if (remoteAlbum.Release.IndexerId == 0)
+            {
+                return null;
+            }
+
+            try
+            {
+                var indexer = _indexerFactory.Get(remoteAlbum.Release.IndexerId);
+                var torrentIndexerSettings = indexer.Settings as ITorrentIndexerSettings;
+
+                if (torrentIndexerSettings != null && torrentIndexerSettings.SeedCriteria != null)
+                {
+                    var seedConfig = new TorrentSeedConfiguration
+                    {
+                        Ratio = torrentIndexerSettings.SeedCriteria.SeedRatio
+                    };
+
+                    var seedTime = remoteAlbum.ParsedAlbumInfo.Discography ? torrentIndexerSettings.SeedCriteria.DiscographySeedTime : torrentIndexerSettings.SeedCriteria.SeedTime;
+                    if (seedTime.HasValue)
+                    {
+                        seedConfig.SeedTime = TimeSpan.FromMinutes(seedTime.Value);
+                    }
+
+                    return seedConfig;
+                }
+            }
+            catch (ModelNotFoundException)
+            {
+                return null;
+            }
+
+            return null;
+        }
+    }
+}