69 lines
2.3 KiB
C#
69 lines
2.3 KiB
C#
using System.Data;
|
|
using System.IO;
|
|
using System.Reflection;
|
|
using Autofac;
|
|
using NLog;
|
|
using NzbDrone.Common;
|
|
using NzbDrone.Common.Eventing;
|
|
using NzbDrone.Core.Datastore;
|
|
using NzbDrone.Core.ExternalNotification;
|
|
using NzbDrone.Core.IndexerSearch;
|
|
using NzbDrone.Core.Indexers;
|
|
using NzbDrone.Core.Instrumentation;
|
|
|
|
namespace NzbDrone.Core
|
|
{
|
|
public static class ContainerExtensions
|
|
{
|
|
|
|
private static readonly Logger logger = LogManager.GetLogger("ServiceRegistration");
|
|
|
|
public static void RegisterCoreServices(this ContainerBuilder containerBuilder)
|
|
{
|
|
containerBuilder.RegisterAssembly("NzbDrone.Common");
|
|
containerBuilder.RegisterAssembly("NzbDrone.Core");
|
|
|
|
containerBuilder.InitDatabase();
|
|
|
|
|
|
containerBuilder.RegisterType<EventAggregator>()
|
|
.As<IEventAggregator>().SingleInstance();
|
|
|
|
containerBuilder.RegisterModule<LogInjectionModule>();
|
|
}
|
|
|
|
|
|
private static void RegisterAssembly(this ContainerBuilder container, string assemblyName)
|
|
{
|
|
|
|
container.RegisterAssemblyTypes(assemblyName);
|
|
|
|
var assembly = Assembly.Load(assemblyName);
|
|
|
|
container.RegisterAssemblyTypes(assembly)
|
|
.Where(t => t.IsSubclassOf(typeof(IndexerBase)))
|
|
.As<IndexerBase>();
|
|
|
|
container.RegisterAssemblyTypes(assembly)
|
|
.Where(t => t.IsSubclassOf(typeof(IndexerSearchBase)))
|
|
.As<IndexerSearchBase>();
|
|
|
|
container.RegisterAssemblyTypes(assembly)
|
|
.Where(t => t.IsSubclassOf(typeof(ExternalNotificationBase)))
|
|
.As<ExternalNotificationBase>();
|
|
}
|
|
|
|
private static void InitDatabase(this ContainerBuilder container)
|
|
{
|
|
logger.Info("Registering Database...");
|
|
|
|
var environmentProvider = new EnvironmentProvider();
|
|
var appDataPath = environmentProvider.GetAppDataPath();
|
|
if (!Directory.Exists(appDataPath)) Directory.CreateDirectory(appDataPath);
|
|
|
|
container.Register(c => c.Resolve<IDbFactory>().Create(environmentProvider.GetNzbDroneDatabase())).As<IDatabase>();
|
|
|
|
container.RegisterGeneric(typeof(BasicRepository<>)).As(typeof(IBasicRepository<>));
|
|
}
|
|
}
|
|
} |