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.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) { BindKernel(); } return _kernel; } } public static void BindKernel() { lock (KernelLock) { Logger.Debug("Binding Ninject's Kernel"); _kernel = new StandardKernel(); //Sqlite var appDataPath = new DirectoryInfo(Path.Combine(AppPath, "App_Data")); if (!appDataPath.Exists) appDataPath.Create(); string connectionString = String.Format("Data Source={0};Version=3;", Path.Combine(appDataPath.FullName, "nzbdrone.db")); var dbProvider = ProviderFactory.GetProvider(connectionString, "System.Data.SQLite"); string logConnectionString = String.Format("Data Source={0};Version=3;", Path.Combine(appDataPath.FullName, "log.db")); var logDbProvider = ProviderFactory.GetProvider(logConnectionString, "System.Data.SQLite"); //SQLExpress //string logConnectionString = String.Format(@"server=.\SQLExpress; database=NzbDroneLogs; Trusted_Connection=True;"); //var logDbProvider = ProviderFactory.GetProvider(logConnectionString, "System.Data.SqlClient"); var logRepository = new SimpleRepository(logDbProvider, SimpleRepositoryOptions.RunMigrations); //dbProvider.ExecuteQuery(new QueryCommand("VACUUM", dbProvider)); //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().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().ToMethod( c => new SimpleRepository(dbProvider, SimpleRepositoryOptions.RunMigrations)).InSingletonScope(); _kernel.Bind().ToConstant(logRepository).WhenInjectedInto(). InSingletonScope(); _kernel.Bind().ToConstant(logRepository).WhenInjectedInto().InSingletonScope(); LogConfiguration.Setup(); ForceMigration(_kernel.Get()); SetupDefaultQualityProfiles(_kernel.Get()); //Setup the default QualityProfiles on start-up BindIndexers(); BindJobs(); BindExternalNotifications(); } } 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.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(); 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.Bluray720 }, 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); } } } }