feat(backend): модуль «замена фона» и хранилище своих картинок
Отключаемый в админке модуль `virtual_background` (дефолт — выключен, чтобы
обновление не меняло продукт у тех, кто ничего не просил). Флаг едет клиенту
двумя путями: на публичные страницы входа — через `GET /public/settings`,
участнику комнаты — в join-ответе (`JoinOut`), потому что значение нужно на
руках ДО первого рендера комнаты, а `/admin/settings` доступен только админу.
Свои картинки пользователя (`/users/me/backgrounds`, GET/POST/DELETE):
файлы на диске (`backgrounds/{user_id}/{id}.{ext}`), в БД только путь — как у
аватаров, «чтобы не грузили БД». Лимит в 10 штук проверяется на сервере под
блокировкой строки пользователя: две одновременные загрузки иначе обе увидели
бы «уже девять» и обе прошли бы. Удаление сносит и запись, и файл; чужую
картинку по её id удалить нельзя — владелец в условии запроса.
Валидация загрузки (допустимые форматы, магические байты, реальный размер)
выделена из `services/avatars.py` в общий `services/images.py`: правила у
аватара и фона одни и те же, а разъехавшись, они дали бы дыру ровно там, ради
чего проверка и написана. Публичный API аватаров не изменился.
Сжимает картинку клиент (Pillow на бэкенде нет), но серверная валидация
остаётся полноценной — запрос может прийти и мимо интерфейса.
Новый ключ настройки вписан в `_MANAGED_KEYS` тестов: без этого включённый
в общей dev-БД модуль ронял чужие тесты, которые считают себя изолированными.
This commit is contained in:
@@ -3,80 +3,49 @@
|
||||
Файл лежит на диске `MEDIA_ROOT/avatars/{user_id}.{ext}`; в БД (`users.avatar_path`)
|
||||
хранится путь относительно `MEDIA_ROOT` (`avatars/{user_id}.{ext}`) — тот же
|
||||
приём, что и у записей аудиотреков (`recordings_dir`, `core/config.py`).
|
||||
|
||||
Сама проверка содержимого (допустимые форматы, магические байты, реальный
|
||||
размер) живёт в `services/images.py` — она общая с картинками фона видео
|
||||
(`services/backgrounds.py`).
|
||||
"""
|
||||
|
||||
import uuid
|
||||
from collections.abc import Callable
|
||||
from pathlib import Path
|
||||
|
||||
from fastapi import UploadFile
|
||||
|
||||
from services.images import (
|
||||
ImageInvalidTypeError,
|
||||
ImageTooLargeError,
|
||||
read_and_validate_image,
|
||||
)
|
||||
|
||||
# Лимит размера загружаемого аватара — 2 МБ.
|
||||
MAX_AVATAR_SIZE_BYTES = 2 * 1024 * 1024
|
||||
|
||||
# Читаем файл чанками, не доверяя заголовку `Content-Length` (клиент может
|
||||
# солгать о размере) — реальный размер считается по факту прочитанных байт.
|
||||
_CHUNK_SIZE_BYTES = 64 * 1024
|
||||
|
||||
# Допустимые типы изображений -> расширение файла на диске.
|
||||
_ALLOWED_CONTENT_TYPES: dict[str, str] = {
|
||||
"image/jpeg": "jpg",
|
||||
"image/png": "png",
|
||||
"image/webp": "webp",
|
||||
}
|
||||
|
||||
# Магические байты (сигнатуры) форматов — заголовку `Content-Type` от клиента
|
||||
# доверять нельзя (легко подделать), реальный формат определяется по
|
||||
# содержимому файла.
|
||||
_MAGIC_CHECKS: dict[str, Callable[[bytes], bool]] = {
|
||||
"image/jpeg": lambda head: head[:3] == b"\xff\xd8\xff",
|
||||
"image/png": lambda head: head[:8] == b"\x89PNG\r\n\x1a\n",
|
||||
"image/webp": lambda head: head[:4] == b"RIFF" and head[8:12] == b"WEBP",
|
||||
}
|
||||
|
||||
# Достаточно первых 12 байт, чтобы проверить все сигнатуры выше (WebP —
|
||||
# самая длинная проверка, требует байты 8..11 включительно).
|
||||
_MAGIC_HEAD_SIZE = 12
|
||||
|
||||
|
||||
class AvatarTooLargeError(Exception):
|
||||
class AvatarTooLargeError(ImageTooLargeError):
|
||||
"""Загружаемый файл превышает `MAX_AVATAR_SIZE_BYTES` (413)."""
|
||||
|
||||
|
||||
class AvatarInvalidTypeError(Exception):
|
||||
class AvatarInvalidTypeError(ImageInvalidTypeError):
|
||||
"""`Content-Type` не входит в список допустимых либо не совпадает с содержимым (415)."""
|
||||
|
||||
|
||||
async def read_and_validate_avatar(file: UploadFile) -> tuple[bytes, str]:
|
||||
"""Прочитать содержимое файла аватара чанками и провалидировать тип/размер.
|
||||
"""Прочитать содержимое файла аватара и провалидировать тип/размер.
|
||||
|
||||
Возвращает `(содержимое, расширение)`. Порядок проверок: сначала
|
||||
заявленный `Content-Type` (быстрый отсев), затем фактический размер по
|
||||
мере чтения, затем магические байты содержимого — заявленный тип должен
|
||||
совпасть с реальным (иначе подделка `Content-Type` не даст загрузить,
|
||||
например, исполняемый файл под видом `image/png`).
|
||||
Возвращает `(содержимое, расширение)`. Ошибки общего валидатора
|
||||
перезаворачиваются в «аватарные» — вызывающий код (`api/users.py`,
|
||||
`api/admin.py`) отображает их в 413/415 и не должен знать про
|
||||
`services/images.py`.
|
||||
"""
|
||||
declared_type = file.content_type
|
||||
if declared_type not in _ALLOWED_CONTENT_TYPES:
|
||||
raise AvatarInvalidTypeError(f"unsupported_content_type: {declared_type}")
|
||||
|
||||
chunks: list[bytes] = []
|
||||
total_size = 0
|
||||
while True:
|
||||
chunk = await file.read(_CHUNK_SIZE_BYTES)
|
||||
if not chunk:
|
||||
break
|
||||
total_size += len(chunk)
|
||||
if total_size > MAX_AVATAR_SIZE_BYTES:
|
||||
raise AvatarTooLargeError(f"file exceeds {MAX_AVATAR_SIZE_BYTES} bytes")
|
||||
chunks.append(chunk)
|
||||
content = b"".join(chunks)
|
||||
|
||||
magic_check = _MAGIC_CHECKS[declared_type]
|
||||
if not magic_check(content[:_MAGIC_HEAD_SIZE]):
|
||||
raise AvatarInvalidTypeError("content_does_not_match_declared_content_type")
|
||||
|
||||
return content, _ALLOWED_CONTENT_TYPES[declared_type]
|
||||
try:
|
||||
return await read_and_validate_image(file, MAX_AVATAR_SIZE_BYTES)
|
||||
except ImageTooLargeError as exc:
|
||||
raise AvatarTooLargeError(str(exc)) from exc
|
||||
except ImageInvalidTypeError as exc:
|
||||
raise AvatarInvalidTypeError(str(exc)) from exc
|
||||
|
||||
|
||||
def _avatar_relative_path(user_id: uuid.UUID, ext: str) -> str:
|
||||
|
||||
Reference in New Issue
Block a user