103 lines
3.7 KiB
C#
103 lines
3.7 KiB
C#
using MareSynchronosShared.Authentication;
|
|
using MareSynchronosShared.Data;
|
|
using MareSynchronosShared.Metrics;
|
|
using MareSynchronosShared.Utils;
|
|
using Microsoft.AspNetCore.Authentication;
|
|
using Microsoft.AspNetCore.Authorization;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Prometheus;
|
|
|
|
namespace MareSynchronosStaticFilesServer;
|
|
|
|
public class Startup
|
|
{
|
|
private bool _isMain;
|
|
public Startup(IConfiguration configuration)
|
|
{
|
|
Configuration = configuration;
|
|
var mareSettings = Configuration.GetRequiredSection("MareSynchronos");
|
|
_isMain = string.IsNullOrEmpty(mareSettings.GetValue(nameof(StaticFilesServerConfiguration.RemoteCacheSourceUri), string.Empty));
|
|
}
|
|
|
|
public IConfiguration Configuration { get; }
|
|
|
|
public void ConfigureServices(IServiceCollection services)
|
|
{
|
|
services.AddHttpContextAccessor();
|
|
|
|
services.AddLogging();
|
|
|
|
services.Configure<MareConfigurationAuthBase>(Configuration.GetRequiredSection("MareSynchronos"));
|
|
services.Configure<StaticFilesServerConfiguration>(Configuration.GetRequiredSection("MareSynchronos"));
|
|
services.AddSingleton(Configuration);
|
|
|
|
var mareSettings = Configuration.GetRequiredSection("MareSynchronos");
|
|
|
|
services.AddControllers();
|
|
|
|
services.AddSingleton(m => new MareMetrics(m.GetService<ILogger<MareMetrics>>(), new List<string>
|
|
{
|
|
MetricsAPI.CounterAuthenticationCacheHits,
|
|
MetricsAPI.CounterAuthenticationFailures,
|
|
MetricsAPI.CounterAuthenticationRequests,
|
|
MetricsAPI.CounterAuthenticationSuccesses
|
|
}, new List<string>
|
|
{
|
|
MetricsAPI.GaugeFilesTotalSize,
|
|
MetricsAPI.GaugeFilesTotal,
|
|
MetricsAPI.GaugeFilesUniquePastDay,
|
|
MetricsAPI.GaugeFilesUniquePastDaySize,
|
|
MetricsAPI.GaugeFilesUniquePastHour,
|
|
MetricsAPI.GaugeFilesUniquePastHourSize
|
|
}));
|
|
services.AddSingleton<CachedFileProvider>();
|
|
services.AddSingleton<FileStatisticsService>();
|
|
|
|
services.AddHostedService(m => m.GetService<FileStatisticsService>());
|
|
services.AddHostedService<FileCleanupService>();
|
|
|
|
services.AddSingleton<SecretKeyAuthenticatorService>();
|
|
services.AddDbContextPool<MareDbContext>(options =>
|
|
{
|
|
options.UseNpgsql(Configuration.GetConnectionString("DefaultConnection"), builder =>
|
|
{
|
|
builder.MigrationsHistoryTable("_efmigrationshistory", "public");
|
|
}).UseSnakeCaseNamingConvention();
|
|
options.EnableThreadSafetyChecks(false);
|
|
}, mareSettings.GetValue(nameof(MareConfigurationBase.DbContextPoolSize), 1024));
|
|
|
|
services.AddAuthentication(options =>
|
|
{
|
|
options.DefaultScheme = SecretKeyAuthenticationHandler.AuthScheme;
|
|
})
|
|
.AddScheme<AuthenticationSchemeOptions, SecretKeyAuthenticationHandler>(SecretKeyAuthenticationHandler.AuthScheme, options => { });
|
|
services.AddAuthorization(options => options.FallbackPolicy = new AuthorizationPolicyBuilder().RequireAuthenticatedUser().Build());
|
|
|
|
services.AddGrpc(o =>
|
|
{
|
|
o.MaxReceiveMessageSize = null;
|
|
});
|
|
}
|
|
|
|
public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
|
|
{
|
|
app.UseHttpLogging();
|
|
|
|
app.UseRouting();
|
|
|
|
var metricServer = new KestrelMetricServer(4981);
|
|
metricServer.Start();
|
|
|
|
app.UseHttpMetrics();
|
|
|
|
app.UseAuthentication();
|
|
app.UseAuthorization();
|
|
|
|
app.UseEndpoints(e =>
|
|
{
|
|
if(_isMain)
|
|
e.MapGrpcService<GrpcFileService>();
|
|
e.MapControllers();
|
|
});
|
|
}
|
|
} |