add configurator for better extension control

Sample/MauiProgram.cs

@@ -40,7 +40,7 @@ public static MauiApp CreateMauiApp()
         builder.Logging.AddDebug();
         builder.Services.AddBlazorWebViewDeveloperTools();
 #endif
-        builder.Services.AddShinyMediator<MauiEventCollector>();
+        builder.Services.AddShinyMediator(x => x.UseMaui());
         builder.Services.AddSingletonAsImplementedInterfaces<SingletonEventHandler>();
         builder.Services.AddSingletonAsImplementedInterfaces<SingletonRequestHandler>();
 

readme.md

@@ -133,7 +133,7 @@ First, let's install [Shiny.Mediator.Maui](https://www.nuget.org/packages/Shiny.
 Now...let's go back to our MauiProgram.cs and alter the AddShinyMediator
 
 ```csharp
-builder.Services.AddShinyMediator<MauiEventCollector>();
+builder.Services.AddShinyMediator(cfg => cfg.UseMaui());
 ```
 
 Now your viewmodel (or page) can simply implement the IEventHandler<T> interface to participate

src/Shiny.Mediator.Maui/MauiEventCollector.cs

@@ -1,5 +1,10 @@
 namespace Shiny.Mediator;
 
+public static class MauiEventCollectorExtensions
+{
+    public static ShinyConfigurator UseMaui(this ShinyConfigurator cfg) => cfg.AddEventCollector<MauiEventCollector>();
+}
+
 public class MauiEventCollector : IEventCollector
 {
     public IReadOnlyList<IEventHandler<TEvent>> GetHandlers<TEvent>() where TEvent : IEvent

src/Shiny.Mediator/MediatorExtensions.cs

@@ -2,6 +2,7 @@
 using Microsoft.Extensions.DependencyInjection.Extensions;
 using Shiny.Mediator.Impl;
 using Shiny.Mediator.Infrastructure;
+using Shiny.Mediator.Middleware;
 
 namespace Shiny.Mediator;
 
@@ -15,20 +16,23 @@ public static void RunOffThread(this Task task, Action<Exception> onError)
                 onError(x.Exception);
         });
 
-    public static IServiceCollection AddShinyMediator(this IServiceCollection services)
+
+    public static IServiceCollection AddShinyMediator(this IServiceCollection services, Action<ShinyConfigurator>? configurator = null)
     {
         services.TryAddSingleton<IMediator, Impl.Mediator>();
         services.TryAddSingleton<IRequestSender, DefaultRequestSender>();
         services.TryAddSingleton<IEventPublisher, DefaultEventPublisher>();
+        configurator?.Invoke(new ShinyConfigurator(services));
+
         return services;
     }
 
 
-    public static IServiceCollection AddShinyMediator<TEventCollector>(this IServiceCollection services) where TEventCollector : class, IEventCollector
-    { 
-        services.AddSingleton<IEventCollector, TEventCollector>();
-        return services.AddShinyMediator();
-    }
+    public static ShinyConfigurator AddExceptionHandling(this ShinyConfigurator cfg)
+        => cfg.AddOpenRequestMiddleware(typeof(ExceptionHandlerMiddleware<,>));
+
+    public static ShinyConfigurator AddTimedMiddleware(this ShinyConfigurator cfg)
+        => cfg.AddOpenRequestMiddleware(typeof(TimedMiddleware<,>));
 
 
     public static IServiceCollection AddSingletonAsImplementedInterfaces<TImplementation>(this IServiceCollection services) where TImplementation : class

src/Shiny.Mediator/ShinyConfigurator.cs

@@ -0,0 +1,45 @@
+using Microsoft.Extensions.DependencyInjection;
+
+namespace Shiny.Mediator;
+
+
+public sealed class ShinyConfigurator(IServiceCollection services)
+{
+    public ShinyConfigurator AddRequestMiddleware<TRequest, TResult, TImpl>(
+        ServiceLifetime lifetime = ServiceLifetime.Scoped)
+        where TRequest : IRequest<TResult>
+        where TImpl : class, IRequestMiddleware<TRequest, TResult>
+    {
+        switch (lifetime)
+        {
+            case ServiceLifetime.Transient:
+                services.AddTransient<IRequestMiddleware<TRequest, TResult>, TImpl>();
+                break;
+
+            case ServiceLifetime.Scoped:
+                services.AddScoped<IRequestMiddleware<TRequest, TResult>, TImpl>();
+                break;
+
+            case ServiceLifetime.Singleton:
+                services.AddSingleton<IRequestMiddleware<TRequest, TResult>, TImpl>();
+                break;
+        }
+
+        return this;
+    }
+
+
+    public ShinyConfigurator AddOpenRequestMiddleware(Type implementationType, ServiceLifetime lifetime = ServiceLifetime.Scoped) 
+    {
+        // validate open generic
+        services.Add(new ServiceDescriptor(typeof(IRequestMiddleware<,>), null, implementationType, lifetime));
+        return this;
+    }
+
+
+    public ShinyConfigurator AddEventCollector<TImpl>() where TImpl : class, IEventCollector
+    {
+        services.AddSingleton<IEventCollector, TImpl>();
+        return this;
+    }
+}

tests/Shiny.Mediator.Tests/EventHandlerTests.cs

@@ -15,7 +15,7 @@ public EventHandlerTests()
     public async Task SubscriptionFired()
     {
         var services = new ServiceCollection();
-        services.AddShinyMediator();
+        services.AddShinyMediator(cfg => {});
         var sp = services.BuildServiceProvider();
         var mediator = sp.GetRequiredService<IMediator>();
 

tests/Shiny.Mediator.Tests/MiddlewareTests.cs

@@ -17,7 +17,10 @@ public MiddlewareTests()
     public async Task ConstrainedAndOpen(bool addConstrained, bool addOpen)
     {
         var services = new ServiceCollection();
-        services.AddShinyMediator();
+        services.AddShinyMediator(cfg =>
+        {
+
+        });
         services.AddSingletonAsImplementedInterfaces<MiddlewareRequestResultHandler>();
 
         if (addConstrained)

tests/Shiny.Mediator.Tests/RequestHandlerTests.cs

@@ -9,7 +9,7 @@ public async Task Missing_RequestHandler()
         try
         {
             var services = new ServiceCollection();
-            services.AddShinyMediator();
+            services.AddShinyMediator(cfg => { });
             var sp = services.BuildServiceProvider();
             var mediator = sp.GetRequiredService<IMediator>();
             await mediator.Send(new TestRequest());

tests/Shiny.Mediator.Tests/SourceGeneratorTests.cs

@@ -9,9 +9,11 @@ public class SourceGeneratorTests
     public void DidRegister()
     {
         var services = new ServiceCollection();
-        services.AddShinyMediator();
-        // TODO: source gen call
-        // services.AddDiscoveredMediatorHandlers();
+        services.AddShinyMediator(cfg =>
+        {
+            // TODO: source gen call
+            // services.AddDiscoveredMediatorHandlers();
+        });
         var sp = services.BuildServiceProvider();
 
         sp.GetService<IEventHandler<SourceGenEvent>>().Should().NotBeNull("Event Handler not found");