53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
"""Утилиты для паролей и JWT-токенов."""
|
|
|
|
from datetime import datetime, timedelta
|
|
from typing import Optional
|
|
|
|
import bcrypt
|
|
from jose import JWTError, jwt
|
|
|
|
from .config import settings
|
|
|
|
ALGORITHM = "HS256"
|
|
# bcrypt ограничивает длину входа 72 байтами. Длиннее — обрезаем.
|
|
_BCRYPT_MAX_BYTES = 72
|
|
|
|
|
|
def _normalize_for_bcrypt(password: str) -> bytes:
|
|
"""bcrypt принимает максимум 72 байта. Длинные пароли обрезаем,
|
|
чтобы не получать ValueError при хешировании."""
|
|
encoded = password.encode("utf-8")
|
|
return encoded[:_BCRYPT_MAX_BYTES]
|
|
|
|
|
|
def hash_password(password: str) -> str:
|
|
"""Возвращает bcrypt-хеш пароля (ASCII-строка)."""
|
|
salt = bcrypt.gensalt(rounds=12)
|
|
return bcrypt.hashpw(_normalize_for_bcrypt(password), salt).decode("utf-8")
|
|
|
|
|
|
def verify_password(plain: str, hashed: str) -> bool:
|
|
"""Проверяет пароль против bcrypt-хеша."""
|
|
try:
|
|
return bcrypt.checkpw(_normalize_for_bcrypt(plain), hashed.encode("utf-8"))
|
|
except (ValueError, TypeError):
|
|
return False
|
|
|
|
|
|
def create_access_token(data: dict, expires_minutes: Optional[int] = None) -> str:
|
|
"""Создаёт JWT с подписью HS256 и заданным сроком жизни."""
|
|
to_encode = data.copy()
|
|
expire = datetime.utcnow() + timedelta(
|
|
minutes=expires_minutes or settings.access_token_expire_minutes
|
|
)
|
|
to_encode.update({"exp": expire})
|
|
return jwt.encode(to_encode, settings.secret_key, algorithm=ALGORITHM)
|
|
|
|
|
|
def decode_access_token(token: str) -> Optional[dict]:
|
|
"""Декодирует JWT. Возвращает None при любой ошибке (просрочен, подделан и т.п.)."""
|
|
try:
|
|
return jwt.decode(token, settings.secret_key, algorithms=[ALGORITHM])
|
|
except JWTError:
|
|
return None
|