Merge pull request #10 from unic/feature/improve-servicesconfigurator

Add flexibility to change the base path and add extension points
merkle-open · Aug 24, 2017 · bc0505f · bc0505f
2 parents 7045641 + b710cf3
commit bc0505f
Showing 1 changed file with 17 additions and 7 deletions.
diff --git a/NitroNet.Sitecore.Microsoft.DependencyInjection/NitroNetServicesConfigurator.cs b/NitroNet.Sitecore.Microsoft.DependencyInjection/NitroNetServicesConfigurator.cs
@@ -1,5 +1,4 @@
-using System.Web;
-using System.Web.Hosting;
+using System.Web.Hosting;
 using Microsoft.Extensions.DependencyInjection;
 using NitroNet.Mvc;
 using NitroNet.Sitecore.Caching;
@@ -11,6 +10,8 @@
 using NitroNet.ViewEngine.TemplateHandler;
 using NitroNet.ViewEngine.TemplateHandler.Grid;
 using NitroNet.ViewEngine.ViewEngines;
+using Sitecore;
+using Sitecore.Configuration;
 using Sitecore.DependencyInjection;
 using Sitecore.Mvc.Common;
 using Veil;
@@ -19,6 +20,7 @@
 
 namespace NitroNet.Sitecore.Microsoft.DependencyInjection
 {
+    [UsedImplicitly]
     public class NitroNetServicesConfigurator : IServicesConfigurator
     {
         public void Configure(IServiceCollection serviceCollection)
@@ -27,7 +29,7 @@ public void Configure(IServiceCollection serviceCollection)
             RegisterNitroNetSitecore(serviceCollection);
         }
 
-        private static void RegisterNitroNetSitecore(IServiceCollection serviceCollection)
+        protected virtual void RegisterNitroNetSitecore(IServiceCollection serviceCollection)
         {
             serviceCollection.AddTransient<SitecoreNitroNetViewEngine>();
             serviceCollection.AddTransient<ISitecoreRenderingRepository, SitecoreRenderingRepository>();
@@ -36,13 +38,13 @@ private static void RegisterNitroNetSitecore(IServiceCollection serviceCollectio
             serviceCollection.AddSingleton<INitroTemplateHandlerFactory, SitecoreMvcNitroTemplateHandlerFactory>();
         }
 
-        private static void RegisterNitroNet(IServiceCollection serviceCollection)
+        protected virtual void RegisterNitroNet(IServiceCollection serviceCollection)
         {
-            var rootPath = HostingEnvironment.MapPath("~/");
+            var basePath = this.GetNitroNetBasePath();
 
-            var config = ConfigurationLoader.LoadNitroConfiguration(rootPath);
+            var config = ConfigurationLoader.LoadNitroConfiguration(basePath);
             serviceCollection.AddSingleton(config);
-            serviceCollection.AddSingleton<IFileSystem>(new FileSystem(rootPath, config));
+            serviceCollection.AddSingleton<IFileSystem>(new FileSystem(basePath, config));
 
             serviceCollection.AddSingleton<IHelperHandlerFactory, DefaultRenderingHelperHandlerFactory>();
             serviceCollection.AddTransient<IMemberLocator, MemberLocatorFromNamingRule>();
@@ -54,5 +56,13 @@ private static void RegisterNitroNet(IServiceCollection serviceCollection)
             serviceCollection.AddSingleton<ITemplateRepository, NitroTemplateRepository>();
             serviceCollection.AddSingleton<INitroTemplateHandlerFactory, MvcNitroTemplateHandlerFactory>();
         }
+
+        protected virtual string GetNitroNetBasePath()
+        {
+            var rootPath = HostingEnvironment.MapPath("~/");
+            var basePath = PathInfo.Combine(PathInfo.Create(rootPath), PathInfo.Create(Settings.GetAppSetting("NitroNet.BasePath")));
+
+            return basePath.ToString();
+        }
     }
 }