up

src/Policy/__Libraries/StellaOps.Policy.Storage.Postgres/ServiceCollectionExtensions.cs

@@ -2,6 +2,8 @@ using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
 using StellaOps.Infrastructure.Postgres;
 using StellaOps.Infrastructure.Postgres.Options;
+using StellaOps.Policy.Scoring.Receipts;
+using StellaOps.Policy.Storage.Postgres.Repositories;
 
 namespace StellaOps.Policy.Storage.Postgres;
 
@@ -25,6 +27,15 @@ public static class ServiceCollectionExtensions
         services.Configure<PostgresOptions>(sectionName, configuration.GetSection(sectionName));
         services.AddSingleton<PolicyDataSource>();
 
+        // Register repositories
+        services.AddScoped<IPackRepository, PackRepository>();
+        services.AddScoped<IPackVersionRepository, PackVersionRepository>();
+        services.AddScoped<IRuleRepository, RuleRepository>();
+        services.AddScoped<IRiskProfileRepository, RiskProfileRepository>();
+        services.AddScoped<IEvaluationRunRepository, EvaluationRunRepository>();
+        services.AddScoped<IExceptionRepository, ExceptionRepository>();
+        services.AddScoped<IReceiptRepository, PostgresReceiptRepository>();
+
         return services;
     }
 
@@ -41,6 +52,15 @@ public static class ServiceCollectionExtensions
         services.Configure(configureOptions);
         services.AddSingleton<PolicyDataSource>();
 
+        // Register repositories
+        services.AddScoped<IPackRepository, PackRepository>();
+        services.AddScoped<IPackVersionRepository, PackVersionRepository>();
+        services.AddScoped<IRuleRepository, RuleRepository>();
+        services.AddScoped<IRiskProfileRepository, RiskProfileRepository>();
+        services.AddScoped<IEvaluationRunRepository, EvaluationRunRepository>();
+        services.AddScoped<IExceptionRepository, ExceptionRepository>();
+        services.AddScoped<IReceiptRepository, PostgresReceiptRepository>();
+
         return services;
     }
 }