refactor: cleanup; feat: connection to agent

1 files changed, 13 insertions, 1 deletions

diff --git a/src/marlin.web/Program.cs b/src/marlin.web/Program.cs
index f079a3c..700e696 100644
--- a/src/marlin.web/Program.cs
+++ b/src/marlin.web/Program.cs
@@ -13,6 +13,16 @@ builder.Services.AddDbContext<ApplicationDbContext>(options => options
 
 builder.Services.AddScoped<ISourceSecretStore, DatabaseSourceSecretStore>();
 
+builder.Services
+    .AddAuthentication(AgentSchemeOptions.Name)
+    .AddScheme<AgentSchemeOptions, AgentAuthenticationHandler>(AgentSchemeOptions.Name, null);
+
+builder.Services.AddAuthorization(options =>
+{
+    options.AddPolicy("Agent", p => p.RequireClaim("agent_id"));
+    options.AddPolicy("JobScoped", p => p.RequireClaim("job_id"));
+});
+
 builder.Services.AddHostedService<DatabaseUpgrader>();
 
 builder.Services.AddSingleton<VersionControlSyncService>();
@@ -20,7 +30,8 @@ builder.Services.AddHostedService<VersionControlSyncService>(sp => sp
     .GetRequiredService<VersionControlSyncService>());
 
 // Add services to the container.
-builder.Services.AddControllersWithViews();
+builder.Services.AddControllersWithViews(options =>
+    options.OutputFormatters.RemoveType<Microsoft.AspNetCore.Mvc.Formatters.HttpNoContentOutputFormatter>());
 
 var app = builder.Build();
 
@@ -35,6 +46,7 @@ if (!app.Environment.IsDevelopment())
 app.UseHttpsRedirection();
 app.UseRouting();
 
+app.UseAuthentication();
 app.UseAuthorization();
 
 app.MapStaticAssets();