takahe/api/views/media.py

from django.core.files import File
from django.shortcuts import get_object_or_404
from hatchway import ApiError, QueryOrBody, api_view

from activities.models import PostAttachment, PostAttachmentStates
from api import schemas
from core.files import blurhash_image, resize_image

from ..decorators import scope_required


@scope_required("write:media")
@api_view.post
def upload_media(
    request,
    file: File,
    description: QueryOrBody[str] = "",
    focus: QueryOrBody[str] = "0,0",
) -> schemas.MediaAttachment:
    main_file = resize_image(
        file,
        size=(2000, 2000),
        cover=False,
    )
    thumbnail_file = resize_image(
        file,
        size=(400, 225),
        cover=True,
    )
    attachment = PostAttachment.objects.create(
        blurhash=blurhash_image(thumbnail_file),
        mimetype="image/webp",
        width=main_file.image.width,
        height=main_file.image.height,
        name=description or None,
        state=PostAttachmentStates.fetched,
        author=request.identity,
    )
    attachment.file.save(
        main_file.name,
        main_file,
    )
    attachment.thumbnail.save(
        thumbnail_file.name,
        thumbnail_file,
    )
    attachment.save()
    return schemas.MediaAttachment.from_post_attachment(attachment)


@scope_required("read:media")
@api_view.get
def get_media(
    request,
    id: str,
) -> schemas.MediaAttachment:
    attachment = get_object_or_404(PostAttachment, pk=id)
    if attachment.post:
        if attachment.post.author != request.identity:
            raise ApiError(401, "Not the author of this attachment")
    elif attachment.author and attachment.author != request.identity:
        raise ApiError(401, "Not the author of this attachment")
    return schemas.MediaAttachment.from_post_attachment(attachment)


@scope_required("write:media")
@api_view.put
def update_media(
    request,
    id: str,
    description: QueryOrBody[str] = "",
    focus: QueryOrBody[str] = "0,0",
) -> schemas.MediaAttachment:
    attachment = get_object_or_404(PostAttachment, pk=id)
    if attachment.post:
        if attachment.post.author != request.identity:
            raise ApiError(401, "Not the author of this attachment")
    elif attachment.author != request.identity:
        raise ApiError(401, "Not the author of this attachment")
    attachment.name = description or None
    attachment.save()
    return schemas.MediaAttachment.from_post_attachment(attachment)