"""Schemas for WeCom API."""

from __future__ import annotations

from pydantic import BaseModel, Field


class WeComConfigRequest(BaseModel):
    """企业微信配置请求"""

    corp_id: str = Field(..., description="企业 ID")
    agent_id: str = Field(..., description="应用 ID")
    app_secret: str = Field(..., description="应用 Secret")
    token: str = Field(..., description="回调验证 Token")
    encoding_aes_key: str = Field(..., description="消息加密密钥")


class WeComSendMessageRequest(BaseModel):
    """发送消息请求"""

    chat_id: str = Field(..., description="群聊 ID")
    message: str = Field(..., description="消息内容")