from pydantic import BaseModel, Field
from typing import Optional
from datetime import datetime


class UserBase(BaseModel):
    """用户基础模型"""

    username: str = Field(
        ..., 
        min_length=3, 
        max_length=50, 
        description="登录名",
        json_schema_extra={
            "error_messages": {
                "min_length": "用户名长度至少为3个字符",
                "max_length": "用户名长度不能超过50个字符"
            }
        }
    )
    nickname: str = Field(..., max_length=50, description="昵称")
    email: Optional[str] = Field(None, max_length=50, description="电子邮箱")
    phone: Optional[str] = Field(None, max_length=11, description="手机号码")
    wx_openid: Optional[str] = Field(None, max_length=100, description="微信OpenID")
    avatar: Optional[str] = Field(None, max_length=255, description="头像路径")
    role_id: Optional[int] = Field(2, description="角色ID")

    model_config = {"extra": "ignore"}


class UserCreateRequest(UserBase):
    """用户创建请求模型"""

    password: str = Field(
        ..., 
        min_length=6, 
        max_length=50, 
        description="密码",
        json_schema_extra={
            "error_messages": {
                "min_length": "密码长度至少为6个字符",
                "max_length": "密码长度不能超过50个字符"
            }
        }
    )


class UserCreate(UserBase):
    """用户创建模型"""

    password_hash: str = Field(
        ..., min_length=6, max_length=128, description="密码哈希值"
    )


class UserUpdate(BaseModel):
    """用户更新模型"""

    username: str = Field(..., min_length=3, max_length=50, description="登录名")
    nickname: str = Field(..., max_length=50, description="昵称")
    email: Optional[str] = Field(None, max_length=50, description="电子邮箱")
    phone: Optional[str] = Field(None, max_length=11, description="手机号码")
    wx_openid: Optional[str] = Field(None, max_length=100, description="微信OpenID")
    avatar: Optional[str] = Field(None, max_length=255, description="头像路径")
    role_id: Optional[int] = Field(2, description="角色ID")
    password: Optional[str] = Field(None, max_length=128, description="密码")


class UserLogin(BaseModel):
    """用户登录模型"""

    username: str = Field(..., description="登录名")
    password: str = Field(..., description="密码")
    captcha_id: str = Field(..., description="验证码ID")
    captcha_code: str = Field(..., description="验证码")


class UserResponse(BaseModel):
    """用户响应模型"""

    id: int = Field(..., description="用户ID")
    username: str = Field(..., description="登录名")
    nickname: str = Field(..., description="昵称")
    email: Optional[str] = Field(None, description="电子邮箱")
    phone: Optional[str] = Field(None, description="手机号码")
    wx_openid: Optional[str] = Field(None, description="微信OpenID")
    avatar: Optional[str] = Field(None, description="头像路径")
    role_id: int = Field(..., description="角色ID")
    createtime: datetime = Field(..., description="创建时间")
    updatetime: datetime = Field(..., description="更新时间")
    lastlogintime: Optional[datetime] = Field(None, description="最后登录时间")

    class Config:
        from_attributes = True


class Token(BaseModel):
    """令牌模型"""

    access_token: str = Field(..., description="访问令牌")
    token_type: str = Field(..., description="令牌类型")


class TokenData(BaseModel):
    """令牌数据模型"""

    user_id: Optional[int] = None
    email: Optional[str] = None


class RoleBase(BaseModel):
    """角色基础模型"""

    name: str = Field(..., max_length=50, description="角色名称")


class RoleCreate(RoleBase):
    """角色创建模型"""

    creator: Optional[str] = Field(None, max_length=50, description="创建人")


class RoleResponse(RoleBase):
    """角色响应模型"""

    id: int = Field(..., description="角色ID")
    createtime: datetime = Field(..., description="创建时间")
    updatetime: datetime = Field(..., description="更新时间")
    creator: Optional[str] = Field(None, description="创建人")

    class Config:
        from_attributes = True


# 权限相关模型
class PermissionBase(BaseModel):
    """权限基础模型"""

    name: str = Field(..., max_length=100, description="权限名称")
    code: str = Field(..., max_length=100, description="权限编码")
    parentid: int = Field(0, description="父权限ID（0表示顶级模块）")
    description: Optional[str] = Field(None, max_length=255, description="权限描述")


class PermissionCreate(PermissionBase):
    """权限创建模型"""
    pass


class PermissionResponse(PermissionBase):
    """权限响应模型"""

    id: int = Field(..., description="权限ID")
    createtime: datetime = Field(..., description="创建时间")
    updatetime: datetime = Field(..., description="更新时间")

    class Config:
        from_attributes = True


class PermissionTreeResponse(BaseModel):
    """权限树响应模型"""

    id: int = Field(..., description="权限ID")
    name: str = Field(..., description="权限名称")
    code: str = Field(..., description="权限编码")
    parentid: int = Field(..., description="父权限ID")
    description: Optional[str] = Field(None, description="权限描述")
    children: list = Field([], description="子权限列表")


# 角色权限关联相关模型
class RolePermissionAssign(BaseModel):
    """角色权限分配模型"""

    role_id: int = Field(..., description="角色ID")
    permission_ids: list[int] = Field(..., description="权限ID列表")


class RolePermissionResponse(BaseModel):
    """角色权限响应模型"""

    role_id: int = Field(..., description="角色ID")
    role_name: str = Field(..., description="角色名称")
    permissions: list[PermissionResponse] = Field(..., description="权限列表")


# 系统设置相关模型
class SystemSettingBase(BaseModel):
    """系统设置基础模型"""

    key: str = Field(..., max_length=100, description="设置键名")
    value: str = Field(..., max_length=1000, description="设置值")
    description: Optional[str] = Field(None, max_length=255, description="设置描述")

    model_config = {"extra": "ignore"}


class SystemSettingCreate(SystemSettingBase):
    """系统设置创建模型"""
    pass


class SystemSettingUpdate(SystemSettingBase):
    """系统设置更新模型"""
    key: Optional[str] = Field(None, max_length=100, description="设置键名")


class SystemSettingResponse(BaseModel):
    """系统设置响应模型"""

    id: int = Field(..., description="设置ID")
    key: str = Field(..., description="设置键名")
    value: str = Field(..., description="设置值")
    description: Optional[str] = Field(None, description="设置描述")
    createtime: datetime = Field(..., description="创建时间")
    updatetime: datetime = Field(..., description="更新时间")

    class Config:
        from_attributes = True