42 lines
1.7 KiB
C#
42 lines
1.7 KiB
C#
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Options;
|
|
using StellaOps.Concelier.Connector.Common.Http;
|
|
using StellaOps.Concelier.Connector.Cve.Configuration;
|
|
using StellaOps.Concelier.Connector.Cve.Internal;
|
|
|
|
namespace StellaOps.Concelier.Connector.Cve;
|
|
|
|
public static class CveServiceCollectionExtensions
|
|
{
|
|
public static IServiceCollection AddCveConnector(this IServiceCollection services, Action<CveOptions> configure)
|
|
{
|
|
ArgumentNullException.ThrowIfNull(services);
|
|
ArgumentNullException.ThrowIfNull(configure);
|
|
|
|
services.AddOptions<CveOptions>()
|
|
.Configure(configure)
|
|
.PostConfigure(static opts => opts.Validate());
|
|
|
|
services.AddSourceHttpClient(CveOptions.HttpClientName, (sp, clientOptions) =>
|
|
{
|
|
var options = sp.GetRequiredService<IOptions<CveOptions>>().Value;
|
|
clientOptions.BaseAddress = options.BaseEndpoint;
|
|
clientOptions.Timeout = TimeSpan.FromSeconds(30);
|
|
clientOptions.UserAgent = "StellaOps.Concelier.Cve/1.0";
|
|
clientOptions.AllowedHosts.Clear();
|
|
clientOptions.AllowedHosts.Add(options.BaseEndpoint.Host);
|
|
clientOptions.DefaultRequestHeaders["Accept"] = "application/json";
|
|
if (options.HasCredentials())
|
|
{
|
|
clientOptions.DefaultRequestHeaders["CVE-API-ORG"] = options.ApiOrg;
|
|
clientOptions.DefaultRequestHeaders["CVE-API-USER"] = options.ApiUser;
|
|
clientOptions.DefaultRequestHeaders["CVE-API-KEY"] = options.ApiKey;
|
|
}
|
|
});
|
|
|
|
services.AddSingleton<CveDiagnostics>();
|
|
services.AddTransient<CveConnector>();
|
|
return services;
|
|
}
|
|
}
|