from __future__ import annotations

from fastapi import APIRouter, Depends, Query
from sqlalchemy.orm import Session

from app.core.database import get_db
from app.core.dependencies import get_current_user
from app.core.responses import api_success
from app.models.user import User
from app.schemas.user import UserProfile
from app.services.entitlement_service import EntitlementService, entitlement_dict
from app.services.growth_profile_service import GrowthProfileService, growth_profile_dict
from app.services.periodic_report_service import PeriodicReportService, periodic_report_user_dict
from app.services.periodic_report_lazy_service import PeriodicReportLazyService

router = APIRouter()


@router.get("/profile")
def profile(
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
) -> dict:
    data = UserProfile.model_validate(current_user).model_dump(mode="json")
    view = EntitlementService.active_entitlement(db, current_user)
    data["entitlement"] = entitlement_dict(view)
    return api_success(data)


@router.get("/growth-profile")
def growth_profile(
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
) -> dict:
    profile = GrowthProfileService.get_growth_profile(db, current_user.id)
    summaries = GrowthProfileService.recent_topic_summaries(db, current_user.id, limit=10)
    return api_success(
        {
            "profile": growth_profile_dict(profile),
            "recentSummaries": [
                {
                    "id": item.id,
                    "topicSessionId": item.topic_session_id,
                    "schemaVersion": item.schema_version,
                    "summary": item.summary,
                    "currentFocus": item.main_events,
                    "nextObservation": item.next_observation,
                    "status": item.status,
                    "errorMessage": item.error_message,
                    "attemptCount": item.attempt_count,
                    "maxAttempts": item.max_attempts,
                    "nextRunAt": item.next_run_at,
                    "lastStartedAt": item.last_started_at,
                    "finishedAt": item.finished_at,
                    "generatedAt": item.generated_at,
                }
                for item in summaries
            ],
        }
    )


@router.get("/periodic-report/list")
def periodic_reports(
    limit: int = 20,
    ensure: bool = Query(default=False),
    db: Session = Depends(get_db),
    current_user: User = Depends(get_current_user),
) -> dict:
    entitlement = EntitlementService.active_entitlement(db, current_user)
    if not entitlement.enable_periodic_reports:
        return api_success([])
    if ensure:
        PeriodicReportLazyService.enqueue_missing_reports(db, user=current_user)
        db.commit()
    reports = PeriodicReportService.list_user_reports(
        db,
        user_id=current_user.id,
        limit=max(1, min(limit, 50)),
        statuses=("pending", "running", "success", "failed", "empty"),
    )
    return api_success([periodic_report_user_dict(item) for item in reports])