84 lines
3.5 KiB
Python
84 lines
3.5 KiB
Python
"""Маршруты аутентификации."""
|
||
|
||
from __future__ import annotations
|
||
|
||
from datetime import timedelta
|
||
|
||
from fastapi import APIRouter, Depends, HTTPException, status
|
||
from sqlalchemy.orm import Session
|
||
|
||
from app.api.deps import get_current_user
|
||
from app.core.config import settings
|
||
from app.core.security import create_access_token, verify_password
|
||
from app.db.session import get_db
|
||
from app.models.user import User
|
||
from app.schemas.auth import EmailConfirm, EmailRequest, Token
|
||
from app.schemas.user import UserLogin, UserRead
|
||
from app.services.email_confirmation import confirm_email as mark_confirmed
|
||
from app.services.email_confirmation import issue_confirmation_token
|
||
|
||
router = APIRouter(prefix="/auth", tags=["auth"])
|
||
|
||
|
||
@router.post("/login", response_model=Token, summary="Авторизация по email и паролю")
|
||
def login(user_in: UserLogin, db: Session = Depends(get_db)) -> Token:
|
||
"""Проверяем логин и выдаём JWT."""
|
||
|
||
user = db.query(User).filter(User.email == user_in.email).first()
|
||
if not user or not verify_password(user_in.password, user.hashed_password):
|
||
raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Неверные данные")
|
||
|
||
if settings.require_email_confirmation and not user.is_email_confirmed:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_403_FORBIDDEN,
|
||
detail="Подтвердите e-mail, прежде чем войти",
|
||
)
|
||
|
||
token = create_access_token(user.email, timedelta(minutes=settings.access_token_expire_minutes))
|
||
return Token(access_token=token)
|
||
|
||
|
||
@router.get("/me", response_model=UserRead, summary="Текущий пользователь")
|
||
def read_current_user(current_user: User = Depends(get_current_user)) -> UserRead:
|
||
"""Простая проверка токена."""
|
||
|
||
return current_user
|
||
|
||
|
||
@router.post("/request-confirmation", summary="Повторная отправка письма с кодом")
|
||
def request_confirmation(payload: EmailRequest, db: Session = Depends(get_db)) -> dict[str, str | None]:
|
||
"""Генерируем код подтверждения и в режиме разработки возвращаем его в ответе."""
|
||
|
||
user = db.query(User).filter(User.email == payload.email).first()
|
||
if not user:
|
||
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Пользователь не найден")
|
||
|
||
token = issue_confirmation_token(user, db)
|
||
|
||
hint = None
|
||
if settings.environment != 'production':
|
||
hint = token
|
||
|
||
return {
|
||
"detail": "Письмо с подтверждением отправлено. Проверьте почту.",
|
||
"debug_token": hint,
|
||
}
|
||
|
||
|
||
@router.post("/confirm", summary="Подтверждаем e-mail по коду")
|
||
def confirm_email(payload: EmailConfirm, db: Session = Depends(get_db)) -> dict[str, str]:
|
||
"""Активируем почту, если код совпал."""
|
||
|
||
user = db.query(User).filter(User.email == payload.email).first()
|
||
if not user:
|
||
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Пользователь не найден")
|
||
|
||
if not user.email_confirmation_token:
|
||
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Код не запрошен")
|
||
|
||
if user.email_confirmation_token != payload.token:
|
||
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Неверный код")
|
||
|
||
mark_confirmed(user, db)
|
||
return {"detail": "E-mail подтверждён"}
|