Merge pull request #18 from enisn/feature/controller-property-activators

♻️Set controller context in ControllerFactory
enisn · Dec 9, 2021 · 4804b40 · 4804b40
2 parents d8949f5 + 5b3ab75
commit 4804b40
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/DotNurse.Injector.AspNetCore/DotNurseControllerFactory.cs b/src/DotNurse.Injector.AspNetCore/DotNurseControllerFactory.cs
@@ -13,7 +13,9 @@ public class DotNurseControllerFactory : IControllerFactory
     private readonly IControllerActivator controllerActivator;
     private readonly IAttributeInjector attributeInjector;
 
-    public DotNurseControllerFactory(IControllerActivator controllerActivator, IAttributeInjector attributeInjector)
+    public DotNurseControllerFactory(
+        IControllerActivator controllerActivator,
+        IAttributeInjector attributeInjector)
     {
         this.controllerActivator = controllerActivator;
         this.attributeInjector = attributeInjector;
@@ -22,7 +24,14 @@ public DotNurseControllerFactory(IControllerActivator controllerActivator, IAttr
     public object CreateController(ControllerContext context)
     {
         var controller = controllerActivator.Create(context);
+
+        if (controller is ControllerBase controllerBase)
+        {
+            controllerBase.ControllerContext = context;
+        }
+
         attributeInjector.InjectIntoMembers(controller, context.HttpContext.RequestServices);
+
         return controller;
     }