from __future__ import annotations

from datetime import datetime
from typing import Any

from sqlalchemy import DateTime, ForeignKey, Integer, JSON, String, Text, func
from sqlalchemy.orm import Mapped, mapped_column

from app.core.database import Base
from app.utils.ids import prefixed_id


class StudyProfile(Base):
    __tablename__ = "study_profiles"

    id: Mapped[str] = mapped_column(
        String(40),
        primary_key=True,
        default=lambda: prefixed_id("sprof"),
    )
    user_id: Mapped[str] = mapped_column(
        String(40),
        ForeignKey("users.id"),
        index=True,
        unique=True,
        nullable=False,
    )
    exam: Mapped[str | None] = mapped_column(String(120), nullable=True)
    board: Mapped[str | None] = mapped_column(String(120), nullable=True)
    grade: Mapped[str | None] = mapped_column(String(60), nullable=True)
    subject: Mapped[str | None] = mapped_column(String(120), nullable=True)
    chapter: Mapped[str | None] = mapped_column(String(180), nullable=True)
    topic: Mapped[str | None] = mapped_column(String(180), nullable=True)
    syllabus_text: Mapped[str | None] = mapped_column(Text, nullable=True)
    source_mode: Mapped[str | None] = mapped_column(String(60), nullable=True)
    uploaded_document_id: Mapped[str | None] = mapped_column(String(40), nullable=True)
    level: Mapped[str | None] = mapped_column(String(60), nullable=True)
    goal: Mapped[str | None] = mapped_column(String(60), nullable=True)
    time_left: Mapped[str | None] = mapped_column(String(60), nullable=True)
    language_preference: Mapped[str | None] = mapped_column(String(60), nullable=True)
    primary_need: Mapped[str | None] = mapped_column(String(60), nullable=True)
    weak_areas: Mapped[list[str]] = mapped_column(JSON, default=list, nullable=False)
    selected_plan: Mapped[str | None] = mapped_column(String(60), nullable=True)
    trial_started: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
    onboarding_completed: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
    extra: Mapped[dict[str, Any]] = mapped_column(JSON, default=dict, nullable=False)
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True),
        server_default=func.now(),
        nullable=False,
    )
    updated_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True),
        server_default=func.now(),
        onupdate=func.now(),
        nullable=False,
    )