1
0
mirror of https://github.com/Sonarr/Sonarr.git synced 2026-04-22 22:16:13 -04:00

New: Option to treat downloads with non-media extensions as failed

Closes #7369
This commit is contained in:
Mark McDowall
2024-11-13 20:13:00 -08:00
parent 8c67a3bdee
commit 776143cc81
25 changed files with 229 additions and 51 deletions
@@ -0,0 +1,69 @@
using System;
using System.Collections.Generic;
using System.Linq;
using NzbDrone.Common.Cache;
using NzbDrone.Core.Messaging.Events;
using NzbDrone.Core.ThingiProvider.Events;
namespace NzbDrone.Core.Indexers;
public interface ICachedIndexerSettingsProvider
{
CachedIndexerSettings GetSettings(int indexerId);
}
public class CachedIndexerSettingsProvider : ICachedIndexerSettingsProvider, IHandle<ProviderUpdatedEvent<IIndexer>>
{
private readonly IIndexerFactory _indexerFactory;
private readonly ICached<CachedIndexerSettings> _cache;
public CachedIndexerSettingsProvider(IIndexerFactory indexerFactory, ICacheManager cacheManager)
{
_indexerFactory = indexerFactory;
_cache = cacheManager.GetRollingCache<CachedIndexerSettings>(GetType(), "settingsByIndexer", TimeSpan.FromHours(1));
}
public CachedIndexerSettings GetSettings(int indexerId)
{
if (indexerId == 0)
{
return null;
}
return _cache.Get(indexerId.ToString(), () => FetchIndexerSettings(indexerId));
}
private CachedIndexerSettings FetchIndexerSettings(int indexerId)
{
var indexer = _indexerFactory.Get(indexerId);
var indexerSettings = indexer.Settings as IIndexerSettings;
if (indexerSettings == null)
{
return null;
}
var settings = new CachedIndexerSettings
{
FailDownloads = indexerSettings.FailDownloads.Select(f => (FailDownloads)f).ToHashSet()
};
if (indexer.Settings is ITorrentIndexerSettings torrentIndexerSettings)
{
settings.SeedCriteriaSettings = torrentIndexerSettings.SeedCriteria;
}
return settings;
}
public void Handle(ProviderUpdatedEvent<IIndexer> message)
{
_cache.Clear();
}
}
public class CachedIndexerSettings
{
public HashSet<FailDownloads> FailDownloads { get; set; }
public SeedCriteriaSettings SeedCriteriaSettings { get; set; }
}