该提交实现了支持企业微信、微信公众号、个人微信桥接三种模式的完整微信渠道适配器,包含以下核心模块: 1. 基础认证与配置相关:auth_adapter、config_reload、setup_contract等 2. 消息处理与格式转换:format、attachment_adapter、outbound_adapter等 3. 多模式客户端支持:wecom/mp子模块,包含加解密、消息收发能力 4. 辅助能力:限速器、防抖、会话绑定、事件映射、模板渲染等 5. 扩展能力:二维码登录、消息读取、特权用户、心跳监控等 实现了完整的微信生态对接能力,支持消息收发、事件处理、API调用限流、配置热重载等功能。
77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
WECHAT_MCP_TOOLS: dict[str, dict[str, Any]] = {
|
|
"wechat-send-message": {
|
|
"name": "wechat_send_message",
|
|
"description": "Send a message to a WeChat user or group",
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"target": {"type": "string", "description": "wx:<open_id> or wx:group:<chat_id>"},
|
|
"content": {"type": "string", "description": "Message content"},
|
|
},
|
|
"required": ["target", "content"],
|
|
},
|
|
},
|
|
"wechat-send-media": {
|
|
"name": "wechat_send_media",
|
|
"description": "Send media (image/video/voice) to a WeChat user or group",
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"target": {"type": "string"},
|
|
"media_type": {"type": "string", "enum": ["image", "video", "voice", "file"]},
|
|
"media_data": {"type": "string", "description": "Base64-encoded media data"},
|
|
},
|
|
"required": ["target", "media_type", "media_data"],
|
|
},
|
|
},
|
|
"wechat-read-message": {
|
|
"name": "wechat_read_message",
|
|
"description": "Read a message by ID",
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"msg_id": {"type": "string"},
|
|
},
|
|
"required": ["msg_id"],
|
|
},
|
|
},
|
|
"wechat-list-peers": {
|
|
"name": "wechat_list_peers",
|
|
"description": "List available WeChat contacts",
|
|
"parameters": {"type": "object", "properties": {}},
|
|
},
|
|
"wechat-list-groups": {
|
|
"name": "wechat_list_groups",
|
|
"description": "List available WeChat groups",
|
|
"parameters": {"type": "object", "properties": {}},
|
|
},
|
|
}
|
|
|
|
WECHAT_MCP_RESOURCES: dict[str, dict[str, Any]] = {
|
|
"wechat-channel-info": {
|
|
"uri": "wechat://channel/info",
|
|
"name": "WeChat Channel Info",
|
|
"description": "Current WeChat channel configuration and status",
|
|
},
|
|
}
|
|
|
|
|
|
def list_mcp_tools() -> list[dict[str, Any]]:
|
|
return list(WECHAT_MCP_TOOLS.values())
|
|
|
|
|
|
def list_mcp_resources() -> list[dict[str, Any]]:
|
|
return list(WECHAT_MCP_RESOURCES.values())
|
|
|
|
|
|
def get_mcp_tool(tool_name: str) -> dict[str, Any] | None:
|
|
return WECHAT_MCP_TOOLS.get(tool_name)
|
|
|
|
|
|
def get_mcp_resource(resource_uri: str) -> dict[str, Any] | None:
|
|
return WECHAT_MCP_RESOURCES.get(resource_uri)
|