From d45076f2a97d9f0abbd5e395c15ef5f055ed68fe Mon Sep 17 00:00:00 2001
From: TheXamlGuy <dan_clark@outlook.com>
Date: Thu, 4 Jan 2024 20:53:49 +0000
Subject: [PATCH] get plugin system working

---
 .../ContextualCommandBuilder.cs               | 11 ++++
 .../ContextualCommandView.xaml                |  6 +--
 .../ContextualCommandView.xaml.cs             |  2 +-
 .../ContextualCommandViewModel.cs             |  5 +-
 .../Hyperbar.Desktop.Contextual.csproj        | 29 ++++++++++
 Hyperbar.Desktop/App.xaml.cs                  | 39 ++++++++++----
 Hyperbar.Desktop/Hyperbar.Desktop.csproj      |  1 +
 .../Templates/IServiceCollectionExtensions.cs | 26 +++++++++
 .../Templates/ITemplateGeneratorFactory.cs    |  1 +
 .../Templates/TemplateGeneratorFactory.cs     |  8 +--
 Hyperbar.Desktop/Views/CommandViewModel.cs    | 19 +++----
 Hyperbar.sln                                  | 18 +++++++
 Hyperbar/IServiceCollectionExtensions.cs      | 27 ----------
 Hyperbar/{ => Lifecycles}/AppService.cs       |  0
 Hyperbar/Lifecycles/CommandContext.cs         |  8 +++
 Hyperbar/Lifecycles/ICommandBuilder.cs        |  8 +++
 Hyperbar/Lifecycles/ICommandContext.cs        |  6 +++
 Hyperbar/Lifecycles/ICommandViewModel.cs      |  7 +++
 Hyperbar/{ => Lifecycles}/IInitializer.cs     |  0
 .../IServiceCollectionExtensions.cs           | 53 +++++++++++++++++++
 .../{ => Templates}/DataTemplateDescriptor.cs |  0
 .../IDataTemplateDescriptor.cs                |  0
 Hyperbar/{ => Templates}/ITemplateFactory.cs  |  0
 .../{ => Templates}/ITemplatedViewModel.cs    |  0
 24 files changed, 219 insertions(+), 55 deletions(-)
 create mode 100644 Hyperbar.Desktop.Contextual/ContextualCommandBuilder.cs
 rename {Hyperbar.Desktop/Views => Hyperbar.Desktop.Contextual}/ContextualCommandView.xaml (73%)
 rename {Hyperbar.Desktop/Views => Hyperbar.Desktop.Contextual}/ContextualCommandView.xaml.cs (80%)
 rename {Hyperbar.Desktop/Views => Hyperbar.Desktop.Contextual}/ContextualCommandViewModel.cs (78%)
 create mode 100644 Hyperbar.Desktop.Contextual/Hyperbar.Desktop.Contextual.csproj
 create mode 100644 Hyperbar.Desktop/Templates/IServiceCollectionExtensions.cs
 delete mode 100644 Hyperbar/IServiceCollectionExtensions.cs
 rename Hyperbar/{ => Lifecycles}/AppService.cs (100%)
 create mode 100644 Hyperbar/Lifecycles/CommandContext.cs
 create mode 100644 Hyperbar/Lifecycles/ICommandBuilder.cs
 create mode 100644 Hyperbar/Lifecycles/ICommandContext.cs
 create mode 100644 Hyperbar/Lifecycles/ICommandViewModel.cs
 rename Hyperbar/{ => Lifecycles}/IInitializer.cs (100%)
 create mode 100644 Hyperbar/Lifecycles/IServiceCollectionExtensions.cs
 rename Hyperbar/{ => Templates}/DataTemplateDescriptor.cs (100%)
 rename Hyperbar/{ => Templates}/IDataTemplateDescriptor.cs (100%)
 rename Hyperbar/{ => Templates}/ITemplateFactory.cs (100%)
 rename Hyperbar/{ => Templates}/ITemplatedViewModel.cs (100%)

