git.stella-ops.org/src/StellaOps.Concelier.Storage.Mongo/ServiceCollectionExtensions.cs

using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.DependencyInjection.Extensions;
using Microsoft.Extensions.Options;
using MongoDB.Driver;
using StellaOps.Concelier.Core.Jobs;
using StellaOps.Concelier.Storage.Mongo.Advisories;
using StellaOps.Concelier.Storage.Mongo.Aliases;
using StellaOps.Concelier.Storage.Mongo.ChangeHistory;
using StellaOps.Concelier.Storage.Mongo.Documents;
using StellaOps.Concelier.Storage.Mongo.Dtos;
using StellaOps.Concelier.Storage.Mongo.Exporting;
using StellaOps.Concelier.Storage.Mongo.JpFlags;
using StellaOps.Concelier.Storage.Mongo.MergeEvents;
using StellaOps.Concelier.Storage.Mongo.Conflicts;
using StellaOps.Concelier.Storage.Mongo.PsirtFlags;
using StellaOps.Concelier.Storage.Mongo.Statements;
using StellaOps.Concelier.Storage.Mongo.Events;
using StellaOps.Concelier.Core.Events;
using StellaOps.Concelier.Storage.Mongo.Migrations;
using StellaOps.Concelier.Storage.Mongo.Observations;
using StellaOps.Concelier.Core.Observations;

namespace StellaOps.Concelier.Storage.Mongo;

public static class ServiceCollectionExtensions
{
    public static IServiceCollection AddMongoStorage(this IServiceCollection services, Action<MongoStorageOptions> configureOptions)
    {
        ArgumentNullException.ThrowIfNull(services);
        ArgumentNullException.ThrowIfNull(configureOptions);

        services.AddOptions<MongoStorageOptions>()
            .Configure(configureOptions)
            .PostConfigure(static options => options.EnsureValid());

        services.TryAddSingleton(TimeProvider.System);

        services.AddSingleton<IMongoClient>(static sp =>
        {
            var options = sp.GetRequiredService<IOptions<MongoStorageOptions>>().Value;
            return new MongoClient(options.ConnectionString);
        });

        services.AddSingleton(static sp =>
        {
            var options = sp.GetRequiredService<IOptions<MongoStorageOptions>>().Value;
            var client = sp.GetRequiredService<IMongoClient>();
            var settings = new MongoDatabaseSettings
            {
                ReadConcern = ReadConcern.Majority,
                WriteConcern = WriteConcern.WMajority,
                ReadPreference = ReadPreference.PrimaryPreferred,
            };

            var database = client.GetDatabase(options.GetDatabaseName(), settings);
            var writeConcern = database.Settings.WriteConcern.With(wTimeout: options.CommandTimeout);
            return database.WithWriteConcern(writeConcern);
        });

        services.AddScoped<IMongoSessionProvider, MongoSessionProvider>();

        services.AddSingleton<MongoBootstrapper>();
        services.AddSingleton<IJobStore, MongoJobStore>();
        services.AddSingleton<ILeaseStore, MongoLeaseStore>();
        services.AddSingleton<ISourceStateRepository, MongoSourceStateRepository>();
        services.AddSingleton<IDocumentStore, DocumentStore>();
        services.AddSingleton<IDtoStore, DtoStore>();
        services.AddSingleton<IAdvisoryStore, AdvisoryStore>();
        services.AddSingleton<IAliasStore, AliasStore>();
        services.AddSingleton<IChangeHistoryStore, MongoChangeHistoryStore>();
        services.AddSingleton<IJpFlagStore, JpFlagStore>();
        services.AddSingleton<IPsirtFlagStore, PsirtFlagStore>();
        services.AddSingleton<IMergeEventStore, MergeEventStore>();
        services.AddSingleton<IAdvisoryStatementStore, AdvisoryStatementStore>();
        services.AddSingleton<IAdvisoryConflictStore, AdvisoryConflictStore>();
        services.AddSingleton<IAdvisoryObservationStore, AdvisoryObservationStore>();
        services.AddSingleton<IAdvisoryObservationLookup, AdvisoryObservationLookup>();
        services.AddSingleton<IAdvisoryEventRepository, MongoAdvisoryEventRepository>();
        services.AddSingleton<IAdvisoryEventLog, AdvisoryEventLog>();
        services.AddSingleton<IExportStateStore, ExportStateStore>();
        services.TryAddSingleton<ExportStateManager>();

        services.AddSingleton<IMongoCollection<JobRunDocument>>(static sp =>
        {
            var database = sp.GetRequiredService<IMongoDatabase>();
            return database.GetCollection<JobRunDocument>(MongoStorageDefaults.Collections.Jobs);
        });

        services.AddSingleton<IMongoCollection<JobLeaseDocument>>(static sp =>
        {
            var database = sp.GetRequiredService<IMongoDatabase>();
            return database.GetCollection<JobLeaseDocument>(MongoStorageDefaults.Collections.Locks);
        });

        services.AddSingleton<IMongoCollection<AdvisoryObservationDocument>>(static sp =>
        {
            var database = sp.GetRequiredService<IMongoDatabase>();
            return database.GetCollection<AdvisoryObservationDocument>(MongoStorageDefaults.Collections.AdvisoryObservations);
        });

        services.AddHostedService<RawDocumentRetentionService>();

        services.AddSingleton<MongoMigrationRunner>();
        services.AddSingleton<IMongoMigration, EnsureDocumentExpiryIndexesMigration>();
        services.AddSingleton<IMongoMigration, EnsureGridFsExpiryIndexesMigration>();
        services.AddSingleton<IMongoMigration, EnsureAdvisoryEventCollectionsMigration>();
        services.AddSingleton<IMongoMigration, SemVerStyleBackfillMigration>();

        return services;
    }
}