"""
分析管理API
"""
from fastapi import APIRouter, Depends, HTTPException, BackgroundTasks
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select

from app.database import get_db
from app.models import Report, AnalysisStatus
from app.schemas import MessageResponse, AnalysisRequest
from app.services.ai_analyzer import ai_analyzer
from app.services.pdf_extractor import pdf_extractor

router = APIRouter(prefix="/api/analysis", tags=["AI分析"])


async def run_analysis_task(report_id: int):
    """后台分析任务"""
    async with AsyncSession(get_db) as db:  # 注意：这里需要正确的获取session方式，但在后台任务中比较复杂
        # 简化处理：在后台任务中重新创建session
        from app.database import AsyncSessionLocal
        async with AsyncSessionLocal() as session:
            stmt = select(Report).where(Report.id == report_id)
            result = await session.execute(stmt)
            report = result.scalar_one_or_none()
            
            if report:
                # 1. 确保已提取
                if not report.is_extracted:
                    await pdf_extractor.extract_and_save(session, report)
                
                # 2. 执行分析
                await ai_analyzer.analyze_report(session, report, force=True)


@router.post("/run", response_model=MessageResponse)
async def run_analysis(
    request: AnalysisRequest,
    background_tasks: BackgroundTasks,
    db: AsyncSession = Depends(get_db)
):
    """触发AI分析"""
    try:
        stmt = select(Report).where(Report.id == request.report_id)
        result = await db.execute(stmt)
        report = result.scalar_one_or_none()
        
        if not report:
            raise HTTPException(status_code=404, detail="报告不存在")
        
        if not report.is_downloaded:
            raise HTTPException(status_code=400, detail="报告尚未下载，无法分析")
        
        # 更新状态
        report.analysis_status = AnalysisStatus.PENDING.value
        await db.commit()
        
        # 启动后台任务
        import asyncio
        asyncio.create_task(run_analysis_with_new_session(request.report_id, request.force))
        
        return MessageResponse(success=True, message="已开始分析任务")
    except HTTPException:
        raise
    except Exception as e:
        import traceback
        error_msg = f"启动分析失败: {str(e)}\n{traceback.format_exc()}"
        print(error_msg) # 打印到控制台
        raise HTTPException(status_code=500, detail=str(e)) # 返回给前端


async def run_analysis_with_new_session(report_id: int, force: bool):
    from app.database import AsyncSessionLocal
    from app.models import Report, AnalysisStatus # 显式导入以防未加载
    
    async with AsyncSessionLocal() as session:
        report = None
        try:
            stmt = select(Report).where(Report.id == report_id)
            result = await session.execute(stmt)
            report = result.scalar_one_or_none()
            
            if report:
                # 1. 确保已提取
                if not report.is_extracted:
                    await pdf_extractor.extract_and_save(session, report)
                
                # 2. 执行分析
                await ai_analyzer.analyze_report(session, report, force=force)
                
        except Exception as e:
            import traceback
            error_trace = traceback.format_exc()
            print(f"后台分析任务严重异常 (report_id={report_id}): {e}\n{error_trace}")
            
            # 标记为失败，防止前端永久卡死
            if report:
                try:
                    # 重新获取对象以确保 session 仍然有效或在新的 commit 中
                    report.analysis_status = AnalysisStatus.FAILED.value
                    await session.commit()
                except Exception as commit_err:
                    print(f"标记任务失败时发生二次错误: {commit_err}")
        finally:
            await session.close()