Merge pull request #567 from stacklok/uagent

Log User-agents from clients

Author: Luke Hinds

src/codegate/providers/copilot/provider.py

@@ -243,6 +243,8 @@ def get_headers_dict(self, complete_request) -> Dict[str, str]:
                 try:
                     name, value = header.decode("utf-8").split(":", 1)
                     headers_dict[name.strip().lower()] = value.strip()
+                    if name == "user-agent":
+                        logger.debug(f"User-Agent header received: {value} from {self.peername}")
                 except ValueError:
                     continue
 

src/codegate/server.py

@@ -1,7 +1,7 @@
 import traceback
 
 import structlog
-from fastapi import APIRouter, FastAPI
+from fastapi import APIRouter, FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 from starlette.middleware.errors import ServerErrorMiddleware
@@ -35,6 +35,15 @@ def init_app(pipeline_factory: PipelineFactory) -> FastAPI:
         description=__description__,
         version=__version__,
     )
+
+    @app.middleware("http")
+    async def log_user_agent(request: Request, call_next):
+        user_agent = request.headers.get("user-agent")
+        client_host = request.client.host if request.client else "unknown"
+        logger.debug(f"User-Agent header received: {user_agent} from {client_host}")
+        response = await call_next(request)
+        return response
+
     app.add_middleware(
         CORSMiddleware,
         allow_origins=["*"],