"""Pydantic schemas for the Report Generator plugin."""

from __future__ import annotations

from datetime import datetime

from pydantic import BaseModel, Field


class TemplateCreate(BaseModel):
    name: str = Field(..., min_length=1, max_length=255)
    description: str = Field("", max_length=2000)
    template_type: str = Field("jinja2", pattern="^(jinja2|sql)$")
    content: str = Field(..., min_length=1)
    output_format: str = Field("csv", pattern="^(csv|excel|json)$")


class TemplateUpdate(BaseModel):
    name: str | None = Field(None, min_length=1, max_length=255)
    description: str | None = Field(None, max_length=2000)
    template_type: str | None = Field(None, pattern="^(jinja2|sql)$")
    content: str | None = None
    output_format: str | None = Field(None, pattern="^(csv|excel|json)$")


class TemplateResponse(BaseModel):
    id: str
    name: str
    description: str
    template_type: str
    content: str
    output_format: str
    created_by: str
    deleted_at: datetime | None = None
    created_at: datetime | None = None
    updated_at: datetime | None = None


class ReportGenerateRequest(BaseModel):
    template_id: str = Field(..., min_length=1)
    data: dict = Field(default_factory=dict)


class ReportResponse(BaseModel):
    id: str
    template_id: str
    status: str
    output_path: str | None = None
    error_message: str | None = None
    created_by: str
    created_at: datetime | None = None
    updated_at: datetime | None = None