"""
文档库管理 API 接口

提供文档列表、详情查询和删除功能
"""
from typing import Optional, List

from fastapi import APIRouter, HTTPException, Query
from pydantic import BaseModel

from app.core.database import mongodb

router = APIRouter(prefix="/documents", tags=["文档库"])


class DocumentItem(BaseModel):
    doc_id: str
    filename: str
    original_filename: str
    doc_type: str
    file_size: int
    created_at: str
    metadata: Optional[dict] = None


@router.get("")
async def get_documents(
    doc_type: Optional[str] = Query(None, description="文档类型过滤"),
    limit: int = Query(50, ge=1, le=100, description="返回数量")
):
    """
    获取文档列表

    Returns:
        文档列表
    """
    try:
        # 构建查询条件
        query = {}
        if doc_type:
            query["doc_type"] = doc_type

        # 查询文档
        cursor = mongodb.documents.find(query).sort("created_at", -1).limit(limit)

        documents = []
        async for doc in cursor:
            documents.append({
                "doc_id": str(doc["_id"]),
                "filename": doc.get("metadata", {}).get("filename", ""),
                "original_filename": doc.get("metadata", {}).get("original_filename", ""),
                "doc_type": doc.get("doc_type", ""),
                "file_size": doc.get("metadata", {}).get("file_size", 0),
                "created_at": doc.get("created_at", "").isoformat() if doc.get("created_at") else "",
                "metadata": {
                    "row_count": doc.get("metadata", {}).get("row_count"),
                    "column_count": doc.get("metadata", {}).get("column_count"),
                    "columns": doc.get("metadata", {}).get("columns", [])[:10]  # 只返回前10列
                }
            })

        return {
            "success": True,
            "documents": documents,
            "total": len(documents)
        }

    except Exception as e:
        raise HTTPException(status_code=500, detail=f"获取文档列表失败: {str(e)}")


@router.get("/{doc_id}")
async def get_document(doc_id: str):
    """
    获取文档详情

    Args:
        doc_id: 文档ID

    Returns:
        文档详情
    """
    try:
        doc = await mongodb.get_document(doc_id)

        if not doc:
            raise HTTPException(status_code=404, detail="文档不存在")

        return {
            "success": True,
            "document": {
                "doc_id": str(doc["_id"]),
                "filename": doc.get("metadata", {}).get("filename", ""),
                "original_filename": doc.get("metadata", {}).get("original_filename", ""),
                "doc_type": doc.get("doc_type", ""),
                "file_size": doc.get("metadata", {}).get("file_size", 0),
                "created_at": doc.get("created_at", "").isoformat() if doc.get("created_at") else "",
                "content": doc.get("content", ""),  # 原始文本内容
                "structured_data": doc.get("structured_data"),  # 结构化数据(如果有)
                "metadata": doc.get("metadata", {})
            }
        }

    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"获取文档详情失败: {str(e)}")


@router.delete("/{doc_id}")
async def delete_document(doc_id: str):
    """
    删除文档

    Args:
        doc_id: 文档ID

    Returns:
        删除结果
    """
    try:
        # 从 MongoDB 删除
        deleted = await mongodb.delete_document(doc_id)

        if not deleted:
            raise HTTPException(status_code=404, detail="文档不存在")

        # TODO: 从 MySQL 删除相关数据(如果是Excel)
        # TODO: 从 RAG 删除相关索引

        return {
            "success": True,
            "message": "文档已删除"
        }

    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"删除失败: {str(e)}")