"""Schemas — Timed aptitude / logical reasoning tests (SOW)."""

from typing import Any
from uuid import UUID

from pydantic import BaseModel, Field

from app.schemas.common import StubMeta
from app.schemas.questions import Difficulty, GeneratedQuestion


class AptitudeGenerateRequest(BaseModel):
    """POST /api/v1/aptitude/generate"""

    correlation_id: UUID | None = None
    domain: str = Field(default="logical_reasoning")
    role_title: str | None = None
    difficulty: Difficulty = Difficulty.MEDIUM
    count: int = Field(default=10, ge=1, le=50)
    time_limit_minutes: int = Field(default=30, ge=5, le=180)
    language: str = Field(default="en", min_length=2, max_length=10)
    options: dict[str, Any] = Field(default_factory=dict)


class AptitudeGenerateResponse(StubMeta):
    correlation_id: UUID | None = None
    aptitude_id: str | None = None
    domain: str
    time_limit_minutes: int
    questions: list[GeneratedQuestion] = Field(default_factory=list)


class AptitudeGradeRequest(BaseModel):
    """POST /api/v1/aptitude/grade"""

    correlation_id: UUID | None = None
    aptitude_id: str
    answers: list[dict[str, Any]] = Field(..., min_length=1)
    time_taken_seconds: int | None = None
    language: str = Field(default="en", min_length=2, max_length=10)
    options: dict[str, Any] = Field(default_factory=dict)


class AptitudeGradeResponse(StubMeta):
    correlation_id: UUID | None = None
    aptitude_id: str
    total_score: float = 0.0
    max_total_score: float = 0.0
    passed: bool | None = None
    percentile_estimate: float | None = None
    feedback_markdown: str | None = None
    item_results: list[dict[str, Any]] = Field(default_factory=list)