mirror of
https://github.com/bitwarden/server
synced 2025-12-16 00:03:54 +00:00
30 lines
1.1 KiB
C#
30 lines
1.1 KiB
C#
using System.Reflection;
|
|
using Bit.Seeder;
|
|
using Microsoft.Extensions.DependencyInjection.Extensions;
|
|
|
|
namespace Bit.SeederApi.Extensions;
|
|
|
|
public static class ServiceCollectionExtensions
|
|
{
|
|
/// <summary>
|
|
/// Dynamically registers all scene types that implement IScene<TRequest> from the Seeder assembly.
|
|
/// Scenes are registered as keyed scoped services using their class name as the key.
|
|
/// </summary>
|
|
public static IServiceCollection AddScenes(this IServiceCollection services)
|
|
{
|
|
var seederAssembly = Assembly.Load("Seeder");
|
|
var sceneTypes = seederAssembly.GetTypes()
|
|
.Where(t => t is { IsClass: true, IsAbstract: false } &&
|
|
t.GetInterfaces().Any(i => i.IsGenericType &&
|
|
i.GetGenericTypeDefinition().Name == "IScene`1"));
|
|
|
|
foreach (var sceneType in sceneTypes)
|
|
{
|
|
services.TryAddScoped(sceneType);
|
|
services.TryAddKeyedScoped(typeof(IScene), sceneType.Name, (sp, key) => sp.GetRequiredService(sceneType));
|
|
}
|
|
|
|
return services;
|
|
}
|
|
}
|