import base64

from fastapi import APIRouter, Depends, HTTPException, status, Request

from starlette.responses import Response

from arq.connections import ArqRedis
from asyncpg import exceptions

from app.depends import check_token
from app.models import CachedFile as CachedFileDB
from app.serializers import CachedFile, CreateCachedFile
from app.services.cache_updater import cache_file_by_book_id
from app.services.caption_getter import get_caption
from app.services.downloader import get_filename
from app.services.files_client import download_file as download_file_from_cache
from app.services.library_client import get_book


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


@router.get("/{object_id}/{object_type}", response_model=CachedFile)
async def get_cached_file(object_id: int, object_type: str):
    cached_file = await CachedFileDB.objects.get_or_none(
        object_id=object_id, object_type=object_type
    )

    if not cached_file:
        cached_file = await cache_file_by_book_id({}, object_id, object_type)

    if not cached_file:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)

    return cached_file


@router.get("/download/{object_id}/{object_type}")
async def download_cached_file(object_id: int, object_type: str):
    cached_file = await CachedFileDB.objects.get_or_none(
        object_id=object_id, object_type=object_type
    )

    if not cached_file:
        cached_file = await cache_file_by_book_id({}, object_id, object_type)

    if not cached_file:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)

    cache_data = cached_file.data

    data = await download_file_from_cache(
        cache_data["chat_id"], cache_data["message_id"]
    )

    if data is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)

    filename = await get_filename(object_id, object_type)

    book = await get_book(object_id)

    return Response(
        data,
        headers={
            "Content-Disposition": f"attachment; filename={filename}",
            "X-Caption-B64": base64.b64encode(get_caption(book).encode("utf-8")).decode(
                "latin-1"
            ),
        },
    )


@router.delete("/{object_id}/{object_type}", response_model=CachedFile)
async def delete_cached_file(object_id: int, object_type: str):
    cached_file = await CachedFileDB.objects.get(
        object_id=object_id, object_type=object_type
    )

    if not cached_file:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)

    await cached_file.delete()

    return cached_file


@router.post("/", response_model=CachedFile)
async def create_or_update_cached_file(data: CreateCachedFile):
    try:
        return await CachedFileDB.objects.create(**data.dict())
    except exceptions.UniqueViolationError:
        data_dict = data.dict()
        object_id = data_dict.pop("object_id")
        object_type = data_dict.pop("object_type")
        cached_file = await CachedFileDB.objects.get(
            object_id=object_id, object_type=object_type
        )
        cached_file.update_from_dict(data_dict)
        return await cached_file.update()


@router.post("/update_cache")
async def update_cache(request: Request):
    arq_pool: ArqRedis = request.app.state.arq_pool
    await arq_pool.enqueue_job("check_books")

    return "Ok!"