Enabling Injection of Custom Validation Attributes in MVC Projects

LightInject.Mvc/LightInject.Mvc.cs

@@ -25,6 +25,7 @@ LightInject.Mvc version 1.0.0.3
     http://seesharper.github.io/LightInject/
     http://twitter.com/bernhardrichter    
 ******************************************************************************/
+
 [module: System.Diagnostics.CodeAnalysis.SuppressMessage("StyleCop.CSharp.ReadabilityRules", "SA1126:PrefixCallsCorrectly", Justification = "Reviewed")]
 [module: System.Diagnostics.CodeAnalysis.SuppressMessage("StyleCop.CSharp.ReadabilityRules", "SA1101:PrefixLocalCallsWithThis", Justification = "No inheritance")]
 [module: System.Diagnostics.CodeAnalysis.SuppressMessage("StyleCop.CSharp.MaintainabilityRules", "SA1402:FileMayOnlyContainASingleClass", Justification = "Single source file deployment.")]
@@ -38,6 +39,7 @@ namespace LightInject
     using System.Reflection;
     using System.Web.Mvc;
     using LightInject.Mvc;
+
 
     /// <summary>
     /// Extends the <see cref="IServiceContainer"/> interface with a method that 
@@ -81,6 +83,9 @@ public static void EnableMvc(this IServiceContainer serviceContainer)
             ((ServiceContainer)serviceContainer).EnablePerWebRequestScope();
             SetDependencyResolver(serviceContainer);
             InitializeFilterAttributeProvider(serviceContainer);
+
+			ModelValidatorProviders.Providers.Remove(ModelValidatorProviders.Providers.OfType<DataAnnotationsModelValidatorProvider>().First());
+			ModelValidatorProviders.Providers.Add(new LightInjectModelValidatorProvider(serviceContainer));
         }
 
         private static void SetDependencyResolver(IServiceContainer serviceContainer)
@@ -113,6 +118,7 @@ namespace LightInject.Mvc
     using System.Collections.Generic;
     using System.Linq;
     using System.Web.Mvc;
+	using System.Reflection;
 
     /// <summary>
     /// An <see cref="IDependencyResolver"/> adapter for the LightInject service container.
@@ -194,4 +200,48 @@ public override IEnumerable<Filter> GetFilters(ControllerContext controllerConte
             return filters;
         }                
     }
-}
+
+	/// <summary>
+	/// A Model Validator Provider for Light Inject
+	/// </summary>
+	internal class LightInjectModelValidatorProvider: DataAnnotationsModelValidatorProvider
+	{
+		/// <summary>
+		/// An Instance of a service container
+		/// </summary>
+		private readonly IServiceContainer _serviceContainer;
+
+		/// <summary>
+		/// Initializes a new instance of the <see cref="LightInjectModelValidatorProvider"/> class.
+        /// </summary>
+		internal LightInjectModelValidatorProvider(IServiceContainer serviceContainer)
+        {
+	        _serviceContainer = serviceContainer;
+        }
+
+        /// <summary>
+        /// Gets a list of validators.
+        /// </summary>
+        /// <param name="metadata">The metadata.</param>
+        /// <param name="context">The context.</param>
+        /// <param name="attributes">The list of validation attributes.</param>
+        /// <returns>A list of validators.</returns>
+        protected override IEnumerable<ModelValidator> GetValidators(ModelMetadata metadata, ControllerContext context, IEnumerable<Attribute> attributes)
+        {
+			var attributeMethodInfo =
+				typeof(DataAnnotationsModelValidator).GetMethod("get_Attribute", BindingFlags.NonPublic | BindingFlags.DeclaredOnly | BindingFlags.Instance);
+
+            var validators = base.GetValidators(metadata, context, attributes).ToList();
+            foreach (var modelValidator in validators.OfType<DataAnnotationsModelValidator>())
+            {
+	            var attribute = attributeMethodInfo.Invoke(modelValidator, new object[0]);
+
+				if (attribute.GetType().GetConstructors().Any(x => !x.GetParameters().Any()))
+				{
+					_serviceContainer.InjectProperties(attribute);
+				}
+            }
+            return validators;
+        }
+    }
+}

LightInject.Mvc/LightInject.Mvc.csproj

@@ -34,6 +34,7 @@
   </PropertyGroup>
   <ItemGroup>
     <Reference Include="System" />
+    <Reference Include="System.ComponentModel.DataAnnotations" />
     <Reference Include="System.Core" />
     <Reference Include="System.Web" />
     <Reference Include="System.Web.Mvc, Version=4.0.0.0, Culture=neutral, PublicKeyToken=31bf3856ad364e35, processorArchitecture=MSIL" />