up

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

@@ -0,0 +1,37 @@
+using Microsoft.Extensions.Configuration;
+using Microsoft.Extensions.DependencyInjection;
+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.Time.Services;
+
+namespace StellaOps.AirGap.Controller.DependencyInjection;
+
+public static class AirGapControllerServiceCollectionExtensions
+{
+    public static IServiceCollection AddAirGapController(this IServiceCollection services, IConfiguration configuration)
+    {
+        services.Configure<AirGapControllerMongoOptions>(configuration.GetSection("AirGap:Mongo"));
+
+        services.AddSingleton<StalenessCalculator>();
+        services.AddSingleton<AirGapStateService>();
+
+        services.AddSingleton<IAirGapStateStore>(sp =>
+        {
+            var opts = sp.GetRequiredService<IOptions<AirGapControllerMongoOptions>>().Value;
+            if (string.IsNullOrWhiteSpace(opts.ConnectionString))
+            {
+                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);
+            return new MongoAirGapStateStore(collection);
+        });
+
+        return services;
+    }
+}