from dataclasses import dataclass

from fastapi import Query
from pydantic import BaseModel, ConfigDict, Field, field_validator

from app.common.enums import QueueEnum
from app.core.base_params import BaseQueryParam, TenantByQueryParam, UserByQueryParam
from app.core.base_schema import BaseSchema, TenantBySchema, UserBySchema


class RuleCreateSchema(BaseModel):
    """规则创建模型"""

    name: str = Field(..., min_length=1, max_length=128, description="规则名称")
    category_id: int | None = Field(default=None, ge=0, description="规则分类ID")
    description: str | None = Field(default=None, max_length=2000, description="规则描述")
    prompt_template: str = Field(..., min_length=1, description="提示词模板")
    status: int = Field(default=0, ge=0, le=1, description="状态 0:启用 1:停用")

    @field_validator("name")
    @classmethod
    def validate_name(cls, value: str) -> str:
        value = value.strip()
        if not value:
            raise ValueError("规则名称不能为空")
        return value

    @field_validator("category_id")
    @classmethod
    def normalize_category_id(cls, value: int | None) -> int | None:
        return None if value == 0 else value

    @field_validator("prompt_template")
    @classmethod
    def validate_prompt_template(cls, value: str) -> str:
        value = value.strip()
        if not value:
            raise ValueError("提示词模板不能为空")
        return value

    @field_validator("status")
    @classmethod
    def validate_status(cls, value: int) -> int:
        if value not in {0, 1}:
            raise ValueError("状态仅支持 0(启用) 或 1(停用)")
        return value


class RuleUpdateSchema(RuleCreateSchema):
    """规则更新模型"""


class RuleOutSchema(RuleCreateSchema, BaseSchema, UserBySchema, TenantBySchema):
    """规则响应模型"""

    model_config = ConfigDict(from_attributes=True)

    category_name_snapshot: str | None = Field(default=None, max_length=64, description="规则分类名称快照")
    category_name: str | None = Field(default=None, max_length=64, description="规则分类名称")


@dataclass
class RuleQueryParam(BaseQueryParam, UserByQueryParam, TenantByQueryParam):
    """规则查询参数"""

    name: str | None = Query(None, description="规则名称")
    category_id: int | None = Query(None, ge=0, description="规则分类ID")
    status: int | None = Query(None, ge=0, le=1, description="状态 0:启用 1:停用")

    def __post_init__(self) -> None:
        self.name = (QueueEnum.like.value, self.name)
        if isinstance(self.category_id, int):
            self.category_id = (QueueEnum.eq.value, None if self.category_id == 0 else self.category_id)
        if isinstance(self.status, int):
            self.status = (QueueEnum.eq.value, self.status)