diff --git a/Hyperbar.Desktop.Contextual/ContextualCommandBuilder.cs b/Hyperbar.Desktop.Contextual/ContextualCommandBuilder.cs
new file mode 100644
index 0000000..da53c81
--- /dev/null
+++ b/Hyperbar.Desktop.Contextual/ContextualCommandBuilder.cs
@@ -0,0 +1,11 @@
+using Microsoft.Extensions.DependencyInjection;
+
+namespace Hyperbar.Desktop.Contextual;
+
+public class ContextualCommandBuilder : ICommandBuilder
+{
+    public void Create(IServiceCollection services)
+    {
+        services.AddCommandTemplate<ContextualCommandViewModel, ContextualCommandView>();
+    }
+}
diff --git a/Hyperbar.Desktop/Views/ContextualCommandView.xaml b/Hyperbar.Desktop.Contextual/ContextualCommandView.xaml
similarity index 73%
rename from Hyperbar.Desktop/Views/ContextualCommandView.xaml
rename to Hyperbar.Desktop.Contextual/ContextualCommandView.xaml
index e608e52..cd71865 100644
--- a/Hyperbar.Desktop/Views/ContextualCommandView.xaml
+++ b/Hyperbar.Desktop.Contextual/ContextualCommandView.xaml
@@ -1,15 +1,15 @@
 <?xml version="1.0" encoding="utf-8" ?>
 <Page
-    x:Class="Hyperbar.Desktop.ContextualCommandView"
+    x:Class="Hyperbar.Desktop.Contextual.ContextualCommandView"
     xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
     xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
     xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
-    xmlns:local="using:Hyperbar.Desktop"
+    xmlns:local="using:Hyperbar.Desktop.Contextual"
     xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
     Background="{ThemeResource ApplicationPageBackgroundThemeBrush}"
     mc:Ignorable="d">
 
     <Grid>
-        <Button />
+        <Button Content="This is a test" />
     </Grid>
 </Page>
diff --git a/Hyperbar.Desktop/Views/ContextualCommandView.xaml.cs b/Hyperbar.Desktop.Contextual/ContextualCommandView.xaml.cs
similarity index 80%
rename from Hyperbar.Desktop/Views/ContextualCommandView.xaml.cs
rename to Hyperbar.Desktop.Contextual/ContextualCommandView.xaml.cs
index fb83147..7a86f81 100644
--- a/Hyperbar.Desktop/Views/ContextualCommandView.xaml.cs
+++ b/Hyperbar.Desktop.Contextual/ContextualCommandView.xaml.cs
@@ -1,6 +1,6 @@
 using Microsoft.UI.Xaml.Controls;
 
-namespace Hyperbar.Desktop;
+namespace Hyperbar.Desktop.Contextual;
 
 public sealed partial class ContextualCommandView : Page
 {
diff --git a/Hyperbar.Desktop/Views/ContextualCommandViewModel.cs b/Hyperbar.Desktop.Contextual/ContextualCommandViewModel.cs
similarity index 78%
rename from Hyperbar.Desktop/Views/ContextualCommandViewModel.cs
rename to Hyperbar.Desktop.Contextual/ContextualCommandViewModel.cs
index e6b6f64..fee2451 100644
--- a/Hyperbar.Desktop/Views/ContextualCommandViewModel.cs
+++ b/Hyperbar.Desktop.Contextual/ContextualCommandViewModel.cs
@@ -1,6 +1,7 @@
-namespace Hyperbar.Desktop;
+namespace Hyperbar.Desktop.Contextual;
 
 public class ContextualCommandViewModel :
+    ICommandViewModel,
     ITemplatedViewModel
 {
     public ContextualCommandViewModel(ITemplateFactory templateFactory)
@@ -9,4 +10,4 @@ public class ContextualCommandViewModel :
     }
 
     public ITemplateFactory TemplateFactory { get; }
-}
\ No newline at end of file
+}
diff --git a/Hyperbar.Desktop.Contextual/Hyperbar.Desktop.Contextual.csproj b/Hyperbar.Desktop.Contextual/Hyperbar.Desktop.Contextual.csproj
new file mode 100644
index 0000000..613c60d
--- /dev/null
+++ b/Hyperbar.Desktop.Contextual/Hyperbar.Desktop.Contextual.csproj
@@ -0,0 +1,29 @@
+<Project Sdk="Microsoft.NET.Sdk">
+  <PropertyGroup>
+    <TargetFramework>net8.0-windows10.0.19041.0</TargetFramework>
+    <TargetPlatformMinVersion>10.0.17763.0</TargetPlatformMinVersion>
+    <RootNamespace>Hyperbar.Desktop.Contextual</RootNamespace>
+    <RuntimeIdentifiers>win10-x86;win10-x64;win10-arm64</RuntimeIdentifiers>
+    <UseWinUI>true</UseWinUI>
+	  <UseRidGraph>true</UseRidGraph>
+  </PropertyGroup>
+  <ItemGroup>
+    <None Remove="ContextualCommandView.xaml" />
+  </ItemGroup>
+
+  <ItemGroup>
+	  <PackageReference Include="Microsoft.Extensions.Hosting" Version="8.0.0" />
+	  <PackageReference Include="Microsoft.WindowsAppSDK" Version="1.5.231202003-experimental1" />
+	  <PackageReference Include="Microsoft.Windows.SDK.BuildTools" Version="10.0.25936-preview" />
+  </ItemGroup>
+
+  <ItemGroup>
+    <ProjectReference Include="..\Hyperbar\Hyperbar.csproj" />
+  </ItemGroup>
+
+  <ItemGroup>
+    <Page Update="ContextualCommandView.xaml">
+      <Generator>MSBuild:Compile</Generator>
+    </Page>
+  </ItemGroup>
+</Project>
diff --git a/Hyperbar.Desktop/App.xaml.cs b/Hyperbar.Desktop/App.xaml.cs
index bc8a243..69c427b 100644
--- a/Hyperbar.Desktop/App.xaml.cs
+++ b/Hyperbar.Desktop/App.xaml.cs
@@ -1,8 +1,10 @@
-using Hyperbar.Desktop.Controls;
+using Hyperbar.Desktop.Contextual;
+using Hyperbar.Desktop.Controls;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Hosting;
 using Microsoft.UI.Xaml;
 using System;
