from pydantic import BaseModel, ConfigDict, Field, field_validator

from app.core.base_schema import BaseQueryParam, BaseSchema, UserByQueryParam, UserBySchema
from app.core.validator import validate_required_code


class DeptCreateSchema(BaseModel):
    """部门创建模型"""

    name: str = Field(..., min_length=1, max_length=64, description="部门名称")
    order: int = Field(default=1, ge=0, description="显示顺序")
    code: str = Field(..., min_length=2, max_length=64, description="部门编码")
    parent_id: int | None = Field(default=None, ge=1, description="父部门ID")
    status: int = Field(default=0, ge=0, le=1, description="状态(0:启动 1:停用)")
    description: str | None = Field(default=None, max_length=255, description="备注")

    @field_validator("name")
    @classmethod
    def validate_name(cls, value: str):
        """校验部门名称：不能为空"""
        if not value or not value.strip():
            raise ValueError("部门名称不能为空")
        return value.strip()

    @field_validator("code")
    @classmethod
    def validate_code(cls, value: str):
        """校验部门编码：字母开头，2-64 位，仅含字母/数字/下划线"""
        return validate_required_code(value)

    @field_validator("status")
    @classmethod
    def validate_status(cls, value: int):
        """校验状态：仅支持 0(正常)、1(禁用)"""
        if value not in {0, 1}:
            raise ValueError("状态仅支持 0(正常) 或 1(禁用)")
        return value


class DeptUpdateSchema(DeptCreateSchema):
    """部门更新模型"""


class DeptOutSchema(DeptCreateSchema, BaseSchema, UserBySchema):
    """部门详情响应模型（不含 children，用于详情和更新）"""

    model_config = ConfigDict(from_attributes=True)

    parent_name: str | None = Field(default=None, max_length=64, description="父部门名称")


class DeptTreeOutSchema(DeptOutSchema):
    """部门树形响应模型（含 children，用于树形列表）"""

    children: list["DeptTreeOutSchema"] | None = Field(default=None, description="子部门列表")


class DeptQueryParam(BaseQueryParam, UserByQueryParam):
    """部门管理查询参数"""

    name: str | None = Field(None, description="部门名称", json_schema_extra={"q": "like"})
    status: int | None = Field(None, ge=0, le=1, description="状态(0:启动 1:停用)", json_schema_extra={"q": "eq"})