diff --git a/compliance-agent/src/api/server.rs b/compliance-agent/src/api/server.rs index 9b89714..99d9248 100644 --- a/compliance-agent/src/api/server.rs +++ b/compliance-agent/src/api/server.rs @@ -8,7 +8,7 @@ use tower_http::set_header::SetResponseHeaderLayer; use tower_http::trace::TraceLayer; use crate::agent::ComplianceAgent; -use crate::api::auth_middleware::{require_jwt_auth, JwksState}; +use crate::api::auth_middleware::{require_jwt_auth, require_tenant_status, JwksState}; use crate::api::routes; use crate::error::AgentError; @@ -44,9 +44,14 @@ pub async fn start_api_server(agent: ComplianceAgent, port: u16) -> Result<(), A jwks_url, }; tracing::info!("Keycloak JWT auth enabled for realm '{kc_realm}'"); + // Layers execute outermost-first. The Extension must run before + // require_jwt_auth so that middleware can read JwksState from + // request extensions, and the status gate must run after the + // JWT auth so TenantContext is in extensions. app = app - .layer(Extension(jwks_state)) - .layer(middleware::from_fn(require_jwt_auth)); + .layer(middleware::from_fn(require_tenant_status)) + .layer(middleware::from_fn(require_jwt_auth)) + .layer(Extension(jwks_state)); } else { tracing::warn!("Keycloak not configured - API endpoints are unprotected"); }