+using System.Collections.Generic;
 
 namespace Hyperbar.Desktop;
 
@@ -17,18 +19,37 @@ public partial class App :
 
         IHost? host = Host.CreateDefaultBuilder()
             .UseContentRoot(AppContext.BaseDirectory)
-            .ConfigureServices(args => 
+            .ConfigureServices(services =>
             {
-                args.AddTransient<IInitializer, AppInitializer>();
-                args.AddTransient<DesktopFlyout>();
+                services.AddHostedService<AppService>();
 
-                args.AddTransient<ITemplateFactory, TemplateFactory>();
-                args.AddTransient<ITemplateGeneratorFactory, TemplateGeneratorFactory>();
+                services.AddTransient<IInitializer, AppInitializer>();
+                services.AddTransient<DesktopFlyout>();
 
-                args.AddDataTemplate<CommandViewModel, CommandView>("Commands");
-                args.AddDataTemplate<ContextualCommandViewModel, ContextualCommandView>();
+                services.AddTransient<ITemplateFactory, TemplateFactory>();
+                services.AddTransient<ITemplateGeneratorFactory, TemplateGeneratorFactory>();
 
-                args.AddHostedService<AppService>();
+                services.AddDataTemplate<CommandViewModel, CommandView>("Commands");
+
+                // Commands
+                services.AddSomething<ContextualCommandBuilder>();
+ 
+                services.AddTransient(provider =>
+                {
+                    IEnumerable<ICommandViewModel> Resolve(IServiceProvider services)
+                    {
+                        foreach (ICommandContext commandContext in services.GetServices<ICommandContext>())
+                        {
+                            if (commandContext.ServiceProvider.GetService<ICommandViewModel>() is 
+                                ICommandViewModel commandViewModel)
+                            {
+                                yield return commandViewModel;
+                            }
+                        }
+                    }
+
+                    return Resolve(provider);                
+                });
             })
             .Build();
         
diff --git a/Hyperbar.Desktop/Hyperbar.Desktop.csproj b/Hyperbar.Desktop/Hyperbar.Desktop.csproj
index 85810a1..a88a7ad 100644
--- a/Hyperbar.Desktop/Hyperbar.Desktop.csproj
+++ b/Hyperbar.Desktop/Hyperbar.Desktop.csproj
@@ -41,6 +41,7 @@
 		</Page>
 	</ItemGroup>
 	<ItemGroup>
+	  <ProjectReference Include="..\Hyperbar.Desktop.Contextual\Hyperbar.Desktop.Contextual.csproj" />
 	  <ProjectReference Include="..\Hyperbar.Desktop.Controls\Hyperbar.Desktop.Controls.csproj" />
 	  <ProjectReference Include="..\Hyperbar\Hyperbar.csproj" />
 	</ItemGroup>
diff --git a/Hyperbar.Desktop/Templates/IServiceCollectionExtensions.cs b/Hyperbar.Desktop/Templates/IServiceCollectionExtensions.cs
new file mode 100644
index 0000000..a0b8a37
--- /dev/null
+++ b/Hyperbar.Desktop/Templates/IServiceCollectionExtensions.cs
@@ -0,0 +1,26 @@
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Hosting;
+
+namespace Hyperbar.Desktop
+{
+    public static class IServiceCollectionExtensions
+    {
+        public static IServiceCollection AddSomething<TCommandBuilder>(this IServiceCollection services)
+            where TCommandBuilder :
+            ICommandBuilder, new()
+        {
+            TCommandBuilder builder = new();
+            IHost? host = new HostBuilder()
+                .ConfigureServices(services =>
+                {
+                    services.AddTransient<ITemplateFactory, TemplateFactory>();
+                    services.AddTransient<ITemplateGeneratorFactory, TemplateGeneratorFactory>();
+
+                    builder.Create(services);
+                }).Build();
+
+            services.AddTransient<ICommandContext>(provider => new CommandContext(host.Services));
+            return services;
+        }
+    }
+}
\ No newline at end of file
diff --git a/Hyperbar.Desktop/Templates/ITemplateGeneratorFactory.cs b/Hyperbar.Desktop/Templates/ITemplateGeneratorFactory.cs
index 2bd3eaf..8657ac4 100644
--- a/Hyperbar.Desktop/Templates/ITemplateGeneratorFactory.cs
+++ b/Hyperbar.Desktop/Templates/ITemplateGeneratorFactory.cs
@@ -2,6 +2,7 @@
 
 namespace Hyperbar.Desktop
 {
+
     public interface ITemplateGeneratorFactory
     {
         DataTemplate Create();
diff --git a/Hyperbar.Desktop/Templates/TemplateGeneratorFactory.cs b/Hyperbar.Desktop/Templates/TemplateGeneratorFactory.cs
index 7be05df..b580cbe 100644
--- a/Hyperbar.Desktop/Templates/TemplateGeneratorFactory.cs
+++ b/Hyperbar.Desktop/Templates/TemplateGeneratorFactory.cs
@@ -9,10 +9,10 @@ public class TemplateGeneratorFactory :
     public DataTemplate Create()
     {
         string xamlString = @"
-                    <DataTemplate xmlns='http://schemas.microsoft.com/winfx/2006/xaml/presentation' 
-                                  xmlns:desktop='using:Hyperbar.Desktop'>
-                        <desktop:TemplateGeneratorControl />
-                    </DataTemplate>";
+                <DataTemplate xmlns='http://schemas.microsoft.com/winfx/2006/xaml/presentation' 
+                                xmlns:desktop='using:Hyperbar.Desktop'>
+                    <desktop:TemplateGeneratorControl />
+                </DataTemplate>";
 
         return (DataTemplate)XamlReader.Load(xamlString);
     }
diff --git a/Hyperbar.Desktop/Views/CommandViewModel.cs b/Hyperbar.Desktop/Views/CommandViewModel.cs
index c8e4d56..b88fb50 100644
--- a/Hyperbar.Desktop/Views/CommandViewModel.cs
+++ b/Hyperbar.Desktop/Views/CommandViewModel.cs
@@ -1,20 +1,21 @@
-namespace Hyperbar.Desktop;
+using System.Collections;
+using System.Collections.Generic;
+
+namespace Hyperbar.Desktop;
 
 public partial class CommandViewModel : 
     ObservableCollectionViewModel,
     ITemplatedViewModel
 {
-    public CommandViewModel(ITemplateFactory templateFactory)
+    public CommandViewModel(ITemplateFactory templateFactory, 
+        IEnumerable<ICommandViewModel> commands)
     {
         TemplateFactory = templateFactory;
 
-        this.Add(new ContextualCommandViewModel(templateFactory));
-        this.Add(new ContextualCommandViewModel(templateFactory));
-        this.Add(new ContextualCommandViewModel(templateFactory));
-        this.Add(new ContextualCommandViewModel(templateFactory));
-        this.Add(new ContextualCommandViewModel(templateFactory));
-
-        var d = Items;
+        foreach (var command in commands)
+        {
+            this.Add(command);
+        }
     }
 
     public ITemplateFactory TemplateFactory { get; }
diff --git a/Hyperbar.sln b/Hyperbar.sln
index fbdbf31..464d92a 100644
--- a/Hyperbar.sln
+++ b/Hyperbar.sln
@@ -11,6 +11,8 @@ Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "Hyperbar.Desktop.Win32", "H
 EndProject
 Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "Hyperbar", "Hyperbar\Hyperbar.csproj", "{E5795878-C7E3-4386-86FA-33681BCF8D5B}"
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Hyperbar.Desktop.Contextual", "Hyperbar.Desktop.Contextual\Hyperbar.Desktop.Contextual.csproj", "{C32D4073-2A9B-4257-8895-09951FAD8E7A}"
+EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|Any CPU = Debug|Any CPU
@@ -95,6 +97,22 @@ Global
 		{E5795878-C7E3-4386-86FA-33681BCF8D5B}.Release|x64.Build.0 = Release|Any CPU
 		{E5795878-C7E3-4386-86FA-33681BCF8D5B}.Release|x86.ActiveCfg = Release|Any CPU
 		{E5795878-C7E3-4386-86FA-33681BCF8D5B}.Release|x86.Build.0 = Release|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Debug|ARM64.ActiveCfg = Debug|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Debug|ARM64.Build.0 = Debug|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Debug|x64.ActiveCfg = Debug|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Debug|x64.Build.0 = Debug|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Debug|x86.ActiveCfg = Debug|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Debug|x86.Build.0 = Debug|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Release|Any CPU.Build.0 = Release|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Release|ARM64.ActiveCfg = Release|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Release|ARM64.Build.0 = Release|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Release|x64.ActiveCfg = Release|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Release|x64.Build.0 = Release|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Release|x86.ActiveCfg = Release|Any CPU
+		{C32D4073-2A9B-4257-8895-09951FAD8E7A}.Release|x86.Build.0 = Release|Any CPU
 	EndGlobalSection
 	GlobalSection(SolutionProperties) = preSolution
 		HideSolutionNode = FALSE
diff --git a/Hyperbar/IServiceCollectionExtensions.cs b/Hyperbar/IServiceCollectionExtensions.cs
deleted file mode 100644
index 357a6a3..0000000
--- a/Hyperbar/IServiceCollectionExtensions.cs
+++ /dev/null
@@ -1,27 +0,0 @@
-using Microsoft.Extensions.DependencyInjection;
-
-namespace Hyperbar;
-
-public static class IServiceCollectionExtensions
-{
-    public static IServiceCollection AddDataTemplate<TData, TTemplate>(this IServiceCollection services, 
-        object? key = null)
-    {
-        Type dataType = typeof(TData);
-        Type templateType = typeof(TTemplate);
-
-        key ??= dataType.Name;
-
-        services.AddKeyedTransient(dataType, key);
-        services.AddKeyedTransient(templateType, key);
-
-        services.AddTransient<IDataTemplateDescriptor>(provider => new DataTemplateDescriptor
-        {
-            DataType = dataType,
-            TemplateType = templateType,
-            Key = key
-        });
-
-        return services;
-    }
-}
diff --git a/Hyperbar/AppService.cs b/Hyperbar/Lifecycles/AppService.cs
similarity index 100%
rename from Hyperbar/AppService.cs
rename to Hyperbar/Lifecycles/AppService.cs
diff --git a/Hyperbar/Lifecycles/CommandContext.cs b/Hyperbar/Lifecycles/CommandContext.cs
new file mode 100644
index 0000000..029c0de
--- /dev/null
+++ b/Hyperbar/Lifecycles/CommandContext.cs
@@ -0,0 +1,8 @@
+
+namespace Hyperbar;
+
+public class CommandContext(IServiceProvider serviceProvider) :
+    ICommandContext
+{
+    public IServiceProvider ServiceProvider => serviceProvider;
+}
diff --git a/Hyperbar/Lifecycles/ICommandBuilder.cs b/Hyperbar/Lifecycles/ICommandBuilder.cs
new file mode 100644
index 0000000..1954cb0
--- /dev/null
+++ b/Hyperbar/Lifecycles/ICommandBuilder.cs
@@ -0,0 +1,8 @@
+using Microsoft.Extensions.DependencyInjection;
+
+namespace Hyperbar;
+
+public interface ICommandBuilder
+{
+    void Create(IServiceCollection services);
+}
diff --git a/Hyperbar/Lifecycles/ICommandContext.cs b/Hyperbar/Lifecycles/ICommandContext.cs
new file mode 100644
index 0000000..05422be
--- /dev/null
+++ b/Hyperbar/Lifecycles/ICommandContext.cs
@@ -0,0 +1,6 @@
+namespace Hyperbar;
+
+public interface ICommandContext
+{
+    IServiceProvider ServiceProvider { get; }
+}
diff --git a/Hyperbar/Lifecycles/ICommandViewModel.cs b/Hyperbar/Lifecycles/ICommandViewModel.cs
new file mode 100644
index 0000000..747fda2
--- /dev/null
+++ b/Hyperbar/Lifecycles/ICommandViewModel.cs
@@ -0,0 +1,7 @@
+namespace Hyperbar;
+
+public interface ICommandViewModel
+{
+
+}
+
diff --git a/Hyperbar/IInitializer.cs b/Hyperbar/Lifecycles/IInitializer.cs
similarity index 100%
rename from Hyperbar/IInitializer.cs
rename to Hyperbar/Lifecycles/IInitializer.cs
diff --git a/Hyperbar/Lifecycles/IServiceCollectionExtensions.cs b/Hyperbar/Lifecycles/IServiceCollectionExtensions.cs
new file mode 100644
index 0000000..1a4c737
--- /dev/null
+++ b/Hyperbar/Lifecycles/IServiceCollectionExtensions.cs
@@ -0,0 +1,53 @@
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Hosting;
+
+namespace Hyperbar;
+
+public static class IServiceCollectionExtensions
+{
+    public static IServiceCollection AddCommandTemplate<TCommand, TCommandTemplate>(this IServiceCollection services)
+        where TCommand :
+        ICommandViewModel
+    {
+        Type dataType = typeof(TCommand);
+        Type templateType = typeof(TCommandTemplate);
+
+        var key = dataType.Name;
+
+        services.AddTransient(typeof(ICommandViewModel), dataType);
+        services.AddTransient(templateType);
+
+        services.AddKeyedTransient(typeof(ICommandViewModel), key, dataType);
+        services.AddKeyedTransient(templateType, key);
+
+        services.AddTransient<IDataTemplateDescriptor>(provider => new DataTemplateDescriptor
+        {
+            DataType = dataType,
+            TemplateType = templateType,
+            Key = key
+        });
+
+        return services;
+    }
+
+    public static IServiceCollection AddDataTemplate<TData, TTemplate>(this IServiceCollection services,
+        object? key = null)
+    {
+        Type dataType = typeof(TData);
+        Type templateType = typeof(TTemplate);
+
+        key ??= dataType.Name;
+
+        services.AddKeyedTransient(dataType, key);
+        services.AddKeyedTransient(templateType, key);
+
+        services.AddTransient<IDataTemplateDescriptor>(provider => new DataTemplateDescriptor
+        {
+            DataType = dataType,
+            TemplateType = templateType,
+            Key = key
+        });
+
+        return services;
+    }
+}
diff --git a/Hyperbar/DataTemplateDescriptor.cs b/Hyperbar/Templates/DataTemplateDescriptor.cs
similarity index 100%
rename from Hyperbar/DataTemplateDescriptor.cs
rename to Hyperbar/Templates/DataTemplateDescriptor.cs
diff --git a/Hyperbar/IDataTemplateDescriptor.cs b/Hyperbar/Templates/IDataTemplateDescriptor.cs
similarity index 100%
rename from Hyperbar/IDataTemplateDescriptor.cs
rename to Hyperbar/Templates/IDataTemplateDescriptor.cs
diff --git a/Hyperbar/ITemplateFactory.cs b/Hyperbar/Templates/ITemplateFactory.cs
similarity index 100%
rename from Hyperbar/ITemplateFactory.cs
rename to Hyperbar/Templates/ITemplateFactory.cs
diff --git a/Hyperbar/ITemplatedViewModel.cs b/Hyperbar/Templates/ITemplatedViewModel.cs
similarity index 100%
rename from Hyperbar/ITemplatedViewModel.cs
rename to Hyperbar/Templates/ITemplatedViewModel.cs