from pydantic import BaseModel, Field, field_validator

from app.api.v1.module_energy.variable.schema import VariableItemSchema


class TemplateChapterRuleSchema(BaseModel):
    """模板章节规则配置项。"""

    id: int | None = Field(default=None, ge=1, description="章节规则标注ID，存在则更新，不传则新增")
    chapter_name: str = Field(..., min_length=1, max_length=128, description="章节名称")
    chapter_order: int = Field(default=999, ge=0, description="章节排序")
    rule_id: int = Field(..., ge=1, description="关联规则ID")
    description: str | None = Field(default=None, max_length=500, description="描述")

    @field_validator("chapter_name")
    @classmethod
    def validate_chapter_name(cls, value: str) -> str:
        value = value.strip()
        if not value:
            raise ValueError("章节名称不能为空")
        return value


class TemplateConfigSaveSchema(BaseModel):
    """模板章节规则和变量关联统一保存模型。"""

    template_id: int = Field(..., ge=1, description="模板ID")
    chapters: list[TemplateChapterRuleSchema] = Field(default_factory=list, description="章节规则标注列表")
    variables: list[VariableItemSchema] = Field(default_factory=list, description="变量关联列表")


class TemplateSingleChapterConfigSaveSchema(TemplateConfigSaveSchema):
    """单个章节标注和变量关联保存模型。"""

    @field_validator("chapters")
    @classmethod
    def validate_single_chapter(cls, value: list[TemplateChapterRuleSchema]) -> list[TemplateChapterRuleSchema]:
        if len(value) != 1:
            raise ValueError("chapters 只能传入一条章节规则标注")
        return value