git.stella-ops.org/src/VexLens/StellaOps.VexLens/Extensions/VexLensServiceCollectionExtensions.cs

using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.DependencyInjection.Extensions;
using StellaOps.VexLens.Api;
using StellaOps.VexLens.Consensus;
using StellaOps.VexLens.Integration;
using StellaOps.VexLens.Mapping;
using StellaOps.VexLens.Normalization;
using StellaOps.VexLens.Observability;
using StellaOps.VexLens.Options;
using StellaOps.VexLens.Storage;
using StellaOps.VexLens.Trust;
using StellaOps.VexLens.Verification;

namespace StellaOps.VexLens.Extensions;

/// <summary>
/// Extension methods for registering VexLens services.
/// </summary>
public static class VexLensServiceCollectionExtensions
{
    /// <summary>
    /// Adds VexLens consensus engine services to the service collection.
    /// </summary>
    public static IServiceCollection AddVexLens(
        this IServiceCollection services,
        IConfiguration configuration)
    {
        var section = configuration.GetSection(VexLensOptions.SectionName);
        services.Configure<VexLensOptions>(section);

        var options = section.Get<VexLensOptions>() ?? new VexLensOptions();

        return services.AddVexLensCore(options);
    }

    /// <summary>
    /// Adds VexLens consensus engine services with explicit options.
    /// </summary>
    public static IServiceCollection AddVexLens(
        this IServiceCollection services,
        Action<VexLensOptions> configure)
    {
        var options = new VexLensOptions();
        configure(options);

        services.Configure(configure);

        return services.AddVexLensCore(options);
    }

    /// <summary>
    /// Adds VexLens services for testing with in-memory storage.
    /// </summary>
    public static IServiceCollection AddVexLensForTesting(this IServiceCollection services)
    {
        var options = new VexLensOptions
        {
            Storage = { Driver = "memory" },
            Telemetry = { MetricsEnabled = false, TracingEnabled = false }
        };

        return services.AddVexLensCore(options);
    }

    private static IServiceCollection AddVexLensCore(
        this IServiceCollection services,
        VexLensOptions options)
    {
        // Normalization
        services.TryAddSingleton<IVexNormalizerRegistry>(sp =>
        {
            var registry = new VexNormalizerRegistry();
            RegisterNormalizers(registry, options.Normalization);
            return registry;
        });

        // Product mapping
        services.TryAddSingleton<IProductMapper, ProductMapper>();

        // Verification
        services.TryAddSingleton<ISignatureVerifier, SignatureVerifier>();

        // Issuer directory - use in-memory by default, can be replaced
        services.TryAddSingleton<IIssuerDirectory, InMemoryIssuerDirectory>();

        // Trust engine
        services.TryAddSingleton<ITrustWeightEngine, TrustWeightEngine>();

        // Consensus engine
        services.TryAddSingleton<IVexConsensusEngine, VexConsensusEngine>();

        // Storage
        RegisterStorage(services, options.Storage);

        // Event emitter - in-memory for now
        services.TryAddSingleton<IConsensusEventEmitter, InMemoryConsensusEventEmitter>();

        // API service
        services.TryAddScoped<IVexLensApiService, VexLensApiService>();

        // Rationale service for AI/ML consumption
        services.TryAddScoped<IConsensusRationaleService, ConsensusRationaleService>();

        // Integration services
        services.TryAddScoped<IPolicyEngineIntegration, PolicyEngineIntegration>();
        services.TryAddScoped<IVulnExplorerIntegration, VulnExplorerIntegration>();

        // Metrics
        if (options.Telemetry.MetricsEnabled)
        {
            services.TryAddSingleton<VexLensMetrics>();
        }

        return services;
    }

    private static void RegisterNormalizers(
        VexNormalizerRegistry registry,
        VexLensNormalizationOptions options)
    {
        var enabledFormats = new HashSet<string>(
            options.EnabledFormats,
            StringComparer.OrdinalIgnoreCase);

        if (enabledFormats.Contains("OpenVEX"))
        {
            registry.Register(new OpenVexNormalizer());
        }

        if (enabledFormats.Contains("CSAF"))
        {
            registry.Register(new CsafVexNormalizer());
        }

        if (enabledFormats.Contains("CycloneDX"))
        {
            registry.Register(new CycloneDxVexNormalizer());
        }
    }

    private static void RegisterStorage(
        IServiceCollection services,
        VexLensStorageOptions options)
    {
        switch (options.Driver.ToLowerInvariant())
        {
            case "memory":
                services.TryAddSingleton<IConsensusProjectionStore>(sp =>
                {
                    var emitter = sp.GetRequiredService<IConsensusEventEmitter>();
                    return new InMemoryConsensusProjectionStore(emitter);
                });
                break;

            case "mongo":
                // MongoDB storage would be registered here
                // For now, fall back to in-memory
                services.TryAddSingleton<IConsensusProjectionStore>(sp =>
                {
                    var emitter = sp.GetRequiredService<IConsensusEventEmitter>();
                    return new InMemoryConsensusProjectionStore(emitter);
                });
                break;

            default:
                throw new InvalidOperationException(
                    $"Unknown VexLens storage driver: {options.Driver}");
        }
    }
}