Kris
05abecc02b
该提交实现了支持企业微信、微信公众号、个人微信桥接三种模式的完整微信渠道适配器，包含以下核心模块： 1. 基础认证与配置相关：auth_adapter、config_reload、setup_contract等 2. 消息处理与格式转换：format、attachment_adapter、outbound_adapter等 3. 多模式客户端支持：wecom/mp子模块，包含加解密、消息收发能力 4. 辅助能力：限速器、防抖、会话绑定、事件映射、模板渲染等 5. 扩展能力：二维码登录、消息读取、特权用户、心跳监控等 实现了完整的微信生态对接能力，支持消息收发、事件处理、API调用限流、配置热重载等功能。
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import re
|
|
from typing import Any
|
|
|
|
|
|
class WeChatMentionAdapter:
|
|
@staticmethod
|
|
def strip_regexes(ctx: dict[str, Any], cfg: dict[str, Any], agent_id: str) -> list[re.Pattern]:
|
|
patterns: list[re.Pattern] = []
|
|
mode = ctx.get("mode", "personal")
|
|
|
|
if mode == "mp":
|
|
patterns.append(re.compile(r"@[\u4e00-\u9fff\w]+"))
|
|
elif mode == "wecom":
|
|
agent_name = cfg.get("agent_name", "")
|
|
if agent_name:
|
|
patterns.append(re.compile(rf"@{re.escape(agent_name)}"))
|
|
patterns.append(re.compile(r"@all"))
|
|
elif mode == "personal":
|
|
patterns.append(re.compile(r"@[\u4e00-\u9fff\w]+"))
|
|
|
|
return patterns
|
|
|
|
@staticmethod
|
|
def strip_patterns(ctx: dict[str, Any], cfg: dict[str, Any], agent_id: str) -> list[str]:
|
|
patterns: list[str] = []
|
|
mode = ctx.get("mode", "personal")
|
|
|
|
if mode == "wecom":
|
|
agent_name = cfg.get("agent_name", "")
|
|
if agent_name:
|
|
patterns.append(f"@{agent_name}")
|
|
|
|
return patterns
|
|
|
|
@staticmethod
|
|
def strip_mentions(text: str, ctx: dict[str, Any], cfg: dict[str, Any], agent_id: str) -> str:
|
|
mode = ctx.get("mode", "personal")
|
|
agent_name = cfg.get("agent_name", "")
|
|
|
|
if mode == "wecom" and agent_name:
|
|
text = re.sub(rf"@{re.escape(agent_name)}\s*", "", text)
|
|
|
|
text = re.sub(r"@all\s*", "", text, flags=re.IGNORECASE)
|
|
|
|
def _wrap_other_mention(m: re.Match) -> str:
|
|
name = m.group(0).strip().lstrip("@").strip()
|
|
if mode == "wecom" and name == agent_name:
|
|
return ""
|
|
return f"<at>{name}</at>"
|
|
|
|
text = re.sub(r"@[\u4e00-\u9fff\w]+\s*", _wrap_other_mention, text)
|
|
|
|
return text.strip()
|