Radarr/NzbDrone.Core/CentralDispatch.cs

229 lines
9.1 KiB
C#
Raw Normal View History

2010-09-23 03:19:47 +00:00
using System;
using System.Linq;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Web.Hosting;
2010-09-23 03:19:47 +00:00
using Ninject;
2011-04-10 02:44:01 +00:00
using NLog;
2011-05-23 23:29:14 +00:00
using NzbDrone.Core.Datastore;
using NzbDrone.Core.Instrumentation;
using NzbDrone.Core.Providers;
2011-04-04 03:50:12 +00:00
using NzbDrone.Core.Providers.Core;
using NzbDrone.Core.Providers.ExternalNotification;
using NzbDrone.Core.Providers.Indexer;
using NzbDrone.Core.Providers.Jobs;
using NzbDrone.Core.Repository;
using NzbDrone.Core.Repository.Quality;
2010-09-23 03:19:47 +00:00
using SubSonic.DataProviders;
using SubSonic.Repository;
namespace NzbDrone.Core
{
2010-10-05 06:21:18 +00:00
public static class CentralDispatch
2010-09-23 03:19:47 +00:00
{
2011-03-30 06:18:35 +00:00
private static StandardKernel _kernel;
private static readonly Object KernelLock = new object();
2010-10-10 19:00:07 +00:00
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
2010-10-07 22:17:24 +00:00
2011-04-10 02:44:01 +00:00
public static String AppPath
{
get
{
if (!String.IsNullOrWhiteSpace(HostingEnvironment.ApplicationPhysicalPath))
{
return HostingEnvironment.ApplicationPhysicalPath;
}
return Directory.GetCurrentDirectory();
}
}
public static StandardKernel NinjectKernel
{
get
{
if (_kernel == null)
{
2011-05-23 23:29:14 +00:00
InitializeApp();
2011-04-10 02:44:01 +00:00
}
return _kernel;
}
}
2011-05-23 23:29:14 +00:00
private static void InitializeApp()
{
BindKernel();
2011-05-23 23:29:14 +00:00
LogConfiguration.Setup();
2011-05-23 23:29:14 +00:00
Migrations.Run();
ForceMigration(_kernel.Get<IRepository>());
2011-05-23 23:29:14 +00:00
SetupDefaultQualityProfiles(_kernel.Get<IRepository>()); //Setup the default QualityProfiles on start-up
BindIndexers();
BindJobs();
BindExternalNotifications();
}
2010-10-10 19:00:07 +00:00
public static void BindKernel()
2010-09-23 03:19:47 +00:00
{
lock (KernelLock)
2010-10-10 19:00:07 +00:00
{
Logger.Debug("Binding Ninject's Kernel");
_kernel = new StandardKernel();
//dbProvider.Log = new NlogWriter();
_kernel.Bind<QualityProvider>().ToSelf().InSingletonScope();
_kernel.Bind<TvDbProvider>().ToSelf().InTransientScope();
_kernel.Bind<HttpProvider>().ToSelf().InSingletonScope();
2011-04-08 23:55:23 +00:00
_kernel.Bind<SeriesProvider>().ToSelf().InSingletonScope();
_kernel.Bind<SeasonProvider>().ToSelf().InSingletonScope();
_kernel.Bind<EpisodeProvider>().ToSelf().InSingletonScope();
_kernel.Bind<UpcomingEpisodesProvider>().ToSelf().InSingletonScope();
_kernel.Bind<DiskProvider>().ToSelf().InSingletonScope();
_kernel.Bind<SabProvider>().ToSelf().InSingletonScope();
_kernel.Bind<HistoryProvider>().ToSelf().InSingletonScope();
_kernel.Bind<RootDirProvider>().ToSelf().InSingletonScope();
_kernel.Bind<ExternalNotificationProvider>().ToSelf().InSingletonScope();
_kernel.Bind<XbmcProvider>().ToSelf().InSingletonScope();
_kernel.Bind<ConfigProvider>().To<ConfigProvider>().InSingletonScope();
_kernel.Bind<SyncProvider>().ToSelf().InSingletonScope();
2011-04-09 00:08:03 +00:00
_kernel.Bind<RenameProvider>().ToSelf().InSingletonScope();
2011-04-08 23:55:23 +00:00
_kernel.Bind<NotificationProvider>().ToSelf().InSingletonScope();
_kernel.Bind<LogProvider>().ToSelf().InSingletonScope();
_kernel.Bind<MediaFileProvider>().ToSelf().InSingletonScope();
_kernel.Bind<JobProvider>().ToSelf().InSingletonScope();
_kernel.Bind<IndexerProvider>().ToSelf().InSingletonScope();
_kernel.Bind<WebTimer>().ToSelf().InSingletonScope();
_kernel.Bind<AutoConfigureProvider>().ToSelf().InSingletonScope();
2011-05-23 23:29:14 +00:00
_kernel.Bind<IRepository>().ToConstant(Connection.MainDataRepository).InSingletonScope();
_kernel.Bind<IRepository>().ToConstant(Connection.LogDataRepository).WhenInjectedInto<SubsonicTarget>().InSingletonScope();
_kernel.Bind<IRepository>().ToConstant(Connection.LogDataRepository).WhenInjectedInto<LogProvider>().InSingletonScope();
2010-10-10 19:00:07 +00:00
}
2010-09-23 03:19:47 +00:00
}
private static void BindIndexers()
{
_kernel.Bind<IndexerBase>().To<NzbsOrg>().InSingletonScope();
_kernel.Bind<IndexerBase>().To<NzbMatrix>().InSingletonScope();
_kernel.Bind<IndexerBase>().To<NzbsRUs>().InSingletonScope();
_kernel.Bind<IndexerBase>().To<Newzbin>().InSingletonScope();
}
private static void BindJobs()
2011-04-20 02:17:28 +00:00
{
_kernel.Bind<IJob>().To<RssSyncJob>().InTransientScope();
_kernel.Bind<IJob>().To<ImportNewSeriesJob>().InTransientScope();
_kernel.Bind<IJob>().To<UpdateInfoJob>().InTransientScope();
_kernel.Bind<IJob>().To<DiskScanJob>().InTransientScope();
_kernel.Bind<IJob>().To<DeleteSeriesJob>().InTransientScope();
2011-04-23 19:47:05 +00:00
_kernel.Get<JobProvider>().Initialize();
_kernel.Get<WebTimer>().StartTimer(30);
2011-04-20 02:17:28 +00:00
}
private static void BindExternalNotifications()
{
_kernel.Bind<ExternalNotificationProviderBase>().To<XbmcNotificationProvider>().InSingletonScope();
var notifiers = _kernel.GetAll<ExternalNotificationProviderBase>();
_kernel.Get<ExternalNotificationProvider>().InitializeNotifiers(notifiers.ToList());
}
2010-10-05 06:21:18 +00:00
private static void ForceMigration(IRepository repository)
{
2011-04-23 19:47:05 +00:00
repository.All<Series>().Count();
repository.All<Season>().Count();
repository.All<Episode>().Count();
repository.All<EpisodeFile>().Count();
repository.All<QualityProfile>().Count();
repository.All<History>().Count();
repository.All<IndexerSetting>().Count();
2010-10-05 06:21:18 +00:00
}
/// <summary>
2011-04-20 01:20:20 +00:00
/// Forces IISExpress process to exit with the host application
/// </summary>
public static void DedicateToHost()
{
try
{
2010-10-24 17:35:58 +00:00
Logger.Debug("Attaching to parent process for automatic termination.");
2011-04-10 02:44:01 +00:00
var pc = new PerformanceCounter("Process", "Creating Process ID",
Process.GetCurrentProcess().ProcessName);
var pid = (int)pc.NextValue();
var hostProcess = Process.GetProcessById(pid);
hostProcess.EnableRaisingEvents = true;
hostProcess.Exited += (delegate
2011-04-10 02:44:01 +00:00
{
Logger.Info("Host has been terminated. Shutting down web server.");
ShutDown();
});
2010-10-24 17:35:58 +00:00
Logger.Debug("Successfully Attached to host. Process ID: {0}", pid);
}
catch (Exception e)
{
Logger.Fatal(e);
}
}
private static void ShutDown()
{
Logger.Info("Shutting down application.");
Process.GetCurrentProcess().Kill();
}
private static void SetupDefaultQualityProfiles(IRepository repository)
{
var sd = new QualityProfile
2011-04-10 02:44:01 +00:00
{
Name = "SD",
2011-05-23 17:20:43 +00:00
Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD },
Cutoff = QualityTypes.SDTV
2011-04-10 02:44:01 +00:00
};
var hd = new QualityProfile
2011-04-10 02:44:01 +00:00
{
Name = "HD",
Allowed =
2011-05-26 03:13:39 +00:00
new List<QualityTypes> { QualityTypes.HDTV, QualityTypes.WEBDL, QualityTypes.Bluray720p },
2011-04-10 02:44:01 +00:00
Cutoff = QualityTypes.HDTV
};
//Add or Update SD
Logger.Debug(String.Format("Checking for default QualityProfile: {0}", sd.Name));
var sdDb = repository.Single<QualityProfile>(i => i.Name == sd.Name);
if (sdDb == null)
{
Logger.Debug(String.Format("Adding new default QualityProfile: {0}", sd.Name));
repository.Add(sd);
}
else
{
Logger.Debug(String.Format("Updating default QualityProfile: {0}", sd.Name));
sd.QualityProfileId = sdDb.QualityProfileId;
repository.Update(sd);
}
//Add or Update HD
Logger.Debug(String.Format("Checking for default QualityProfile: {0}", hd.Name));
var hdDb = repository.Single<QualityProfile>(i => i.Name == hd.Name);
if (hdDb == null)
{
Logger.Debug(String.Format("Adding new default QualityProfile: {0}", hd.Name));
repository.Add(hd);
}
else
{
Logger.Debug(String.Format("Updating default QualityProfile: {0}", hd.Name));
hd.QualityProfileId = hdDb.QualityProfileId;
repository.Update(hd);
}
}
2010-09-23 03:19:47 +00:00
}
2010-09-28 05:58:49 +00:00
}