92 lines
3.9 KiB
Python
92 lines
3.9 KiB
Python
from fastapi import Request, Response
|
||
from fastapi.responses import JSONResponse
|
||
from starlette.middleware.base import BaseHTTPMiddleware
|
||
from starlette.types import ASGIApp
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from app.core.database import async_session_maker
|
||
from app.repositories.session_repository import SessionRepository
|
||
from app.models.session import Session
|
||
import logging
|
||
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class SessionMiddleware(BaseHTTPMiddleware):
|
||
"""Middleware для автоматического управления сессиями"""
|
||
|
||
def __init__(self, app: ASGIApp, cookie_name: str = "session_id"):
|
||
super().__init__(app)
|
||
self.cookie_name = cookie_name
|
||
|
||
async def dispatch(self, request: Request, call_next):
|
||
# Пропускаем статические файлы, служебные эндпоинты и OPTIONS запросы
|
||
if (request.url.path.startswith(("/docs", "/redoc", "/openapi.json", "/health", "/favicon.ico")) or
|
||
request.method == "OPTIONS"):
|
||
return await call_next(request)
|
||
|
||
# Получаем session_id из cookie или заголовка
|
||
session_id = request.cookies.get(self.cookie_name) or request.headers.get("X-Session-ID")
|
||
|
||
session_obj = None
|
||
|
||
try:
|
||
# Работаем с БД в рамках одной async сессии
|
||
async with async_session_maker() as db_session:
|
||
session_repo = SessionRepository(db_session)
|
||
|
||
# Проверяем существующую сессию
|
||
if session_id:
|
||
session_obj = await session_repo.get_by_session_id(session_id)
|
||
if session_obj and not session_obj.is_expired():
|
||
# Обновляем время последней активности
|
||
await session_repo.update_last_activity(session_id)
|
||
else:
|
||
session_obj = None
|
||
|
||
# Создаем новую сессию, если нет действующей
|
||
if not session_obj:
|
||
user_agent = request.headers.get("User-Agent")
|
||
client_ip = getattr(request.client, 'host', None) if request.client else None
|
||
session_obj = await session_repo.create_session(
|
||
user_agent=user_agent,
|
||
ip_address=client_ip
|
||
)
|
||
logger.info(f"Created new session: {session_obj.session_id}")
|
||
|
||
# Добавляем сессию в контекст запроса
|
||
request.state.session = session_obj
|
||
|
||
except Exception as e:
|
||
logger.error(f"Session middleware error: {e}")
|
||
return JSONResponse(
|
||
status_code=500,
|
||
content={"error": "Session management error"}
|
||
)
|
||
|
||
# Выполняем запрос
|
||
response = await call_next(request)
|
||
|
||
# Устанавливаем cookie с session_id в ответе
|
||
if session_obj and isinstance(response, Response):
|
||
response.set_cookie(
|
||
key=self.cookie_name,
|
||
value=session_obj.session_id,
|
||
max_age=30 * 24 * 60 * 60, # 30 дней
|
||
httponly=True,
|
||
secure=False, # Для dev среды
|
||
samesite="lax"
|
||
)
|
||
|
||
return response
|
||
|
||
|
||
async def get_current_session(request: Request) -> Session:
|
||
"""Получить текущую сессию из контекста запроса"""
|
||
return getattr(request.state, 'session', None)
|
||
|
||
|
||
async def get_db_session() -> AsyncSession:
|
||
"""Получить новую сессию БД для использования в эндпоинтах"""
|
||
async with async_session_maker() as session:
|
||
yield session |