MeterVault/src/Infrastructure/DependencyInjection.cs

using MeterVault.Core.Normalization;
using MeterVault.Infrastructure.Import;
using MeterVault.Infrastructure.Ingestion;
using MeterVault.Infrastructure.Normalization;
using MeterVault.Infrastructure.Persistence;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.DependencyInjection;

namespace MeterVault.Infrastructure;

/// <summary>Composition root for the infrastructure layer (persistence, ingestion, import).</summary>
public static class DependencyInjection
{
    public static IServiceCollection AddMeterVaultInfrastructure(
        this IServiceCollection services,
        string connectionString)
    {
        // A factory (for per-operation contexts in Blazor components/read services — a shared
        // circuit-scoped DbContext is not thread-safe) plus a scoped context (from the factory)
        // for request/tick-scoped services (API, workers, importers) that inject it directly.
        services.AddDbContextFactory<MeterVaultDbContext>(options =>
            options
                .UseNpgsql(connectionString, npgsql =>
                    npgsql.MigrationsAssembly(typeof(MeterVaultDbContext).Assembly.FullName))
                .UseSnakeCaseNamingConvention());
        services.AddScoped<MeterVaultDbContext>(sp =>
            sp.GetRequiredService<IDbContextFactory<MeterVaultDbContext>>().CreateDbContext());

        services.AddSingleton<INormalizationEngine>(_ => NormalizationEngine.CreateDefault());
        services.AddScoped<NormalizationService>();
        services.AddScoped<CsvImporter>();
        services.AddScoped<ImportService>();
        services.AddScoped<ReferenceDataImporter>();
        services.AddScoped<IngestionService>();
        services.AddScoped<MqttMessageRouter>();
        // HttpClient + HA tester are available even with live ingestion off, so the admin
        // "Test connection" works without the background workers running.
        services.AddHttpClient();
        services.AddScoped<HaConnectionTester>();
        // Singleton: wraps one IDataProtector, and the ingestion workers (themselves singletons)
        // resolve connector secrets on every reconnect.
        services.AddSingleton<Security.SecretProtector>();
        // Singleton: it caches the last answer so the dashboard never waits on a remote call.
        services.AddSingleton<Update.UpdateCheckService>();
        services.AddSingleton<Update.UpdateRunner>();
        services.AddScoped<Costing.CostService>();
        services.AddScoped<Dashboard.DashboardService>();
        services.AddScoped<Dashboard.SolarService>();
        services.AddScoped<Dashboard.MeterPeriodService>();
        services.AddScoped<Dashboard.ConsumableService>();
        services.AddScoped<Dashboard.MeterDetailService>();
        services.AddScoped<Dashboard.FlowService>();
        services.AddScoped<Backup.ExportService>();

        return services;
    }

    /// <summary>
    /// Registers the live-ingestion background workers (MQTT/Tasmota + Home Assistant). Kept
    /// separate from <see cref="AddMeterVaultInfrastructure"/> so tests can opt out of brokers.
    /// </summary>
    public static IServiceCollection AddMeterVaultIngestion(this IServiceCollection services)
    {
        services.AddHttpClient();
        services.AddHostedService<MqttIngestionWorker>();
        services.AddHostedService<HomeAssistantWorker>();
        services.AddHostedService<HomeAssistantWebSocketWorker>();
        return services;
    }
}