from fastapi import APIRouter, HTTPException, Depends
from sqlalchemy.orm import Session
from database import get_db

router_documento = APIRouter(prefix="/documentos", tags=["Documentos"])


@router_documento.post("/")
def criar_documento(
    nome_arquivo: str,
    caminho_arquivo: str,
    usuario_id: int,
    db: Session = Depends(get_db),
):
    documento = documento(
        nome_arquivo=nome_arquivo,
        caminho_arquivo=caminho_arquivo,
        usuario_id=usuario_id,
    )
    db.add(documento)
    db.commit()
    db.refresh(documento)
    return documento


@router_documento.get("/{documento_id}")
def obter_documento(documento_id: str, db: Session = Depends(get_db)):
    documento = db.query(documento).filter(documento.id == documento_id).first()
    if not documento:
        raise HTTPException(status_code=404, detail="Documento não encontrado")
    return documento