from uuid import UUID

from app.schema.media import MediaFile
from pydantic import BaseModel


class MediaFileResponse(BaseModel):
    id: UUID
    title: str | None = None
    file_name: str | None = None
    cloud_link: str | None = None
    url: str
    review: bool = False
    should_download: bool = False

class Link(BaseModel):
    url: str

def get_file_details(mediafile: MediaFile) -> MediaFileResponse | None:
    response = MediaFileResponse(id=mediafile.id,
                                 title=mediafile.title,
                                 file_name=mediafile.file_name,
                                 cloud_link=mediafile.cloud_link,
                                 url=str(mediafile.url),
                                 review=(mediafile.review == 1),
                                 should_download=(mediafile.should_download == 1))
    return response