Jackett/src/Jackett/Services/IndexerManagerService.cs

102 lines
3.4 KiB
C#
Raw Normal View History

2015-07-19 00:27:41 +00:00
using Autofac;
2015-07-19 13:22:50 +00:00
using Jackett.Indexers;
2015-07-19 00:27:41 +00:00
using Jackett.Models;
2015-07-19 00:59:30 +00:00
using Newtonsoft.Json.Linq;
2015-07-19 00:27:41 +00:00
using NLog;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace Jackett.Services
{
public interface IIndexerManagerService
{
void TestIndexer(string name);
void DeleteIndexer(string name);
2015-07-19 13:22:50 +00:00
IIndexer GetIndexer(string name);
IEnumerable<IIndexer> GetAllIndexers();
void SaveConfig(IIndexer indexer, JToken obj);
void InitIndexers();
2015-07-19 00:27:41 +00:00
}
public class IndexerManagerService : IIndexerManagerService
{
private IContainer container;
private IConfigurationService configService;
private Logger logger;
public IndexerManagerService(IContainer c, IConfigurationService config, Logger l)
{
container = c;
configService = config;
logger = l;
}
2015-07-19 13:22:50 +00:00
public void InitIndexers()
2015-07-19 00:27:41 +00:00
{
2015-07-19 13:22:50 +00:00
// Load the existing config for each indexer
foreach (var indexer in GetAllIndexers())
{
var configFilePath = GetIndexerConfigFilePath(indexer);
if (File.Exists(configFilePath))
{
var jsonString = JObject.Parse(File.ReadAllText(configFilePath));
indexer.LoadFromSavedConfiguration(jsonString);
}
}
2015-07-19 00:27:41 +00:00
}
2015-07-19 13:22:50 +00:00
public IIndexer GetIndexer(string name)
2015-07-19 00:27:41 +00:00
{
2015-07-19 13:22:50 +00:00
var indexer = GetAllIndexers().Where(i => string.Equals(i.DisplayName, name, StringComparison.InvariantCultureIgnoreCase)).FirstOrDefault();
if (indexer == null)
{
logger.Error("Request for unknown indexer: " + name);
throw new Exception("Unknown indexer: " + name);
}
return indexer;
}
public IEnumerable<IIndexer> GetAllIndexers()
{
return container.Resolve<IEnumerable<IIndexer>>().OrderBy(_ => _.DisplayName);
2015-07-19 00:27:41 +00:00
}
public async void TestIndexer(string name)
{
var indexer = GetIndexer(name);
var browseQuery = new TorznabQuery();
var results = await indexer.PerformQuery(browseQuery);
logger.Debug(string.Format("Found {0} releases from {1}", results.Length, indexer.DisplayName));
if (results.Length == 0)
throw new Exception("Found no results while trying to browse this tracker");
}
public void DeleteIndexer(string name)
{
var indexer = GetIndexer(name);
var configPath = configService.GetIndexerConfigDir();
File.Delete(configPath);
//Indexers.Remove(name);
//LoadMissingIndexers();
}
2015-07-19 13:22:50 +00:00
private string GetIndexerConfigFilePath(IIndexer indexer)
2015-07-19 00:27:41 +00:00
{
return Path.Combine(configService.GetIndexerConfigDir(), indexer.GetType().Name.ToLower() + ".json");
}
2015-07-19 00:59:30 +00:00
2015-07-19 13:22:50 +00:00
public void SaveConfig(IIndexer indexer, JToken obj)
2015-07-19 00:59:30 +00:00
{
var configFilePath = GetIndexerConfigFilePath(indexer);
if (!Directory.Exists(configService.GetIndexerConfigDir()))
Directory.CreateDirectory(configService.GetIndexerConfigDir());
File.WriteAllText(configFilePath, obj.ToString());
}
2015-07-19 00:27:41 +00:00
}
}