Respect X-Forwarded-Proto for building URLs

owl-corp · Aug 19, 2024 · 79b00e3 · 79b00e3
1 parent f787db9
commit 79b00e3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/thallium-backend/src/app.py b/thallium-backend/src/app.py
@@ -34,7 +34,9 @@ def health_check() -> JSONResponse:
 
 def build_url(request: Request, path: str = "") -> str:
  """Build a URL from a request, for OpenAPI + Scalar."""
- return f"{request.url.scheme}://{request.headers['Host']}{CONFIG.app_prefix}{path}"
+ scheme = request.headers.get("x-forwarded-proto", request.url.scheme)
+
+ return f"{scheme}://{request.headers['Host']}{CONFIG.app_prefix}{path}"
 
 
 @fastapi_app.get("/docs", include_in_schema=False)