"""
报告管理API
"""
from typing import List, Optional
from fastapi import APIRouter, Depends, HTTPException, Query
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select, desc
from sqlalchemy.orm import selectinload

from app.database import get_db
from app.models import Report, ExtractedContent, AnalysisResult
from app.schemas import ReportResponse, ReportDetail, MessageResponse

router = APIRouter(prefix="/api/reports", tags=["报告管理"])


@router.get("", response_model=List[ReportResponse])
async def list_reports(
    company_id: Optional[int] = None,
    report_type: Optional[str] = None,
    year: Optional[int] = None,
    is_analyzed: Optional[bool] = None,
    page: int = 1,
    page_size: int = 20,
    db: AsyncSession = Depends(get_db)
):
    """获取报告列表"""
    stmt = select(Report).options(selectinload(Report.company))
    
    if company_id:
        stmt = stmt.where(Report.company_id == company_id)
    if report_type:
        stmt = stmt.where(Report.report_type == report_type)
    if year:
        stmt = stmt.where(Report.report_year == year)
    if is_analyzed is not None:
        stmt = stmt.where(Report.is_analyzed == is_analyzed)
        
    stmt = stmt.order_by(desc(Report.created_at))
    stmt = stmt.offset((page - 1) * page_size).limit(page_size)
    
    result = await db.execute(stmt)
    reports = result.scalars().all()
    
    response = []
    for report in reports:
        report_data = ReportResponse.model_validate(report)
        # 补充关联信息
        report_data.company_name = report.company.company_name
        report_data.stock_code = report.company.stock_code
        response.append(report_data)
        
    return response


@router.get("/{report_id}", response_model=ReportDetail)
async def get_report(report_id: int, db: AsyncSession = Depends(get_db)):
    """获取报告详情（包含提取内容和分析结果）"""
    stmt = select(Report).where(Report.id == report_id).options(
        selectinload(Report.company),
        selectinload(Report.extracted_contents),
        selectinload(Report.analysis_results)
    )
    result = await db.execute(stmt)
    report = result.scalar_one_or_none()
    
    if not report:
        raise HTTPException(status_code=404, detail="报告不存在")
        
    report_data = ReportDetail.model_validate(report)
    report_data.company_name = report.company.company_name
    report_data.stock_code = report.company.stock_code
    
    return report_data


@router.delete("/{report_id}", response_model=MessageResponse)
async def delete_report(report_id: int, db: AsyncSession = Depends(get_db)):
    """删除报告"""
    stmt = select(Report).where(Report.id == report_id)
    result = await db.execute(stmt)
    report = result.scalar_one_or_none()
    
    if not report:
        raise HTTPException(status_code=404, detail="报告不存在")
    
    await db.delete(report)
    await db.commit()
    
    return MessageResponse(success=True, message="删除成功")