lucretia/embeddingsearch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improved configurability for Swagger and ApiKeyMiddleware

Browse Source
This commit is contained in:
LD-Reborn
2025-07-12 21:23:06 +02:00
parent 3feebed5de
commit 58b40fac84
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/Server/Program.cs
View File

@@ -35,7 +35,10 @@ List<string>? allowedIps = builder.Configuration.GetSection("Embeddingsearch:Elm
app.Use(async (context, next) => app.Use(async (context, next) =>
{ {
if (context.Request.Path.StartsWithSegments("/elmah")) bool requestIsElmah = context.Request.Path.StartsWithSegments("/elmah");
bool requestIsSwagger = context.Request.Path.StartsWithSegments("/swagger");
if (requestIsElmah || requestIsSwagger)
{ {
var remoteIp = context.Connection.RemoteIpAddress?.ToString(); var remoteIp = context.Connection.RemoteIpAddress?.ToString();
bool blockRequest = allowedIps is null bool blockRequest = allowedIps is null
@@ -56,14 +59,18 @@ app.UseElmah();
app.MapHealthChecks("/healthz"); app.MapHealthChecks("/healthz");
bool IsDevelopment = app.Environment.IsDevelopment();
bool useSwagger = app.Configuration.GetValue<bool>("UseSwagger");
bool? UseMiddleware = app.Configuration.GetValue<bool?>("UseMiddleware");
// Configure the HTTP request pipeline. // Configure the HTTP request pipeline.
if (app.Environment.IsDevelopment()) if (IsDevelopment || useSwagger)
{ {
app.UseSwagger(); app.UseSwagger();
app.UseSwaggerUI(); app.UseSwaggerUI();
//app.UseElmahExceptionPage(); // Messes with JSON response for API calls. Leaving this here so I don't accidentally put this in again later on. //app.UseElmahExceptionPage(); // Messes with JSON response for API calls. Leaving this here so I don't accidentally put this in again later on.
} }
else if (UseMiddleware == true || IsDevelopment)
{ {
app.UseMiddleware<ApiKeyMiddleware>(); app.UseMiddleware<ApiKeyMiddleware>();
} }