41 lines
1.1 KiB
C#
41 lines
1.1 KiB
C#
using System.Security.Claims;
|
|
|
|
namespace StellaOps.Gateway.WebService.Middleware;
|
|
|
|
public sealed class TenantMiddleware
|
|
{
|
|
private readonly RequestDelegate _next;
|
|
private readonly ILogger<TenantMiddleware> _logger;
|
|
|
|
public TenantMiddleware(RequestDelegate next, ILogger<TenantMiddleware> logger)
|
|
{
|
|
_next = next;
|
|
_logger = logger;
|
|
}
|
|
|
|
public async Task InvokeAsync(HttpContext context)
|
|
{
|
|
if (GatewayRoutes.IsSystemPath(context.Request.Path))
|
|
{
|
|
await _next(context);
|
|
return;
|
|
}
|
|
|
|
var tenantId = context.User.FindFirstValue("tid");
|
|
if (!string.IsNullOrWhiteSpace(tenantId))
|
|
{
|
|
context.Items[GatewayContextKeys.TenantId] = tenantId;
|
|
if (!context.Request.Headers.ContainsKey("tid"))
|
|
{
|
|
context.Request.Headers["tid"] = tenantId;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
_logger.LogDebug("No tenant claim found on request {TraceId}", context.TraceIdentifier);
|
|
}
|
|
|
|
await _next(context);
|
|
}
|
|
}
|