2019-06-14 03:54:25 +00:00
|
|
|
using System;
|
2020-03-19 14:47:25 +00:00
|
|
|
using NzbDrone.Common.Cache;
|
2019-06-14 03:54:25 +00:00
|
|
|
using NzbDrone.Core.Datastore;
|
|
|
|
using NzbDrone.Core.Download.Clients;
|
2020-03-19 14:47:25 +00:00
|
|
|
using NzbDrone.Core.Messaging.Events;
|
2019-06-14 03:54:25 +00:00
|
|
|
using NzbDrone.Core.Parser.Model;
|
2021-04-23 17:18:09 +00:00
|
|
|
using NzbDrone.Core.ThingiProvider.Events;
|
2019-06-14 03:54:25 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.Indexers
|
|
|
|
{
|
|
|
|
public interface ISeedConfigProvider
|
|
|
|
{
|
|
|
|
TorrentSeedConfiguration GetSeedConfiguration(RemoteMovie release);
|
2020-03-19 14:47:25 +00:00
|
|
|
TorrentSeedConfiguration GetSeedConfiguration(int indexerId);
|
2019-06-14 03:54:25 +00:00
|
|
|
}
|
|
|
|
|
2021-04-23 17:18:09 +00:00
|
|
|
public class SeedConfigProvider : ISeedConfigProvider, IHandle<ProviderUpdatedEvent<IIndexer>>
|
2019-06-14 03:54:25 +00:00
|
|
|
{
|
|
|
|
private readonly IIndexerFactory _indexerFactory;
|
2020-03-19 14:47:25 +00:00
|
|
|
private readonly ICached<SeedCriteriaSettings> _cache;
|
2019-06-14 03:54:25 +00:00
|
|
|
|
2020-03-19 14:47:25 +00:00
|
|
|
public SeedConfigProvider(IIndexerFactory indexerFactory, ICacheManager cacheManager)
|
2019-06-14 03:54:25 +00:00
|
|
|
{
|
|
|
|
_indexerFactory = indexerFactory;
|
2020-03-19 14:47:25 +00:00
|
|
|
_cache = cacheManager.GetRollingCache<SeedCriteriaSettings>(GetType(), "criteriaByIndexer", TimeSpan.FromHours(1));
|
2019-06-14 03:54:25 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public TorrentSeedConfiguration GetSeedConfiguration(RemoteMovie remoteMovie)
|
|
|
|
{
|
2019-12-22 22:08:53 +00:00
|
|
|
if (remoteMovie.Release.DownloadProtocol != DownloadProtocol.Torrent)
|
|
|
|
{
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (remoteMovie.Release.IndexerId == 0)
|
|
|
|
{
|
|
|
|
return null;
|
|
|
|
}
|
2019-06-14 03:54:25 +00:00
|
|
|
|
2020-03-19 14:47:25 +00:00
|
|
|
return GetSeedConfiguration(remoteMovie.Release.IndexerId);
|
|
|
|
}
|
|
|
|
|
|
|
|
public TorrentSeedConfiguration GetSeedConfiguration(int indexerId)
|
|
|
|
{
|
|
|
|
if (indexerId == 0)
|
|
|
|
{
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
var seedCriteria = _cache.Get(indexerId.ToString(), () => FetchSeedCriteria(indexerId));
|
|
|
|
|
|
|
|
if (seedCriteria == null)
|
|
|
|
{
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
var seedConfig = new TorrentSeedConfiguration
|
|
|
|
{
|
|
|
|
Ratio = seedCriteria.SeedRatio
|
|
|
|
};
|
|
|
|
|
|
|
|
var seedTime = seedCriteria.SeedTime;
|
|
|
|
|
|
|
|
if (seedTime.HasValue)
|
|
|
|
{
|
|
|
|
seedConfig.SeedTime = TimeSpan.FromMinutes(seedTime.Value);
|
|
|
|
}
|
|
|
|
|
|
|
|
return seedConfig;
|
|
|
|
}
|
|
|
|
|
|
|
|
private SeedCriteriaSettings FetchSeedCriteria(int indexerId)
|
|
|
|
{
|
2019-06-14 03:54:25 +00:00
|
|
|
try
|
|
|
|
{
|
2020-03-19 14:47:25 +00:00
|
|
|
var indexer = _indexerFactory.Get(indexerId);
|
2019-06-14 03:54:25 +00:00
|
|
|
var torrentIndexerSettings = indexer.Settings as ITorrentIndexerSettings;
|
|
|
|
|
2020-03-19 14:47:25 +00:00
|
|
|
return torrentIndexerSettings?.SeedCriteria;
|
2019-06-14 03:54:25 +00:00
|
|
|
}
|
|
|
|
catch (ModelNotFoundException)
|
|
|
|
{
|
|
|
|
return null;
|
|
|
|
}
|
2020-03-19 14:47:25 +00:00
|
|
|
}
|
2019-06-14 03:54:25 +00:00
|
|
|
|
2021-04-23 17:18:09 +00:00
|
|
|
public void Handle(ProviderUpdatedEvent<IIndexer> message)
|
2020-03-19 14:47:25 +00:00
|
|
|
{
|
|
|
|
_cache.Clear();
|
2019-06-14 03:54:25 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|