main.py

"""
FastAPI应用入口文件
"""
from contextlib import asynccontextmanager

import structlog
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from app import __version__, __description__
from app.api import convert_router, status_router, static_router, health_router, auth_router, async_convert_router
from app.config import settings
from app.database import init_database
from app.middleware import (
    RequestLoggingMiddleware,
    ResponseTimeMiddleware,
    RequestSizeMiddleware,
    ErrorHandlingMiddleware,
    RateLimitMiddleware,
    APIKeyAuthMiddleware
)


logger = structlog.get_logger(__name__)


@asynccontextmanager
async def lifespan(app: FastAPI):
    """应用生命周期管理"""
    # 启动事件
    logger.info("Starting pdf2othersAPI", version=__version__)
    
    try:
        # 初始化数据库
        await init_database()
        logger.info("Database initialized")
        
        # 初始化队列管理器
        from app.core.queue_manager import initialize_queue_manager
        await initialize_queue_manager()
        logger.info("Queue manager initialized")
        
        # 启动后台处理进程
        from app.core.process_pool import start_background_processing
        start_background_processing()
        logger.info("Background processing started")
        
        # 验证外部依赖
        from app.core import mineru_client, pandoc_client
        
        # 检查vLLM服务
        vllm_healthy = await mineru_client.check_vllm_health()
        if vllm_healthy:
            logger.info("vLLM service is available")
        else:
            logger.warning("vLLM service is not available")
        
        # 检查Pandoc
        pandoc_available = await pandoc_client.check_pandoc_availability()
        if pandoc_available:
            pandoc_version = await pandoc_client.get_pandoc_version()
            logger.info("Pandoc is available", version=pandoc_version)
        else:
            logger.warning("Pandoc is not available")
        
        # 确保存储目录存在
        settings.storage.pdfs_path.mkdir(parents=True, exist_ok=True)
        settings.storage.tmp_trs_dir.mkdir(parents=True, exist_ok=True)
        logger.info("Storage directories initialized")
        
    except Exception as e:
        logger.error("Failed to initialize application", error=str(e))
        raise
    
    yield
    
    # 关闭事件
    logger.info("Shutting down pdf2othersAPI")
    
    # 停止后台处理进程
    try:
        from app.core.process_pool import stop_background_processing
        stop_background_processing()
        logger.info("Background processing stopped")
    except Exception as e:
        logger.error("Failed to stop background processing", error=str(e))


# 创建FastAPI应用
app = FastAPI(
    title="pdf2othersAPI",
    description=__description__,
    version=__version__,
    lifespan=lifespan,
    docs_url="/docs",
    redoc_url="/redoc",
    openapi_url="/openapi.json"
)

# 添加CORS中间件
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # 生产环境应该限制源
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 添加自定义中间件（顺序很重要）
app.add_middleware(ErrorHandlingMiddleware)
app.add_middleware(RateLimitMiddleware, max_requests=100, window_seconds=60)
app.add_middleware(RequestSizeMiddleware)
app.add_middleware(ResponseTimeMiddleware)
app.add_middleware(RequestLoggingMiddleware)

# 添加认证中间件（在其他中间件之后）
if settings.auth_enabled:
    app.add_middleware(
        APIKeyAuthMiddleware,
        public_paths=["/", "/docs", "/redoc", "/openapi.json", "/health", "/favicon.ico", "/auth/permissions"],
        api_key_header=settings.api_key_header
    )

# 注册路由
app.include_router(health_router)
app.include_router(auth_router)  # 认证路由（在其他路由之前）
app.include_router(convert_router)
app.include_router(async_convert_router)  # 异步转换路由
app.include_router(status_router)
app.include_router(static_router)


@app.get("/")
async def root():
    """根端点"""
    return {
        "name": "pdf2othersAPI",
        "version": __version__,
        "description": __description__,
        "docs": "/docs",
        "health": "/health"
    }


if __name__ == "__main__":
    import uvicorn
    
    uvicorn.run(
        "main:app",
        host=settings.api.host,
        port=settings.api.port,
        reload=True,
        log_level=settings.log_level.lower()
    )