using System; using System.Linq; using System.Collections.Generic; using System.Diagnostics; using System.IO; using System.Web.Hosting; using Ninject; using NLog; using NzbDrone.Core.Datastore; using NzbDrone.Core.Instrumentation; using NzbDrone.Core.Providers; 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; using SubSonic.DataProviders; using SubSonic.Repository; namespace NzbDrone.Core { public static class CentralDispatch { private static StandardKernel _kernel; private static readonly Object KernelLock = new object(); private static readonly Logger Logger = LogManager.GetCurrentClassLogger(); public static String AppPath { get { if (!String.IsNullOrWhiteSpace(HostingEnvironment.ApplicationPhysicalPath)) { return HostingEnvironment.ApplicationPhysicalPath; } return Directory.GetCurrentDirectory(); } } public static StandardKernel NinjectKernel { get { if (_kernel == null) { InitializeApp(); } return _kernel; } } private static void InitializeApp() { BindKernel(); LogConfiguration.Setup(); Migrations.Run(Connection.MainConnectionString, true); ForceMigration(_kernel.Get()); SetupDefaultQualityProfiles(_kernel.Get()); //Setup the default QualityProfiles on start-up BindIndexers(); BindJobs(); BindExternalNotifications(); } public static void BindKernel() { lock (KernelLock) { Logger.Debug("Binding Ninject's Kernel"); _kernel = new StandardKernel(); //dbProvider.Log = new NlogWriter(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InTransientScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().To().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToSelf().InSingletonScope(); _kernel.Bind().ToConstant(Connection.CreateSimpleRepository(Connection.MainConnectionString)).InSingletonScope(); _kernel.Bind().ToConstant(Connection.CreateSimpleRepository(Connection.LogConnectionString)).WhenInjectedInto().InSingletonScope(); _kernel.Bind().ToConstant(Connection.CreateSimpleRepository(Connection.LogConnectionString)).WhenInjectedInto().InSingletonScope(); } } private static void BindIndexers() { _kernel.Bind().To().InSingletonScope(); _kernel.Bind().To().InSingletonScope(); _kernel.Bind().To().InSingletonScope(); _kernel.Bind().To().InSingletonScope(); var indexers = _kernel.GetAll(); _kernel.Get().InitializeIndexers(indexers.ToList()); } private static void BindJobs() { _kernel.Bind().To().InTransientScope(); _kernel.Bind().To().InTransientScope(); _kernel.Bind().To().InTransientScope(); _kernel.Bind().To().InTransientScope(); _kernel.Bind().To().InTransientScope(); _kernel.Bind().To().InTransientScope(); _kernel.Get().Initialize(); _kernel.Get().StartTimer(30); } private static void BindExternalNotifications() { _kernel.Bind().To().InSingletonScope(); var notifiers = _kernel.GetAll(); _kernel.Get().InitializeNotifiers(notifiers.ToList()); } private static void ForceMigration(IRepository repository) { repository.All().Count(); repository.All().Count(); repository.All().Count(); repository.All().Count(); repository.All().Count(); repository.All().Count(); } /// /// Forces IISExpress process to exit with the host application /// public static void DedicateToHost() { try { Logger.Debug("Attaching to parent process for automatic termination."); 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 { Logger.Info("Host has been terminated. Shutting down web server."); ShutDown(); }); 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 { Name = "SD", Allowed = new List { QualityTypes.SDTV, QualityTypes.DVD }, Cutoff = QualityTypes.SDTV }; var hd = new QualityProfile { Name = "HD", Allowed = new List { QualityTypes.HDTV, QualityTypes.WEBDL, QualityTypes.Bluray720p }, Cutoff = QualityTypes.HDTV }; //Add or Update SD Logger.Debug(String.Format("Checking for default QualityProfile: {0}", sd.Name)); var sdDb = repository.Single(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(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); } } } }