import uuid
from datetime import datetime

from pydantic import BaseModel, Field


class GeneratePdfRequest(BaseModel):
    title: str = Field(min_length=1, max_length=255)
    document_ids: list[uuid.UUID] = []
    chat_id: uuid.UUID | None = None
    template_id: uuid.UUID | None = None


class PdfResponse(BaseModel):
    id: uuid.UUID
    user_id: uuid.UUID
    title: str
    source_document_ids: list[uuid.UUID] | None
    source_chat_id: uuid.UUID | None
    template_id: uuid.UUID | None
    created_at: datetime

    model_config = {"from_attributes": True}


class PdfListResponse(BaseModel):
    pdfs: list[PdfResponse]