From 2b7bd315a53ec29e252115e0c2ade57ca178512d Mon Sep 17 00:00:00 2001
From: Daniel Clark <dan_clark@outlook.com>
Date: Fri, 16 Dec 2022 09:06:14 +0000
Subject: [PATCH] Fix manual handler registrations

---
 Framework/Avalonia/Contents/ContentHandler.cs                 | 1 -
 Framework/Avalonia/Extensions/IServiceCollectionExtensions.cs | 3 ++-
 .../Avalonia/Navigation/ContentControlNavigationHandler.cs    | 4 ++--
 .../Foundation/Extensions/IServiceCollectionExtensions.cs     | 1 +
 4 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/Framework/Avalonia/Contents/ContentHandler.cs b/Framework/Avalonia/Contents/ContentHandler.cs
index cf144e4..d259885 100644
--- a/Framework/Avalonia/Contents/ContentHandler.cs
+++ b/Framework/Avalonia/Contents/ContentHandler.cs
@@ -1,6 +1,5 @@
 using Avalonia;
 using Mediator;
-using System.Diagnostics;
 using Toolkit.Framework.Foundation;
 
 namespace Toolkit.Framework.Avalonia;
diff --git a/Framework/Avalonia/Extensions/IServiceCollectionExtensions.cs b/Framework/Avalonia/Extensions/IServiceCollectionExtensions.cs
index ad8dcfc..54ef11d 100644
--- a/Framework/Avalonia/Extensions/IServiceCollectionExtensions.cs
+++ b/Framework/Avalonia/Extensions/IServiceCollectionExtensions.cs
@@ -6,10 +6,11 @@ namespace Toolkit.Framework.Avalonia;
 
 public static class IServiceCollectionExtensions
 {
-    public static IServiceCollection AddNavigation(this IServiceCollection serviceCollection)
+    public static IServiceCollection AddAvalonia(this IServiceCollection serviceCollection)
     {
         serviceCollection.TryAddSingleton<INavigationRouteDescriptorCollection, NavigationRouteDescriptorCollection>();
 
+        serviceCollection.AddHandler<ContentHandler>();
         serviceCollection.AddHandler<NavigationRouteHandler>();
         serviceCollection.AddHandler<NavigateHandler>();
         serviceCollection.AddHandler<FrameNavigationHandler>();
diff --git a/Framework/Avalonia/Navigation/ContentControlNavigationHandler.cs b/Framework/Avalonia/Navigation/ContentControlNavigationHandler.cs
index 5ee2098..78d6b2f 100644
--- a/Framework/Avalonia/Navigation/ContentControlNavigationHandler.cs
+++ b/Framework/Avalonia/Navigation/ContentControlNavigationHandler.cs
@@ -5,7 +5,7 @@ namespace Toolkit.Framework.Avalonia;
 
 public class ContentControlNavigationHandler : IRequestHandler<ContentControlNavigation, bool>
 {
-    public async ValueTask<bool> Handle(ContentControlNavigation request, CancellationToken cancellationToken)
+    public ValueTask<bool> Handle(ContentControlNavigation request, CancellationToken cancellationToken)
     {
         if (request.Template is TemplatedControl control)
         {
@@ -13,6 +13,6 @@ public class ContentControlNavigationHandler : IRequestHandler<ContentControlNav
             request.Route.Content = control;
         }
 
-        return await Task.FromResult(true);
+        return new ValueTask<bool>(true);
     }
 }
\ No newline at end of file
diff --git a/Framework/Foundation/Extensions/IServiceCollectionExtensions.cs b/Framework/Foundation/Extensions/IServiceCollectionExtensions.cs
index 017478e..fd4d095 100644
--- a/Framework/Foundation/Extensions/IServiceCollectionExtensions.cs
+++ b/Framework/Foundation/Extensions/IServiceCollectionExtensions.cs
@@ -37,6 +37,7 @@ public static class IServiceCollectionExtensions
             .AddSingleton<IMediator, Mediator>()
             .AddHandler<InitializeHandler>()
             .AddSingleton<IServiceFactory>(provider => new ServiceFactory(provider.GetService, (instanceType, parameters) => ActivatorUtilities.CreateInstance(provider, instanceType, parameters!)))
+            .AddHandler<ServiceFactoryHandler>()
             .AddSingleton<IInitialization, Initialization>(provider => new Initialization(() =>
             {
                 return serviceCollection.Where(x => x.ServiceType.GetInterfaces()