up

src/AirGap/StellaOps.AirGap.Controller/DependencyInjection/AirGapControllerServiceCollectionExtensions.cs

@@ -1,10 +1,12 @@
 using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Options;
 using MongoDB.Driver;
 using StellaOps.AirGap.Controller.Options;
 using StellaOps.AirGap.Controller.Services;
 using StellaOps.AirGap.Controller.Stores;
+using StellaOps.AirGap.Importer.Validation;
 using StellaOps.AirGap.Time.Services;
 
 namespace StellaOps.AirGap.Controller.DependencyInjection;
@@ -14,24 +16,33 @@ public static class AirGapControllerServiceCollectionExtensions
     public static IServiceCollection AddAirGapController(this IServiceCollection services, IConfiguration configuration)
     {
         services.Configure<AirGapControllerMongoOptions>(configuration.GetSection("AirGap:Mongo"));
+        services.Configure<AirGapStartupOptions>(configuration.GetSection("AirGap:Startup"));
 
+        services.AddSingleton<AirGapTelemetry>();
         services.AddSingleton<StalenessCalculator>();
         services.AddSingleton<AirGapStateService>();
+        services.AddSingleton<TufMetadataValidator>();
+        services.AddSingleton<RootRotationPolicy>();
 
         services.AddSingleton<IAirGapStateStore>(sp =>
         {
             var opts = sp.GetRequiredService<IOptions<AirGapControllerMongoOptions>>().Value;
+            var logger = sp.GetRequiredService<ILogger<MongoAirGapStateStore>>();
             if (string.IsNullOrWhiteSpace(opts.ConnectionString))
             {
+                logger.LogInformation("AirGap controller using in-memory state store (Mongo connection string not configured).");
                 return new InMemoryAirGapStateStore();
             }
 
             var mongoClient = new MongoClient(opts.ConnectionString);
             var database = mongoClient.GetDatabase(string.IsNullOrWhiteSpace(opts.Database) ? "stellaops_airgap" : opts.Database);
             var collection = MongoAirGapStateStore.EnsureCollection(database);
+            logger.LogInformation("AirGap controller using Mongo state store (db={Database}, collection={Collection}).", opts.Database, opts.Collection);
             return new MongoAirGapStateStore(collection);
         });
 
+        services.AddHostedService<AirGapStartupDiagnosticsHostedService>();
+
         return services;
     }
 }