from fastapi import APIRouter, HTTPException, status

from dispatch.database.core import DbSession
from dispatch.database.service import CommonParameters, search_filter_sort_paginate
from dispatch.models import PrimaryKey

from .models import DocumentCreate, DocumentPagination, DocumentRead, DocumentUpdate
from .service import create, delete, get, update

router = APIRouter()


@router.get("", response_model=DocumentPagination)
def get_documents(common: CommonParameters):
    """Get all documents."""
    return search_filter_sort_paginate(model="Document", **common)


@router.get("/{document_id}", response_model=DocumentRead)
def get_document(db_session: DbSession, document_id: PrimaryKey):
    """Update a document."""
    document = get(db_session=db_session, document_id=document_id)
    if not document:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail=[{"msg": "A document with this id does not exist."}],
        )
    return document


@router.post("", response_model=DocumentRead)
def create_document(db_session: DbSession, document_in: DocumentCreate):
    """Create a new document."""
    return create(db_session=db_session, document_in=document_in)


@router.put("/{document_id}", response_model=DocumentRead)
def update_document(db_session: DbSession, document_id: PrimaryKey, document_in: DocumentUpdate):
    """Update a document."""
    document = get(db_session=db_session, document_id=document_id)
    if not document:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail=[{"msg": "A document with this id does not exist."}],
        )
    document = update(db_session=db_session, document=document, document_in=document_in)
    return document


@router.delete("/{document_id}", response_model=None)
def delete_document(db_session: DbSession, document_id: PrimaryKey):
    """Delete a document."""
    document = get(db_session=db_session, document_id=document_id)
    if not document:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail=[{"msg": "A document with this id does not exist."}],
        )
    delete(db_session=db_session, document_id=document_id)