Toolkit2/Toolkit.Foundation/Mediator.cs

using System.Reflection;

namespace Toolkit.Foundation;

public class Mediator(IServiceProvider provider) :
    IMediator
{
    public Task<TResponse?> SendAsync<TResponse>(IRequest<TResponse> request,
        CancellationToken cancellationToken = default)
    {
        Type handlerType = typeof(HandlerWrapper<,>).MakeGenericType(request.GetType(),
            typeof(TResponse));

        if (provider.GetService(handlerType) 
            is object handler)
        {
            if (handlerType.GetMethod("Handle") is MethodInfo handleMethod)
            {
                return (Task<TResponse?>)handleMethod.Invoke(handler, new object[] { request, cancellationToken })!;
            }
        }

        return Task.FromResult<TResponse?>(default);
    }

    public Task<object?> SendAsync(object message,
        CancellationToken cancellationToken = default)
    {
        if (message.GetType().GetInterface(typeof(IRequest<>).Name) is Type requestType &&
            requestType.GetGenericArguments().Length == 1)
        {
            Type responseType = requestType.GetGenericArguments()[0];
            Type handlerType = typeof(HandlerWrapper<,>).MakeGenericType(message.GetType(), 
                responseType);

            if (provider.GetService(handlerType) 
                is object handler)
            {
                if (handlerType.GetMethod("Handle") is MethodInfo handleMethod)
                {
                    return (Task<object?>)handleMethod.Invoke(handler, new object[] { message, cancellationToken })!;
                }
            }
        }

        return Task.FromResult<object?>(default);
    }
}