"""Утилиты безопасности: JWT, хэширование паролей, dependency для получения текущего пользователя.""" from datetime import datetime, timedelta, timezone from typing import Any from fastapi import Depends, HTTPException, status from fastapi.security import OAuth2PasswordBearer from jose import JWTError, jwt from passlib.context import CryptContext from sqlalchemy import select from sqlalchemy.ext.asyncio import AsyncSession from app.config import settings from app.database import get_db # Контекст хэширования паролей (bcrypt) pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto") # OAuth2 схема oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/api/auth/login") def hash_password(password: str) -> str: """Хэшировать пароль через bcrypt.""" return pwd_context.hash(password) def verify_password(plain_password: str, hashed_password: str) -> bool: """Проверить соответствие пароля его хэшу.""" return pwd_context.verify(plain_password, hashed_password) def create_access_token(data: dict[str, Any], expires_delta: timedelta | None = None) -> str: """Создать JWT токен доступа.""" to_encode = data.copy() expire = datetime.now(timezone.utc) + ( expires_delta or timedelta(minutes=settings.ACCESS_TOKEN_EXPIRE_MINUTES) ) to_encode.update({"exp": expire}) return jwt.encode(to_encode, settings.SECRET_KEY, algorithm=settings.ALGORITHM) def verify_token(token: str) -> dict[str, Any]: """ Декодировать и верифицировать JWT токен. Raises: HTTPException: если токен невалиден или просрочен. """ try: payload = jwt.decode(token, settings.SECRET_KEY, algorithms=[settings.ALGORITHM]) user_id: int | None = payload.get("sub") if user_id is None: raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail="Неверный токен аутентификации", headers={"WWW-Authenticate": "Bearer"}, ) return payload except JWTError: raise HTTPException( status_code=status.HTTP_401_UNAUTHORIZED, detail="Невалидный или просроченный токен", headers={"WWW-Authenticate": "Bearer"}, ) async def get_current_user( token: str = Depends(oauth2_scheme), db: AsyncSession = Depends(get_db), ): """ FastAPI dependency: получить текущего авторизованного пользователя из JWT токена. Raises: HTTPException 401: если токен невалиден. HTTPException 404: если пользователь не найден. """ from app.models.user import User payload = verify_token(token) user_id: int = int(payload.get("sub")) result = await db.execute(select(User).where(User.id == user_id)) user = result.scalar_one_or_none() if user is None: raise HTTPException( status_code=status.HTTP_404_NOT_FOUND, detail="Пользователь не найден", ) return user async def get_current_verified_user( current_user=Depends(get_current_user), ): """ FastAPI dependency: только верифицированные пользователи. Raises: HTTPException 403: если email не подтверждён. """ if not current_user.is_verified: raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail="Необходимо подтвердить email адрес", ) return current_user