from fastapi import Request, Response
from sqlalchemy.ext.asyncio import AsyncSession

from database import get_async_session


# from starlette.middleware.base import BaseHTTPMiddleware

# from database import SessionLocal

#
# class DatabaseSessionMiddleware(BaseHTTPMiddleware):
#     async def dispatch(self, request: Request, call_next):
#         response = Response("Internal server error", status_code=500)
#         try:
#             request.state.db = SessionLocal()
#             response = await call_next(request)
#         finally:
#             request.state.db.close()
#         return response


def get_db(request: Request):
    print(request)
    return request.state.db