LMS/backend/app/schemas.py

from datetime import datetime
from pydantic import BaseModel, EmailStr, field_validator
import re

from .models import UserRole, UserStatus


# ── Auth ─────────────────────────────────────────────────────────────────────

class UserRegister(BaseModel):
    username: str
    email: EmailStr
    password: str
    role: UserRole = UserRole.student

    @field_validator("username")
    @classmethod
    def username_valid(cls, v: str) -> str:
        v = v.strip()
        if not re.match(r"^[A-Za-z0-9_]{3,64}$", v):
            raise ValueError(
                "Username must be 3–64 characters, letters/numbers/underscore only."
            )
        return v

    @field_validator("password")
    @classmethod
    def password_strength(cls, v: str) -> str:
        if len(v) < 8:
            raise ValueError("Password must be at least 8 characters.")
        return v


class UserLogin(BaseModel):
    username: str
    password: str


class UserOut(BaseModel):
    id: int
    username: str
    email: str
    role: UserRole
    status: UserStatus
    created_at: datetime

    model_config = {"from_attributes": True}


class UserApprove(BaseModel):
    status: UserStatus


class UserRoleUpdate(BaseModel):
    role: UserRole


class ChangePassword(BaseModel):
    current_password: str
    new_password: str

    @field_validator("new_password")
    @classmethod
    def password_strength(cls, v: str) -> str:
        if len(v) < 8:
            raise ValueError("Password must be at least 8 characters.")
        return v


class TokenPayload(BaseModel):
    sub: int   # user id
    exp: int


# ── PDFs ──────────────────────────────────────────────────────────────────────

class PDFOut(BaseModel):
    id: int
    user_id: int
    owner_username: str
    title: str
    total_pages: int | None
    created_at: datetime

    model_config = {"from_attributes": True}


class PDFUploadResult(BaseModel):
    filename: str
    success: bool
    pdf: PDFOut | None = None
    error: str | None = None


# ── Annotations ───────────────────────────────────────────────────────────────

class AnnotationIn(BaseModel):
    canvas_data: dict


class AnnotationOut(BaseModel):
    id: int
    pdf_id: int
    page_number: int
    canvas_data: dict
    updated_at: datetime

    model_config = {"from_attributes": True}