48 lines
2.1 KiB
C#
48 lines
2.1 KiB
C#
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.DependencyInjection.Extensions;
|
|
using StellaOps.Telemetry.Federation.Aggregation;
|
|
using StellaOps.Telemetry.Federation.Bundles;
|
|
using StellaOps.Telemetry.Federation.Consent;
|
|
using StellaOps.Telemetry.Federation.Intelligence;
|
|
using StellaOps.Telemetry.Federation.Privacy;
|
|
using StellaOps.Telemetry.Federation.Security;
|
|
using StellaOps.Telemetry.Federation.Sync;
|
|
|
|
namespace StellaOps.Telemetry.Federation;
|
|
|
|
public static class FederationServiceCollectionExtensions
|
|
{
|
|
public static IServiceCollection AddFederatedTelemetry(
|
|
this IServiceCollection services,
|
|
Action<FederatedTelemetryOptions>? configureOptions = null)
|
|
{
|
|
ArgumentNullException.ThrowIfNull(services);
|
|
|
|
services.AddOptions<FederatedTelemetryOptions>()
|
|
.Configure(options => configureOptions?.Invoke(options));
|
|
|
|
services.TryAddSingleton<IPrivacyBudgetTracker, PrivacyBudgetTracker>();
|
|
services.TryAddSingleton<ITelemetryAggregator, TelemetryAggregator>();
|
|
services.TryAddSingleton<HmacFederationDsseEnvelopeService>();
|
|
services.TryAddSingleton<IFederationDsseEnvelopeSigner>(sp => sp.GetRequiredService<HmacFederationDsseEnvelopeService>());
|
|
services.TryAddSingleton<IFederationDsseEnvelopeVerifier>(sp => sp.GetRequiredService<HmacFederationDsseEnvelopeService>());
|
|
services.TryAddSingleton<IConsentManager, ConsentManager>();
|
|
services.TryAddSingleton<IFederatedTelemetryBundleBuilder, FederatedTelemetryBundleBuilder>();
|
|
services.TryAddSingleton<IExploitIntelligenceMerger, ExploitIntelligenceMerger>();
|
|
services.TryAddSingleton<FederatedIntelligenceNormalizer>();
|
|
services.TryAddSingleton<IEgressPolicyIntegration, EgressPolicyIntegration>();
|
|
|
|
return services;
|
|
}
|
|
|
|
public static IServiceCollection AddFederatedTelemetrySync(
|
|
this IServiceCollection services)
|
|
{
|
|
ArgumentNullException.ThrowIfNull(services);
|
|
|
|
services.AddHostedService<FederatedTelemetrySyncService>();
|
|
|
|
return services;
|
|
}
|
|
}
|