import time
from starlette.middleware.base import BaseHTTPMiddleware
from app.db import crud
from fastapi import Request


class LoggingMiddleware(BaseHTTPMiddleware):
    async def dispatch(self, request: Request, call_next):
        # Ignore les routes non enregistrées (ex: /docs, /openapi.json)
        if request.url.path not in request.app.state.logged_paths:
            return await call_next(request)
        start = time.time()
        response = await call_next(request)
        response_time_ms = (time.time() - start) * 1000
        prediction_id = getattr(request.state, "prediction_id", None)
        db = request.app.state.SessionLocal()
        try:
            crud.log_request(
                db=db,
                endpoint=request.url.path,
                method=request.method,
                status_code=response.status_code,
                response_time_ms=round(response_time_ms, 2),
                prediction_id=prediction_id,
            )
        finally:
            db.close()
        return response