Eveldee/Akari.Prototype
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update commands registration

Browse Source 
Now use reflection to load all commands directly into the DI container
This commit is contained in:
Eveldee
2021-06-05 11:02:33 +02:00
parent b040f5b0b2
commit 933d123d44
2 changed files with 34 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
Akari.Prototype.Server/Program.cs
View File

@@ -1,5 +1,6 @@
using Akari.Prototype.Server.Cli.Commands;
using Akari.Prototype.Server.Options;
using Akari.Prototype.Server.Utils.Extensions;
using CliFx;
using Microsoft.AspNetCore.Hosting;
using Microsoft.Extensions.Configuration;
@@ -74,9 +75,7 @@ namespace Akari.Prototype.Server
{
Startup.ConfigureStandardServices(hostContext.Configuration, services);
services.AddTransient<FingerprintCommands>();
services.AddTransient<FingerprintCommands.RemoveFingerprintCommand>();
services.AddTransient<FingerprintCommands.AddFingerprintCommand>();
services.AddCommandsFromThisAssembly();
});
public static IHostBuilder CreateWebHostBuilder(string[] args) =>

32
Akari.Prototype.Server/Utils/Extensions/IServiceCollectionExtensions.cs Normal file
View File

@@ -0,0 +1,32 @@
using CliFx;
using CliFx.Attributes;
using Microsoft.Extensions.DependencyInjection;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Threading.Tasks;
namespace Akari.Prototype.Server.Utils.Extensions
{
public static class IServiceCollectionExtensions
{
public static IServiceCollection AddCommandsFromThisAssembly(this IServiceCollection services) => services.AddCommandsFrom(Assembly.GetCallingAssembly());
public static IServiceCollection AddCommandsFrom(this IServiceCollection services, Assembly assembly)
{
static bool IsCommandType(Type type) =>
type.GetInterfaces().Contains(typeof(ICommand)) &&
type.IsDefined(typeof(CommandAttribute)) &&
!type.IsAbstract &&
!type.IsInterface;
foreach (var commandType in assembly.ExportedTypes.Where(IsCommandType))
{
services.AddTransient(commandType);
}
return services;
}
}
}