from typing import Optional

from fastapi import File, UploadFile, Depends, Form
from starlette import status
from fastapi import APIRouter, HTTPException

from app.models import UploadedFile as UploadedFileDB
from app.serializers import UploadedFile, CreateUploadedFile
from app.services.file_uploader import FileUploader
from app.depends import check_token


router = APIRouter(
    prefix="/api/v1/files",
    dependencies=[Depends(check_token)],
    tags=["files"]
)


@router.get("/", response_model=list[UploadedFile])
async def get_files():
    return await UploadedFileDB.objects.all()


@router.get("/{file_id}", response_model=UploadedFile, responses={
    404: {},
})
async def get_file(file_id: int):
    uploaded_file = await UploadedFileDB.objects.get_or_none(id=file_id)

    if not uploaded_file:
        raise HTTPException(status.HTTP_404_NOT_FOUND)

    return uploaded_file


@router.post("/", response_model=UploadedFile)
async def create_file(data: CreateUploadedFile):
    return await UploadedFileDB.objects.create(
        **data.dict()
    )


@router.post("/upload/", response_model=UploadedFile)
async def upload_file(file: UploadFile = File({}), caption: Optional[str] = Form({})):
    return await FileUploader.upload(file, caption=caption)


@router.delete("/{file_id}", response_model=UploadedFile, responses={
    400: {}
})
async def delete_file(file_id: int):
    uploaded_file = await UploadedFileDB.objects.get_or_none(id=file_id)

    if not uploaded_file:
        raise HTTPException(status.HTTP_400_BAD_REQUEST)

    await uploaded_file.delete()
    return uploaded_file