from __future__ import annotations

from fastapi import APIRouter, Depends, HTTPException, Response

from sqlalchemy import select

from sqlalchemy.orm import Session

from app.api.deps import Perm, get_db, require

from app.models import AuditLog, Report, User

from app.schemas import ReportRequest

from app.services.reporting import (

    build_report, render_csv_timeline, render_json, render_markdown,

)

router = APIRouter(tags=["reports"])

@router.post("/investigations/{inv_id}/reports")

def create_report(inv_id: str, payload: ReportRequest, db: Session = Depends(get_db),

                  user: User = Depends(require(Perm.EXPORT))):

    try:

        report = build_report(db, inv_id, payload.sections)

    except ValueError as e:

        raise HTTPException(404, str(e))

    rec = Report(investigation_id=inv_id, title=payload.title or report["title"],

                 format=payload.format, sections=report["sections"], created_by=user.id)

    db.add(rec)

    db.flush()

    db.add(AuditLog(actor_user_id=user.id, action="report.create",

                    entity_type="report", entity_id=rec.id, meta={"format": payload.format}))

    db.commit()

    db.refresh(rec)

    return {"report_id": rec.id, "format": payload.format, "report": report}

@router.get("/reports/{report_id}/download")

def download_report(report_id: str, format: str = "markdown", db: Session = Depends(get_db),

                    _: User = Depends(require(Perm.EXPORT))):

    rec = db.get(Report, report_id)

    if not rec:

        raise HTTPException(404, "report not found")

    report = {"title": rec.title, "investigation_id": rec.investigation_id, "sections": rec.sections}

    if format == "json":

        return Response(render_json(report), media_type="application/json")

    if format == "csv":

        return Response(render_csv_timeline(report), media_type="text/csv",

                        headers={"Content-Disposition": f"attachment; filename=report_{report_id}.csv"})

    if format == "pdf":

        from app.services.report_pdf import render_pdf

        return Response(render_pdf(report), media_type="application/pdf",

                        headers={"Content-Disposition": f"attachment; filename=report_{report_id}.pdf"})

    return Response(render_markdown(report), media_type="text/markdown")