from fastapi import APIRouter, Depends, HTTPException
from fastapi.responses import FileResponse
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select
from app.core.database import get_db
from app.core.security import get_current_user_id
from app.models.documento import Documento
from app.models.falecido import Falecido
from app.models.familia import Familia, MembroFamilia
from app.schemas.schemas import DocumentoOut, DocumentoGerarRequest
from app.services.document_generator import GENERATORS
import os

router = APIRouter(prefix="/familias/{familia_id}/documentos", tags=["documentos"])

@router.get("/", response_model=list[DocumentoOut])
async def listar(familia_id: int, db: AsyncSession = Depends(get_db)):
    result = await db.execute(select(Documento).where(Documento.familia_id == familia_id))
    return result.scalars().all()

@router.post("/gerar", response_model=DocumentoOut)
async def gerar(familia_id: int, data: DocumentoGerarRequest, db: AsyncSession = Depends(get_db)):
    gen = GENERATORS.get(data.tipo)
    if not gen:
        raise HTTPException(400, f"Tipo de documento inválido: {data.tipo}")
    
    fam = await db.execute(select(Familia).where(Familia.id == familia_id))
    familia = fam.scalar_one_or_none()
    if not familia:
        raise HTTPException(404, "Família não encontrada")
    
    fal = await db.execute(select(Falecido).where(Falecido.id == data.falecido_id))
    falecido = fal.scalar_one_or_none()
    if not falecido:
        raise HTTPException(404, "Falecido não encontrado")
    
    membros = await db.execute(select(MembroFamilia).where(MembroFamilia.familia_id == familia_id))
    membro = membros.scalars().first()
    membro_nome = membro.nome if membro else "Representante da Família"
    
    path = gen(familia.nome, falecido.nome, membro_nome, falecido.cpf or "000.000.000-00")
    
    doc = Documento(
        familia_id=familia_id,
        falecido_id=data.falecido_id,
        tipo=data.tipo,
        nome=f"{data.tipo.replace('_',' ').title()} - {falecido.nome}",
        arquivo_path=path,
        status="gerado",
    )
    db.add(doc)
    await db.commit()
    await db.refresh(doc)
    return doc

@router.get("/{doc_id}/download")
async def download(familia_id: int, doc_id: int, db: AsyncSession = Depends(get_db)):
    result = await db.execute(select(Documento).where(Documento.id == doc_id, Documento.familia_id == familia_id))
    doc = result.scalar_one_or_none()
    if not doc or not doc.arquivo_path or not os.path.exists(doc.arquivo_path):
        raise HTTPException(404, "Documento não encontrado")
    return FileResponse(doc.arquivo_path, filename=os.path.basename(doc.arquivo_path), media_type="application/pdf")