from typing import Union

from fastapi import APIRouter, Depends, HTTPException, status

from fastapi_pagination import Params
from fastapi_pagination.ext.ormar import paginate
from app.utils.pagination import CustomPage

from app.models import Book as BookDB, Author as AuthorDB, AuthorAnnotation as AuthorAnnotationDB
from app.serializers.book import Book, CreateBook, UpdateBook, CreateRemoteBook
from app.services.book import BookTGRMSearchService, BookCreator
from app.depends import check_token


book_router = APIRouter(
    prefix="/api/v1/books",
    tags=["book"],
    dependencies=[Depends(check_token)],
)


@book_router.get("/", response_model=CustomPage[Book], dependencies=[Depends(Params)])
async def get_books():
    return await paginate(
        BookDB.objects.select_related("authors")
    )


@book_router.post("/", response_model=Book)
async def create_book(data: Union[CreateBook, CreateRemoteBook]):
    book = await BookCreator.create(data)

    return await BookDB.objects.select_related("authors").get(id=book.id)


@book_router.get("/{id}", response_model=Book)
async def get_book(id: int):
    book = await BookDB.objects.select_related("authors").get_or_none(id=id)
    
    if book is None:
        raise HTTPException(status.HTTP_404_NOT_FOUND)

    return book


@book_router.get("/{source_id}/{remote_id}", response_model=Book)
async def get_remote_book(source_id: int, remote_id: int):
    book = await BookDB.objects.select_related("authors").get_or_none(
        source=source_id,
        remote_id=remote_id
    )

    if book is None:
        raise HTTPException(status.HTTP_404_NOT_FOUND)

    return book


@book_router.put("/{id}", response_model=Book)
async def update_book(id: int, data: UpdateBook):
    book = await BookDB.objects.select_related("authors").get_or_none(id=id)

    if book is None:
        raise HTTPException(status.HTTP_404_NOT_FOUND)

    for author in list(book.authors):
        await book.authors.remove(author)

    data_dict = data.dict()

    author_ids = data_dict.pop("authors", [])
    authors = await AuthorDB.objects.filter(id__in=author_ids).all()

    book = await BookDB.objects.create(
        **data_dict
    )

    for author in authors:
        await book.authors.add(author)

    return book


@book_router.get("/{id}/annotation")
async def get_book_annotation(id: int):
    annotation = await AuthorAnnotationDB.objects.get(book__id=id)

    if annotation is None:
        raise HTTPException(status.HTTP_404_NOT_FOUND)

    return annotation


@book_router.get("/search/{query}", response_model=CustomPage[Book], dependencies=[Depends(Params)])
async def search_books(query: str):
    return await BookTGRMSearchService.get(query)