Vladless
101 lines
4.3 KiB
Python
101 lines
4.3 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from api.depends import get_session, verify_identity_token
|
|
from api.v2.schemas.identities import (
|
|
IdentityResponse,
|
|
LinkTelegramRequest,
|
|
LoginRequest,
|
|
LoginResponse,
|
|
LoginTelegramRequest,
|
|
RegisterByEmailRequest,
|
|
RegisterResponse,
|
|
)
|
|
from config import API_TOKEN_TTL_DAYS, API_TOKEN
|
|
from database import identities as idb
|
|
from utils.telegram_login import verify_telegram_login
|
|
|
|
router = APIRouter(prefix="/auth", tags=["Auth"])
|
|
TOKEN_TTL_HINT = "бессрочно" if API_TOKEN_TTL_DAYS is None else f"{API_TOKEN_TTL_DAYS} дн."
|
|
TELEGRAM_LOGIN_MAX_AGE = 86400 # 24 часа
|
|
|
|
|
|
@router.post("/register", response_model=RegisterResponse)
|
|
async def register_by_email(
|
|
body: RegisterByEmailRequest,
|
|
session: AsyncSession = Depends(get_session),
|
|
):
|
|
(
|
|
"""Регистрация по почте и паролю: создаётся идентичность, выдаётся токен. Срок действия токена: """
|
|
+ TOKEN_TTL_HINT
|
|
+ "."
|
|
)
|
|
email = body.email.strip().lower()
|
|
if not email:
|
|
raise HTTPException(status_code=400, detail="Email обязателен")
|
|
if not body.password or len(body.password) < 8:
|
|
raise HTTPException(status_code=400, detail="Пароль минимум 8 символов")
|
|
existing = await idb.get_identity_by_email(session, email)
|
|
if existing:
|
|
raise HTTPException(status_code=409, detail="Идентичность с таким email уже существует")
|
|
identity, token = await idb.create_identity_with_token(session, email=email, password=body.password)
|
|
return RegisterResponse(identity_id=identity.id, token=token)
|
|
|
|
|
|
@router.post("/login", response_model=LoginResponse)
|
|
async def login(
|
|
body: LoginRequest,
|
|
session: AsyncSession = Depends(get_session),
|
|
):
|
|
"""Вход по email и паролю. Возвращает identity_id и новый токен. Срок действия токена: """ + TOKEN_TTL_HINT + "."
|
|
email = body.email.strip().lower()
|
|
if not email:
|
|
raise HTTPException(status_code=400, detail="Email обязателен")
|
|
result = await idb.login_by_email(session, email, body.password)
|
|
if not result:
|
|
raise HTTPException(status_code=401, detail="Неверный email или пароль")
|
|
identity, token = result
|
|
return LoginResponse(identity_id=identity.id, token=token)
|
|
|
|
|
|
@router.post("/login-telegram", response_model=LoginResponse)
|
|
async def login_telegram(
|
|
body: LoginTelegramRequest,
|
|
session: AsyncSession = Depends(get_session),
|
|
):
|
|
(
|
|
"""Вход через Telegram Login Widget (кнопка на сайте). По tg_id находим или создаём Identity, выдаём токен. Срок действия токена: """
|
|
+ TOKEN_TTL_HINT
|
|
+ "."
|
|
)
|
|
payload = body.model_dump(mode="json")
|
|
if not verify_telegram_login(payload, API_TOKEN, max_age_seconds=TELEGRAM_LOGIN_MAX_AGE):
|
|
raise HTTPException(status_code=401, detail="Неверная подпись или устаревшие данные от Telegram")
|
|
identity = await idb.get_or_create_identity_for_tg(session, body.id)
|
|
token = await idb.issue_token_for_identity(session, identity)
|
|
return LoginResponse(identity_id=identity.id, token=token)
|
|
|
|
|
|
@router.post("/link-telegram", response_model=IdentityResponse)
|
|
async def link_telegram(
|
|
body: LinkTelegramRequest,
|
|
session: AsyncSession = Depends(get_session),
|
|
identity=Depends(verify_identity_token),
|
|
):
|
|
"""Привязывает Telegram (tg_id) к текущей идентичности. Требуется X-Identity-Id и X-Token."""
|
|
result = await idb.attach_telegram(session, identity.id, body.tg_id)
|
|
if not result:
|
|
raise HTTPException(
|
|
status_code=409,
|
|
detail="Этот Telegram уже привязан к другой идентичности",
|
|
)
|
|
return IdentityResponse.model_validate(result)
|
|
|
|
|
|
@router.get("/me", response_model=IdentityResponse)
|
|
async def me(
|
|
identity=Depends(verify_identity_token),
|
|
):
|
|
"""Текущая идентичность по заголовкам X-Identity-Id и X-Token."""
|
|
return IdentityResponse.model_validate(identity)
|