ForbiddenStarsApp/backend/app/routers/users.py

"""Роутер текущего пользователя."""
from __future__ import annotations

from fastapi import APIRouter, Depends, File, Query, Request, UploadFile
from fastapi.responses import FileResponse
from sqlmodel import Session

from app.auth.deps import get_current_user
from app.core.errors import NotFoundError, ValidationError
from app.db.session import get_session
from app.models import User
from app.schemas import api as s
from app.services import audit_service, group_service, stats_service, user_service

router = APIRouter(prefix="/users", tags=["users"])

_AVATAR_MAX_BYTES = 2 * 1024 * 1024  # 2 МБ


def build_me(session: Session, user: User) -> s.MeRead:
    groups = [
        s.GroupBrief(id=g.id, name=g.name, role=role)
        for g, role in group_service.list_user_groups(session, user.id)  # type: ignore[arg-type]
    ]
    return s.MeRead(
        id=user.id,  # type: ignore[arg-type]
        nickname=user.nickname,
        role=user.role,
        auth_provider=user.auth_provider,
        telegram_id=user.telegram_id,
        active_group_id=user.active_group_id,
        bio=user.bio,
        avatar_url=user_service.avatar_url_for(user.id, user.avatar_path, user.updated_at),  # type: ignore[arg-type]
        groups=groups,
    )


@router.get("/me", response_model=s.MeRead)
def get_me(
    session: Session = Depends(get_session),
    user: User = Depends(get_current_user),
) -> s.MeRead:
    return build_me(session, user)


@router.patch("/me", response_model=s.UserRead)
def update_me(
    body: s.NicknameUpdate,
    request: Request,
    session: Session = Depends(get_session),
    user: User = Depends(get_current_user),
) -> s.UserRead:
    user_service.update_nickname(session, user, body.nickname)
    audit_service.record(
        session,
        actor_id=user.id,
        action="update",
        entity_type="user",
        entity_id=user.id,
        payload={"nickname": user.nickname},
        ip=request.client.host if request.client else None,
        user_agent=request.headers.get("user-agent"),
    )
    session.commit()
    return s.UserRead(
        id=user.id,  # type: ignore[arg-type]
        nickname=user.nickname,
        role=user.role,
        auth_provider=user.auth_provider,
        telegram_id=user.telegram_id,
        active_group_id=user.active_group_id,
        bio=user.bio,
        avatar_url=user_service.avatar_url_for(user.id, user.avatar_path, user.updated_at),  # type: ignore[arg-type]
    )


@router.patch("/me/profile", response_model=s.MeRead)
def update_my_profile(
    body: s.ProfileUpdate,
    request: Request,
    session: Session = Depends(get_session),
    user: User = Depends(get_current_user),
) -> s.MeRead:
    user_service.update_bio(session, user, body.bio)
    audit_service.record(
        session,
        actor_id=user.id,
        action="update",
        entity_type="user",
        entity_id=user.id,
        payload={"bio": True},
        ip=request.client.host if request.client else None,
    )
    session.commit()
    return build_me(session, user)


@router.put("/me/avatar", response_model=s.MeRead)
def upload_my_avatar(
    file: UploadFile = File(...),
    session: Session = Depends(get_session),
    user: User = Depends(get_current_user),
) -> s.MeRead:
    content = file.file.read(_AVATAR_MAX_BYTES + 1)
    if len(content) > _AVATAR_MAX_BYTES:
        raise ValidationError("Файл слишком большой (макс. 2 МБ).")
    ext = user_service.sniff_image_ext(content)
    if ext is None:
        raise ValidationError("Поддерживаются только изображения PNG, JPEG или WebP.")
    user_service.set_avatar(session, user, content, ext)
    return build_me(session, user)


@router.delete("/me/avatar", response_model=s.MeRead)
def delete_my_avatar(
    session: Session = Depends(get_session),
    user: User = Depends(get_current_user),
) -> s.MeRead:
    user_service.clear_avatar(session, user)
    return build_me(session, user)


@router.get("/{user_id}/avatar")
def get_user_avatar(
    user_id: int,
    session: Session = Depends(get_session),
    _user: User = Depends(get_current_user),
) -> FileResponse:
    target = session.get(User, user_id)
    path = user_service.avatar_abs_path(target.avatar_path) if target else None
    if path is None or not path.exists():
        raise NotFoundError("Аватар не найден.")
    ext = path.suffix.lstrip(".")
    return FileResponse(
        path,
        media_type=user_service.avatar_media_type(ext),
        headers={"Cache-Control": "public, max-age=3600"},
    )


@router.get("/{user_id}/profile", response_model=s.PublicProfile)
def get_user_profile(
    user_id: int,
    session: Session = Depends(get_session),
    _user: User = Depends(get_current_user),
) -> dict:
    return user_service.public_profile(session, user_id)


@router.get("/me/stats", response_model=s.ProfileStats)
def my_stats(
    session: Session = Depends(get_session),
    user: User = Depends(get_current_user),
) -> dict:
    return stats_service.profile_stats(session, user.id)  # type: ignore[arg-type]


@router.put("/me/active-group", response_model=s.MeRead)
def set_active_group(
    body: s.ActiveGroupUpdate,
    session: Session = Depends(get_session),
    user: User = Depends(get_current_user),
) -> s.MeRead:
    user_service.set_active_group(session, user, body.group_id)
    return build_me(session, user)


@router.get("/nickname-available", response_model=s.NicknameAvailable)
def nickname_available(
    value: str = Query(..., min_length=1),
    session: Session = Depends(get_session),
    user: User = Depends(get_current_user),
) -> s.NicknameAvailable:
    available = user_service.nickname_available(session, value.strip(), exclude_user_id=user.id)
    return s.NicknameAvailable(available=available)