"""Knowledge base ingestion API."""

from pathlib import Path

from fastapi import APIRouter, Depends, File, UploadFile
from sqlalchemy.ext.asyncio import AsyncSession

from app.config import get_settings
from app.db import get_db
from app.models import KnowledgeDoc
from app.schemas import KnowledgeIngestResponse
from app.services.rag import get_knowledge_store, ingest_path

router = APIRouter(prefix="/api/knowledge", tags=["knowledge"])


@router.get("/status")
async def knowledge_status() -> dict:
    """Return current knowledge chunk count."""
    store = get_knowledge_store()
    return {"chunk_count": store.chunk_count}


@router.post(
    "/ingest",
    response_model=KnowledgeIngestResponse,
)
async def ingest_upload(
    file: UploadFile = File(...),
    db: AsyncSession = Depends(get_db),
) -> KnowledgeIngestResponse:
    """Upload and ingest a knowledge document."""
    settings = get_settings()
    target_dir = Path(settings.knowledge_path)
    target_dir.mkdir(parents=True, exist_ok=True)
    filename = Path(file.filename or "upload.txt").name
    dest = target_dir / filename
    content = await file.read()
    dest.write_bytes(content)

    source = dest.suffix.lower().lstrip(".") or "text"
    count = await ingest_path(dest, source)
    db.add(
        KnowledgeDoc(
            filename=filename,
            source_type=source,
            chunk_count=count,
        )
    )
    await db.commit()
    return KnowledgeIngestResponse(
        filename=filename,
        chunk_count=count,
        message=f"Ingested {count} chunks from {filename}",
    )