Mediator work

Framework/Foundation/Lifecycles/Mediator.cs

@@ -1,5 +1,4 @@
 using Mediator;
-using System.Diagnostics;
 
 namespace Toolkit.Framework.Foundation;
 
@@ -44,8 +43,7 @@ public class Mediator : IMediator
 
     public ValueTask<TResponse> Send<TResponse>(IRequest<TResponse> request, CancellationToken cancellationToken = default)
     {
-        dynamic? handler = factory.GetService(typeof(RequestClassHandlerWrapper<,>).MakeGenericType(request.GetType(), typeof(TResponse)));
-        
+        dynamic? handler = factory.GetService(typeof(RequestClassHandlerWrapper<,>).MakeGenericType(request.GetType(), typeof(TResponse)));  
         if (handler is not null)
         {
             return handler.Handle((dynamic)request, cancellationToken);
@@ -56,16 +54,70 @@ public class Mediator : IMediator
 
     public ValueTask<TResponse> Send<TResponse>(ICommand<TResponse> command, CancellationToken cancellationToken = default)
     {
-        throw new NotImplementedException();
+        dynamic? handler = factory.GetService(typeof(CommandClassHandlerWrapper<,>).MakeGenericType(command.GetType(), typeof(TResponse)));
+        if (handler is not null)
+        {
+            return handler.Handle((dynamic)command, cancellationToken);
+        }
+
+        return default;
     }
 
     public ValueTask<TResponse> Send<TResponse>(IQuery<TResponse> query, CancellationToken cancellationToken = default)
     {
-        throw new NotImplementedException();
+        dynamic? handler = factory.GetService(typeof(QueryClassHandlerWrapper<,>).MakeGenericType(query.GetType(), typeof(TResponse)));
+        if (handler is not null)
+        {
+            return handler.Handle((dynamic)query, cancellationToken);
+        }
+
+        return default;
     }
 
     public ValueTask<object?> Send(object message, CancellationToken cancellationToken = default)
     {
-        throw new NotImplementedException();
+        if (message.GetType().GetInterface(typeof(IRequest<>).Name) is { } requestInterface)
+        {
+            if (requestInterface.GetGenericArguments() is { Length: 1 } arguments)
+            {
+                Type responseType = arguments[0];
+
+                dynamic? handler = factory.GetService(typeof(RequestClassHandlerWrapper<,>).MakeGenericType(message.GetType(), responseType));
+                if (handler is not null)
+                {
+                    return handler.Handle((dynamic)message, cancellationToken);
+                }
+            }
+        }
+
+        if (message.GetType().GetInterface(typeof(ICommand<>).Name) is { } commandInterface)
+        {
+            if (commandInterface.GetGenericArguments() is { Length: 1 } arguments)
+            {
+                Type responseType = arguments[0];
+
+                dynamic? handler = factory.GetService(typeof(CommandClassHandlerWrapper<,>).MakeGenericType(message.GetType(), responseType));
+                if (handler is not null)
+                {
+                    return handler.Handle((dynamic)message, cancellationToken);
+                }
+            }
+        }
+
+        if (message.GetType().GetInterface(typeof(IQuery<>).Name) is { } queryInterface)
+        {
+            if (queryInterface.GetGenericArguments() is { Length: 1 } arguments)
+            {
+                Type responseType = arguments[0];
+
+                dynamic? handler = factory.GetService(typeof(QueryClassHandlerWrapper<,>).MakeGenericType(message.GetType(), responseType));
+                if (handler is not null)
+                {
+                    return handler.Handle((dynamic)message, cancellationToken);
+                }
+            }
+        }
+
+        return default;
     }
 }