Kris
05abecc02b
该提交实现了支持企业微信、微信公众号、个人微信桥接三种模式的完整微信渠道适配器，包含以下核心模块： 1. 基础认证与配置相关：auth_adapter、config_reload、setup_contract等 2. 消息处理与格式转换：format、attachment_adapter、outbound_adapter等 3. 多模式客户端支持：wecom/mp子模块，包含加解密、消息收发能力 4. 辅助能力：限速器、防抖、会话绑定、事件映射、模板渲染等 5. 扩展能力：二维码登录、消息读取、特权用户、心跳监控等 实现了完整的微信生态对接能力，支持消息收发、事件处理、API调用限流、配置热重载等功能。
122 lines
4.0 KiB
Python
122 lines
4.0 KiB
Python
from __future__ import annotations
|
|
|
|
import time
|
|
from dataclasses import dataclass
|
|
from typing import Any
|
|
|
|
from yuxi.utils.logging_config import logger
|
|
|
|
|
|
@dataclass
|
|
class InteractiveEvent:
|
|
event_key: str
|
|
event_type: str
|
|
channel_user_id: str
|
|
chat_id: str
|
|
task_id: str | None = None
|
|
selections: dict[str, list[str]] | None = None
|
|
timestamp: float = 0.0
|
|
|
|
|
|
class WeChatInteractiveAdapter:
|
|
def __init__(self):
|
|
self._enabled: bool = False
|
|
self._pending_events: dict[str, InteractiveEvent] = {}
|
|
self._processed_events: set[str] = set()
|
|
self._event_handlers: dict[str, list[callable]] = {}
|
|
|
|
def configure(self, config: dict[str, Any]) -> None:
|
|
self._enabled = config.get("interactive", {}).get("enabled", False)
|
|
logger.info(f"[WeChat/Interactive] Configured: enabled={self._enabled}")
|
|
|
|
def register_handler(self, event_key: str, handler: callable) -> None:
|
|
self._event_handlers.setdefault(event_key, []).append(handler)
|
|
|
|
def parse_click_event(self, payload: dict[str, Any]) -> InteractiveEvent | None:
|
|
if not self._enabled:
|
|
return None
|
|
|
|
msg_type = payload.get("MsgType", "")
|
|
event_type = payload.get("Event", "")
|
|
if msg_type != "event" and event_type not in ("click", "view"):
|
|
return None
|
|
|
|
event_key = payload.get("EventKey", "")
|
|
from_user = payload.get("FromUserName", "")
|
|
chat_id = payload.get("ToUserName", "")
|
|
|
|
if not event_key or not from_user:
|
|
return None
|
|
|
|
event = InteractiveEvent(
|
|
event_key=event_key,
|
|
event_type=event_type,
|
|
channel_user_id=from_user,
|
|
chat_id=chat_id,
|
|
task_id=payload.get("TaskId"),
|
|
timestamp=time.time(),
|
|
)
|
|
logger.info(f"[WeChat/Interactive] Parsed event: key={event_key}, user={from_user}")
|
|
return event
|
|
|
|
def parse_submit_event(self, payload: dict[str, Any]) -> InteractiveEvent | None:
|
|
if not self._enabled:
|
|
return None
|
|
|
|
msg_type = payload.get("MsgType", "")
|
|
event = payload.get("Event", "")
|
|
|
|
if msg_type != "event" or event != "template_card_event":
|
|
return None
|
|
|
|
card_event = payload.get("TemplateCardEvent", {})
|
|
event_key = card_event.get("EventKey", "")
|
|
task_id = card_event.get("TaskId", "")
|
|
|
|
selections_raw = card_event.get("SelectedItems", {})
|
|
selections: dict[str, list[str]] = {}
|
|
if isinstance(selections_raw, dict):
|
|
for key, value in selections_raw.items():
|
|
if isinstance(value, list):
|
|
selections[key] = value
|
|
elif isinstance(value, str):
|
|
selections[key] = [value]
|
|
|
|
interactive_event = InteractiveEvent(
|
|
event_key=event_key,
|
|
event_type="submit",
|
|
channel_user_id=payload.get("FromUserName", ""),
|
|
chat_id=payload.get("ToUserName", ""),
|
|
task_id=task_id,
|
|
selections=selections,
|
|
timestamp=time.time(),
|
|
)
|
|
logger.info(f"[WeChat/Interactive] Parsed submit event: key={event_key}, task={task_id}")
|
|
return interactive_event
|
|
|
|
async def dispatch_event(self, event: InteractiveEvent) -> list[dict[str, Any]]:
|
|
if not self._enabled:
|
|
return []
|
|
|
|
event_id = f"{event.event_key}:{event.channel_user_id}:{event.timestamp}"
|
|
if event_id in self._processed_events:
|
|
return []
|
|
self._processed_events.add(event_id)
|
|
|
|
results: list[dict[str, Any]] = []
|
|
handlers = self._event_handlers.get(event.event_key, [])
|
|
for handler in handlers:
|
|
try:
|
|
result = await handler(event)
|
|
results.append(result)
|
|
except Exception as e:
|
|
logger.error(f"[WeChat/Interactive] Handler error for '{event.event_key}': {e}")
|
|
|
|
return results
|
|
|
|
def is_enabled(self) -> bool:
|
|
return self._enabled
|
|
|
|
def clear_processed(self) -> None:
|
|
self._processed_events.clear()
|