"""Pydantic schemas for document models."""

from datetime import datetime
from typing import Optional
from uuid import UUID

from pydantic import BaseModel

from app.models.document import DocumentStatus, DocumentType


class DocumentBase(BaseModel):
    document_type: DocumentType
    file_name: str
    file_url: str
    file_size: Optional[int] = None
    mime_type: Optional[str] = None
    description: Optional[str] = None


class DocumentCreate(DocumentBase):
    user_id: Optional[UUID] = None
    application_id: Optional[UUID] = None


class DocumentResponse(DocumentBase):
    id: UUID
    user_id: Optional[UUID] = None
    application_id: Optional[UUID] = None
    status: DocumentStatus
    verified_by: Optional[UUID] = None
    verified_at: Optional[datetime] = None
    verification_notes: Optional[str] = None
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True


class DocumentUpdate(BaseModel):
    status: Optional[DocumentStatus] = None
    verification_notes: Optional[str] = None