use IServiceCollection instead of IMvcBuilder

README.md

@@ -236,8 +236,7 @@ Install-Package PipelineNet.ServiceProvider.AspNetCore
 It includes extension methods for registering factories and automatic middleware registration through assembly scanning:
 ```C#
 builder.Services.AddScoped<IMyService, MyService>();
-builder.Services.AddControllers()
-    .AddPipelineNet(typeof(RoudCornersAsyncMiddleware).Assembly); // Add pipeline and chain of responsibility factories and all middleware from the assembly
+builder.Services.AddPipelineNetAspNetCore(typeof(RoudCornersAsyncMiddleware).Assembly); // Add pipeline and chain of responsibility factories and all middleware from the assembly
 
 public interface IMyService
 {
@@ -285,7 +284,7 @@ public class RoudCornersAsyncMiddleware : IAsyncMiddleware<Bitmap>
 }
 ```
 
-Note that `PipelineNet.ServiceProvider.AspNetCore` uses `IHttpContextAccessor` and thus by default, cannot be used outside of HTTP request. You can create your own implementation though.
+Note that `PipelineNet.ServiceProvider.AspNetCore` uses `IHttpContextAccessor` so it cannot be used outside of the HTTP request.
 
 ### Unity implementation
 

src/PipelineNet.ServiceProvider.AspNetCore.Tests/Program.cs

@@ -8,11 +8,12 @@
 
 // Add services to the container.
 
+builder.Services.AddPipelineNetAspNetCore(typeof(RoudCornersAsyncMiddleware).Assembly);
+
 builder.Services.AddLogging(builder =>
     builder.Services.AddSingleton<ILoggerProvider, TestOutputHelperLoggerProvider>());
 
-builder.Services.AddControllers()
-    .AddPipelineNet(typeof(RoudCornersAsyncMiddleware).Assembly);
+builder.Services.AddControllers();
 
 var app = builder.Build();
 

src/PipelineNet.ServiceProvider.AspNetCore/ServiceCollectionExtensions.cs

@@ -0,0 +1,68 @@
+using Microsoft.AspNetCore.Http;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.DependencyInjection.Extensions;
+using PipelineNet.MiddlewareResolver;
+using PipelineNet.ServiceProvider.AspNetCore.MiddlewareResolver;
+using PipelineNet.ServiceProvider.ChainsOfResponsibility.Factories;
+using PipelineNet.ServiceProvider.Pipelines.Factories;
+using System;
+using System.Collections.Generic;
+using System.Reflection;
+
+namespace PipelineNet.ServiceProvider.AspNetCore
+{
+    /// <summary>
+    /// Extension methods to the service collection.
+    /// </summary>
+    public static class ServiceCollectionExtensions
+    {
+        /// <summary>
+        /// Adds pipeline and chain of responsibility factories. Adds all middleware from assemblies.
+        /// </summary>
+        /// <param name="services">The service collection.</param>
+        /// <param name="assemblies">Assemblies to scan.</param>
+        /// <param name="lifetime">The lifetime of the registered middleware.</param>
+        /// <returns>The service collection.</returns>
+        /// <remarks>Middleware instances will be resolved using the <see cref="IHttpContextAccessor"/>.</remarks>
+        public static IServiceCollection AddPipelineNetAspNetCore(
+            this IServiceCollection services,
+            IEnumerable<Assembly> assemblies,
+            ServiceLifetime lifetime = ServiceLifetime.Scoped)
+        {
+            if (services == null) throw new ArgumentNullException("services");
+            if (assemblies == null) throw new ArgumentNullException("assemblies");
+
+            services.AddMiddlewareFromAssemblies(assemblies, lifetime);
+
+            services.AddHttpContextAccessor();
+
+            services.TryAddSingleton<IMiddlewareResolver, HttpContextAccessorMiddlewareResolver>();
+
+            services.TryAddSingleton(typeof(IPipelineFactory<>), typeof(AsyncPipelineFactory<>));
+            services.TryAddSingleton(typeof(IAsyncPipelineFactory<>), typeof(AsyncPipelineFactory<>));
+            services.TryAddSingleton(typeof(IResponsibilityChainFactory<,>), typeof(ResponsibilityChainFactory<,>));
+            services.TryAddSingleton(typeof(IAsyncResponsibilityChainFactory<,>), typeof(AsyncResponsibilityChainFactory<,>));
+
+            return services;
+        }
+
+        /// <summary>
+        /// Adds pipeline and chain of responsibility factories. Adds all middleware from the assembly.
+        /// </summary>
+        /// <param name="services">The service collection.</param>
+        /// <param name="assembly">The assembly to scan.</param>
+        /// <param name="lifetime">The lifetime of the registered middleware.</param>
+        /// <returns>The service collection.</returns>
+        /// <remarks>Middleware instances will be resolved using the <see cref="IHttpContextAccessor"/>.</remarks>
+        public static IServiceCollection AddPipelineNetAspNetCore(
+            this IServiceCollection services,
+            Assembly assembly,
+            ServiceLifetime lifetime = ServiceLifetime.Scoped)
+        {
+            if (services == null) throw new ArgumentNullException("services");
+            if (assembly == null) throw new ArgumentNullException("assembly");
+
+            return services.AddPipelineNetAspNetCore(new[] { assembly }, lifetime);
+        }
+    }
+}