diff --git a/src/Configuration.cs b/src/Configuration.cs index 9837788..7ed3be6 100644 --- a/src/Configuration.cs +++ b/src/Configuration.cs @@ -7,11 +7,20 @@ namespace Unity.Microsoft.DependencyInjection { - internal static class Configuration + public static class Configuration { - internal static IUnityContainer AddServices(this IUnityContainer container, IServiceCollection services) + public static IUnityContainer AddServices(this IUnityContainer container, IServiceCollection services) { - var lifetime = ((UnityContainer)container).Configure().Lifetime; + var extension = ((UnityContainer)container).Configure(); + + if (extension == null) + { + extension = new MdiExtension(); + container.AddExtension(extension); + } + + var lifetime = extension.Lifetime; + var registerFunc = ((UnityContainer)container).Register; ((UnityContainer)container).Register = ((UnityContainer)container).AppendNew;