🔨 fix: ip logging (related: #14)

Author: Zerohertz

.gitignore

@@ -173,6 +173,7 @@ cython_debug/
 # Credentials
 local.env
 prod.env
+local.yaml
 
 # Etc
 junit.xml

Makefile

@@ -17,8 +17,8 @@ test:
 
 .PHONY: local
 local:
-	export DESCRIPTION=$$(cat README.md) && source ./envs/local.env && uv run uvicorn app.main:app --host 0.0.0.0 --port $${PORT} --env-file ./envs/local.env --reload --log-level debug
+	export DESCRIPTION=$$(cat README.md) && source ./envs/local.env && uv run uvicorn app.main:app --host 0.0.0.0 --port $${PORT} --env-file ./envs/local.env --proxy-headers --forwarded-allow-ips='*' --reload
 
 .PHONY: prod
 prod:
-	export DESCRIPTION=$$(cat README.md) && uv run uvicorn app.main:app --host 0.0.0.0 --env-file ./envs/prod.env
+	export DESCRIPTION=$$(cat README.md) && uv run uvicorn app.main:app --host 0.0.0.0 --env-file ./envs/prod.env --proxy-headers --forwarded-allow-ips='*'

app/core/middlewares.py

@@ -11,14 +11,22 @@ class LoggingMiddleware(BaseHTTPMiddleware):
     async def dispatch(
         self, request: Request, call_next: RequestResponseEndpoint
     ) -> Response:
-        if request.client:
-            ip = ansi_format(
-                f"{request.client.host}:{request.client.port}",
-                bg_color=ANSI_BG_COLOR.LIGHT_BLACK,
-                style=[ANSI_STYLE.UNDERLINE, ANSI_STYLE.BOLD],
-            )
+        if request.headers.get("x-real-ip"):
+            ip = request.headers.get("x-real-ip")
+            logger.info(ip)
+        elif request.headers.get("x-forwarded-for"):
+            ip = request.headers.get("x-forwarded-for")
+            logger.info(ip)
+        elif request.client:
+            ip = request.client.host
+            logger.info(ip)
         else:
             ip = "None"
+        ip = ansi_format(
+            f"{ip}",
+            bg_color=ANSI_BG_COLOR.LIGHT_BLACK,
+            style=[ANSI_STYLE.UNDERLINE, ANSI_STYLE.BOLD],
+        )
         url = ansi_format(
             str(request.url),
             bg_color=ANSI_BG_COLOR.LIGHT_BLACK,