Radarr/NzbDrone.Core/ContainerExtensions.cs

64 lines
2.2 KiB
C#
Raw Normal View History

2013-03-24 04:16:00 +00:00
using System.Data;
using System.IO;
using System.Reflection;
using Autofac;
using NLog;
using NzbDrone.Common;
using NzbDrone.Core.Datastore;
using NzbDrone.Core.ExternalNotification;
2013-03-07 03:45:36 +00:00
using NzbDrone.Core.IndexerSearch;
2013-02-23 04:37:23 +00:00
using NzbDrone.Core.Indexers;
using NzbDrone.Core.Instrumentation;
namespace NzbDrone.Core
{
public static class ContainerExtensions
{
2013-01-19 18:54:00 +00:00
private static readonly Logger logger = LogManager.GetLogger("ServiceRegistration");
public static void RegisterCoreServices(this ContainerBuilder containerBuilder)
{
2013-02-19 07:20:51 +00:00
containerBuilder.RegisterAssembly("NzbDrone.Common");
2013-02-19 01:57:08 +00:00
containerBuilder.RegisterAssembly("NzbDrone.Core");
containerBuilder.InitDatabase();
containerBuilder.RegisterModule<LogInjectionModule>();
}
2013-02-19 01:57:08 +00:00
private static void RegisterAssembly(this ContainerBuilder container, string assemblyName)
{
2013-01-19 19:42:06 +00:00
2013-02-19 01:57:08 +00:00
container.RegisterAssemblyTypes(assemblyName);
2013-02-19 01:57:08 +00:00
var assembly = Assembly.Load(assemblyName);
container.RegisterAssemblyTypes(assembly)
2013-01-19 18:54:00 +00:00
.Where(t => t.IsSubclassOf(typeof(IndexerBase)))
.As<IndexerBase>().SingleInstance();
container.RegisterAssemblyTypes(assembly)
.Where(t => t.IsSubclassOf(typeof(IndexerSearchBase)))
.As<IndexerSearchBase>().SingleInstance();
container.RegisterAssemblyTypes(assembly)
2013-01-19 18:54:00 +00:00
.Where(t => t.IsSubclassOf(typeof(ExternalNotificationBase)))
.As<ExternalNotificationBase>().SingleInstance();
}
2013-01-19 19:42:06 +00:00
private static void InitDatabase(this ContainerBuilder container)
{
2013-01-19 18:54:00 +00:00
logger.Info("Registering Database...");
var environmentProvider = new EnvironmentProvider();
var appDataPath = environmentProvider.GetAppDataPath();
if (!Directory.Exists(appDataPath)) Directory.CreateDirectory(appDataPath);
2013-03-26 06:55:27 +00:00
container.Register(c => c.Resolve<IDbFactory>().Create(environmentProvider.GetNzbDroneDatabase())).As<IDatabase>().SingleInstance();
container.RegisterGeneric(typeof(BasicRepository<>)).As(typeof(IBasicRepository<>));
}
}
}