feat: add tenant dependency and register TenantMiddleware

fiqrikm18 · fiqrikm18 · commit 80b5f37a5dfd · 2026-06-26T11:53:36.000+07:00
diff --git a/src/core/bootstrap/middleware.py b/src/core/bootstrap/middleware.py
@@ -12,6 +12,7 @@
 from src.core.middleware.request_size import LimitRequestSizeMiddleware
 from src.core.middleware.security_headers import SecurityHeadersMiddleware
 from src.core.middleware.structured_logging import StructuredLoggingMiddleware
+from src.core.middleware.tenant import TenantMiddleware
 from src.core.security.providers import JWTAuthProvider
 
 settings = get_settings()
@@ -42,6 +43,7 @@ def register_middleware(app: FastAPI):
     app.add_middleware(MetricsMiddleware)
     app.add_middleware(StructuredLoggingMiddleware)
     app.add_middleware(AuditLoggingMiddleware)
+    app.add_middleware(TenantMiddleware)
     app.add_middleware(
         AuthenticationMiddleware,
         providers=[JWTAuthProvider()],
diff --git a/src/core/dependency/tenant.py b/src/core/dependency/tenant.py
@@ -0,0 +1,18 @@
+from uuid import UUID
+
+from fastapi import HTTPException, Request
+from starlette.status import HTTP_400_BAD_REQUEST
+
+
+def get_current_tenant_id(request: Request) -> UUID:
+    tenant_id: UUID | None = getattr(request.state, "tenant_id", None)
+    if tenant_id is None:
+        raise HTTPException(
+            status_code=HTTP_400_BAD_REQUEST,
+            detail="Tenant not identified",
+        )
+    return tenant_id
+
+
+def get_optional_tenant_id(request: Request) -> UUID | None:
+    return getattr(request.state, "tenant_id", None)
